Applied formatting

Author: syrusakbary

src/graphql_server/asgi/__init__.py

@@ -45,7 +45,12 @@
 )
 
 if TYPE_CHECKING:
-    from collections.abc import AsyncGenerator, AsyncIterator, Mapping, Sequence  # pragma: no cover
+    from collections.abc import (  # pragma: no cover
+        AsyncGenerator,
+        AsyncIterator,
+        Mapping,
+        Sequence,
+    )
 
     from graphql.type import GraphQLSchema  # pragma: no cover
     from starlette.types import Receive, Scope, Send  # pragma: no cover
@@ -112,7 +117,9 @@ async def iter_json(
     async def send_json(self, message: Mapping[str, object]) -> None:
         try:
             await self.ws.send_text(self.view.encode_json(message))
-        except WebSocketDisconnect as exc:  # pragma: no cover - network errors mocked elsewhere
+        except (
+            WebSocketDisconnect
+        ) as exc:  # pragma: no cover - network errors mocked elsewhere
             raise WebSocketDisconnected from exc
 
     async def close(self, code: int, reason: str) -> None:

src/graphql_server/channels/testing.py

@@ -71,7 +71,9 @@ def __init__(
             subprotocols: an ordered list of preferred subprotocols to be sent to the server.
             **kwargs: additional arguments to be passed to the `WebsocketCommunicator` constructor.
         """
-        if connection_params is None:  # pragma: no cover - tested via custom initialisation
+        if (
+            connection_params is None
+        ):  # pragma: no cover - tested via custom initialisation
             connection_params = {}
         self.protocol = protocol
         subprotocols = kwargs.get("subprotocols", [])
@@ -139,7 +141,9 @@ async def subscribe(
                 },
             }
 
-            if variables is not None:  # pragma: no cover - exercised in higher-level tests
+            if (
+                variables is not None
+            ):  # pragma: no cover - exercised in higher-level tests
                 start_message["payload"]["variables"] = variables
 
             await self.send_json_to(start_message)
@@ -155,7 +159,9 @@ async def subscribe(
                     ret.errors = self.process_errors(payload.get("errors") or [])
                 ret.extensions = payload.get("extensions", None)
                 yield ret
-            elif message["type"] == "error":  # pragma: no cover - network failures untested
+            elif (
+                message["type"] == "error"
+            ):  # pragma: no cover - network failures untested
                 error_payload = message["payload"]
                 yield ExecutionResult(
                     data=None, errors=self.process_errors(error_payload)

src/tests/http/clients/chalice.py

@@ -3,7 +3,7 @@
 import urllib.parse
 from io import BytesIO
 from json import dumps
-from typing import Any, Optional, Union
+from typing import Any, Optional
 from typing_extensions import Literal
 
 from graphql import ExecutionResult
@@ -84,7 +84,7 @@ async def _graphql_request(
         headers: Optional[dict[str, str]] = None,
         extensions: Optional[dict[str, Any]] = None,
         **kwargs: Any,
-        ) -> Response:
+    ) -> Response:
         body = self._build_body(
             query=query,
             operation_name=operation_name,

src/tests/http/test_upload.py

@@ -1,4 +1,3 @@
-import contextlib
 import json
 from io import BytesIO
 

src/tests/test/test_client_utils.py

@@ -8,7 +8,9 @@
 
 class DummyClient(BaseGraphQLTestClient):
     def request(self, body, headers=None, files=None):
-        return types.SimpleNamespace(content=json.dumps(body).encode(), json=lambda: body)
+        return types.SimpleNamespace(
+            content=json.dumps(body).encode(), json=lambda: body
+        )
 
 
 def test_build_body_with_variables_and_files():

src/tests/test/test_runtime.py

@@ -1,57 +1,58 @@
 import pytest
 from graphql import (
     ExecutionResult,
+    GraphQLError,
     GraphQLField,
     GraphQLObjectType,
     GraphQLSchema,
     GraphQLString,
-    GraphQLError,
     parse,
 )
 
-import graphql_server.runtime as runtime
-
+from graphql_server import runtime
 
 schema = GraphQLSchema(
-    query=GraphQLObjectType('Query', {'hello': GraphQLField(GraphQLString)})
+    query=GraphQLObjectType("Query", {"hello": GraphQLField(GraphQLString)})
 )
 
 
 def test_validate_document_with_rules():
     from graphql.validation.rules.no_unused_fragments import NoUnusedFragmentsRule
 
-    doc = parse('query Test { hello }')
+    doc = parse("query Test { hello }")
     assert runtime.validate_document(schema, doc, (NoUnusedFragmentsRule,)) == []
 
 
 def test_get_custom_context_kwargs(monkeypatch):
-    assert runtime._get_custom_context_kwargs({'a': 1}) == {'operation_extensions': {'a': 1}}
-    monkeypatch.setattr(runtime, 'IS_GQL_33', False)
+    assert runtime._get_custom_context_kwargs({"a": 1}) == {
+        "operation_extensions": {"a": 1}
+    }
+    monkeypatch.setattr(runtime, "IS_GQL_33", False)
     try:
-        assert runtime._get_custom_context_kwargs({'a': 1}) == {}
+        assert runtime._get_custom_context_kwargs({"a": 1}) == {}
     finally:
-        monkeypatch.setattr(runtime, 'IS_GQL_33', True)
+        monkeypatch.setattr(runtime, "IS_GQL_33", True)
 
 
 def test_get_operation_type_multiple_operations():
-    doc = parse('query A{hello} query B{hello}')
+    doc = parse("query A{hello} query B{hello}")
     with pytest.raises(Exception):
         runtime._get_operation_type(doc)
 
 
 def test_parse_and_validate_document_node():
-    doc = parse('query Q { hello }')
+    doc = parse("query Q { hello }")
     res = runtime._parse_and_validate(schema, doc, None)
     assert res == doc
 
 
 def test_introspect_success_and_failure(monkeypatch):
     data = runtime.introspect(schema)
-    assert '__schema' in data
+    assert "__schema" in data
 
     def fake_execute_sync(schema, query):
-        return ExecutionResult(data=None, errors=[GraphQLError('boom')])
+        return ExecutionResult(data=None, errors=[GraphQLError("boom")])
 
-    monkeypatch.setattr(runtime, 'execute_sync', fake_execute_sync)
+    monkeypatch.setattr(runtime, "execute_sync", fake_execute_sync)
     with pytest.raises(ValueError):
         runtime.introspect(schema)

src/tests/types/test_unset.py

@@ -14,4 +14,4 @@ def test_deprecated_is_unset_and_getattr():
     with pytest.warns(DeprecationWarning):
         assert unset.is_unset(unset.UNSET)
     with pytest.raises(AttributeError):
-        getattr(unset, "missing")
+        unset.missing

src/tests/utils/test_debug.py

@@ -2,7 +2,10 @@
 
 import pytest
 
-from graphql_server.utils.debug import GraphQLJSONEncoder, pretty_print_graphql_operation
+from graphql_server.utils.debug import (
+    GraphQLJSONEncoder,
+    pretty_print_graphql_operation,
+)
 
 
 def test_graphql_json_encoder_default():

src/tests/websockets/test_graphql_transport_ws.py

@@ -423,7 +423,7 @@ async def test_duplicated_operation_ids(ws: WebSocketClient):
 
 async def test_reused_operation_ids(ws: WebSocketClient):
     """Test that an operation id can be reused after it has been
-        previously used for a completed operation.
+    previously used for a completed operation.
     """
     # Use sub1 as an id for an operation
     await ws.send_message(