diff options
| -rwxr-xr-x | smoketest/scripts/cli/test_protocols_static.py | 154 |
1 files changed, 154 insertions, 0 deletions
diff --git a/smoketest/scripts/cli/test_protocols_static.py b/smoketest/scripts/cli/test_protocols_static.py new file mode 100755 index 000000000..100fd3387 --- /dev/null +++ b/smoketest/scripts/cli/test_protocols_static.py @@ -0,0 +1,154 @@ +#!/usr/bin/env python3 +# +# Copyright (C) 2021 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 re +import os +import json +import unittest + +from netifaces import interfaces + +from vyos.configsession import ConfigSession +from vyos.configsession import ConfigSessionError +from vyos.ifconfig import Interface +from vyos.ifconfig import Section +from vyos.template import is_ipv6 +from vyos.util import cmd +from vyos.util import read_file +from vyos.validate import is_intf_addr_assigned + +dummy_if = 'dum08765' +base_path = ['protocols', 'static'] + +routes = { + '10.0.0.0/8' : { + 'next_hop' : '192.0.2.2', + 'distance' : '200', + }, + '172.16.0.0/12' : { + 'next_hop' : '192.0.2.3', + }, + '192.168.0.0/16' : { + 'next_hop' : '192.0.2.3', + }, + '2001:db8:1000::/48' : { + 'next_hop' : '2001:db8::1000', + }, + '2001:db8:2000::/48' : { + 'next_hop' : '2001:db8::2000', + }, +} + +interface_routes = { + '10.0.0.0/8' : { + 'next_hop' : dummy_if, + 'distance' : '200', + }, + '172.16.0.0/12' : { + 'next_hop' : dummy_if, + }, + '192.168.0.0/16' : { + 'next_hop' : dummy_if, + }, + '2001:db8:1000::/48' : { + 'next_hop' : dummy_if, + }, + '2001:db8:2000::/48' : { + 'next_hop' : dummy_if, + }, +} + + +class StaticRouteTest(unittest.TestCase): + def setUp(self): + self.session = ConfigSession(os.getpid()) + # we need an alive next-hop interface + self.session.set(['interfaces', 'dummy', dummy_if, 'address', '192.0.2.1/24']) + self.session.set(['interfaces', 'dummy', dummy_if, 'address', '2001:db8::1/64']) + + def tearDown(self): + self.session.delete(['interfaces', 'dummy', dummy_if]) + self.session.commit() + + def test_static_routes(self): + for route, route_config in routes.items(): + route_type = 'route' + if is_ipv6(route): + route_type = 'route6' + self.session.set(base_path + [route_type, route, 'next-hop', route_config['next_hop']]) + if 'distance' in route_config: + self.session.set(base_path + [route_type, route, 'next-hop', route_config['next_hop'], 'distance', route_config['distance']]) + + # commit changes + self.session.commit() + + # Verify routes + for route, route_config in routes.items(): + ip_ver = '-4' + if is_ipv6(route): + ip_ver = '-6' + tmp = json.loads(cmd(f'ip {ip_ver} -d -j route show {route}')) + + found = False + for result in tmp: + # unfortunately iproute2 does not return the distance + if 'dst' in result and result['dst'] == route: + if 'gateway' in result and result['gateway'] == route_config['next_hop']: + found = True + + self.assertTrue(found) + + route_type = 'route' + if is_ipv6(route): + route_type = 'route6' + self.session.delete(base_path + [route_type, route]) + + def test_interface_routes(self): + for route, route_config in interface_routes.items(): + route_type = 'interface-route' + if is_ipv6(route): + route_type = 'interface-route6' + self.session.set(base_path + [route_type, route, 'next-hop-interface', route_config['next_hop']]) + if 'distance' in route_config: + self.session.set(base_path + [route_type, route, 'next-hop-interface', route_config['next_hop'], 'distance', route_config['distance']]) + + # commit changes + self.session.commit() + + # Verify routes + for route, route_config in interface_routes.items(): + ip_ver = '-4' + if is_ipv6(route): + ip_ver = '-6' + tmp = json.loads(cmd(f'ip {ip_ver} -d -j route show {route}')) + + found = False + for result in tmp: + # unfortunately iproute2 does not return the distance + if 'dst' in result and result['dst'] == route: + if 'dev' in result and result['dev'] == route_config['next_hop']: + found = True + break + + self.assertTrue(found) + + route_type = 'interface-route' + if is_ipv6(route): + route_type = 'interface-route6' + self.session.delete(base_path + [route_type, route]) + +if __name__ == '__main__': + unittest.main(verbosity=2, failfast=True) |