mirror of
https://github.com/onionshare/onionshare.git
synced 2025-05-06 16:35:03 -04:00
Add onionshare CLI to cli folder, move GUI to desktop folder, and start refactoring it to work with briefcase
This commit is contained in:
Micah Lee
parent
93e90c89ae
commit
a54f99adf6
583 changed files with 14871 additions and 474 deletions
252
desktop/tests/test_gui_receive.py
Normal file
252
desktop/tests/test_gui_receive.py
Normal file
|
|
@ -0,0 +1,252 @@
|
|||
import pytest
|
||||
import os
|
||||
import requests
|
||||
import shutil
|
||||
import sys
|
||||
from datetime import datetime, timedelta
|
||||
|
||||
from PyQt5 import QtCore, QtTest
|
||||
|
||||
from .gui_base_test import GuiBaseTest
|
||||
|
||||
|
||||
class TestReceive(GuiBaseTest):
|
||||
# Shared test methods
|
||||
|
||||
def upload_file(
|
||||
self, tab, file_to_upload, expected_basename, identical_files_at_once=False
|
||||
):
|
||||
"""Test that we can upload the file"""
|
||||
|
||||
# Wait 2 seconds to make sure the filename, based on timestamp, isn't accidentally reused
|
||||
QtTest.QTest.qWait(2000)
|
||||
|
||||
files = {"file[]": open(file_to_upload, "rb")}
|
||||
url = f"http://127.0.0.1:{tab.app.port}/upload"
|
||||
if tab.settings.get("general", "public"):
|
||||
requests.post(url, files=files)
|
||||
if identical_files_at_once:
|
||||
# Send a duplicate upload to test for collisions
|
||||
requests.post(url, files=files)
|
||||
else:
|
||||
requests.post(
|
||||
url,
|
||||
files=files,
|
||||
auth=requests.auth.HTTPBasicAuth(
|
||||
"onionshare", tab.get_mode().web.password
|
||||
),
|
||||
)
|
||||
if identical_files_at_once:
|
||||
# Send a duplicate upload to test for collisions
|
||||
requests.post(
|
||||
url,
|
||||
files=files,
|
||||
auth=requests.auth.HTTPBasicAuth(
|
||||
"onionshare", tab.get_mode().web.password
|
||||
),
|
||||
)
|
||||
|
||||
QtTest.QTest.qWait(1000)
|
||||
|
||||
# Make sure the file is within the last 10 seconds worth of fileames
|
||||
exists = False
|
||||
now = datetime.now()
|
||||
for _ in range(10):
|
||||
date_dir = now.strftime("%Y-%m-%d")
|
||||
if identical_files_at_once:
|
||||
time_dir = now.strftime("%H.%M.%S-1")
|
||||
else:
|
||||
time_dir = now.strftime("%H.%M.%S")
|
||||
receive_mode_dir = os.path.join(
|
||||
tab.settings.get("receive", "data_dir"), date_dir, time_dir
|
||||
)
|
||||
expected_filename = os.path.join(receive_mode_dir, expected_basename)
|
||||
if os.path.exists(expected_filename):
|
||||
exists = True
|
||||
break
|
||||
now = now - timedelta(seconds=1)
|
||||
|
||||
self.assertTrue(exists)
|
||||
|
||||
def upload_file_should_fail(self, tab):
|
||||
"""Test that we can't upload the file when permissions are wrong, and expected content is shown"""
|
||||
QtTest.QTest.qWait(1000)
|
||||
|
||||
files = {"file[]": open(self.tmpfile_test, "rb")}
|
||||
url = f"http://127.0.0.1:{tab.app.port}/upload"
|
||||
if tab.settings.get("general", "public"):
|
||||
r = requests.post(url, files=files)
|
||||
else:
|
||||
r = requests.post(
|
||||
url,
|
||||
files=files,
|
||||
auth=requests.auth.HTTPBasicAuth(
|
||||
"onionshare", tab.get_mode().web.password
|
||||
),
|
||||
)
|
||||
|
||||
def accept_dialog():
|
||||
window = tab.common.gui.qtapp.activeWindow()
|
||||
if window:
|
||||
window.close()
|
||||
|
||||
QtCore.QTimer.singleShot(1000, accept_dialog)
|
||||
self.assertTrue("Error uploading, please inform the OnionShare user" in r.text)
|
||||
|
||||
def try_without_auth_in_non_public_mode(self, tab):
|
||||
r = requests.post(f"http://127.0.0.1:{tab.app.port}/upload")
|
||||
self.assertEqual(r.status_code, 401)
|
||||
r = requests.get(f"http://127.0.0.1:{tab.app.port}/close")
|
||||
self.assertEqual(r.status_code, 401)
|
||||
|
||||
# 'Grouped' tests follow from here
|
||||
|
||||
def run_all_receive_mode_setup_tests(self, tab):
|
||||
"""Set up a share in Receive mode and start it"""
|
||||
self.history_is_not_visible(tab)
|
||||
self.click_toggle_history(tab)
|
||||
self.history_is_visible(tab)
|
||||
self.server_working_on_start_button_pressed(tab)
|
||||
self.server_status_indicator_says_starting(tab)
|
||||
self.server_is_started(tab)
|
||||
self.web_server_is_running(tab)
|
||||
self.have_a_password(tab)
|
||||
self.url_description_shown(tab)
|
||||
self.have_copy_url_button(tab)
|
||||
self.have_show_qr_code_button(tab)
|
||||
self.server_status_indicator_says_started(tab)
|
||||
self.web_page(tab, "Select the files you want to send, then click")
|
||||
|
||||
def run_all_receive_mode_tests(self, tab):
|
||||
"""Upload files in receive mode and stop the share"""
|
||||
self.run_all_receive_mode_setup_tests(tab)
|
||||
if not tab.settings.get("general", "public"):
|
||||
self.try_without_auth_in_non_public_mode(tab)
|
||||
self.upload_file(tab, self.tmpfile_test, "test.txt")
|
||||
self.history_widgets_present(tab)
|
||||
self.counter_incremented(tab, 1)
|
||||
self.upload_file(tab, self.tmpfile_test, "test.txt")
|
||||
self.counter_incremented(tab, 2)
|
||||
self.upload_file(tab, self.tmpfile_test2, "test2.txt")
|
||||
self.counter_incremented(tab, 3)
|
||||
self.upload_file(tab, self.tmpfile_test2, "test2.txt")
|
||||
self.counter_incremented(tab, 4)
|
||||
# Test uploading the same file twice at the same time, and make sure no collisions
|
||||
self.upload_file(tab, self.tmpfile_test, "test.txt", True)
|
||||
self.counter_incremented(tab, 6)
|
||||
self.history_indicator(tab, "2")
|
||||
self.server_is_stopped(tab)
|
||||
self.web_server_is_stopped(tab)
|
||||
self.server_status_indicator_says_closed(tab)
|
||||
self.server_working_on_start_button_pressed(tab)
|
||||
self.server_is_started(tab)
|
||||
self.history_indicator(tab, "2")
|
||||
|
||||
def run_all_clear_all_button_tests(self, tab):
|
||||
"""Test the Clear All history button"""
|
||||
self.run_all_receive_mode_setup_tests(tab)
|
||||
self.upload_file(tab, self.tmpfile_test, "test.txt")
|
||||
self.history_widgets_present(tab)
|
||||
self.clear_all_history_items(tab, 0)
|
||||
self.upload_file(tab, self.tmpfile_test, "test.txt")
|
||||
self.clear_all_history_items(tab, 2)
|
||||
|
||||
def run_all_upload_non_writable_dir_tests(self, tab):
|
||||
"""Test uploading a file when the data_dir is non-writable"""
|
||||
upload_dir = os.path.join(self.tmpdir.name, "OnionShare")
|
||||
shutil.rmtree(upload_dir, ignore_errors=True)
|
||||
os.makedirs(upload_dir, 0o700)
|
||||
|
||||
# Set the upload dir setting
|
||||
tab.get_mode().data_dir_lineedit.setText(upload_dir)
|
||||
tab.settings.set("receive", "data_dir", upload_dir)
|
||||
|
||||
self.run_all_receive_mode_setup_tests(tab)
|
||||
os.chmod(upload_dir, 0o400)
|
||||
self.upload_file_should_fail(tab)
|
||||
self.server_is_stopped(tab)
|
||||
self.web_server_is_stopped(tab)
|
||||
self.server_status_indicator_says_closed(tab)
|
||||
os.chmod(upload_dir, 0o700)
|
||||
|
||||
# Tests
|
||||
|
||||
@pytest.mark.gui
|
||||
def test_clear_all_button(self):
|
||||
"""
|
||||
Clear all history items should work
|
||||
"""
|
||||
tab = self.new_receive_tab()
|
||||
|
||||
self.run_all_common_setup_tests()
|
||||
self.run_all_clear_all_button_tests(tab)
|
||||
|
||||
self.close_all_tabs()
|
||||
|
||||
@pytest.mark.gui
|
||||
def test_autostop_timer(self):
|
||||
"""
|
||||
Test autostop timer
|
||||
"""
|
||||
tab = self.new_receive_tab()
|
||||
tab.get_mode().mode_settings_widget.toggle_advanced_button.click()
|
||||
tab.get_mode().mode_settings_widget.autostop_timer_checkbox.click()
|
||||
|
||||
self.run_all_common_setup_tests()
|
||||
self.run_all_receive_mode_setup_tests(tab)
|
||||
self.set_timeout(tab, 5)
|
||||
self.autostop_timer_widget_hidden(tab)
|
||||
self.server_timed_out(tab, 15000)
|
||||
self.web_server_is_stopped(tab)
|
||||
|
||||
self.close_all_tabs()
|
||||
|
||||
@pytest.mark.gui
|
||||
def test_upload(self):
|
||||
"""
|
||||
Test uploading files
|
||||
"""
|
||||
tab = self.new_receive_tab()
|
||||
|
||||
self.run_all_common_setup_tests()
|
||||
self.run_all_receive_mode_tests(tab)
|
||||
|
||||
self.close_all_tabs()
|
||||
|
||||
@pytest.mark.gui
|
||||
@pytest.mark.skipif(sys.platform == "win32", reason="Windows doesn't have chmod")
|
||||
def test_upload_non_writable_dir(self):
|
||||
"""
|
||||
Test uploading files to a non-writable directory
|
||||
"""
|
||||
tab = self.new_receive_tab()
|
||||
|
||||
self.run_all_upload_non_writable_dir_tests(tab)
|
||||
|
||||
self.close_all_tabs()
|
||||
|
||||
@pytest.mark.gui
|
||||
def test_public_upload(self):
|
||||
"""
|
||||
Test uploading files in public mode
|
||||
"""
|
||||
tab = self.new_receive_tab()
|
||||
tab.get_mode().mode_settings_widget.public_checkbox.click()
|
||||
|
||||
self.run_all_common_setup_tests()
|
||||
self.run_all_receive_mode_tests(tab)
|
||||
|
||||
self.close_all_tabs()
|
||||
|
||||
@pytest.mark.gui
|
||||
@pytest.mark.skipif(sys.platform == "win32", reason="Windows doesn't have chmod")
|
||||
def test_public_upload_non_writable_dir(self):
|
||||
"""
|
||||
Test uploading files to a non-writable directory in public mode
|
||||
"""
|
||||
tab = self.new_receive_tab()
|
||||
tab.get_mode().mode_settings_widget.public_checkbox.click()
|
||||
|
||||
self.run_all_upload_non_writable_dir_tests(tab)
|
||||
|
||||
self.close_all_tabs()
|
||||
Loading…
Add table
Add a link
Reference in a new issue