mirror of
https://github.com/freedombox/FreedomBox.git
synced 2026-01-28 08:03:36 +00:00
ace339eabf
This introduces flake8 and fixes a bunch of flake8 errors. flake8 is run with: ./venv/bin/flake8 plinth if you're using a python3 venv. We can eventually further integrate this with gitlab ci. https://salsa.debian.org/freedombox-team/plinth/issues/58 Reviewed-by: Sunil Mohan Adapa <sunil@medhas.org>
328 lines
12 KiB
Python
328 lines
12 KiB
Python
#
|
|
# This file is part of FreedomBox.
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
|
|
"""
|
|
Test module for network configuration utilities.
|
|
"""
|
|
|
|
import copy
|
|
import time
|
|
|
|
import pytest
|
|
|
|
ethernet_settings = {
|
|
'common': {
|
|
'type': '802-3-ethernet',
|
|
'name': 'plinth_test_eth',
|
|
'interface': 'eth0',
|
|
'zone': 'internal',
|
|
},
|
|
'ipv4': {
|
|
'method': 'auto',
|
|
'dns': '',
|
|
'second_dns': '',
|
|
},
|
|
'ipv6': {
|
|
'method': 'auto',
|
|
'dns': '',
|
|
'second_dns': '',
|
|
},
|
|
}
|
|
|
|
wifi_settings = {
|
|
'common': {
|
|
'type': '802-11-wireless',
|
|
'name': 'plinth_test_wifi',
|
|
'interface': 'wlan0',
|
|
'zone': 'external',
|
|
},
|
|
'ipv4': {
|
|
'method': 'auto',
|
|
'dns': '',
|
|
'second_dns': '',
|
|
},
|
|
'ipv6': {
|
|
'method': 'auto',
|
|
'dns': '',
|
|
'second_dns': '',
|
|
},
|
|
'wireless': {
|
|
'ssid': 'plinthtestwifi',
|
|
'mode': 'adhoc',
|
|
'auth_mode': 'open',
|
|
'band': 'a',
|
|
'channel': 0,
|
|
'bssid': 'a0:86:c6:08:11:02',
|
|
},
|
|
}
|
|
|
|
pppoe_settings = {
|
|
'common': {
|
|
'type': 'pppoe',
|
|
'name': 'plinth_test_pppoe',
|
|
'interface': 'eth1',
|
|
'zone': 'internal',
|
|
},
|
|
'pppoe': {
|
|
'username': 'x-user',
|
|
'password': 'x-password',
|
|
},
|
|
}
|
|
|
|
|
|
@pytest.fixture(name='network')
|
|
def fixture_network(needs_root):
|
|
"""Return the network module. Load it conservatively."""
|
|
from plinth import network as network_module
|
|
return network_module
|
|
|
|
|
|
def _connection(network, settings):
|
|
"""Create, yield and delete a connection."""
|
|
uuid = network.add_connection(settings)
|
|
time.sleep(0.1)
|
|
yield uuid
|
|
network.delete_connection(uuid)
|
|
|
|
|
|
@pytest.fixture(name='ethernet_uuid')
|
|
def fixture_ethernet_uuid(network):
|
|
"""Test fixture to setup and teardown an Ethernet connection."""
|
|
yield from _connection(network, ethernet_settings)
|
|
|
|
|
|
@pytest.fixture(name='wifi_uuid')
|
|
def fixture_wifi_uuid(network):
|
|
"""Test fixture to setup and teardown an Wi-Fi connection."""
|
|
yield from _connection(network, wifi_settings)
|
|
|
|
|
|
@pytest.fixture(name='pppoe_uuid')
|
|
def fixture_pppoe_uuid(network):
|
|
"""Test fixture to setup and teardown an PPPoE connection."""
|
|
yield from _connection(network, pppoe_settings)
|
|
|
|
|
|
@pytest.mark.usefixtures('ethernet_uuid', 'wifi_uuid', 'pppoe_uuid')
|
|
def test_get_connection_list(network):
|
|
"""Check that we can get a list of available connections."""
|
|
connections = network.get_connection_list()
|
|
|
|
assert 'plinth_test_eth' in [x['name'] for x in connections]
|
|
assert 'plinth_test_wifi' in [x['name'] for x in connections]
|
|
assert 'plinth_test_pppoe' in [x['name'] for x in connections]
|
|
|
|
|
|
def test_get_connection(network, ethernet_uuid, wifi_uuid):
|
|
"""Check that we can get a connection by name."""
|
|
connection = network.get_connection(ethernet_uuid)
|
|
assert connection.get_id() == 'plinth_test_eth'
|
|
|
|
connection = network.get_connection(wifi_uuid)
|
|
assert connection.get_id() == 'plinth_test_wifi'
|
|
|
|
with pytest.raises(network.ConnectionNotFound):
|
|
network.get_connection('x-invalid-network-id')
|
|
|
|
|
|
def test_edit_ethernet_connection(network, ethernet_uuid):
|
|
"""Check that we can update an ethernet connection."""
|
|
connection = network.get_connection(ethernet_uuid)
|
|
ethernet_settings2 = copy.deepcopy(ethernet_settings)
|
|
ethernet_settings2['common']['name'] = 'plinth_test_eth_new'
|
|
ethernet_settings2['common']['interface'] = 'eth1'
|
|
ethernet_settings2['common']['zone'] = 'external'
|
|
ethernet_settings2['ipv4']['method'] = 'auto'
|
|
network.edit_connection(connection, ethernet_settings2)
|
|
|
|
connection = network.get_connection(ethernet_uuid)
|
|
assert connection.get_id() == 'plinth_test_eth_new'
|
|
|
|
settings_connection = connection.get_setting_connection()
|
|
assert settings_connection.get_interface_name() == 'eth1'
|
|
assert settings_connection.get_zone() == 'external'
|
|
|
|
settings_ipv4 = connection.get_setting_ip4_config()
|
|
assert settings_ipv4.get_method() == 'auto'
|
|
|
|
|
|
def test_edit_pppoe_connection(network, pppoe_uuid):
|
|
"""Check that we can update a PPPoE connection."""
|
|
connection = network.get_connection(pppoe_uuid)
|
|
pppoe_settings2 = copy.deepcopy(pppoe_settings)
|
|
pppoe_settings2['common']['name'] = 'plinth_test_pppoe_new'
|
|
pppoe_settings2['common']['interface'] = 'eth2'
|
|
pppoe_settings2['common']['zone'] = 'external'
|
|
pppoe_settings2['pppoe']['username'] = 'x-user-new'
|
|
pppoe_settings2['pppoe']['password'] = 'x-password-new'
|
|
network.edit_connection(connection, pppoe_settings2)
|
|
|
|
connection = network.get_connection(pppoe_uuid)
|
|
assert connection.get_id() == 'plinth_test_pppoe_new'
|
|
|
|
settings_connection = connection.get_setting_connection()
|
|
assert settings_connection.get_interface_name() == 'eth2'
|
|
assert settings_connection.get_zone() == 'external'
|
|
|
|
settings_pppoe = connection.get_setting_pppoe()
|
|
assert settings_pppoe.get_username() == 'x-user-new'
|
|
secrets = connection.get_secrets('pppoe')
|
|
assert secrets['pppoe']['password'] == 'x-password-new'
|
|
|
|
settings_ppp = connection.get_setting_ppp()
|
|
assert settings_ppp.get_lcp_echo_failure() == 5
|
|
assert settings_ppp.get_lcp_echo_interval() == 30
|
|
|
|
|
|
def test_edit_wifi_connection(network, wifi_uuid):
|
|
"""Check that we can update a wifi connection."""
|
|
connection = network.get_connection(wifi_uuid)
|
|
wifi_settings2 = copy.deepcopy(wifi_settings)
|
|
wifi_settings2['common']['name'] = 'plinth_test_wifi_new'
|
|
wifi_settings2['common']['interface'] = 'wlan1'
|
|
wifi_settings2['common']['zone'] = 'external'
|
|
wifi_settings2['ipv4']['method'] = 'auto'
|
|
wifi_settings2['wireless']['ssid'] = 'plinthtestwifi2'
|
|
wifi_settings2['wireless']['mode'] = 'infrastructure'
|
|
wifi_settings2['wireless']['auth_mode'] = 'wpa'
|
|
wifi_settings2['wireless']['passphrase'] = 'secretpassword'
|
|
network.edit_connection(connection, wifi_settings2)
|
|
|
|
connection = network.get_connection(wifi_uuid)
|
|
|
|
assert connection.get_id() == 'plinth_test_wifi_new'
|
|
|
|
settings_connection = connection.get_setting_connection()
|
|
assert settings_connection.get_interface_name() == 'wlan1'
|
|
assert settings_connection.get_zone() == 'external'
|
|
|
|
settings_wireless = connection.get_setting_wireless()
|
|
assert settings_wireless.get_ssid().get_data() == b'plinthtestwifi2'
|
|
assert settings_wireless.get_mode() == 'infrastructure'
|
|
|
|
wifi_sec = connection.get_setting_wireless_security()
|
|
assert wifi_sec.get_key_mgmt() == 'wpa-psk'
|
|
|
|
secrets = connection.get_secrets('802-11-wireless-security')
|
|
assert secrets['802-11-wireless-security']['psk'] == 'secretpassword'
|
|
|
|
|
|
def test_ethernet_manual_ipv4_address(network, ethernet_uuid):
|
|
"""Check that we can manually set IPv4 address on ethernet."""
|
|
connection = network.get_connection(ethernet_uuid)
|
|
ethernet_settings2 = copy.deepcopy(ethernet_settings)
|
|
ethernet_settings2['ipv4']['method'] = 'manual'
|
|
ethernet_settings2['ipv4']['address'] = '169.254.0.1'
|
|
ethernet_settings2['ipv4']['netmask'] = '255.255.254.0'
|
|
ethernet_settings2['ipv4']['gateway'] = '169.254.0.254'
|
|
ethernet_settings2['ipv4']['dns'] = '1.2.3.4'
|
|
ethernet_settings2['ipv4']['second_dns'] = '1.2.3.5'
|
|
network.edit_connection(connection, ethernet_settings2)
|
|
|
|
connection = network.get_connection(ethernet_uuid)
|
|
settings_ipv4 = connection.get_setting_ip4_config()
|
|
assert settings_ipv4.get_method() == 'manual'
|
|
|
|
address = settings_ipv4.get_address(0)
|
|
assert address.get_address() == '169.254.0.1'
|
|
assert address.get_prefix() == 23
|
|
assert settings_ipv4.get_gateway() == '169.254.0.254'
|
|
assert settings_ipv4.get_num_dns() == 2
|
|
assert settings_ipv4.get_dns(0) == '1.2.3.4'
|
|
assert settings_ipv4.get_dns(1) == '1.2.3.5'
|
|
|
|
|
|
def test_ethernet_manual_ipv6_address(network, ethernet_uuid):
|
|
"""Check that we can manually set IPv6 address on ethernet."""
|
|
connection = network.get_connection(ethernet_uuid)
|
|
ethernet_settings2 = copy.deepcopy(ethernet_settings)
|
|
ethernet_settings2['ipv6']['method'] = 'manual'
|
|
ethernet_settings2['ipv6']['address'] = '::ffff:169.254.0.1'
|
|
ethernet_settings2['ipv6']['prefix'] = '63'
|
|
ethernet_settings2['ipv6']['gateway'] = '::ffff:169.254.0.254'
|
|
ethernet_settings2['ipv6']['dns'] = '::ffff:1.2.3.4'
|
|
ethernet_settings2['ipv6']['second_dns'] = '::ffff:1.2.3.5'
|
|
network.edit_connection(connection, ethernet_settings2)
|
|
|
|
connection = network.get_connection(ethernet_uuid)
|
|
settings_ipv6 = connection.get_setting_ip6_config()
|
|
assert settings_ipv6.get_method() == 'manual'
|
|
|
|
address = settings_ipv6.get_address(0)
|
|
assert address.get_address() == '::ffff:169.254.0.1'
|
|
assert address.get_prefix() == 63
|
|
assert settings_ipv6.get_gateway() == '::ffff:169.254.0.254'
|
|
assert settings_ipv6.get_num_dns() == 2
|
|
assert settings_ipv6.get_dns(0) == '::ffff:1.2.3.4'
|
|
assert settings_ipv6.get_dns(1) == '::ffff:1.2.3.5'
|
|
|
|
|
|
def test_wifi_manual_ipv4_address(network, wifi_uuid):
|
|
"""Check that we can manually set IPv4 address on wifi."""
|
|
connection = network.get_connection(wifi_uuid)
|
|
wifi_settings2 = copy.deepcopy(wifi_settings)
|
|
wifi_settings2['ipv4']['method'] = 'manual'
|
|
wifi_settings2['ipv4']['address'] = '169.254.0.2'
|
|
wifi_settings2['ipv4']['netmask'] = '255.255.254.0'
|
|
wifi_settings2['ipv4']['gateway'] = '169.254.0.254'
|
|
wifi_settings2['ipv4']['dns'] = '1.2.3.4'
|
|
wifi_settings2['ipv4']['second_dns'] = '1.2.3.5'
|
|
wifi_settings2['wireless']['ssid'] = 'plinthtestwifi'
|
|
wifi_settings2['wireless']['mode'] = 'adhoc'
|
|
wifi_settings2['wireless']['auth_mode'] = 'open'
|
|
network.edit_connection(connection, wifi_settings2)
|
|
|
|
connection = network.get_connection(wifi_uuid)
|
|
settings_ipv4 = connection.get_setting_ip4_config()
|
|
assert settings_ipv4.get_method() == 'manual'
|
|
|
|
address = settings_ipv4.get_address(0)
|
|
assert address.get_address() == '169.254.0.2'
|
|
assert address.get_prefix() == 23
|
|
assert settings_ipv4.get_gateway() == '169.254.0.254'
|
|
assert settings_ipv4.get_num_dns() == 2
|
|
assert settings_ipv4.get_dns(0) == '1.2.3.4'
|
|
assert settings_ipv4.get_dns(1) == '1.2.3.5'
|
|
|
|
|
|
def test_wifi_manual_ipv6_address(network, wifi_uuid):
|
|
"""Check that we can manually set IPv6 address on wifi."""
|
|
connection = network.get_connection(wifi_uuid)
|
|
wifi_settings2 = copy.deepcopy(wifi_settings)
|
|
wifi_settings2['ipv6']['method'] = 'manual'
|
|
wifi_settings2['ipv6']['address'] = '::ffff:169.254.0.2'
|
|
wifi_settings2['ipv6']['prefix'] = 63
|
|
wifi_settings2['ipv6']['gateway'] = '::ffff:169.254.0.254'
|
|
wifi_settings2['ipv6']['dns'] = '::ffff:1.2.3.4'
|
|
wifi_settings2['ipv6']['second_dns'] = '::ffff:1.2.3.5'
|
|
wifi_settings2['wireless']['ssid'] = 'plinthtestwifi'
|
|
wifi_settings2['wireless']['mode'] = 'adhoc'
|
|
wifi_settings2['wireless']['auth_mode'] = 'open'
|
|
network.edit_connection(connection, wifi_settings2)
|
|
|
|
connection = network.get_connection(wifi_uuid)
|
|
settings_ipv6 = connection.get_setting_ip6_config()
|
|
assert settings_ipv6.get_method() == 'manual'
|
|
|
|
address = settings_ipv6.get_address(0)
|
|
assert address.get_address() == '::ffff:169.254.0.2'
|
|
assert address.get_prefix() == 63
|
|
assert settings_ipv6.get_gateway() == '::ffff:169.254.0.254'
|
|
assert settings_ipv6.get_num_dns() == 2
|
|
assert settings_ipv6.get_dns(0) == '::ffff:1.2.3.4'
|
|
assert settings_ipv6.get_dns(1) == '::ffff:1.2.3.5'
|