diff options
| -rwxr-xr-x | scripts/cli/test_bond.py | 53 | ||||
| -rwxr-xr-x | scripts/cli/test_interfaces.py | 95 |
2 files changed, 95 insertions, 53 deletions
diff --git a/scripts/cli/test_bond.py b/scripts/cli/test_bond.py deleted file mode 100755 index fb948ffdf..000000000 --- a/scripts/cli/test_bond.py +++ /dev/null @@ -1,53 +0,0 @@ -#!/usr/bin/env python3 -# -# Copyright (C) 2019-2020 VyOS maintainers and contributors -# -# This program is free software; you can redistribute it and/or modify -# it under the terms of the GNU General Public License version 2 or later as -# published by the Free Software Foundation. -# -# 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 General Public License for more details. -# -# You should have received a copy of the GNU General Public License -# along with this program. If not, see <http://www.gnu.org/licenses/>. - -import os -import unittest -import vyos.config -import vyos.configsession - -from netifaces import ifaddresses, AF_INET, AF_INET6 -from vyos.validate import is_intf_addr_assigned - -base_path = ['interfaces', 'bonding'] -test_addr = ['192.0.2.1/25', '2001:db8:1::ffff/64'] -interfaces = ['bond0'] - -class TestInterfacesBond(unittest.TestCase): - def setUp(self): - self.session = vyos.configsession.ConfigSession(os.getpid()) - env = self.session.get_session_env() - self.config = vyos.config.Config(session_env=env) - - def tearDown(self): - # Delete existing interfaces - self.session.delete(base_path) - self.session.commit() - - def test_add_address(self): - """ Check if address is added to interface """ - for intf in interfaces: - for addr in test_addr: - self.session.set(base_path + [intf, 'address', addr]) - self.session.commit() - - for intf in interfaces: - for af in AF_INET, AF_INET6: - for addr in ifaddresses(intf)[af]: - self.assertTrue(is_intf_addr_assigned(intf, addr['addr'])) - -if __name__ == '__main__': - unittest.main() diff --git a/scripts/cli/test_interfaces.py b/scripts/cli/test_interfaces.py new file mode 100755 index 000000000..1f96bdfc6 --- /dev/null +++ b/scripts/cli/test_interfaces.py @@ -0,0 +1,95 @@ +#!/usr/bin/env python3 +# +# Copyright (C) 2019-2020 VyOS maintainers and contributors +# +# This program is free software; you can redistribute it and/or modify +# it under the terms of the GNU General Public License version 2 or later as +# published by the Free Software Foundation. +# +# 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 General Public License for more details. +# +# You should have received a copy of the GNU General Public License +# along with this program. If not, see <http://www.gnu.org/licenses/>. + +import os +import unittest +import vyos.config +import vyos.configsession + +from netifaces import ifaddresses, AF_INET, AF_INET6 +from vyos.validate import is_intf_addr_assigned, is_ipv6_link_local +from vyos.interfaces import list_interfaces_of_type + +class BasicInterfaceTest: + class BaseTest(unittest.TestCase): + def setUp(self): + self.session = vyos.configsession.ConfigSession(os.getpid()) + env = self.session.get_session_env() + self.config = vyos.config.Config(session_env=env) + self._test_addr = ['192.0.2.1/25', '2001:db8:1::ffff/64'] + + def tearDown(self): + self.session.delete(self._base_path) + self.session.commit() + + def test_add_address(self): + """ Check if address can be added to interface """ + + # Add address + for intf in self._interfaces: + for addr in self._test_addr: + self.session.set(self._base_path + [intf, 'address', addr]) + self.session.commit() + + # Validate address + for intf in self._interfaces: + for af in AF_INET, AF_INET6: + for addr in ifaddresses(intf)[af]: + # checking link local addresses makes no sense + if is_ipv6_link_local(addr['addr']): + continue + + self.assertTrue(is_intf_addr_assigned(intf, addr['addr'])) + + +class BondInterfaceTest(BasicInterfaceTest.BaseTest): + def setUp(self): + super().setUp() + self._base_path = ['interfaces', 'bonding'] + self._interfaces = ['bond0'] + + def test_add_remove_member(self): + members = [] + # we need to filter out VLAN interfaces identified by a dot (.) + # in their name - just in case! + for tmp in list_interfaces_of_type("ethernet"): + if not '.' in tmp: + members.append(tmp) + + for intf in self._interfaces: + for member in members: + # We can not enslave an interface when there is an address + # assigned - thus we do not allow it in case someone + # runs vyos-smoketest on his production device + self.session.set(self._base_path + [intf, 'member', 'interface', member]) + + self.session.commit() + + for intf in self._interfaces: + self.session.delete(self._base_path + [intf, 'member']) + + self.session.commit() + + +class BridgeInterfaceTest(BasicInterfaceTest.BaseTest): + def setUp(self): + super().setUp() + self._base_path = ['interfaces', 'bridge'] + self._interfaces = ['br0'] + + +if __name__ == '__main__': + unittest.main() |