1 files changed, 25 insertions, 6 deletions
diff --git a/spec/frontend/environments/environment_external_url_spec.js b/spec/frontend/environments/environment_external_url_spec.js
index 4c133665979..5966993166b 100644
--- a/spec/frontend/environments/environment_external_url_spec.js
+++ b/spec/frontend/environments/environment_external_url_spec.js
@@ -1,16 +1,35 @@
 import { mount } from '@vue/test-utils';
+import { s__, __ } from '~/locale';
 import ExternalUrlComp from '~/environments/components/environment_external_url.vue';
+import ModalCopyButton from '~/vue_shared/components/modal_copy_button.vue';
 
 describe('External URL Component', () => {
   let wrapper;
-  const externalUrl = 'https://gitlab.com';
+  let externalUrl;
 
-  beforeEach(() => {
-    wrapper = mount(ExternalUrlComp, { propsData: { externalUrl } });
+  describe('with safe link', () => {
+    beforeEach(() => {
+      externalUrl = 'https://gitlab.com';
+      wrapper = mount(ExternalUrlComp, { propsData: { externalUrl } });
+    });
+
+    it('should link to the provided externalUrl prop', () => {
+      expect(wrapper.attributes('href')).toBe(externalUrl);
+      expect(wrapper.find('a').exists()).toBe(true);
+    });
   });
 
-  it('should link to the provided externalUrl prop', () => {
-    expect(wrapper.attributes('href')).toEqual(externalUrl);
-    expect(wrapper.find('a').exists()).toBe(true);
+  describe('with unsafe link', () => {
+    beforeEach(() => {
+      externalUrl = 'postgres://gitlab';
+      wrapper = mount(ExternalUrlComp, { propsData: { externalUrl } });
+    });
+
+    it('should show a copy button instead', () => {
+      const button = wrapper.findComponent(ModalCopyButton);
+      expect(button.props('text')).toBe(externalUrl);
+      expect(button.text()).toBe(__('Copy URL'));
+      expect(button.props('title')).toBe(s__('Environments|Copy live environment URL'));
+    });
   });
 });