From 518a25430dc2034813d82e9b946dd200e11b8b69 Mon Sep 17 00:00:00 2001
From: mabashian <mabashia@redhat.com>
Date: Fri, 27 Sep 2019 13:47:13 -0400
Subject: [PATCH] Adjust unit test after adding conditional render to modals in
 launch button component

---
 .../LaunchButton/LaunchButton.test.jsx        | 25 ++++++++-----------
 1 file changed, 10 insertions(+), 15 deletions(-)

diff --git a/awx/ui_next/src/components/LaunchButton/LaunchButton.test.jsx b/awx/ui_next/src/components/LaunchButton/LaunchButton.test.jsx
index 2259bf7e62..75a6a554f2 100644
--- a/awx/ui_next/src/components/LaunchButton/LaunchButton.test.jsx
+++ b/awx/ui_next/src/components/LaunchButton/LaunchButton.test.jsx
@@ -1,5 +1,5 @@
 import React from 'react';
-import { mountWithContexts, waitForElement } from '@testUtils/enzymeHelpers';
+import { mountWithContexts } from '@testUtils/enzymeHelpers';
 import { sleep } from '@testUtils/testUtils';
 
 import LaunchButton from './LaunchButton';
@@ -70,20 +70,15 @@ describe('LaunchButton', () => {
     const wrapper = mountWithContexts(
       <LaunchButton resource={resource}>{children}</LaunchButton>
     );
-    const button = wrapper.find('button');
-    button.prop('onClick')();
-    await waitForElement(
-      wrapper,
-      'Modal.at-c-alertModal--danger',
-      el => el.props().isOpen === true && el.props().title === 'Error!'
-    );
-    const modalCloseButton = wrapper.find('ModalBoxCloseButton');
-    modalCloseButton.simulate('click');
-    await waitForElement(
-      wrapper,
-      'Modal.at-c-alertModal--danger',
-      el => el.props().isOpen === false
-    );
+    expect(wrapper.find('Modal').length).toBe(0);
+    wrapper.find('button').prop('onClick')();
+    await sleep(0);
+    wrapper.update();
+    expect(wrapper.find('Modal').length).toBe(1);
+    wrapper.find('ModalBoxCloseButton').simulate('click');
+    await sleep(0);
+    wrapper.update();
+    expect(wrapper.find('Modal').length).toBe(0);
     done();
   });
 });