def test_get_response_failure(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 301 mock_response.text = "Redirect" mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with('get', url, auth=None) self.assertIn(query_runner.response_error, error)
def test_get_response_failure(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 301 mock_response.text = "Redirect" mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with(url, auth=None) self.assertIn(query_runner.response_error, error)
def test_get_response_success(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 200 mock_response.text = "Success" mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with('get', url, auth=None) self.assertEqual(response.status_code, 200) self.assertIsNone(error)
def test_get_response_success(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 200 mock_response.text = "Success" mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with(url, auth=None) self.assertEqual(response.status_code, 200) self.assertIsNone(error)
def test_get_response_success_custom_auth(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 200 mock_response.text = "Success" mock_get.return_value = mock_response url = "https://example.com/" query_runner = BaseHTTPQueryRunner({}) auth = ("username", "password") response, error = query_runner.get_response(url, auth=auth) mock_get.assert_called_once_with("get", url, auth=auth) self.assertEqual(response.status_code, 200) self.assertIsNone(error)
def test_get_response_httperror_exception(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 500 mock_response.text = "Server Error" http_error = requests.HTTPError() mock_response.raise_for_status.side_effect = http_error mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with('get', url, auth=None) self.assertIsNotNone(error) self.assertIn("Failed to execute query", error)
def test_get_response_httperror_exception(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 500 mock_response.text = "Server Error" http_error = requests.HTTPError() mock_response.raise_for_status.side_effect = http_error mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with(url, auth=None) self.assertIsNotNone(error) self.assertIn("Failed to execute query", error)
def test_get_response_requests_exception(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 500 mock_response.text = "Server Error" exception_message = "Some requests exception" requests_exception = requests.RequestException(exception_message) mock_response.raise_for_status.side_effect = requests_exception mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with('get', url, auth=None) self.assertIsNotNone(error) self.assertEqual(exception_message, error)
def test_get_response_requests_exception(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 500 mock_response.text = "Server Error" exception_message = "Some requests exception" requests_exception = requests.RequestException(exception_message) mock_response.raise_for_status.side_effect = requests_exception mock_get.return_value = mock_response url = 'https://example.com/' query_runner = BaseHTTPQueryRunner({}) response, error = query_runner.get_response(url) mock_get.assert_called_once_with(url, auth=None) self.assertIsNotNone(error) self.assertEqual(exception_message, error)