def create_subscription(
pubsub: SimplePubSub,
schema: GraphQLSchema = email_schema,
document: DocumentNode = default_subscription_ast,
):
emails: List[Email] = [
{
"from": "*****@*****.**",
"subject": "Hello",
"message": "Hello World",
"unread": False,
}
]
def transform(new_email):
emails.append(new_email)
return {"importantEmail": {"email": new_email, "inbox": data["inbox"]}}
data: Dict[str, Any] = {
"inbox": {"emails": emails},
"importantEmail": pubsub.get_subscriber(transform),
}
return subscribe(schema, document, data)
async def iterator_aclose_is_idempotent():
pubsub = SimplePubSub()
iterator = pubsub.get_subscriber()
assert iterator.listening
for n in range(3):
await iterator.aclose()
assert not iterator.listening
async def iterator_aclose_empties_pull_queue():
pubsub = SimplePubSub()
assert not pubsub.subscribers
iterator = pubsub.get_subscriber()
assert len(pubsub.subscribers) == 1
assert iterator.listening
for _n in range(3):
await iterator.__anext__()
assert iterator.push_queue.qsize() == 0
assert iterator.pull_queue.qsize() == 3
await iterator.aclose()
assert not pubsub.subscribers
assert iterator.push_queue.qsize() == 0
assert iterator.pull_queue.qsize() == 0
assert not iterator.listening
async def iterator_aclose_empties_push_queue():
pubsub = SimplePubSub()
assert not pubsub.subscribers
iterator = pubsub.get_subscriber()
assert len(pubsub.subscribers) == 1
assert iterator.listening
for value in range(3):
pubsub.emit(value)
await sleep(0)
assert iterator.push_queue.qsize() == 3
assert iterator.pull_queue.qsize() == 0
await iterator.aclose()
assert not pubsub.subscribers
assert iterator.push_queue.qsize() == 0
assert iterator.pull_queue.qsize() == 0
assert not iterator.listening
async def subscribe_async_iterator_mock():
pubsub = SimplePubSub()
iterator = pubsub.get_subscriber()
# Queue up publishes
assert pubsub.emit("Apple") is True
assert pubsub.emit("Banana") is True
# Read payloads
assert await iterator.__anext__() == "Apple"
assert await iterator.__anext__() == "Banana"
# Read ahead
i3 = await iterator.__anext__()
assert isawaitable(i3)
i4 = await iterator.__anext__()
assert isawaitable(i4)
# Publish
assert pubsub.emit("Coconut") is True
assert pubsub.emit("Durian") is True
# Await out of order to get correct results
assert await i4 == "Durian"
assert await i3 == "Coconut"
# Read ahead
i5 = iterator.__anext__()
# Terminate queue
await iterator.aclose()
# Publish is not caught after terminate
assert pubsub.emit("Fig") is False
# Find that cancelled read-ahead got a "done" result
with raises(StopAsyncIteration):
await i5
# And next returns empty completion value
with raises(StopAsyncIteration):
await iterator.__anext__()
def create_subscription(pubsub: SimplePubSub):
document = parse("""
subscription ($priority: Int = 0) {
importantEmail(priority: $priority) {
email {
from
subject
}
inbox {
unread
total
}
}
}
""")
emails: List[Email] = [{
"from": "*****@*****.**",
"subject": "Hello",
"message": "Hello World",
"unread": False,
}]
def transform(new_email):
emails.append(new_email)
return {"importantEmail": {"email": new_email, "inbox": data["inbox"]}}
data: Dict[str, Any] = {
"inbox": {
"emails": emails
},
"importantEmail": pubsub.get_subscriber(transform),
}
return subscribe(email_schema, document, data)