From 887469d73ed633ebe4dff6e62f11a944b14f42c0 Mon Sep 17 00:00:00 2001
From: mabashian <mabashia@redhat.com>
Date: Tue, 4 Feb 2020 11:14:28 -0500
Subject: [PATCH] Adds test for link add/edit/delete modals

---
 .../Modals/LinkModals/LinkAddModal.test.jsx   | 24 ++++++++
 .../Modals/LinkModals/LinkDeleteModal.jsx     |  2 +
 .../LinkModals/LinkDeleteModal.test.jsx       | 60 +++++++++++++++++++
 .../Modals/LinkModals/LinkEditModal.test.jsx  | 32 ++++++++++
 4 files changed, 118 insertions(+)
 create mode 100644 awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkAddModal.test.jsx
 create mode 100644 awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.test.jsx
 create mode 100644 awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkEditModal.test.jsx

diff --git a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkAddModal.test.jsx b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkAddModal.test.jsx
new file mode 100644
index 0000000000..e7b5b94c40
--- /dev/null
+++ b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkAddModal.test.jsx
@@ -0,0 +1,24 @@
+import React from 'react';
+import { mountWithContexts } from '@testUtils/enzymeHelpers';
+import { WorkflowDispatchContext, WorkflowStateContext } from '@contexts/Workflow';
+import LinkAddModal from './LinkAddModal';
+
+const dispatch = jest.fn();
+
+const workflowContext = {
+  linkToEdit: null
+};
+
+describe('LinkAddModal', () => {
+  test('Confirm button dispatches as expected', () => {
+    const wrapper = mountWithContexts(
+      <WorkflowDispatchContext.Provider value={dispatch}>
+        <WorkflowStateContext.Provider value={workflowContext}>
+          <LinkAddModal />
+        </WorkflowStateContext.Provider>
+      </WorkflowDispatchContext.Provider>
+      );
+    wrapper.find('button#link-confirm').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({ type: 'CREATE_LINK', linkType: 'success' });
+  });
+});
diff --git a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.jsx b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.jsx
index 395ac49617..d868e12083 100644
--- a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.jsx
+++ b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.jsx
@@ -19,6 +19,7 @@ function LinkDeleteModal({ i18n }) {
       onClose={() => dispatch({ type: 'SET_LINK_TO_DELETE', value: null })}
       actions={[
         <Button
+          id="confirm-link-removal"
           aria-label={i18n._(t`Confirm link removal`)}
           key="remove"
           onClick={() => dispatch({ type: 'DELETE_LINK' })}
@@ -27,6 +28,7 @@ function LinkDeleteModal({ i18n }) {
           {i18n._(t`Remove`)}
         </Button>,
         <Button
+        id="cancel-link-removal"
           aria-label={i18n._(t`Cancel link removal`)}
           key="cancel"
           onClick={() => dispatch({ type: 'SET_LINK_TO_DELETE', value: null })}
diff --git a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.test.jsx b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.test.jsx
new file mode 100644
index 0000000000..afb7fea396
--- /dev/null
+++ b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkDeleteModal.test.jsx
@@ -0,0 +1,60 @@
+import React from 'react';
+import {
+  WorkflowDispatchContext,
+  WorkflowStateContext
+} from '@contexts/Workflow';
+import { mountWithContexts } from '@testUtils/enzymeHelpers';
+import LinkDeleteModal from './LinkDeleteModal';
+
+let wrapper;
+const dispatch = jest.fn();
+
+const workflowContext = {
+  linkToDelete: {
+    source: {
+      id: 2
+    },
+    target: {
+      id: 3
+    },
+    linkType: 'always'
+  }
+};
+
+describe('LinkDeleteModal', () => {
+  beforeAll(() => {
+    wrapper = mountWithContexts(
+      <WorkflowDispatchContext.Provider value={dispatch}>
+        <WorkflowStateContext.Provider value={workflowContext}>
+          <LinkDeleteModal />
+        </WorkflowStateContext.Provider>
+      </WorkflowDispatchContext.Provider>
+    );
+  });
+
+  afterAll(() => {
+    wrapper.unmount();
+  });
+
+  test('Confirm button dispatches as expected', () => {
+    wrapper.find('button#confirm-link-removal').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({
+      type: 'DELETE_LINK',
+    });
+  });
+
+  test('Cancel button dispatches as expected', () => {
+    wrapper.find('button#cancel-link-removal').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({
+      type: 'SET_LINK_TO_DELETE',
+      value: null,
+    });
+  });
+
+  test('Close button dispatches as expected', () => {
+    wrapper.find('TimesIcon').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({
+      type: 'SET_LINK_TO_DELETE', value: null
+    });
+  });
+});
diff --git a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkEditModal.test.jsx b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkEditModal.test.jsx
new file mode 100644
index 0000000000..6235ce7de9
--- /dev/null
+++ b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/LinkModals/LinkEditModal.test.jsx
@@ -0,0 +1,32 @@
+import React from 'react';
+import { mountWithContexts } from '@testUtils/enzymeHelpers';
+import { WorkflowDispatchContext, WorkflowStateContext } from '@contexts/Workflow';
+import LinkEditModal from './LinkEditModal';
+
+const dispatch = jest.fn();
+
+const workflowContext = {
+  linkToEdit: {
+    source: {
+      id: 2
+    },
+    target: {
+      id: 3
+    },
+    linkType: 'always'
+  }
+};
+
+describe('LinkEditModal', () => {
+  test('Confirm button dispatches as expected', () => {
+    const wrapper = mountWithContexts(
+      <WorkflowDispatchContext.Provider value={dispatch}>
+        <WorkflowStateContext.Provider value={workflowContext}>
+          <LinkEditModal />
+        </WorkflowStateContext.Provider>
+      </WorkflowDispatchContext.Provider>
+      );
+    wrapper.find('button#link-confirm').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({ type: 'UPDATE_LINK', linkType: 'always' });
+  });
+});