def test_eof():
peer = BitcoinPeerProtocol(MAGIC_HEADER)
pt = PeerTransport(None)
peer.connection_made(pt)
COUNT = 0
@asyncio.coroutine
def async_listen(next_message):
count = 0
try:
while True:
name, data = yield from next_message()
count += 1
except EOFError:
pass
assert count == 2
nonlocal COUNT
COUNT += 1
tasks = [
asyncio.Task(async_listen(peer.new_get_next_message_f()))
for i in range(50)
]
peer.data_received(VERSION_MSG_BIN)
peer.data_received(VERACK_MSG_BIN)
# end of stream
peer.connection_lost(None)
# give everyone a chance to run (but no one finishes)
asyncio.get_event_loop().run_until_complete(asyncio.wait(tasks))
assert COUNT == 50
def test_eof():
peer = BitcoinPeerProtocol(MAGIC_HEADER)
pt = PeerTransport(None)
peer.connection_made(pt)
COUNT = 0
@asyncio.coroutine
def async_listen(next_message):
count = 0
try:
while True:
name, data = yield from next_message()
count += 1
except EOFError:
pass
assert count == 2
nonlocal COUNT
COUNT += 1
tasks = [asyncio.Task(async_listen(peer.new_get_next_message_f())) for i in range(50)]
peer.data_received(VERSION_MSG_BIN)
peer.data_received(VERACK_MSG_BIN)
# end of stream
peer.connection_lost(None)
# give everyone a chance to run (but no one finishes)
asyncio.get_event_loop().run_until_complete(asyncio.wait(tasks))
assert COUNT == 50
