Add tests against the daemon for close/reset

This commit is contained in:
mhchia
2019-09-10 17:51:39 +08:00
parent bb0da41eda
commit df87f5adb9
4 changed files with 156 additions and 1 deletions

View File

@ -85,6 +85,17 @@ async def test_net_stream_read_after_remote_reset(net_stream_pair):
await stream_1.read(MAX_READ_LEN)
@pytest.mark.asyncio
async def test_net_stream_read_after_remote_closed_and_reset(net_stream_pair):
stream_0, stream_1 = net_stream_pair
await stream_0.write(DATA)
await stream_0.close()
await stream_0.reset()
# Sleep to let `stream_1` receive the message.
await asyncio.sleep(0.01)
assert (await stream_1.read(MAX_READ_LEN)) == DATA
@pytest.mark.asyncio
async def test_net_stream_write_after_local_closed(net_stream_pair):
stream_0, stream_1 = net_stream_pair