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)
