diff options
| author | Christian Breunig <christian@breunig.cc> | 2023-12-30 23:25:20 +0100 | 
|---|---|---|
| committer | Christian Breunig <christian@breunig.cc> | 2023-12-31 23:49:48 +0100 | 
| commit | 4ef110fd2c501b718344c72d495ad7e16d2bd465 (patch) | |
| tree | e98bf08f93c029ec4431a3b6ca078e7562e0cc58 /smoketest/scripts/cli/test_protocols_segment-routing.py | |
| parent | 2286b8600da6c631b17e1d5b9b341843e50f9abf (diff) | |
| download | vyos-1x-4ef110fd2c501b718344c72d495ad7e16d2bd465.tar.gz vyos-1x-4ef110fd2c501b718344c72d495ad7e16d2bd465.zip | |
T5474: establish common file name pattern for XML conf mode commands
We will use _ as CLI level divider. The XML definition filename and also
the Python helper should match the CLI node.
Example:
set interfaces ethernet -> interfaces_ethernet.xml.in
set interfaces bond -> interfaces_bond.xml.in
set service dhcp-server -> service_dhcp-server-xml.in
Diffstat (limited to 'smoketest/scripts/cli/test_protocols_segment-routing.py')
| -rwxr-xr-x | smoketest/scripts/cli/test_protocols_segment-routing.py | 112 | 
1 files changed, 112 insertions, 0 deletions
| diff --git a/smoketest/scripts/cli/test_protocols_segment-routing.py b/smoketest/scripts/cli/test_protocols_segment-routing.py new file mode 100755 index 000000000..403c05924 --- /dev/null +++ b/smoketest/scripts/cli/test_protocols_segment-routing.py @@ -0,0 +1,112 @@ +#!/usr/bin/env python3 +# +# Copyright (C) 2023 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 + +from base_vyostest_shim import VyOSUnitTestSHIM + +from vyos.configsession import ConfigSessionError +from vyos.ifconfig import Section +from vyos.utils.process import cmd +from vyos.utils.process import process_named_running +from vyos.utils.system import sysctl_read + +base_path = ['protocols', 'segment-routing'] +PROCESS_NAME = 'zebra' + +class TestProtocolsSegmentRouting(VyOSUnitTestSHIM.TestCase): +    @classmethod +    def setUpClass(cls): +        # call base-classes classmethod +        super(TestProtocolsSegmentRouting, cls).setUpClass() +        # Retrieve FRR daemon PID - it is not allowed to crash, thus PID must remain the same +        cls.daemon_pid = process_named_running(PROCESS_NAME) +        # ensure we can also run this test on a live system - so lets clean +        # out the current configuration :) +        cls.cli_delete(cls, base_path) + +    def tearDown(self): +        self.cli_delete(base_path) +        self.cli_commit() + +        # check process health and continuity +        self.assertEqual(self.daemon_pid, process_named_running(PROCESS_NAME)) + +    def test_srv6(self): +        interfaces = Section.interfaces('ethernet', vlan=False) +        locators = { +            'foo' : { 'prefix' : '2001:a::/64' }, +            'foo' : { 'prefix' : '2001:b::/64', 'usid' : {} }, +        } + +        for locator, locator_config in locators.items(): +            self.cli_set(base_path + ['srv6', 'locator', locator, 'prefix', locator_config['prefix']]) +            if 'usid' in locator_config: +                self.cli_set(base_path + ['srv6', 'locator', locator, 'behavior-usid']) + +        # verify() - SRv6 should be enabled on at least one interface! +        with self.assertRaises(ConfigSessionError): +            self.cli_commit() +        for interface in interfaces: +            self.cli_set(base_path + ['interface', interface, 'srv6']) + +        self.cli_commit() + +        for interface in interfaces: +            self.assertEqual(sysctl_read(f'net.ipv6.conf.{interface}.seg6_enabled'), '1') +            self.assertEqual(sysctl_read(f'net.ipv6.conf.{interface}.seg6_require_hmac'), '0') # default + +        frrconfig = self.getFRRconfig(f'segment-routing', daemon='zebra') +        self.assertIn(f'segment-routing', frrconfig) +        self.assertIn(f' srv6', frrconfig) +        self.assertIn(f'  locators', frrconfig) +        for locator, locator_config in locators.items(): +            self.assertIn(f'   locator {locator}', frrconfig) +            self.assertIn(f'    prefix {locator_config["prefix"]} block-len 40 node-len 24 func-bits 16', frrconfig) + +    def test_srv6_sysctl(self): +        interfaces = Section.interfaces('ethernet', vlan=False) + +        # HMAC accept +        for interface in interfaces: +            self.cli_set(base_path + ['interface', interface, 'srv6']) +            self.cli_set(base_path + ['interface', interface, 'srv6', 'hmac', 'ignore']) +        self.cli_commit() + +        for interface in interfaces: +            self.assertEqual(sysctl_read(f'net.ipv6.conf.{interface}.seg6_enabled'), '1') +            self.assertEqual(sysctl_read(f'net.ipv6.conf.{interface}.seg6_require_hmac'), '-1') # ignore + +        # HMAC drop +        for interface in interfaces: +            self.cli_set(base_path + ['interface', interface, 'srv6']) +            self.cli_set(base_path + ['interface', interface, 'srv6', 'hmac', 'drop']) +        self.cli_commit() + +        for interface in interfaces: +            self.assertEqual(sysctl_read(f'net.ipv6.conf.{interface}.seg6_enabled'), '1') +            self.assertEqual(sysctl_read(f'net.ipv6.conf.{interface}.seg6_require_hmac'), '1') # drop + +        # Disable SRv6 on first interface +        first_if = interfaces[-1] +        self.cli_delete(base_path + ['interface', first_if]) +        self.cli_commit() + +        self.assertEqual(sysctl_read(f'net.ipv6.conf.{first_if}.seg6_enabled'), '0') + +if __name__ == '__main__': +    unittest.main(verbosity=2) | 
