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_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_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(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_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(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('get', 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(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('get', url, auth=None) self.assertIsNotNone(error) self.assertEqual(exception_message, error)
def test_get_response_generic_exception(self, mock_get): mock_response = mock.Mock() mock_response.status_code = 500 mock_response.text = "Server Error" exception_message = "Some generic exception" exception = ValueError(exception_message) mock_response.raise_for_status.side_effect = exception mock_get.return_value = mock_response url = "https://example.com/" query_runner = BaseHTTPQueryRunner({}) self.assertRaisesRegex(ValueError, exception_message, query_runner.get_response, url)
def test_get_auth_with_values(self): query_runner = BaseHTTPQueryRunner({ "username": "******", "password": "******" }) self.assertEqual(query_runner.get_auth(), ("username", "password"))
def test_get_auth_with_values(self): query_runner = BaseHTTPQueryRunner({ 'username': '******', 'password': '******' }) self.assertEqual(query_runner.get_auth(), ('username', 'password'))
def test_get_auth_empty(self): query_runner = BaseHTTPQueryRunner({}) self.assertIsNone(query_runner.get_auth())
def test_requires_authentication_default(self): self.assertFalse(BaseHTTPQueryRunner.requires_authentication) schema = BaseHTTPQueryRunner.configuration_schema() self.assertNotIn('username', schema['required']) self.assertNotIn('password', schema['required'])
def test_requires_authentication_default(self): self.assertFalse(BaseHTTPQueryRunner.requires_authentication) schema = BaseHTTPQueryRunner.configuration_schema() self.assertNotIn("username", schema["required"]) self.assertNotIn("password", schema["required"])