undo change to tests

strawberry-graphql · Jun 8, 2023 · eb8c340 · eb8c340
1 parent 838581b
commit eb8c340
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 9 deletions.
diff --git a/tests/schema/test_permission.py b/tests/schema/test_permission.py
@@ -76,6 +76,7 @@ async def user(self, info) -> typing.AsyncGenerator[str, None]:
     query = "subscription { user }"
 
     result = await schema.subscribe(query)
+
     assert result.errors[0].message == "You are not authorized"
 
 

diff --git a/tests/schema/test_subscription.py b/tests/schema/test_subscription.py
@@ -25,9 +25,11 @@ async def example(self) -> typing.AsyncGenerator[str, None]:
 
     query = "subscription { example }"
 
-    async for result in await schema.subscribe(query):
-        assert not result.errors
-        assert result.data["example"] == "Hi"
+    sub = await schema.subscribe(query)
+    result = await sub.__anext__()
+
+    assert not result.errors
+    assert result.data["example"] == "Hi"
 
 
 @pytest.mark.asyncio
@@ -46,9 +48,11 @@ async def example(self, name: str) -> typing.AsyncGenerator[str, None]:
 
     query = 'subscription { example(name: "Nina") }'
 
-    async for result in await schema.subscribe(query):
-        assert not result.errors
-        assert result.data["example"] == "Hi Nina"
+    sub = await schema.subscribe(query)
+    result = await sub.__anext__()
+
+    assert not result.errors
+    assert result.data["example"] == "Hi Nina"
 
 
 requires_builtin_generics = pytest.mark.skipif(
@@ -87,6 +91,8 @@ class Subscription:
 
     query = "subscription { example }"
 
-    async for result in await schema.subscribe(query):
-        assert not result.errors
-        assert result.data["example"] == "Hi"
+    sub = await schema.subscribe(query)
+    result = await sub.__anext__()
+
+    assert not result.errors
+    assert result.data["example"] == "Hi"