File: test_functional.py

package info (click to toggle)
freedombox 26.2
  • links: PTS, VCS
  • area: main
  • in suites: forky
  • size: 82,976 kB
  • sloc: python: 48,504; javascript: 1,736; xml: 481; makefile: 290; sh: 167; php: 32
file content (113 lines) | stat: -rw-r--r-- 4,624 bytes parent folder | download | duplicates (5)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
# SPDX-License-Identifier: AGPL-3.0-or-later
"""
Functional, browser based tests for tor app.
"""

import pytest

from plinth.tests import functional

_TOR_FEATURE_TO_ELEMENT = {
    'relay': 'tor-relay_enabled',
    'bridge-relay': 'tor-bridge_relay_enabled',
    'hidden-services': 'tor-hs_enabled',
}

pytestmark = [pytest.mark.apps, pytest.mark.domain, pytest.mark.tor]


class TestTorApp(functional.BaseAppTests):
    app_name = 'tor'
    has_service = True
    has_web = False

    def test_set_tor_relay_configuration(self, session_browser):
        """Test setting Tor relay configuration."""
        functional.app_enable(session_browser, 'tor')
        _feature_enable(session_browser, 'relay', should_enable=False)
        _feature_enable(session_browser, 'relay', should_enable=True)
        _assert_feature_enabled(session_browser, 'relay', enabled=True)
        _assert_relay_port(session_browser, 'orport')

    def test_set_tor_bridge_relay_configuration(self, session_browser):
        """Test setting Tor bridge relay configuration."""
        functional.app_enable(session_browser, 'tor')
        _feature_enable(session_browser, 'bridge-relay', should_enable=False)
        _feature_enable(session_browser, 'bridge-relay', should_enable=True)
        _assert_feature_enabled(session_browser, 'bridge-relay', enabled=True)
        _assert_relay_port(session_browser, 'obfs3')
        _assert_relay_port(session_browser, 'obfs4')

    def test_set_tor_hidden_services_configuration(self, session_browser):
        """Test setting Tor hidden services configuration."""
        functional.app_enable(session_browser, 'tor')
        _feature_enable(session_browser, 'hidden-services',
                        should_enable=False)
        _feature_enable(session_browser, 'hidden-services', should_enable=True)
        _assert_feature_enabled(session_browser, 'hidden-services',
                                enabled=True)
        _assert_hidden_services(session_browser)

    # TODO: Test more thoroughly by checking same hidden service is restored
    # and by actually connecting using Tor.
    @pytest.mark.backups
    def test_backup_restore(self, session_browser):
        """Test backup and restore of configuration."""
        functional.app_enable(session_browser, 'tor')
        _feature_enable(session_browser, 'relay', should_enable=True)
        _feature_enable(session_browser, 'bridge-relay', should_enable=True)
        _feature_enable(session_browser, 'hidden-services', should_enable=True)
        functional.backup_create(session_browser, 'tor', 'test_tor')

        _feature_enable(session_browser, 'relay', should_enable=False)
        _feature_enable(session_browser, 'hidden-services',
                        should_enable=False)
        functional.backup_restore(session_browser, 'tor', 'test_tor')

        assert functional.service_is_running(session_browser, 'tor')
        _assert_feature_enabled(session_browser, 'relay', enabled=True)
        _assert_feature_enabled(session_browser, 'bridge-relay', enabled=True)
        _assert_feature_enabled(session_browser, 'hidden-services',
                                enabled=True)


def _feature_enable(browser, feature, should_enable):
    """Enable/disable a Tor feature."""
    element_name = _TOR_FEATURE_TO_ELEMENT[feature]
    functional.nav_to_module(browser, 'tor')
    checkbox_element = browser.find_by_name(element_name).first
    if should_enable == checkbox_element.checked:
        return

    if should_enable:
        if feature == 'bridge-relay':
            browser.find_by_name('tor-relay_enabled').first.check()

        checkbox_element.check()
    else:
        checkbox_element.uncheck()

    functional.submit(browser, form_class='form-configuration')
    functional.wait_for_config_update(browser, 'tor')


def _assert_feature_enabled(browser, feature, enabled):
    """Assert whether Tor relay is enabled or disabled."""
    element_name = _TOR_FEATURE_TO_ELEMENT[feature]
    functional.nav_to_module(browser, 'tor')
    assert browser.find_by_name(element_name).first.checked == enabled


def _assert_relay_port(browser, port_name):
    """Assert that port is available in port forwarding info table."""
    functional.nav_to_module(browser, 'tor')
    assert f'tor-{port_name}' in [
        name.text for name in browser.find_by_css(
            '.table-port-forwarding-info td:first-child')
    ]


def _assert_hidden_services(browser):
    """Assert that hidden service information is shown."""
    functional.nav_to_module(browser, 'tor')
    assert browser.find_by_css('.tor-hs .tor-hs-hostname')