replaced-http-statuses-with-HTTPStatus-from-http-stdlib

CHANGELOG.md

@@ -3,6 +3,7 @@
 ## 0.24 (UNRELEASED)
 
 - Added validation for directive declarations in `make_executable_schema` to prevent schema creation with undeclared directives.
+- Replaced http statuses with HTTPStatus from http stdlib
 
 
 ## 0.23 (2024-03-18)

ariadne/asgi/handlers/http.py

@@ -1,3 +1,4 @@
+from http import HTTPStatus
 import json
 from inspect import isawaitable
 from typing import Any, Optional, Type, Union, cast
@@ -164,7 +165,9 @@ async def graphql_http_server(self, request: Request) -> Response:
         try:
             data = await self.extract_data_from_request(request)
         except HttpError as error:
-            return PlainTextResponse(error.message or error.status, status_code=400)
+            return PlainTextResponse(
+                error.message or error.status, status_code=HTTPStatus.BAD_REQUEST
+            )
 
         success, result = await self.execute_graphql_query(request, data)
         return await self.create_json_response(request, result, success)
@@ -402,7 +405,7 @@ async def create_json_response(
 
         `success`: a `bool` specifying if
         """
-        status_code = 200 if success else 400
+        status_code = HTTPStatus.OK if success else HTTPStatus.BAD_REQUEST
         return JSONResponse(result, status_code=status_code)
 
     def handle_not_allowed_method(self, request: Request):
@@ -423,4 +426,4 @@ def handle_not_allowed_method(self, request: Request):
         if request.method == "OPTIONS":
             return Response(headers=allow_header)
 
-        return Response(status_code=405, headers=allow_header)
+        return Response(status_code=HTTPStatus.METHOD_NOT_ALLOWED, headers=allow_header)

ariadne/constants.py

@@ -4,7 +4,3 @@
 CONTENT_TYPE_JSON = "application/json; charset=UTF-8"
 CONTENT_TYPE_TEXT_HTML = "text/html; charset=UTF-8"
 CONTENT_TYPE_TEXT_PLAIN = "text/plain; charset=UTF-8"
-
-HTTP_STATUS_200_OK = "200 OK"
-HTTP_STATUS_400_BAD_REQUEST = "400 Bad Request"
-HTTP_STATUS_405_METHOD_NOT_ALLOWED = "405 Method Not Allowed"

ariadne/exceptions.py

@@ -1,31 +1,31 @@
+from http import HTTPStatus
 import os
 from typing import Optional, Union
 
-from .constants import HTTP_STATUS_400_BAD_REQUEST
-
 
 class HttpError(Exception):
     """Base class for HTTP errors raised inside the ASGI and WSGI servers."""
 
-    status = ""
-
-    def __init__(self, message: Optional[str] = None) -> None:
-        """Initializes the `HttpError` with optional error message.
+    def __init__(self, status: HTTPStatus, message: Optional[str] = None) -> None:
+        """Initializes the `HttpError` with a status and optional error message.
 
-        # Optional arguments
+        # Arguments
 
-        `message`: a `str` with error message to return in response body or
-        `None`.
+        `status`: HTTP status code as `HTTPStatus`.
+        `message`: Optional error message to return in the response body.
         """
         super().__init__()
+        self.status = status
         self.message = message
 
 
 class HttpBadRequestError(HttpError):
     """Raised when request did not contain the data required to execute
     the GraphQL query."""
 
-    status = HTTP_STATUS_400_BAD_REQUEST
+    def __init__(self, message: Optional[str] = None) -> None:
+        """Initializes the `HttpBadRequestError` with optional error message."""
+        super().__init__(HTTPStatus.BAD_REQUEST, message)
 
 
 class GraphQLFileSyntaxError(Exception):

ariadne/wsgi.py

@@ -2,7 +2,7 @@
 from inspect import isawaitable
 from typing import Any, Callable, Dict, List, Optional, Type, Union, cast
 from urllib.parse import parse_qsl
-
+from http import HTTPStatus
 from graphql import (
     ExecutionContext,
     GraphQLError,
@@ -16,9 +16,6 @@
     CONTENT_TYPE_TEXT_PLAIN,
     DATA_TYPE_JSON,
     DATA_TYPE_MULTIPART,
-    HTTP_STATUS_200_OK,
-    HTTP_STATUS_400_BAD_REQUEST,
-    HTTP_STATUS_405_METHOD_NOT_ALLOWED,
 )
 from .exceptions import HttpBadRequestError, HttpError
 from .explorer import Explorer, ExplorerGraphiQL
@@ -205,9 +202,7 @@
 
         `start_response`: a callable used to begin new HTTP response.
         """
-        start_response(
-            HTTP_STATUS_400_BAD_REQUEST, [("Content-Type", CONTENT_TYPE_JSON)]
-        )
+        start_response(HTTPStatus.BAD_REQUEST, [("Content-Type", CONTENT_TYPE_JSON)])
         error_json = {"errors": [{"message": error.message}]}
         return [json.dumps(error_json).encode("utf-8")]
 
@@ -319,7 +314,7 @@
         if not explorer_html:
             return self.handle_not_allowed_method(environ, start_response)
 
-        start_response(HTTP_STATUS_200_OK, [("Content-Type", CONTENT_TYPE_TEXT_HTML)])
+        start_response(HTTPStatus.OK, [("Content-Type", CONTENT_TYPE_TEXT_HTML)])
         return [cast(str, explorer_html).encode("utf-8")]
 
     def handle_post(self, environ: dict, start_response: Callable) -> List[bytes]:
@@ -558,7 +553,7 @@
         `result`: a `GraphQLResult` for this request.
         """
         success, response = result
-        status_str = HTTP_STATUS_200_OK if success else HTTP_STATUS_400_BAD_REQUEST
+        status_str = HTTPStatus.OK if success else HTTPStatus.BAD_REQUEST
         start_response(status_str, [("Content-Type", CONTENT_TYPE_JSON)])
         return [json.dumps(response).encode("utf-8")]
 
@@ -581,9 +576,9 @@
             allowed_methods.append("GET")
 
         if environ["REQUEST_METHOD"] == "OPTIONS":
-            status_str = HTTP_STATUS_200_OK
+            status_str = HTTPStatus.OK
         else:
-            status_str = HTTP_STATUS_405_METHOD_NOT_ALLOWED
+            status_str = HTTPStatus.METHOD_NOT_ALLOWED
 
         headers = [
             ("Content-Type", CONTENT_TYPE_TEXT_PLAIN),

benchmark/test_extensions.py

@@ -1,3 +1,5 @@
+from http import HTTPStatus
+
 from starlette.testclient import TestClient
 
 from ariadne.asgi import GraphQL
@@ -23,7 +25,7 @@ def api_call():
         )
 
     result = benchmark(api_call)
-    assert result.status_code == 200
+    assert result.status_code == HTTPStatus.OK
     assert not result.json().get("errors")
 
 
@@ -45,7 +47,7 @@ def api_call():
         )
 
     result = benchmark(api_call)
-    assert result.status_code == 200
+    assert result.status_code == HTTPStatus.OK
     assert not result.json().get("errors")
 
 
@@ -67,7 +69,7 @@ def api_call():
         )
 
     result = benchmark(api_call)
-    assert result.status_code == 200
+    assert result.status_code == HTTPStatus.OK
     assert not result.json().get("errors")
 
 
@@ -89,5 +91,5 @@ def api_call():
         )
 
     result = benchmark(api_call)
-    assert result.status_code == 200
+    assert result.status_code == HTTPStatus.OK
     assert not result.json().get("errors")

tests/asgi/test_configuration.py

@@ -1,4 +1,5 @@
 # pylint: disable=not-context-manager
+from http import HTTPStatus
 import time
 from datetime import timedelta
 from unittest.mock import ANY, Mock
@@ -391,15 +392,15 @@ def test_query_over_get_fails_if_variables_are_not_json_serialized(schema):
         "&operationName=Hello"
         '&variables={"name" "John"}'
     )
-    assert response.status_code == 400
+    assert response.status_code == HTTPStatus.BAD_REQUEST
     assert response.content == b"Variables query arg is not a valid JSON"
 
 
 def test_query_over_get_is_not_executed_if_not_enabled(schema):
     app = GraphQL(schema, execute_get_queries=False)
     client = TestClient(app)
     response = client.get("/?query={ status }")
-    assert response.status_code == 200
+    assert response.status_code == HTTPStatus.OK
     assert response.headers["CONTENT-TYPE"] == "text/html; charset=utf-8"
 
 

tests/asgi/test_explorer.py

@@ -1,3 +1,5 @@
+from http import HTTPStatus
+
 from starlette.testclient import TestClient
 
 from ariadne.explorer import (
@@ -13,31 +15,31 @@ def test_default_explorer_html_is_served_on_get_request(schema, snapshot):
     app = GraphQL(schema)
     client = TestClient(app)
     response = client.get("/")
-    assert response.status_code == 200
+    assert response.status_code == HTTPStatus.OK
     assert snapshot == response.text
 
 
 def test_apollo_html_is_served_on_get_request(schema, snapshot):
     app = GraphQL(schema, explorer=ExplorerApollo())
     client = TestClient(app)
     response = client.get("/")
-    assert response.status_code == 200
+    assert response.status_code == HTTPStatus.OK
     assert snapshot == response.text
 
 
 def test_graphiql_html_is_served_on_get_request(schema, snapshot):
     app = GraphQL(schema, explorer=ExplorerGraphiQL())
     client = TestClient(app)
     response = client.get("/")
-    assert response.status_code == 200
+    assert response.status_code == HTTPStatus.OK
     assert snapshot == response.text
 
 
 def test_playground_html_is_served_on_get_request(schema, snapshot):
     app = GraphQL(schema, explorer=ExplorerPlayground())
     client = TestClient(app)
     response = client.get("/")
-    assert response.status_code == 200
+    assert response.status_code == HTTPStatus.OK
     assert snapshot == response.text
 
 

tests/asgi/test_http_methods.py

@@ -1,11 +1,13 @@
+from http import HTTPStatus
+
 from starlette.testclient import TestClient
 
 from ariadne.asgi import GraphQL
 
 
 def test_options_method_is_supported(client):
     response = client.options("/")
-    assert response.status_code == 200
+    assert response.status_code == HTTPStatus.OK
     assert response.headers["Allow"] == "OPTIONS, POST, GET"
 
 
@@ -14,7 +16,7 @@ def test_options_response_excludes_get_if_introspection_is_disabled(schema):
     client = TestClient(app)
 
     response = client.options("/")
-    assert response.status_code == 200
+    assert response.status_code == HTTPStatus.OK
     assert response.headers["Allow"] == "OPTIONS, POST"