async def test_merger_active_connection_prevents_ending(
outside_source_stream, mock_merge_strategy, mock_stream_builder,
mock_connection_streams, mock_connections, event_loop):
connection = mock_connections()
connection_stream = mock_connection_streams()
mock_stream_builder.side_effect = [connection_stream]
canonical_stream = outside_source_stream
async def long_read():
await asyncio.sleep(0.05)
return b""
connection_stream.read.side_effect = long_read
connection_stream.ended.side_effect = lambda: connection_stream.read.called
merger = Merger(mock_stream_builder, 0.01, mock_merge_strategy,
canonical_stream)
f = asyncio.ensure_future(merger.wait_for_ended())
h = asyncio.ensure_future(merger.handle_connection(connection))
await asyncio.sleep(0.02)
exhaust_callbacks(event_loop)
assert not f.done()
await h
await asyncio.sleep(0.02)
exhaust_callbacks(event_loop)
assert f.done()
async def test_merger_no_connection_wait_active_connection(
outside_source_stream, mock_merge_strategy, mock_reader_builder,
mock_delayed_stream_builder, mock_readers, mock_connections,
event_loop):
connection = mock_connections()
reader = mock_readers()
mock_reader_builder.side_effect = [reader]
canonical_stream = outside_source_stream
async def long_read():
await asyncio.sleep(0.05)
return
reader.read.side_effect = long_read
merger = Merger(mock_reader_builder, mock_delayed_stream_builder,
mock_merge_strategy, canonical_stream)
f = asyncio.ensure_future(merger.no_connections_for(0.01))
h = asyncio.ensure_future(merger.handle_connection(connection))
await asyncio.sleep(0.02)
exhaust_callbacks(event_loop)
assert not f.done()
await h
exhaust_callbacks(event_loop)
assert not f.done()
await asyncio.sleep(0.02)
exhaust_callbacks(event_loop)
assert f.done()
await f
merger.stop_accepting_connections()
await merger.wait_for_ended()