Fixed tests

Author: syrusakbary

src/graphql_server/__init__.py

@@ -1,5 +1,4 @@
-"""
-GraphQL-Server
+"""GraphQL-Server
 ===================
 
 GraphQL-Server is a base library that serves as a helper
@@ -10,10 +9,10 @@
 from .runtime import (
     execute,
     execute_sync,
+    introspect,
+    process_errors,
     subscribe,
     validate_document,
-    process_errors,
-    introspect,
 )
 from .version import version, version_info
 
@@ -23,12 +22,12 @@
 __version_info__ = version_info
 
 __all__ = [
-    "version",
-    "version_info",
     "execute",
     "execute_sync",
+    "introspect",
+    "process_errors",
     "subscribe",
     "validate_document",
-    "process_errors",
-    "introspect",
+    "version",
+    "version_info",
 ]

src/graphql_server/aiohttp/views.py

@@ -17,6 +17,7 @@
 
 from aiohttp import http, web
 from aiohttp.multipart import BodyPartReader
+from graphql_server.http import GraphQLRequestData
 from graphql_server.http.async_base_view import (
     AsyncBaseHTTPView,
     AsyncHTTPRequestAdapter,
@@ -28,7 +29,6 @@
     NonTextMessageReceived,
     WebSocketDisconnected,
 )
-from graphql_server.http import GraphQLRequestData
 from graphql_server.http.types import FormData, HTTPMethod, QueryParams
 from graphql_server.http.typevars import (
     Context,

src/graphql_server/channels/handlers/http_handler.py

@@ -21,11 +21,11 @@
 
 from channels.db import database_sync_to_async
 from channels.generic.http import AsyncHttpConsumer
+from graphql_server.http import GraphQLRequestData
 from graphql_server.http.async_base_view import (
     AsyncBaseHTTPView,
     AsyncHTTPRequestAdapter,
 )
-from graphql_server.http import GraphQLRequestData
 from graphql_server.http.exceptions import HTTPException
 from graphql_server.http.sync_base_view import SyncBaseHTTPView, SyncHTTPRequestAdapter
 from graphql_server.http.temporal_response import TemporalResponse

src/graphql_server/django/__init__.py

@@ -1,3 +1,3 @@
-from .views import GraphQLView, AsyncGraphQLView
+from .views import AsyncGraphQLView, GraphQLView
 
-__all__ = ["GraphQLView", "AsyncGraphQLView"]
+__all__ = ["AsyncGraphQLView", "GraphQLView"]

src/graphql_server/fastapi/router.py

@@ -29,9 +29,9 @@
 from fastapi.datastructures import Default
 from fastapi.routing import APIRoute
 from fastapi.utils import generate_unique_id
-from graphql_server.http import GraphQLRequestData
 from graphql_server.asgi import ASGIRequestAdapter, ASGIWebSocketAdapter
 from graphql_server.fastapi.context import BaseContext, CustomContext
+from graphql_server.http import GraphQLRequestData
 from graphql_server.http.async_base_view import AsyncBaseHTTPView
 from graphql_server.http.exceptions import HTTPException
 from graphql_server.http.typevars import Context, RootValue

src/graphql_server/flask/views.py

@@ -135,11 +135,7 @@ def render_graphql_ide(
     ) -> Response:
         return render_template_string(
             self.graphql_ide_html,
-            **{
-                "query": request_data.query,
-                "variables": request_data.variables,
-                "operationName": request_data.operation_name,
-            },
+            query=request_data.query, variables=request_data.variables, operationName=request_data.operation_name,
         )  # type: ignore
 
 

src/graphql_server/http/__init__.py

@@ -3,10 +3,11 @@
 import json
 import re
 from dataclasses import dataclass
-from graphql.language import DocumentNode
 from typing import TYPE_CHECKING, Any, Optional
 from typing_extensions import Literal, TypedDict
 
+from graphql.language import DocumentNode
+
 if TYPE_CHECKING:
     from graphql import ExecutionResult
 

src/graphql_server/http/async_base_view.py

@@ -16,7 +16,7 @@
 from typing_extensions import Literal, TypeGuard
 
 from graphql import ExecutionResult, GraphQLError
-from graphql.language import OperationType, DocumentNode
+from graphql.language import OperationType
 from graphql.type import GraphQLSchema
 
 from graphql_server import execute, subscribe

src/graphql_server/http/sync_base_view.py

@@ -5,9 +5,9 @@
     Any,
     Callable,
     Generic,
+    Literal,
     Optional,
     Union,
-    Literal,
 )
 
 from graphql import ExecutionResult, GraphQLError

src/graphql_server/litestar/controller.py

@@ -374,6 +374,19 @@ async def handle_http_post(
             root_value=root_value,
         )
 
+    @websocket()
+    async def websocket_endpoint(
+        self,
+        socket: WebSocket,
+        context_ws: Any,
+        root_value: Any,
+    ) -> None:
+        await self.run(
+            request=socket,
+            context=context_ws,
+            root_value=root_value,
+        )
+
     async def get_context(
         self,
         request: Union[Request[Any, Any, Any], WebSocket],