def test_client_completes_job_requested(mock_requests, mocker): mocker.patch('time.sleep') client = Client() mock_client_id = 9 read_mock_client_id(mocker, mock_client_id) with mock_requests: mock_requests.get(f'{BASE_URL}/get_client_id', json={'client_id': mock_client_id}, status_code=200) mock_requests.get(f'{BASE_URL}/get_job', json={'job': { '1': 'http://test.com' }}, status_code=200) mock_requests.put( f'{BASE_URL}/put_results', json={'success': 'The job was successfully completed'}, status_code=200) mock_requests.get( 'http://test.com', json={'data': { 'id': '1', 'attributes': { 'agencyId': 'NOAA' } }}, status_code=200) try: execute_client_task(client) except requests.exceptions.HTTPError as exception: assert False, f'Raised an exception: {exception}'
def test_client_returns_403_error_to_server(mock_requests, mocker): mocker.patch('time.sleep') client = Client() mock_client_id = 9 read_mock_client_id(mocker, mock_client_id) with mock_requests: mock_requests.get(f'{BASE_URL}/get_client_id', json={'client_id': mock_client_id}, status_code=200) mock_requests.get(f'{BASE_URL}/get_job', json={'job': { '1': 'http://test.com' }}, status_code=200) mock_requests.put( f'{BASE_URL}/put_results', json={'success': 'The job was successfully completed'}, status_code=200) regulation_response = { "errors": [{ "status": "403", "title": "The document ID could not be found." }], "error": "API limit reached." } mock_requests.get('http://test.com', json=regulation_response, status_code=403) try: execute_client_task(client) except requests.exceptions.HTTPError as exception: assert False, f'Raised an exception: {exception}' response = mock_requests.request_history[-1] assert 'errors' in response.json()
def test_client_returns_500_error_to_server(mock_requests, mocker): mocker.patch('time.sleep') client = Client() mock_client_id = 9 read_mock_client_id(mocker, mock_client_id) with mock_requests: mock_requests.get(f'{BASE_URL}/get_client_id', json={'client_id': mock_client_id}, status_code=200) mock_requests.get(f'{BASE_URL}/get_job', json={'job': { '1': 'http://test.com' }}, status_code=200) mock_requests.put( f'{BASE_URL}/put_results', json={'success': 'The job was successfully completed'}, status_code=200) regulation_response = { "errors": [{ "status": "500", "title": "INTERNAL_SERVER_ERROR", "detail": "Incorrect result size: expected 1, actual 2" }] } mock_requests.get('http://test.com', json=regulation_response, status_code=500) try: execute_client_task(client) except requests.exceptions.HTTPError as exception: assert False, f'Raised an exception: {exception}' response = mock_requests.request_history[-1] assert 'errors' in response.json()