forked from goodboy/tractor
1
0
Fork 0

Parametrize with async for style tests

transport_hardening
Tyler Goodlet 2021-05-07 11:41:18 -04:00
parent 66d18be2ec
commit 7d9a551f77
1 changed files with 63 additions and 5 deletions

View File

@ -1,7 +1,8 @@
""" """
Bidirectional streaming and context API. Bidirectional streaming and context API.
"""
"""
import pytest
import trio import trio
import tractor import tractor
@ -86,7 +87,9 @@ async def simple_rpc(
data: int, data: int,
) -> None: ) -> None:
"""Test a small ping-pong server.
"""
# signal to parent that we're up # signal to parent that we're up
await ctx.started(data + 1) await ctx.started(data + 1)
@ -106,7 +109,44 @@ async def simple_rpc(
count += 1 count += 1
def test_simple_rpc(): @tractor.context
async def simple_rpc_with_forloop(
ctx: tractor.Context,
data: int,
) -> None:
"""Same as previous test but using ``async for`` syntax/api.
"""
# signal to parent that we're up
await ctx.started(data + 1)
print('opening stream in callee')
async with ctx.open_stream() as stream:
count = 0
async for msg in stream:
assert msg == 'ping'
print('pong')
await stream.send('pong')
count += 1
else:
assert count == 10
@pytest.mark.parametrize(
'use_async_for',
[True, False],
)
@pytest.mark.parametrize(
'server_func',
[simple_rpc, simple_rpc_with_forloop],
)
def test_simple_rpc(server_func, use_async_for):
"""The simplest request response pattern. """The simplest request response pattern.
""" """
@ -119,7 +159,7 @@ def test_simple_rpc():
) )
async with portal.open_context( async with portal.open_context(
simple_rpc, server_func, # taken from pytest parameterization
data=10, data=10,
) as (ctx, sent): ) as (ctx, sent):
@ -127,11 +167,29 @@ def test_simple_rpc():
async with ctx.open_stream() as stream: async with ctx.open_stream() as stream:
for _ in range(10): if use_async_for:
count = 0
# receive msgs using async for style
print('ping') print('ping')
await stream.send('ping') await stream.send('ping')
assert await stream.receive() == 'pong'
async for msg in stream:
assert msg == 'pong'
print('ping')
await stream.send('ping')
count += 1
if count >= 9:
break
else:
# classic send/receive style
for _ in range(10):
print('ping')
await stream.send('ping')
assert await stream.receive() == 'pong'
# stream should terminate here # stream should terminate here