Run peer and tracker stop asynchronously.
This commit is contained in:
parent
c822c7588a
commit
c0fba12bdc
1 changed files with 3 additions and 4 deletions
|
@ -39,7 +39,7 @@ def test_main():
|
||||||
tracker_started = loop.run_until_complete(tracker.start())
|
tracker_started = loop.run_until_complete(tracker.start())
|
||||||
# spawn 2 peers concurrently
|
# spawn 2 peers concurrently
|
||||||
peer_1_started, peer_2_started = \
|
peer_1_started, peer_2_started = \
|
||||||
loop.run_until_complete(asyncio.gather(peer_1.start(), peer_2.start(), loop=loop))
|
loop.run_until_complete(asyncio.gather(peer_1.start(), peer_2.start()))
|
||||||
assert tracker_started and peer_1_started and peer_2_started
|
assert tracker_started and peer_1_started and peer_2_started
|
||||||
|
|
||||||
# peer1 publish small file and peer2 downloads it
|
# peer1 publish small file and peer2 downloads it
|
||||||
|
@ -77,7 +77,6 @@ def test_main():
|
||||||
os.remove('test_big_file')
|
os.remove('test_big_file')
|
||||||
os.remove('downloaded_small_file')
|
os.remove('downloaded_small_file')
|
||||||
os.remove('downloaded_big_file')
|
os.remove('downloaded_big_file')
|
||||||
peer_1.stop()
|
loop.run_until_complete(asyncio.wait({peer_1.stop(), peer_2.stop()}))
|
||||||
peer_2.stop()
|
loop.run_until_complete(tracker.stop())
|
||||||
tracker.stop()
|
|
||||||
loop.close()
|
loop.close()
|
||||||
|
|
Loading…
Reference in a new issue