def test_wait_for_requests_w_mock_request(self):
mock_r1 = mock.MagicMock()
mock_r1.WaitUntilComplete.return_value = True
mock_r2 = mock.MagicMock()
mock_r2.WaitUntilComplete.return_value = True
requests = [mock_r1, mock_r2]
self.module.wait = True
under_test = ClcBlueprintPackage(self.module)
under_test._wait_for_requests_to_complete(requests)
self.assertFalse(self.module.fail_json.called)
def test_wait_for_requests_w_mock_request_fail(self):
mock_request = mock.MagicMock()
mock_request.WaitUntilComplete.return_value = True
mock_response = mock.MagicMock()
mock_response.Status.return_value = 'Failed'
mock_request.requests = [mock_response]
requests = [mock_request]
self.module.wait = True
under_test = ClcBlueprintPackage(self.module)
under_test._wait_for_requests_to_complete(requests)
self.assertTrue(self.module.fail_json.called)
