#!/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 . import os import unittest from vyos.configsession import ConfigSession from vyos.ifconfig import Section from vyos.util import cmd from vyos.util import process_named_running PROCESS_NAME = 'ripd' acl_in = '198' acl_out = '199' prefix_list_in = 'foo-prefix' prefix_list_out = 'bar-prefix' route_map = 'FooBar123' base_path = ['protocols', 'rip'] def getFRRconfig(): return cmd('vtysh -c "show run" | sed -n "/router rip/,/^!/p"') class TestProtocolsRIP(unittest.TestCase): def setUp(self): self.session = ConfigSession(os.getpid()) self.session.set(['policy', 'access-list', acl_in, 'rule', '10', 'action', 'permit']) self.session.set(['policy', 'access-list', acl_in, 'rule', '10', 'source', 'any']) self.session.set(['policy', 'access-list', acl_in, 'rule', '10', 'destination', 'any']) self.session.set(['policy', 'access-list', acl_out, 'rule', '20', 'action', 'deny']) self.session.set(['policy', 'access-list', acl_out, 'rule', '20', 'source', 'any']) self.session.set(['policy', 'access-list', acl_out, 'rule', '20', 'destination', 'any']) self.session.set(['policy', 'prefix-list', prefix_list_in, 'rule', '100', 'action', 'permit']) self.session.set(['policy', 'prefix-list', prefix_list_in, 'rule', '100', 'prefix', '192.0.2.0/24']) self.session.set(['policy', 'prefix-list', prefix_list_out, 'rule', '200', 'action', 'deny']) self.session.set(['policy', 'prefix-list', prefix_list_out, 'rule', '200', 'prefix', '192.0.2.0/24']) self.session.set(['policy', 'route-map', route_map, 'rule', '10', 'action', 'permit']) def tearDown(self): self.session.delete(base_path) self.session.delete(['policy', 'access-list', acl_in]) self.session.delete(['policy', 'access-list', acl_out]) self.session.delete(['policy', 'prefix-list', prefix_list_in]) self.session.delete(['policy', 'prefix-list', prefix_list_out]) self.session.delete(['policy', 'route-map', route_map]) self.session.commit() del self.session # Check for running process self.assertTrue(process_named_running(PROCESS_NAME)) def test_rip(self): distance = '40' network_distance = '66' metric = '8' interfaces = Section.interfaces('ethernet') neighbors = ['1.2.3.4', '1.2.3.5', '1.2.3.6'] networks = ['10.0.0.0/8', '172.16.0.0/12', '192.168.0.0/16'] redistribute = ['bgp', 'connected', 'kernel', 'ospf', 'static'] timer_garbage = '888' timer_timeout = '1000' timer_update = '90' self.session.set(base_path + ['default-distance', distance]) self.session.set(base_path + ['default-information', 'originate']) self.session.set(base_path + ['default-metric', metric]) self.session.set(base_path + ['distribute-list', 'access-list', 'in', acl_in]) self.session.set(base_path + ['distribute-list', 'access-list', 'out', acl_out]) self.session.set(base_path + ['distribute-list', 'prefix-list', 'in', prefix_list_in]) self.session.set(base_path + ['distribute-list', 'prefix-list', 'out', prefix_list_out]) self.session.set(base_path + ['passive-interface', 'default']) self.session.set(base_path + ['timers', 'garbage-collection', timer_garbage]) self.session.set(base_path + ['timers', 'timeout', timer_timeout]) self.session.set(base_path + ['timers', 'update', timer_update]) for interface in interfaces: self.session.set(base_path + ['interface', interface]) self.session.set(base_path + ['distribute-list', 'interface', interface, 'access-list', 'in', acl_in]) self.session.set(base_path + ['distribute-list', 'interface', interface, 'access-list', 'out', acl_out]) self.session.set(base_path + ['distribute-list', 'interface', interface, 'prefix-list', 'in', prefix_list_in]) self.session.set(base_path + ['distribute-list', 'interface', interface, 'prefix-list', 'out', prefix_list_out]) for neighbor in neighbors: self.session.set(base_path + ['neighbor', neighbor]) for network in networks: self.session.set(base_path + ['network', network]) self.session.set(base_path + ['network-distance', network, 'distance', network_distance]) self.session.set(base_path + ['route', network]) for proto in redistribute: self.session.set(base_path + ['redistribute', proto, 'metric', metric]) self.session.set(base_path + ['redistribute', proto, 'route-map', route_map]) # commit changes self.session.commit() # Verify FRR ospfd configuration frrconfig = getFRRconfig() self.assertIn(f'router rip', frrconfig) self.assertIn(f' distance {distance}', frrconfig) self.assertIn(f' default-information originate', frrconfig) self.assertIn(f' default-metric {metric}', frrconfig) self.assertIn(f' distribute-list {acl_in} in', frrconfig) self.assertIn(f' distribute-list {acl_out} out', frrconfig) self.assertIn(f' distribute-list prefix {prefix_list_in} in', frrconfig) self.assertIn(f' distribute-list prefix {prefix_list_out} out', frrconfig) self.assertIn(f' passive-interface default', frrconfig) self.assertIn(f' timers basic {timer_update} {timer_timeout} {timer_garbage}', frrconfig) for interface in interfaces: self.assertIn(f' network {interface}', frrconfig) self.assertIn(f' distribute-list {acl_in} in {interface}', frrconfig) self.assertIn(f' distribute-list {acl_out} out {interface}', frrconfig) self.assertIn(f' distribute-list prefix {prefix_list_in} in {interface}', frrconfig) self.assertIn(f' distribute-list prefix {prefix_list_out} out {interface}', frrconfig) for neighbor in neighbors: self.assertIn(f' neighbor {neighbor}', frrconfig) for network in networks: self.assertIn(f' network {network}', frrconfig) self.assertIn(f' distance {network_distance} {network}', frrconfig) self.assertIn(f' route {network}', frrconfig) for proto in redistribute: self.assertIn(f' redistribute {proto} metric {metric} route-map {route_map}', frrconfig) if __name__ == '__main__': unittest.main(verbosity=2)