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)