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

Add test for reporthook.

Browse source
This commit is contained in:
Yuxin Wang 2018-10-04 00:01:22 -04:00
parent f5f3335819
commit 13ba10c595
1 changed files with 6 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

7
tests/test_main.py
View file

@ -44,10 +44,15 @@ def test_main():
assert file_list['test_small_file']['size'] == 1000
file_list = peer_2.list_file()
assert 'test_small_file' in file_list
result, msg = peer_2.download('test_small_file', 'downloaded_small_file')
def reporthook(chunk_num, chunk_size, total_size):
reporthook.value = (chunk_num, total_size)
result, msg = peer_2.download('test_small_file', 'downloaded_small_file', reporthook=reporthook)
assert result is True
assert os.path.exists('downloaded_small_file')
assert fmd5('test_small_file') == fmd5('downloaded_small_file')
assert reporthook.value == (1, 1000)
peer_2.publish('test_big_file')
file_list = tracker.file_list()