mirror of
https://github.com/onionshare/onionshare.git
synced 2024-10-01 01:35:40 -04:00
Begin refactoring Tor tests to inherit from local tests
This commit is contained in:
Micah Lee
parent
4710eaee4c
commit
d8c225a9c6
@ -0,0 +1,8 @@
|
||||
from .onionshare_receive_mode_upload_test_public_mode import OnionShareGuiTest as ReceiveMoveUploadTestPublicMode
|
||||
from .onionshare_receive_mode_upload_test import OnionShareGuiTest as ReceiveModeUploadTest
|
||||
from .onionshare_share_mode_download_test_public_mode import OnionShareGuiTest as ShareModeDownloadTestPublicMode
|
||||
from .onionshare_share_mode_download_test import OnionShareGuiTest as ShareModeDownloadTest
|
||||
from .onionshare_share_mode_download_test_stay_open import OnionShareGuiTest as ShareModeDownloadTestStayOpen
|
||||
from .onionshare_slug_persistent_test import OnionShareGuiTest as SlugPersistentTest
|
||||
from .onionshare_timer_test import OnionShareGuiTest as TimerTest
|
||||
from .commontests import CommonTests
|
||||
@ -7,207 +7,13 @@ import zipfile
|
||||
from PyQt5 import QtCore, QtTest
|
||||
from onionshare import strings
|
||||
|
||||
class CommonTests(object):
|
||||
def test_gui_loaded(self):
|
||||
'''Test that the GUI actually is shown'''
|
||||
self.assertTrue(self.gui.show)
|
||||
|
||||
def test_windowTitle_seen(self):
|
||||
'''Test that the window title is OnionShare'''
|
||||
self.assertEqual(self.gui.windowTitle(), 'OnionShare')
|
||||
|
||||
def test_settings_button_is_visible(self):
|
||||
'''Test that the settings button is visible'''
|
||||
self.assertTrue(self.gui.settings_button.isVisible())
|
||||
|
||||
def test_server_status_bar_is_visible(self):
|
||||
'''Test that the status bar is visible'''
|
||||
self.assertTrue(self.gui.status_bar.isVisible())
|
||||
|
||||
def test_info_widget_is_not_visible(self, mode):
|
||||
'''Test that the info widget along top of screen is not shown'''
|
||||
if mode == 'receive':
|
||||
self.assertFalse(self.gui.receive_mode.info_widget.isVisible())
|
||||
if mode == 'share':
|
||||
self.assertFalse(self.gui.share_mode.info_widget.isVisible())
|
||||
|
||||
def test_info_widget_is_visible(self, mode):
|
||||
'''Test that the info widget along top of screen is shown'''
|
||||
if mode == 'receive':
|
||||
self.assertTrue(self.gui.receive_mode.info_widget.isVisible())
|
||||
if mode == 'share':
|
||||
self.assertTrue(self.gui.share_mode.info_widget.isVisible())
|
||||
|
||||
def test_click_mode(self, mode):
|
||||
'''Test that we can switch Mode by clicking the button'''
|
||||
if mode == 'receive':
|
||||
QtTest.QTest.mouseClick(self.gui.receive_mode_button, QtCore.Qt.LeftButton)
|
||||
self.assertTrue(self.gui.mode, self.gui.MODE_RECEIVE)
|
||||
if mode == 'share':
|
||||
QtTest.QTest.mouseClick(self.gui.share_mode_button, QtCore.Qt.LeftButton)
|
||||
self.assertTrue(self.gui.mode, self.gui.MODE_SHARE)
|
||||
|
||||
def test_history_is_visible(self, mode):
|
||||
'''Test that the History section is visible and that the relevant widget is present'''
|
||||
if mode == 'receive':
|
||||
self.assertTrue(self.gui.receive_mode.uploads.isVisible())
|
||||
self.assertTrue(self.gui.receive_mode.uploads.no_uploads_label.isVisible())
|
||||
if mode == 'share':
|
||||
self.assertTrue(self.gui.share_mode.downloads.isVisible())
|
||||
self.assertTrue(self.gui.share_mode.downloads.no_downloads_label.isVisible())
|
||||
|
||||
def test_server_working_on_start_button_pressed(self, mode):
|
||||
'''Test we can start the service'''
|
||||
# Should be in SERVER_WORKING state
|
||||
if mode == 'receive':
|
||||
QtTest.QTest.mouseClick(self.gui.receive_mode.server_status.server_button, QtCore.Qt.LeftButton)
|
||||
self.assertEqual(self.gui.receive_mode.server_status.status, 1)
|
||||
if mode == 'share':
|
||||
QtTest.QTest.mouseClick(self.gui.share_mode.server_status.server_button, QtCore.Qt.LeftButton)
|
||||
self.assertEqual(self.gui.share_mode.server_status.status, 1)
|
||||
|
||||
def test_server_status_indicator_says_starting(self, mode):
|
||||
'''Test that the Server Status indicator shows we are Starting'''
|
||||
if mode == 'receive':
|
||||
self.assertEquals(self.gui.receive_mode.server_status_label.text(), strings._('gui_status_indicator_share_working', True))
|
||||
if mode == 'share':
|
||||
self.assertEquals(self.gui.share_mode.server_status_label.text(), strings._('gui_status_indicator_share_working', True))
|
||||
|
||||
def test_settings_button_is_hidden(self):
|
||||
'''Test that the settings button is hidden when the server starts'''
|
||||
self.assertFalse(self.gui.settings_button.isVisible())
|
||||
|
||||
def test_a_server_is_started(self, mode):
|
||||
'''Test that the server has started'''
|
||||
QtTest.QTest.qWait(45000)
|
||||
# Should now be in SERVER_STARTED state
|
||||
if mode == 'receive':
|
||||
self.assertEqual(self.gui.receive_mode.server_status.status, 2)
|
||||
if mode == 'share':
|
||||
self.assertEqual(self.gui.share_mode.server_status.status, 2)
|
||||
|
||||
def test_a_web_server_is_running(self):
|
||||
'''Test that the web server has started'''
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
|
||||
self.assertEqual(sock.connect_ex(('127.0.0.1',self.gui.app.port)), 0)
|
||||
|
||||
def test_have_a_slug(self, mode, public_mode):
|
||||
'''Test that we have a valid slug'''
|
||||
if mode == 'receive':
|
||||
if not public_mode:
|
||||
self.assertRegex(self.gui.receive_mode.server_status.web.slug, r'(\w+)-(\w+)')
|
||||
else:
|
||||
self.assertIsNone(self.gui.receive_mode.server_status.web.slug, r'(\w+)-(\w+)')
|
||||
if mode == 'share':
|
||||
if not public_mode:
|
||||
self.assertRegex(self.gui.share_mode.server_status.web.slug, r'(\w+)-(\w+)')
|
||||
else:
|
||||
self.assertIsNone(self.gui.share_mode.server_status.web.slug, r'(\w+)-(\w+)')
|
||||
from tests_gui_local import CommonTests as LocalCommonTests
|
||||
|
||||
class CommonTests(LocalCommonTests):
|
||||
def test_have_an_onion_service(self):
|
||||
'''Test that we have a valid Onion URL'''
|
||||
self.assertRegex(self.gui.app.onion_host, r'[a-z2-7].onion')
|
||||
|
||||
def test_url_description_shown(self, mode):
|
||||
'''Test that the URL label is showing'''
|
||||
if mode == 'receive':
|
||||
self.assertTrue(self.gui.receive_mode.server_status.url_description.isVisible())
|
||||
if mode == 'share':
|
||||
self.assertTrue(self.gui.share_mode.server_status.url_description.isVisible())
|
||||
|
||||
def test_have_copy_url_button(self, mode):
|
||||
'''Test that the Copy URL button is shown'''
|
||||
if mode == 'receive':
|
||||
self.assertTrue(self.gui.receive_mode.server_status.copy_url_button.isVisible())
|
||||
if mode == 'share':
|
||||
self.assertTrue(self.gui.share_mode.server_status.copy_url_button.isVisible())
|
||||
|
||||
def test_server_status_indicator_says_started(self, mode):
|
||||
'''Test that the Server Status indicator shows we are started'''
|
||||
if mode == 'receive':
|
||||
self.assertEquals(self.gui.receive_mode.server_status_label.text(), strings._('gui_status_indicator_receive_started', True))
|
||||
if mode == 'share':
|
||||
self.assertEquals(self.gui.receive_mode.server_status_label.text(), strings._('gui_status_indicator_share_started', True))
|
||||
|
||||
def test_web_page(self, mode, string, public_mode):
|
||||
'''Test that the web page contains a string'''
|
||||
(socks_address, socks_port) = self.gui.app.onion.get_tor_socks_port()
|
||||
socks.set_default_proxy(socks.SOCKS5, socks_address, socks_port)
|
||||
|
||||
s = socks.socksocket()
|
||||
s.settimeout(60)
|
||||
s.connect((self.gui.app.onion_host, 80))
|
||||
|
||||
if not public_mode:
|
||||
if mode == 'receive':
|
||||
path = '/{}'.format(self.gui.receive_mode.server_status.web.slug)
|
||||
if mode == 'share':
|
||||
path = '/{}'.format(self.gui.share_mode.server_status.web.slug)
|
||||
else:
|
||||
path = '/'
|
||||
|
||||
http_request = 'GET {} HTTP/1.0\r\n'.format(path)
|
||||
http_request += 'Host: {}\r\n'.format(self.gui.app.onion_host)
|
||||
http_request += '\r\n'
|
||||
s.sendall(http_request.encode('utf-8'))
|
||||
|
||||
with open('/tmp/webpage', 'wb') as file_to_write:
|
||||
while True:
|
||||
data = s.recv(1024)
|
||||
if not data:
|
||||
break
|
||||
file_to_write.write(data)
|
||||
file_to_write.close()
|
||||
|
||||
f = open('/tmp/webpage')
|
||||
self.assertTrue(string in f.read())
|
||||
f.close()
|
||||
|
||||
def test_history_widgets_present(self, mode):
|
||||
'''Test that the relevant widgets are present in the history view after activity has taken place'''
|
||||
if mode == 'receive':
|
||||
self.assertFalse(self.gui.receive_mode.uploads.no_uploads_label.isVisible())
|
||||
self.assertTrue(self.gui.receive_mode.uploads.clear_history_button.isVisible())
|
||||
if mode == 'share':
|
||||
self.assertFalse(self.gui.share_mode.downloads.no_downloads_label.isVisible())
|
||||
self.assertTrue(self.gui.share_mode.downloads.clear_history_button.isVisible())
|
||||
|
||||
def test_counter_incremented(self, mode, count):
|
||||
'''Test that the counter has incremented'''
|
||||
if mode == 'receive':
|
||||
self.assertEquals(self.gui.receive_mode.uploads_completed, count)
|
||||
if mode == 'share':
|
||||
self.assertEquals(self.gui.share_mode.downloads_completed, count)
|
||||
|
||||
def test_server_is_stopped(self, mode, stay_open):
|
||||
'''Test that the server stops when we click Stop'''
|
||||
if mode == 'receive':
|
||||
QtTest.QTest.mouseClick(self.gui.receive_mode.server_status.server_button, QtCore.Qt.LeftButton)
|
||||
self.assertEquals(self.gui.receive_mode.server_status.status, 0)
|
||||
if mode == 'share':
|
||||
if stay_open:
|
||||
QtTest.QTest.mouseClick(self.gui.share_mode.server_status.server_button, QtCore.Qt.LeftButton)
|
||||
self.assertEquals(self.gui.share_mode.server_status.status, 0)
|
||||
|
||||
def test_web_service_is_stopped(self):
|
||||
'''Test that the web server also stopped'''
|
||||
QtTest.QTest.qWait(2000)
|
||||
sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
|
||||
|
||||
# We should be closed by now. Fail if not!
|
||||
self.assertNotEqual(sock.connect_ex(('127.0.0.1',self.gui.app.port)), 0)
|
||||
|
||||
def test_server_status_indicator_says_closed(self, mode, stay_open):
|
||||
'''Test that the Server Status indicator shows we closed'''
|
||||
if mode == 'receive':
|
||||
self.assertEquals(self.gui.receive_mode.server_status_label.text(), strings._('gui_status_indicator_receive_stopped', True))
|
||||
if mode == 'share':
|
||||
if stay_open:
|
||||
self.assertEquals(self.gui.share_mode.server_status_label.text(), strings._('gui_status_indicator_share_stopped', True))
|
||||
else:
|
||||
self.assertEquals(self.gui.share_mode.server_status_label.text(), strings._('closing_automatically', True))
|
||||
|
||||
def test_cancel_the_share(self, mode):
|
||||
'''Test that we can cancel this share before it's started up '''
|
||||
if mode == 'share':
|
||||
@ -222,119 +28,6 @@ class CommonTests(object):
|
||||
QtTest.QTest.mouseRelease(self.gui.receive_mode.server_status.server_button, QtCore.Qt.LeftButton)
|
||||
self.assertEqual(self.gui.receive_mode.server_status.status, 0)
|
||||
|
||||
|
||||
# Auto-stop timer tests
|
||||
def test_set_timeout(self, mode, timeout):
|
||||
'''Test that the timeout can be set'''
|
||||
timer = QtCore.QDateTime.currentDateTime().addSecs(timeout)
|
||||
if mode == 'receive':
|
||||
self.gui.receive_mode.server_status.shutdown_timeout.setDateTime(timer)
|
||||
self.assertTrue(self.gui.receive_mode.server_status.shutdown_timeout.dateTime(), timer)
|
||||
if mode == 'share':
|
||||
self.gui.share_mode.server_status.shutdown_timeout.setDateTime(timer)
|
||||
self.assertTrue(self.gui.share_mode.server_status.shutdown_timeout.dateTime(), timer)
|
||||
|
||||
def test_timeout_widget_hidden(self, mode):
|
||||
'''Test that the timeout widget is hidden when share has started'''
|
||||
if mode == 'receive':
|
||||
self.assertFalse(self.gui.receive_mode.server_status.shutdown_timeout_container.isVisible())
|
||||
if mode == 'share':
|
||||
self.assertFalse(self.gui.share_mode.server_status.shutdown_timeout_container.isVisible())
|
||||
|
||||
def test_server_timed_out(self, mode, wait):
|
||||
'''Test that the server has timed out after the timer ran out'''
|
||||
QtTest.QTest.qWait(wait)
|
||||
# We should have timed out now
|
||||
if mode == 'receive':
|
||||
self.assertEqual(self.gui.receive_mode.server_status.status, 0)
|
||||
if mode == 'share':
|
||||
self.assertEqual(self.gui.share_mode.server_status.status, 0)
|
||||
|
||||
# Receive-specific tests
|
||||
def test_upload_file(self, public_mode, expected_file):
|
||||
'''Test that we can upload the file'''
|
||||
(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)
|
||||
|
||||
files = {'file[]': open('/tmp/test.txt', 'rb')}
|
||||
if not public_mode:
|
||||
path = 'http://{}/{}/upload'.format(self.gui.app.onion_host, self.gui.receive_mode.web.slug)
|
||||
else:
|
||||
path = 'http://{}/upload'.format(self.gui.app.onion_host)
|
||||
response = session.post(path, files=files)
|
||||
QtTest.QTest.qWait(4000)
|
||||
self.assertTrue(os.path.isfile(expected_file))
|
||||
|
||||
# Share-specific tests
|
||||
def test_file_selection_widget_has_a_file(self):
|
||||
'''Test that the number of files in the list is 1'''
|
||||
self.assertEqual(self.gui.share_mode.server_status.file_selection.get_num_files(), 1)
|
||||
|
||||
def test_deleting_only_file_hides_delete_button(self):
|
||||
'''Test that clicking on the file item shows the delete button. Test that deleting the only item in the list hides the delete button'''
|
||||
rect = self.gui.share_mode.server_status.file_selection.file_list.visualItemRect(self.gui.share_mode.server_status.file_selection.file_list.item(0))
|
||||
QtTest.QTest.mouseClick(self.gui.share_mode.server_status.file_selection.file_list.viewport(), QtCore.Qt.LeftButton, pos=rect.center())
|
||||
# Delete button should be visible
|
||||
self.assertTrue(self.gui.share_mode.server_status.file_selection.delete_button.isVisible())
|
||||
# Click delete, and since there's no more files, the delete button should be hidden
|
||||
QtTest.QTest.mouseClick(self.gui.share_mode.server_status.file_selection.delete_button, QtCore.Qt.LeftButton)
|
||||
self.assertFalse(self.gui.share_mode.server_status.file_selection.delete_button.isVisible())
|
||||
|
||||
def test_add_a_file_and_delete_using_its_delete_widget(self):
|
||||
'''Test that we can also delete a file by clicking on its [X] widget'''
|
||||
self.gui.share_mode.server_status.file_selection.file_list.add_file('/etc/hosts')
|
||||
QtTest.QTest.mouseClick(self.gui.share_mode.server_status.file_selection.file_list.item(0).item_button, QtCore.Qt.LeftButton)
|
||||
self.assertEquals(self.gui.share_mode.server_status.file_selection.get_num_files(), 0)
|
||||
|
||||
def test_file_selection_widget_readd_files(self):
|
||||
'''Re-add some files to the list so we can share'''
|
||||
self.gui.share_mode.server_status.file_selection.file_list.add_file('/etc/hosts')
|
||||
self.gui.share_mode.server_status.file_selection.file_list.add_file('/tmp/test.txt')
|
||||
self.assertEqual(self.gui.share_mode.server_status.file_selection.get_num_files(), 2)
|
||||
|
||||
def test_add_delete_buttons_hidden(self):
|
||||
'''Test that the add and delete buttons are hidden when the server starts'''
|
||||
self.assertFalse(self.gui.share_mode.server_status.file_selection.add_button.isVisible())
|
||||
self.assertFalse(self.gui.share_mode.server_status.file_selection.delete_button.isVisible())
|
||||
|
||||
def test_download_share(self, public_mode):
|
||||
'''Test that we can download the share'''
|
||||
(socks_address, socks_port) = self.gui.app.onion.get_tor_socks_port()
|
||||
socks.set_default_proxy(socks.SOCKS5, socks_address, socks_port)
|
||||
|
||||
s = socks.socksocket()
|
||||
s.settimeout(60)
|
||||
s.connect((self.gui.app.onion_host, 80))
|
||||
|
||||
if public_mode:
|
||||
path = '/download'
|
||||
else:
|
||||
path = '{}/download'.format(self.gui.share_mode.web.slug)
|
||||
|
||||
http_request = 'GET {} HTTP/1.0\r\n'.format(path)
|
||||
http_request += 'Host: {}\r\n'.format(self.gui.app.onion_host)
|
||||
http_request += '\r\n'
|
||||
s.sendall(http_request.encode('utf-8'))
|
||||
|
||||
with open('/tmp/download.zip', 'wb') as file_to_write:
|
||||
while True:
|
||||
data = s.recv(1024)
|
||||
if not data:
|
||||
break
|
||||
file_to_write.write(data)
|
||||
file_to_write.close()
|
||||
|
||||
zip = zipfile.ZipFile('/tmp/download.zip')
|
||||
QtTest.QTest.qWait(4000)
|
||||
self.assertEquals('onionshare', zip.read('test.txt').decode('utf-8'))
|
||||
|
||||
def test_add_button_visible(self):
|
||||
'''Test that the add button should be visible'''
|
||||
self.assertTrue(self.gui.share_mode.server_status.file_selection.add_button.isVisible())
|
||||
|
||||
|
||||
# Stealth tests
|
||||
def test_copy_have_hidserv_auth_button(self, mode):
|
||||
'''Test that the Copy HidservAuth button is shown'''
|
||||
|
||||
Loading…
Reference in New Issue
Block a user