masscollaborationlabs/p2pfs
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Run peer and tracker stop asynchronously.

Browse source
This commit is contained in:
Yuxin Wang 2018-10-07 12:07:18 -04:00
parent c822c7588a
commit c0fba12bdc
1 changed files with 3 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_main.py
View file

@ -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()