def test_wait_for_request_w_mock_request(self): mock_r1 = mock.MagicMock() mock_r1.WaitUntilComplete.return_value = True self.module.wait = True under_test = ClcPublicIp(self.module) under_test._wait_for_request_to_complete(mock_r1) self.assertFalse(self.module.fail_json.called)
def test_wait_for_request_no_wait(self): mock_request = mock.MagicMock() mock_request.WaitUntilComplete.return_value = True self.module.params = { 'wait': False } under_test = ClcPublicIp(self.module) under_test._wait_for_request_to_complete(mock_request) self.assertFalse(self.module.fail_json.called)
def test_wait_for_request_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] self.module.wait = True under_test = ClcPublicIp(self.module) under_test._wait_for_request_to_complete(mock_request) self.assertTrue(self.module.fail_json.called)