From 402bc839a2b8a0f5a79bd38526beab5b922c7e7e Mon Sep 17 00:00:00 2001 From: jwcolbert Date: Wed, 16 Aug 2017 19:39:56 -0500 Subject: [PATCH] updated tests to account for new argument being passed --- tests/test_clc_server_snapshot.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/tests/test_clc_server_snapshot.py b/tests/test_clc_server_snapshot.py index cc746a2..06bd05e 100644 --- a/tests/test_clc_server_snapshot.py +++ b/tests/test_clc_server_snapshot.py @@ -283,10 +283,11 @@ def test_wait_for_requests_w_mock_request(self): mock_r2 = mock.MagicMock() mock_r2.WaitUntilComplete.return_value = True requests = [mock_r1, mock_r2] + self.module.wait = True under_test = ClcSnapshot(self.module) - under_test._wait_for_requests_to_complete(requests) + under_test._wait_for_requests_to_complete(requests, mock.MagicMock()) self.assertFalse(self.module.fail_json.called) def test_wait_for_requests_w_mock_request_fail(self): @@ -299,7 +300,7 @@ def test_wait_for_requests_w_mock_request_fail(self): self.module.wait = True under_test = ClcSnapshot(self.module) - under_test._wait_for_requests_to_complete(requests) + under_test._wait_for_requests_to_complete(requests, mock.MagicMock()) self.assertTrue(self.module.fail_json.called) @patch.object(ClcSnapshot, '_get_servers_from_clc') @@ -326,7 +327,7 @@ def test_wait_for_requests_to_complete(self,mock_get_servers): server_ids = ['INVALID'] mock_get_servers.return_value=[mock.MagicMock()] under_test = ClcSnapshot(self.module) - under_test._wait_for_requests_to_complete (mock.MagicMock()) + under_test._wait_for_requests_to_complete (mock.MagicMock(), mock.MagicMock()) self.assertFalse(self.module.fail_json.called) def test_wait_for_requests_no_wait(self): @@ -336,7 +337,7 @@ def test_wait_for_requests_no_wait(self): 'wait': False } under_test = ClcSnapshot(self.module) - under_test._wait_for_requests_to_complete([mock_request]) + under_test._wait_for_requests_to_complete([mock_request], mock.MagicMock()) self.assertFalse(self.module.fail_json.called) def test_create_server_snapshot_exception(self):