phantom/onionshare
1
0
Fork 0
mirror of https://github.com/onionshare/onionshare.git synced 2025-01-25 10:42:58 -03:00
Code Issues Projects Releases Packages Wiki Activity Actions

Add test_clear_all_button

Browse source
This commit is contained in:
Micah Lee 2019-11-08 22:08:08 +08:00
parent f0fd0857ae
commit 1e98ccb696
No known key found for this signature in database
GPG key ID: 403C2657CD994F73
1 changed files with 49 additions and 21 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

70
tests2/test_gui_share.py
View file

@ -19,7 +19,7 @@ class TestShare(GuiBaseTest):
# Share-specific tests # Share-specific tests
def file_selection_widget_has_files(self, tab, num=2): def file_selection_widget_has_files(self, tab, num=3):
"""Test that the number of items in the list is as expected""" """Test that the number of items in the list is as expected"""
self.assertEqual( self.assertEqual(
tab.get_mode().server_status.file_selection.get_num_files(), num tab.get_mode().server_status.file_selection.get_num_files(), num
@ -61,17 +61,19 @@ class TestShare(GuiBaseTest):
def add_a_file_and_delete_using_its_delete_widget(self, tab): def add_a_file_and_delete_using_its_delete_widget(self, tab):
"""Test that we can also delete a file by clicking on its [X] widget""" """Test that we can also delete a file by clicking on its [X] widget"""
num_files = tab.get_mode().server_status.file_selection.get_num_files()
tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[0]) tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[0])
tab.get_mode().server_status.file_selection.file_list.item( tab.get_mode().server_status.file_selection.file_list.item(
0 0
).item_button.click() ).item_button.click()
self.file_selection_widget_has_files(tab, 0) self.file_selection_widget_has_files(tab, num_files)
def file_selection_widget_read_files(self, tab): def file_selection_widget_read_files(self, tab):
"""Re-add some files to the list so we can share""" """Re-add some files to the list so we can share"""
num_files = tab.get_mode().server_status.file_selection.get_num_files()
tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[0]) tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[0])
tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[1]) tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[1])
self.file_selection_widget_has_files(tab, 2) self.file_selection_widget_has_files(tab, num_files + 2)
def add_large_file(self, tab): def add_large_file(self, tab):
"""Add a large file to the share""" """Add a large file to the share"""
@ -113,7 +115,10 @@ class TestShare(GuiBaseTest):
self.assertEqual("onionshare", zip.read("test.txt").decode("utf-8")) self.assertEqual("onionshare", zip.read("test.txt").decode("utf-8"))
def individual_file_is_viewable_or_not(self, tab): def individual_file_is_viewable_or_not(self, tab):
"""Test whether an individual file is viewable (when in autostop_sharing is false) and that it isn't (when not in autostop_sharing is true)""" """
Test that an individual file is viewable (when in autostop_sharing is false) or that it
isn't (when not in autostop_sharing is true)
"""
url = f"http://127.0.0.1:{tab.app.port}" url = f"http://127.0.0.1:{tab.app.port}"
download_file_url = f"http://127.0.0.1:{tab.app.port}/test.txt" download_file_url = f"http://127.0.0.1:{tab.app.port}/test.txt"
if tab.settings.get("general", "public"): if tab.settings.get("general", "public"):
@ -126,9 +131,21 @@ class TestShare(GuiBaseTest):
), ),
) )
if not tab.settings.get("share", "autostop_sharing"): if tab.settings.get("share", "autostop_sharing"):
self.assertFalse('a href="/test.txt"' in r.text)
if tab.settings.get("general", "public"):
r = requests.get(download_file_url)
else:
r = requests.get(
download_file_url,
auth=requests.auth.HTTPBasicAuth(
"onionshare", tab.get_mode().server_status.web.password
),
)
self.assertEqual(r.status_code, 404)
self.download_share(tab)
else:
self.assertTrue('a href="test.txt"' in r.text) self.assertTrue('a href="test.txt"' in r.text)
if tab.settings.get("general", "public"): if tab.settings.get("general", "public"):
r = requests.get(download_file_url) r = requests.get(download_file_url)
else: else:
@ -145,21 +162,8 @@ class TestShare(GuiBaseTest):
with open(tmp_file.name, "r") as f: with open(tmp_file.name, "r") as f:
self.assertEqual("onionshare", f.read()) self.assertEqual("onionshare", f.read())
else:
self.assertFalse('a href="/test.txt"' in r.text)
if tab.settings.get("general", "public"):
r = requests.get(download_file_url)
else:
r = requests.get(
download_file_url,
auth=requests.auth.HTTPBasicAuth(
"onionshare", tab.get_mode().server_status.web.password
),
)
self.assertEqual(r.status_code, 404)
self.download_share(tab)
QtTest.QTest.qWait(50) QtTest.QTest.qWait(500)
def hit_401(self, tab): def hit_401(self, tab):
"""Test that the server stops after too many 401s, or doesn't when in public_mode""" """Test that the server stops after too many 401s, or doesn't when in public_mode"""
@ -227,9 +231,12 @@ class TestShare(GuiBaseTest):
def run_all_share_mode_setup_tests(self, tab): def run_all_share_mode_setup_tests(self, tab):
"""Tests in share mode prior to starting a share""" """Tests in share mode prior to starting a share"""
tab.get_mode().server_status.file_selection.file_list.add_file(
self.tmpfile_test
)
tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[0]) tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[0])
tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[1]) tab.get_mode().server_status.file_selection.file_list.add_file(self.tmpfiles[1])
self.file_selection_widget_has_files(tab, 2) self.file_selection_widget_has_files(tab, 3)
self.history_is_not_visible(tab) self.history_is_not_visible(tab)
self.click_toggle_history(tab) self.click_toggle_history(tab)
self.history_is_visible(tab) self.history_is_visible(tab)
@ -290,7 +297,17 @@ class TestShare(GuiBaseTest):
"""Test the Clear All history button""" """Test the Clear All history button"""
self.run_all_share_mode_setup_tests(tab) self.run_all_share_mode_setup_tests(tab)
self.run_all_share_mode_started_tests(tab) self.run_all_share_mode_started_tests(tab)
print(
"history items: {}".format(
len(tab.get_mode().history.item_list.items.keys())
)
)
self.individual_file_is_viewable_or_not(tab) self.individual_file_is_viewable_or_not(tab)
print(
"history items: {}".format(
len(tab.get_mode().history.item_list.items.keys())
)
)
self.history_widgets_present(tab) self.history_widgets_present(tab)
self.clear_all_history_items(tab, 0) self.clear_all_history_items(tab, 0)
self.individual_file_is_viewable_or_not(tab) self.individual_file_is_viewable_or_not(tab)
@ -427,3 +444,14 @@ class TestShare(GuiBaseTest):
self.cancel_the_share(tab) self.cancel_the_share(tab)
self.close_all_tabs() self.close_all_tabs()
@pytest.mark.gui
def test_clear_all_button(self):
"""
Test canceling a scheduled share
"""
tab = self.new_share_tab()
tab.get_mode().autostop_sharing_checkbox.click()
self.run_all_common_setup_tests()
self.run_all_clear_all_button_tests(tab)