up up

.vscode/tasks.json

@@ -33,6 +33,16 @@
         "rath"
       ],
       "problemMatcher": []
-    }
+    },
+    {
+      "label": "Clean Up",
+      "type": "shell",
+      "command": "poetry run ruff --fix . && poetry run black ."
+    },
+    {
+      "label": "Check",
+      "type": "shell",
+      "command": "poetry run mypy ."
+    },
   ]
 }

rath/links/httpx.py

@@ -88,7 +88,6 @@ async def aexecute(self, operation: Operation) -> AsyncIterator[GraphQLResult]:
             post_kwargs = {"json": payload}
 
         async with httpx.AsyncClient() as client:
-
             response = await client.post(
                 self.endpoint_url, headers=operation.context.headers, **post_kwargs
             )

rath/links/shrink.py

@@ -5,7 +5,8 @@
 class ShrinkingLink(ParsingLink):
 
     """Shrinking Link is a link that shrinks operation variables.
-    It traverses the variables dict, and converts any model that has a .ashrink() method to its id."""
+    It traverses the variables dict, and converts any model that has a .ashrink() method to its id.
+    """
 
     async def aparse(self, operation: Operation) -> Operation:
         """Parses an operation

tests/test_countries_api_httpx.py

@@ -14,7 +14,6 @@ def real_world_link():
 
 @pytest.mark.public
 async def test_query_async(real_world_link):
-
     rath = Rath(link=real_world_link)
 
     async with rath:
@@ -25,7 +24,6 @@ async def test_query_async(real_world_link):
 
 @pytest.mark.public
 def test_query_sync(real_world_link):
-
     rath = Rath(link=real_world_link)
 
     with rath:
@@ -36,7 +34,6 @@ def test_query_sync(real_world_link):
 
 @pytest.mark.public
 async def test_validation(real_world_link):
-
     link = ValidatingLink(allow_introspection=True)
 
     rath = Rath(link=compose(link, real_world_link))

tests/test_dict.py

@@ -18,7 +18,6 @@ def mock_link():
 
 
 async def test_dicting(mock_link):
-
     link = DictingLink()
 
     class DictableX(BaseModel):

tests/test_sign_local.py

@@ -19,7 +19,7 @@ def is_query(op: Operation):
         return op.node.operation == OperationType.QUERY
 
     def has_token_header(op: Operation):
-        return op.context.headers.get("token") is not None
+        return op.context.headers.get("Authorization") is not None
 
     async def load_payload(operation: Operation):
         return {
@@ -30,6 +30,6 @@ async def load_payload(operation: Operation):
         ComposedSignTokenLink(
             private_key="private_key.pem", payload_retriever=load_payload
         ),
-        AssertLink(assertions=[is_query]),
+        AssertLink(assertions=[has_token_header, is_query]),
     )
     await run_basic_query(x)

tests/test_split.py

@@ -29,7 +29,6 @@ def stateful_mock_link():
 
 
 async def test_aquery(mock_link_left, mock_link_right):
-
     rath = Rath(
         link=split(
             mock_link_left,
@@ -51,7 +50,6 @@ async def test_aquery(mock_link_left, mock_link_right):
 
 
 async def test_stateful_mock(stateful_mock_link):
-
     async with Rath(link=stateful_mock_link) as rath:
         await rath.aquery(
             """
@@ -66,9 +64,7 @@ async def test_stateful_mock(stateful_mock_link):
 
 
 def test_stateful_mock_sync(stateful_mock_link):
-
     with Rath(link=stateful_mock_link) as rath:
-
         rath.query(
             """
                     query {
@@ -81,7 +77,6 @@ def test_stateful_mock_sync(stateful_mock_link):
 
 
 async def test_asubscribe(mock_link_left, mock_link_right):
-
     rath = Rath(
         link=split(
             mock_link_left,
@@ -91,7 +86,6 @@ async def test_asubscribe(mock_link_left, mock_link_right):
     )
 
     async with rath as r:
-
         async for ev in r.asubscribe(
             """
             subscription {

tests/test_transpile.py

@@ -40,7 +40,6 @@ def transpile_beasts(value, depth):
     )
 
     async with rath:
-
         x = await acreate_transpiled_beast(non_optional_parameter=1, vectors=[[2, 3]])
 
 
@@ -64,5 +63,4 @@ def transpile_beasts(value, depth):
         )
     )
     with rath:
-
         x = create_transpiled_beast(non_optional_parameter=1, vectors=[[2, 3]])

tests/test_turms.py

@@ -14,7 +14,6 @@ def mock_link():
 
 
 async def test_query_async(mock_link):
-
     rath = Rath(link=mock_link)
 
     async with rath:
@@ -24,7 +23,6 @@ async def test_query_async(mock_link):
 
 
 def test_query_sync(mock_link):
-
     rath = Rath(link=mock_link)
 
     with rath:

tests/test_validation.py

@@ -63,11 +63,9 @@ def mock_link():
 
 
 async def test_validation(mock_link):
-
     link = ValidatingLink(schema_glob=build_relative_glob("/schemas/beasts.graphql"))
 
     async with Rath(link=compose(link, mock_link)) as r:
-
         await r.aquery(
             """
             query {
@@ -80,7 +78,6 @@ async def test_validation(mock_link):
 
 
 async def test_validation_error(mock_link):
-
     link = ValidatingLink(schema_dsl=schema)
 
     rath = Rath(link=compose(link, mock_link))