summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rwxr-xr-xsmoketest/scripts/cli/test_protocols_static.py154
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)