Git-Mirrors/onionshare
1
0
Fork 0
mirror of https://github.com/onionshare/onionshare.git synced 2025-05-02 06:26:10 -04:00
Code Issues Projects Releases Packages Wiki Activity

Implemented test: test_share_mode_close_after_first_download

Browse source
This commit is contained in:
Micah Lee 2018-04-29 18:00:10 -07:00
parent b162104b55
commit 69b799fe80
2 changed files with 26 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

20
test/test_onionshare_web.py
View file

@ -42,6 +42,8 @@ def web_obj(common_obj, recieve_mode, num_files=0):
web = Web(common_obj, False, recieve_mode)
web.generate_slug()
web.stay_open = True
web.running = True
web.app.testing = True
# Share mode
@ -67,22 +69,38 @@ class TestWeb:
with web.app.test_client() as c:
# Load 404 pages
res = c.get('/')
res.get_data()
assert res.status_code == 404
res = c.get('/invalidslug'.format(web.slug))
res.get_data()
assert res.status_code == 404
# Load download page
res = c.get('/{}'.format(web.slug))
res.get_data()
assert res.status_code == 200
# Download
res = c.get('/{}/download'.format(web.slug))
res.get_data()
assert res.status_code == 200
assert res.mimetype == 'application/zip'
def test_share_mode_close_after_first_download(self, common_obj, temp_file_1024):
pass
web = web_obj(common_obj, False, 3)
web.stay_open = False
assert web.running == True
with web.app.test_client() as c:
# Download the first time
res = c.get('/{}/download'.format(web.slug))
res.get_data()
assert res.status_code == 200
assert res.mimetype == 'application/zip'
assert web.running == False
class TestZipWriterDefault: