onionshare/tests/local_onionshare_404_triggers_ratelimit_test.py

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

from .GuiShareTest import GuiShareTest

class Local404RateLimitTest(unittest.TestCase, GuiShareTest):
    @classmethod
    def setUpClass(cls):
        test_settings = {
            "close_after_first_download": False
        }
        cls.gui = GuiShareTest.set_up(test_settings)

    @classmethod
    def tearDownClass(cls):
        GuiShareTest.tear_down()

    @pytest.mark.gui
    def test_gui(self):
        self.run_all_common_setup_tests()
        self.run_all_share_mode_tests(False, True)
        self.hit_404(False)

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