graphql-python/graphene
1
1
Fork 0
mirror of https://github.com/graphql-python/graphene.git synced 2025-09-21 11:22:33 +03:00
Code Issues Packages Projects Releases Wiki Activity

Checked with flake8

Browse Source
This commit is contained in:
Rob Blackbourn 2020-01-08 14:09:36 +00:00
parent cf365cc845
commit f4f9af7490
1 changed files with 8 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tests_asyncio/test_subscribe.py
View File

@ -1,13 +1,14 @@
from pytest import mark from pytest import mark
from datetime import datetime
from graphene import ObjectType, Int, String, Schema, Field from graphene import ObjectType, Int, String, Schema, Field
class Query(ObjectType): class Query(ObjectType):
hello = String() hello = String()
def resolve_hello(root, info): def resolve_hello(root, info):
return 'Hello, world!' return "Hello, world!"
class Subscription(ObjectType): class Subscription(ObjectType):
count_to_ten = Field(Int) count_to_ten = Field(Int)
@ -16,15 +17,17 @@ class Subscription(ObjectType):
count = 0 count = 0
while count < 10: while count < 10:
count += 1 count += 1
yield { 'count_to_ten': count } yield {"count_to_ten": count}
schema = Schema(query=Query, subscription=Subscription) schema = Schema(query=Query, subscription=Subscription)
@mark.asyncio @mark.asyncio
async def test_subscription(): async def test_subscription():
subscription = 'subscription { countToTen }' subscription = "subscription { countToTen }"
result = await schema.subscribe(subscription) result = await schema.subscribe(subscription)
count = 0 count = 0
async for item in result: async for item in result:
count = item.data['countToTen'] count = item.data["countToTen"]
assert count == 10 assert count == 10