mirror of
https://github.com/onionshare/onionshare.git
synced 2024-10-01 01:35:40 -04:00
71 lines
3.0 KiB
Python
71 lines
3.0 KiB
Python
import requests
|
|
import socks
|
|
import zipfile
|
|
from PyQt5 import QtCore, QtTest
|
|
from .TorGuiBaseTest import TorGuiBaseTest
|
|
from .GuiShareTest import GuiShareTest
|
|
|
|
class TorGuiShareTest(TorGuiBaseTest, GuiShareTest):
|
|
def download_share(self, public_mode):
|
|
# Set up connecting to the onion
|
|
(socks_address, socks_port) = self.gui.app.onion.get_tor_socks_port()
|
|
session = requests.session()
|
|
session.proxies = {}
|
|
session.proxies['http'] = 'socks5h://{}:{}'.format(socks_address, socks_port)
|
|
|
|
# Download files
|
|
if public_mode:
|
|
path = "http://{}/download".format(self.gui.app.onion_host)
|
|
else:
|
|
path = "http://{}/{}/download".format(self.gui.app.onion_host, self.gui.share_mode.web.slug)
|
|
response = session.get(path, stream=True)
|
|
QtTest.QTest.qWait(4000)
|
|
|
|
if response.status_code == 200:
|
|
with open('/tmp/download.zip', 'wb') as file_to_write:
|
|
for chunk in response.iter_content(chunk_size=128):
|
|
file_to_write.write(chunk)
|
|
file_to_write.close()
|
|
zip = zipfile.ZipFile('/tmp/download.zip')
|
|
QtTest.QTest.qWait(4000)
|
|
self.assertEquals('onionshare', zip.read('test.txt').decode('utf-8'))
|
|
|
|
# Persistence tests
|
|
def have_same_onion(self, onion):
|
|
'''Test that we have the same onion'''
|
|
self.assertEqual(self.gui.app.onion_host, onion)
|
|
|
|
def run_all_share_mode_started_tests(self, public_mode):
|
|
"""Tests in share mode after starting a share"""
|
|
self.server_working_on_start_button_pressed(self.gui.share_mode)
|
|
self.server_status_indicator_says_starting(self.gui.share_mode)
|
|
self.add_delete_buttons_hidden()
|
|
self.settings_button_is_hidden()
|
|
self.a_server_is_started(self.gui.share_mode)
|
|
self.a_web_server_is_running()
|
|
self.have_an_onion_service()
|
|
self.have_a_slug(self.gui.share_mode, public_mode)
|
|
self.url_description_shown(self.gui.share_mode)
|
|
self.have_copy_url_button(self.gui.share_mode)
|
|
self.server_status_indicator_says_started(self.gui.share_mode)
|
|
|
|
def run_all_share_mode_persistent_tests(self, public_mode, stay_open):
|
|
"""Same as end-to-end share tests but also test the slug is the same on multiple shared"""
|
|
self.run_all_share_mode_setup_tests()
|
|
self.run_all_share_mode_started_tests(public_mode)
|
|
slug = self.gui.share_mode.server_status.web.slug
|
|
onion = self.gui.app.onion_host
|
|
self.run_all_share_mode_download_tests(public_mode, stay_open)
|
|
self.have_same_onion(onion)
|
|
self.have_same_slug(slug)
|
|
|
|
def run_all_share_mode_timer_tests(self, public_mode):
|
|
"""Auto-stop timer tests in share mode"""
|
|
self.run_all_share_mode_setup_tests()
|
|
self.set_timeout(self.gui.share_mode, 120)
|
|
self.run_all_share_mode_started_tests(public_mode)
|
|
self.timeout_widget_hidden(self.gui.share_mode)
|
|
self.server_timed_out(self.gui.share_mode, 125000)
|
|
self.web_service_is_stopped()
|
|
|