nik/FreedomBox
1
0
Fork 0
mirror of https://github.com/freedombox/FreedomBox.git synced 2026-08-26 12:46:08 +00:00
Code Issues Packages Projects Releases Wiki Activity

network: Update tests to work with UUIDs and add a few more checks

Browse Source
This commit is contained in:
Sunil Mohan Adapa 2015-04-06 22:15:54 +05:30
parent 0cb8b4a5fc
commit f00f65f484
1 changed files with 41 additions and 28 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

69
plinth/tests/test_network.py
View File

@ -24,20 +24,25 @@ from plinth import network
class TestNetwork(unittest.TestCase): class TestNetwork(unittest.TestCase):
"""Verify that the network module performs as expected.""" """Verify that the network module performs as expected."""
ethernet_uuid = None
wifi_uuid = None
@classmethod @classmethod
def setUpClass(cls): def setUp(cls):
network.add_ethernet_connection( connection = network.add_ethernet_connection(
'plinth_test_eth', 'internal', 'plinth_test_eth', 'internal',
'auto', '') 'auto', '')
network.add_wifi_connection( cls.ethernet_uuid = connection['connection']['uuid']
connection = network.add_wifi_connection(
'plinth_test_wifi', 'external', 'plinth_test_wifi', 'external',
'plinthtestwifi', 'adhoc', 'open', '', 'plinthtestwifi', 'adhoc', 'open', '',
'auto', '') 'auto', '')
cls.wifi_uuid = connection['connection']['uuid']
@classmethod @classmethod
def tearDownClass(cls): def tearDown(cls):
network.delete_connection('plinth_test_eth') network.delete_connection(cls.ethernet_uuid)
network.delete_connection('plinth_test_wifi') network.delete_connection(cls.wifi_uuid)
def test_get_connection_list(self): def test_get_connection_list(self):
"""Check that we can get a list of available connections.""" """Check that we can get a list of available connections."""
@ -48,45 +53,53 @@ class TestNetwork(unittest.TestCase):
def test_get_connection(self): def test_get_connection(self):
"""Check that we can get a connection by name.""" """Check that we can get a connection by name."""
conn = network.get_connection('plinth_test_eth') connection = network.get_connection(self.ethernet_uuid)
self.assertEqual( self.assertEqual(
conn.GetSettings()['connection']['id'], 'plinth_test_eth') connection.GetSettings()['connection']['id'], 'plinth_test_eth')
conn = network.get_connection('plinth_test_wifi')
connection = network.get_connection(self.wifi_uuid)
self.assertEqual( self.assertEqual(
conn.GetSettings()['connection']['id'], 'plinth_test_wifi') connection.GetSettings()['connection']['id'], 'plinth_test_wifi')
self.assertRaises(network.ConnectionNotFound, network.get_connection,
'x-invalid-network-id')
def test_edit_ethernet_connection(self): def test_edit_ethernet_connection(self):
"""Check that we can update an ethernet connection.""" """Check that we can update an ethernet connection."""
conn = network.get_connection('plinth_test_eth') connection = network.get_connection(self.ethernet_uuid)
network.edit_ethernet_connection( network.edit_ethernet_connection(
conn, 'plinth_test_eth', 'external', 'manual', '169.254.0.1') connection, 'plinth_test_eth_new', 'external', 'manual',
conn = network.get_connection('plinth_test_eth') '169.254.0.1')
self.assertEqual(conn.GetSettings()['connection']['zone'], 'external') connection = network.get_connection(self.ethernet_uuid)
self.assertEqual(conn.GetSettings()['ipv4']['method'], 'manual') settings = connection.GetSettings()
self.assertEqual(settings['connection']['id'], 'plinth_test_eth_new')
self.assertEqual(settings['connection']['zone'], 'external')
self.assertEqual(settings['ipv4']['method'], 'manual')
self.assertEqual(settings['ipv4']['addresses'],
[['169.254.0.1', 24, '0.0.0.0']])
def test_edit_wifi_connection(self): def test_edit_wifi_connection(self):
"""Check that we can update a wifi connection.""" """Check that we can update a wifi connection."""
conn = network.get_connection('plinth_test_wifi') connection = network.get_connection(self.wifi_uuid)
network.edit_wifi_connection( network.edit_wifi_connection(
conn, 'plinth_test_wifi', 'external', connection, 'plinth_test_wifi_new', 'external',
'plinthtestwifi2', 'infrastructure', 'wpa', 'secretpassword', 'plinthtestwifi2', 'infrastructure', 'wpa', 'secretpassword',
'auto', '') 'auto', '')
conn = network.get_connection('plinth_test_wifi')
self.assertEqual(conn.GetSettings()['connection']['zone'], 'external') connection = network.get_connection(self.wifi_uuid)
settings = connection.GetSettings()
self.assertEqual(settings['connection']['id'], 'plinth_test_wifi_new')
self.assertEqual(settings['connection']['zone'], 'external')
self.assertEqual(settings['802-11-wireless']['ssid'],
'plinthtestwifi2')
self.assertEqual(settings['802-11-wireless']['mode'], 'infrastructure')
self.assertEqual(settings['802-11-wireless-security']['key-mgmt'],
'wpa-psk')
self.assertEqual( self.assertEqual(
conn.GetSettings()['802-11-wireless']['ssid'], 'plinthtestwifi2') connection.GetSecrets()['802-11-wireless-security']['psk'],
self.assertEqual(
conn.GetSettings()['802-11-wireless']['mode'], 'infrastructure')
self.assertEqual(
conn.GetSettings()['802-11-wireless-security']['key-mgmt'],
'wpa-psk')
self.assertEqual(
conn.GetSecrets()['802-11-wireless-security']['psk'],
'secretpassword') 'secretpassword')
if __name__ == "__main__": if __name__ == "__main__":
unittest.main() unittest.main()