onionshare/unit_tests/onionshare_receive_mode_upload_test.py

#!/usr/bin/env python3
import os
import sys
import unittest
import pytest
import json

from PyQt5 import QtWidgets

from onionshare.common import Common
from onionshare.web import Web
from onionshare import onion, strings
from onionshare_gui import *

from .commontests import CommonTests

app = QtWidgets.QApplication(sys.argv)

class OnionShareGuiTest(unittest.TestCase):
    '''Test the OnionShare GUI'''
    @classmethod
    def setUpClass(cls):
        '''Create the GUI'''
        # Create our test file
        testfile = open('/tmp/test.txt', 'w')
        testfile.write('onionshare')
        testfile.close()
        common = Common()
        common.define_css()

        # Start the Onion
        strings.load_strings(common)

        testonion = onion.Onion(common)
        global qtapp
        qtapp = Application(common)
        app = OnionShare(common, testonion, True, 0)

        web = Web(common, False, True)

        test_settings = {
            "auth_password": "",
            "auth_type": "no_auth",
            "autoupdate_timestamp": "",
            "close_after_first_download": True,
            "connection_type": "bundled",
            "control_port_address": "127.0.0.1",
            "control_port_port": 9051,
            "downloads_dir": "/tmp/OnionShare",
            "hidservauth_string": "",
            "no_bridges": True, 
            "private_key": "",
            "public_mode": False,
            "receive_allow_receiver_shutdown": True,
            "save_private_key": False,
            "shutdown_timeout": False,
            "slug": "",
            "socks_address": "127.0.0.1",
            "socks_port": 9050,
            "socket_file_path": "/var/run/tor/control",
            "systray_notifications": True,
            "tor_bridges_use_meek_lite_azure": False,
            "tor_bridges_use_meek_lite_amazon": False,
            "tor_bridges_use_custom_bridges": "",
            "tor_bridges_use_obfs4": False,
            "use_stealth": False,
            "use_legacy_v2_onions": False,
            "use_autoupdate": True,
            "version": "1.3.1"
        }
        testsettings = '/tmp/testsettings.json'
        open(testsettings, 'w').write(json.dumps(test_settings))

        cls.gui = OnionShareGui(common, testonion, qtapp, app, ['/tmp/test.txt'], testsettings, True)
        
    @classmethod
    def tearDownClass(cls):
        '''Clean up after tests'''
        os.remove('/tmp/test.txt')
        os.remove('/tmp/OnionShare/test.txt')
        os.remove('/tmp/OnionShare/test-2.txt')

    @pytest.mark.run(order=1)
    def test_gui_loaded(self):
        CommonTests.test_gui_loaded(self)

    @pytest.mark.run(order=2)
    def test_windowTitle_seen(self):
        CommonTests.test_windowTitle_seen(self)

    @pytest.mark.run(order=3)
    def test_settings_button_is_visible(self):
        CommonTests.test_settings_button_is_visible(self)

    @pytest.mark.run(order=4)
    def test_server_status_bar_is_visible(self):
        CommonTests.test_server_status_bar_is_visible(self)

    @pytest.mark.run(order=5)
    def test_info_widget_is_not_visible(self):
        CommonTests.test_info_widget_is_not_visible(self, 'receive')

    @pytest.mark.run(order=6)
    def test_click_mode(self):
        CommonTests.test_click_mode(self, 'receive')

    @pytest.mark.run(order=7)
    def test_history_is_visible(self):
        CommonTests.test_history_is_visible(self, 'receive')

    @pytest.mark.run(order=8)
    def test_server_working_on_start_button_pressed(self):
        CommonTests.test_server_working_on_start_button_pressed(self, 'receive')

    @pytest.mark.run(order=9)
    def test_server_status_indicator_says_starting(self):
        CommonTests.test_server_status_indicator_says_starting(self, 'receive')

    @pytest.mark.run(order=10)
    def test_settings_button_is_hidden(self):
        CommonTests.test_settings_button_is_hidden(self)

    @pytest.mark.run(order=11)
    def test_a_server_is_started(self):
       CommonTests.test_a_server_is_started(self, 'receive')

    @pytest.mark.run(order=12)
    def test_a_web_server_is_running(self):
        CommonTests.test_a_web_server_is_running(self)

    @pytest.mark.run(order=14)
    def test_have_a_slug(self):
       CommonTests.test_have_a_slug(self, 'receive', False)

    @pytest.mark.run(order=15)
    def test_url_description_shown(self):
       CommonTests.test_url_description_shown(self, 'receive')

    @pytest.mark.run(order=16)
    def test_have_copy_url_button(self):
       CommonTests.test_have_copy_url_button(self, 'receive')

    @pytest.mark.run(order=17)
    def test_server_status_indicator_says_started(self):
        CommonTests.test_server_status_indicator_says_started(self, 'receive')

    @pytest.mark.run(order=18)
    def test_web_page(self):
        CommonTests.test_web_page(self, 'receive', 'Select the files you want to send, then click', False)

    @pytest.mark.run(order=19)
    def test_upload_file(self):
        CommonTests.test_upload_file(self, False, '/tmp/OnionShare/test.txt')

    @pytest.mark.run(order=20)
    def test_history_widgets_present(self):
        CommonTests.test_history_widgets_present(self, 'receive')

    @pytest.mark.run(order=21)
    def test_counter_incremented(self):
        CommonTests.test_counter_incremented(self, 'receive', 1)

    @pytest.mark.run(order=22)
    def test_upload_same_file_is_renamed(self):
        CommonTests.test_upload_file(self, False, '/tmp/OnionShare/test-2.txt')

    @pytest.mark.run(order=23)
    def test_upload_count_incremented_again(self):
        CommonTests.test_counter_incremented(self, 'receive', 2)

    @pytest.mark.run(order=24)
    def test_server_is_stopped(self):
        CommonTests.test_server_is_stopped(self, 'receive', False)

    @pytest.mark.run(order=25)
    def test_web_service_is_stopped(self):
        CommonTests.test_web_service_is_stopped(self)

    @pytest.mark.run(order=26)
    def test_server_status_indicator_says_closed(self):
        CommonTests.test_server_status_indicator_says_closed(self, 'receive', False)

if __name__ == "__main__":
    unittest.main()