From f9debb8f94d349216b0686fafe53e70e8adc5fc5 Mon Sep 17 00:00:00 2001
From: mabashian <mabashia@redhat.com>
Date: Tue, 4 Feb 2020 10:43:39 -0500
Subject: [PATCH] Adds tests for delete all nodes and unsaved changes modals

---
 .../Modals/DeleteAllNodesModal.test.jsx       | 44 +++++++++++++++++++
 .../Modals/UnsavedChangesModal.jsx            |  2 +
 .../Modals/UnsavedChangesModal.test.jsx       | 42 ++++++++++++++++++
 3 files changed, 88 insertions(+)
 create mode 100644 awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/DeleteAllNodesModal.test.jsx
 create mode 100644 awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.test.jsx

diff --git a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/DeleteAllNodesModal.test.jsx b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/DeleteAllNodesModal.test.jsx
new file mode 100644
index 0000000000..5c28c06adb
--- /dev/null
+++ b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/DeleteAllNodesModal.test.jsx
@@ -0,0 +1,44 @@
+import React from 'react';
+import {
+  WorkflowDispatchContext,
+} from '@contexts/Workflow';
+import { mountWithContexts } from '@testUtils/enzymeHelpers';
+import DeleteAllNodesModal from './DeleteAllNodesModal';
+
+let wrapper;
+const dispatch = jest.fn();
+
+describe('DeleteAllNodesModal', () => {
+  beforeAll(() => {
+    wrapper = mountWithContexts(
+      <WorkflowDispatchContext.Provider value={dispatch}>
+        <DeleteAllNodesModal />
+      </WorkflowDispatchContext.Provider>
+    );
+  });
+
+  afterAll(() => {
+    wrapper.unmount();
+  });
+
+  test('Delete All button dispatches as expected', () => {
+    wrapper.find('button#confirm-delete-all-nodes').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({
+      type: 'DELETE_ALL_NODES',
+    });
+  });
+
+  test('Cancel button dispatches as expected', () => {
+    wrapper.find('button#cancel-delete-all-nodes').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({
+      type: 'TOGGLE_DELETE_ALL_NODES_MODAL',
+    });
+  });
+
+  test('Close button dispatches as expected', () => {
+    wrapper.find('TimesIcon').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({
+      type: 'TOGGLE_DELETE_ALL_NODES_MODAL',
+    });
+  });
+});
diff --git a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.jsx b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.jsx
index 7a72289302..e594c7a570 100644
--- a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.jsx
+++ b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.jsx
@@ -15,6 +15,7 @@ function UnsavedChangesModal({ i18n, onSaveAndExit, onExit }) {
       onClose={() => dispatch({ type: 'TOGGLE_UNSAVED_CHANGES_MODAL' })}
       actions={[
         <Button
+          id="confirm-exit-without-saving"
           key="exit"
           variant="danger"
           aria-label={i18n._(t`Exit Without Saving`)}
@@ -23,6 +24,7 @@ function UnsavedChangesModal({ i18n, onSaveAndExit, onExit }) {
           {i18n._(t`Exit Without Saving`)}
         </Button>,
         <Button
+          id="confirm-save-and-exit"
           key="save"
           variant="primary"
           aria-label={i18n._(t`Save & Exit`)}
diff --git a/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.test.jsx b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.test.jsx
new file mode 100644
index 0000000000..850ef98e8e
--- /dev/null
+++ b/awx/ui_next/src/screens/Template/WorkflowJobTemplateVisualizer/Modals/UnsavedChangesModal.test.jsx
@@ -0,0 +1,42 @@
+import React from 'react';
+import {
+  WorkflowDispatchContext,
+} from '@contexts/Workflow';
+import { mountWithContexts } from '@testUtils/enzymeHelpers';
+import UnsavedChangesModal from './UnsavedChangesModal';
+
+let wrapper;
+const dispatch = jest.fn();
+const onSaveAndExit = jest.fn();
+const onExit = jest.fn();
+
+describe('UnsavedChangesModal', () => {
+  beforeAll(() => {
+    wrapper = mountWithContexts(
+      <WorkflowDispatchContext.Provider value={dispatch}>
+        <UnsavedChangesModal onSaveAndExit={onSaveAndExit} onExit={onExit} />
+      </WorkflowDispatchContext.Provider>
+    );
+  });
+
+  afterAll(() => {
+    wrapper.unmount();
+  });
+
+  test('Exit Without Saving button dispatches as expected', () => {
+    wrapper.find('button#confirm-exit-without-saving').simulate('click');
+    expect(onExit).toHaveBeenCalled();
+  });
+
+  test('Save and Exit button dispatches as expected', () => {
+    wrapper.find('button#confirm-save-and-exit').simulate('click');
+    expect(onSaveAndExit).toHaveBeenCalled();
+  });
+
+  test('Close button dispatches as expected', () => {
+    wrapper.find('TimesIcon').simulate('click');
+    expect(dispatch).toHaveBeenCalledWith({
+      type: 'TOGGLE_UNSAVED_CHANGES_MODAL',
+    });
+  });
+});