onionshare/tests/onionshare_790_cancel_on_second_share_test.py

#!/usr/bin/env python3
import pytest
import unittest

from .TorGuiShareTest import TorGuiShareTest

# Tests #790 regression
class ShareModeCancelSecondShareTest(unittest.TestCase, TorGuiShareTest):
    @classmethod
    def setUpClass(cls):
        test_settings = {
            "close_after_first_download": True 
        }
        cls.gui = TorGuiShareTest.set_up(test_settings, 'ShareModeCancelSecondShareTest')

    @pytest.mark.tor
    def test_run_all_common_setup_tests(self):
        self.run_all_common_setup_tests()

    @pytest.mark.run(after='test_run_all_common_setup_tests')
    @pytest.mark.tor
    def test_run_share_mode_tests(self):
        self.run_all_share_mode_tests(False, False)

    @pytest.mark.run(after='test_run_share_mode_tests')
    @pytest.mark.tor
    def test_cancel_the_share(self):
        self.cancel_the_share(self.gui.share_mode)

    @pytest.mark.run(after='test_cancel_the_share')
    @pytest.mark.tor
    def test_server_is_stopped_round2(self):
        self.server_is_stopped(self.gui.share_mode, False)

    @pytest.mark.run(after='test_server_is_stopped_round2')
    @pytest.mark.tor
    def test_web_service_is_stopped_round2(self):
        self.web_service_is_stopped()

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