def test_request(mock_request): mock_response = mock.create_autospec(requests.Response) mock_response.status_code = 200 mock_request.return_value = mock_response client = VCloudClient(HOST, VERSION, ORG) client.auth_token = 'test-token' assert client.request( 'test-method', 'test-url', headers={'test-header': 'test'})
def test_request_failure(mock_request): mock_response = mock.create_autospec(requests.Response) mock_response.status_code = 400 mock_request.return_value = mock_response client = VCloudClient(HOST, VERSION, ORG) client.auth_token = 'test-token' with nose.tools.assert_raises(errors.VCloudAPIError): client.request('test-method', 'test-url')
def test_wait_for_task_success(mock_request): mock_task = objectify.Element('Task') mock_task.attrib['operation'] = 'test-operation' mock_task.attrib['status'] = 'success' mock_response = mock.create_autospec(requests.Response) mock_response.status_code = 200 mock_response.content = etree.tostring(mock_task) mock_request.return_value = mock_response client = VCloudClient(HOST, VERSION, ORG) client.auth_token = 'test-token' assert not client.wait_for_task('test-task-url', 2, 0)
def test_wait_for_task_timeout(mock_request): mock_task = objectify.Element('Task') mock_task.attrib['operation'] = 'test-operation' mock_task.attrib['status'] = 'running' mock_response = mock.create_autospec(requests.Response) mock_response.status_code = 200 mock_response.content = etree.tostring(mock_task) mock_request.return_value = mock_response client = VCloudClient(HOST, VERSION, ORG) client.auth_token = 'test-token' with nose.tools.assert_raises(errors.VCloudTimeoutError): client.wait_for_task('test-task-url', 2, 0)