126 lines
4.1 KiB
Python
126 lines
4.1 KiB
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 eval: (blacken-mode 1) -*-
|
|
# SPDX-License-Identifier: ISC
|
|
#
|
|
# Copyright (c) 2024 NFWare Inc.
|
|
#
|
|
# noqa: E501
|
|
#
|
|
"""
|
|
Test static route functionality
|
|
"""
|
|
|
|
import ipaddress
|
|
|
|
import pytest
|
|
from lib.topogen import Topogen
|
|
from lib.common_config import retry
|
|
|
|
pytestmark = [pytest.mark.staticd, pytest.mark.mgmtd]
|
|
|
|
|
|
@pytest.fixture(scope="module")
|
|
def tgen(request):
|
|
"Setup/Teardown the environment and provide tgen argument to tests"
|
|
|
|
topodef = {"s1": ("r1",), "s2": ("r1",)}
|
|
|
|
tgen = Topogen(topodef, request.module.__name__)
|
|
tgen.start_topology()
|
|
|
|
router_list = tgen.routers()
|
|
for rname, router in router_list.items():
|
|
# Setup VRF red
|
|
router.net.add_l3vrf("red", 10)
|
|
router.net.attach_iface_to_l3vrf(rname + "-eth0", "red")
|
|
# Setup VRF blue
|
|
router.net.add_l3vrf("blue", 20)
|
|
router.net.attach_iface_to_l3vrf(rname + "-eth1", "blue")
|
|
# Load configuration
|
|
router.load_frr_config("frr.conf")
|
|
|
|
tgen.start_router()
|
|
yield tgen
|
|
tgen.stop_topology()
|
|
|
|
|
|
@retry(retry_timeout=1, initial_wait=0.1)
|
|
def check_kernel(r1, prefix, nexthops, vrf, expected_p=True, expected_nh=True):
|
|
vrfstr = f" vrf {vrf}" if vrf else ""
|
|
|
|
net = ipaddress.ip_network(prefix)
|
|
if net.version == 6:
|
|
kernel = r1.run(f"ip -6 route show{vrfstr} {prefix}")
|
|
else:
|
|
kernel = r1.run(f"ip -4 route show{vrfstr} {prefix}")
|
|
|
|
if expected_p:
|
|
assert prefix in kernel, f"Failed to find \n'{prefix}'\n in \n'{kernel:.1920}'"
|
|
else:
|
|
assert (
|
|
prefix not in kernel
|
|
), f"Failed found \n'{prefix}'\n in \n'{kernel:.1920}'"
|
|
|
|
if not expected_p:
|
|
return
|
|
|
|
for nh in nexthops:
|
|
if expected_nh:
|
|
assert f"{nh}" in kernel, f"Failed to find \n'{nh}'\n in \n'{kernel:.1920}'"
|
|
else:
|
|
assert (
|
|
f"{nh}" not in kernel
|
|
), f"Failed found \n'{nh}'\n in \n'{kernel:.1920}'"
|
|
|
|
|
|
def test_static_vrf(tgen):
|
|
if tgen.routers_have_failure():
|
|
pytest.skip(tgen.errors)
|
|
|
|
r1 = tgen.gears["r1"]
|
|
|
|
# Check initial configuration
|
|
check_kernel(r1, "198.51.100.1", ["192.0.2.2", "192.0.2.130"], None)
|
|
check_kernel(r1, "198.51.100.2", ["r1-eth0", "r1-eth1"], None)
|
|
check_kernel(r1, "203.0.113.1", ["192.0.2.130"], "red")
|
|
check_kernel(r1, "203.0.113.2", ["r1-eth1"], "red")
|
|
check_kernel(r1, "203.0.113.129", ["192.0.2.2"], "blue")
|
|
check_kernel(r1, "203.0.113.130", ["r1-eth0"], "blue")
|
|
|
|
# Delete VRF red
|
|
r1.net.del_iface("red")
|
|
|
|
# Check that "red" nexthops are removed, "blue" nexthops are still there
|
|
check_kernel(r1, "198.51.100.1", ["192.0.2.2"], None, expected_nh=False)
|
|
check_kernel(r1, "198.51.100.1", ["192.0.2.130"], None)
|
|
check_kernel(r1, "198.51.100.2", ["r1-eth0"], None, expected_nh=False)
|
|
check_kernel(r1, "198.51.100.2", ["r1-eth1"], None)
|
|
check_kernel(r1, "203.0.113.129", ["192.0.2.2"], "blue", expected_p=False)
|
|
check_kernel(r1, "203.0.113.130", ["r1-eth0"], "blue", expected_p=False)
|
|
|
|
# Delete VRF blue
|
|
r1.net.del_iface("blue")
|
|
|
|
# Check that "blue" nexthops are removed
|
|
check_kernel(r1, "198.51.100.1", ["192.0.2.130"], None, expected_p=False)
|
|
check_kernel(r1, "198.51.100.2", ["r1-eth1"], None, expected_p=False)
|
|
|
|
# Add VRF red back, attach "eth0" to it
|
|
r1.net.add_l3vrf("red", 10)
|
|
r1.net.attach_iface_to_l3vrf("r1-eth0", "red")
|
|
|
|
# Check that "red" nexthops are restored
|
|
check_kernel(r1, "198.51.100.1", ["192.0.2.2"], None)
|
|
check_kernel(r1, "198.51.100.2", ["r1-eth0"], None)
|
|
|
|
# Add VRF blue back, attach "eth1" to it
|
|
r1.net.add_l3vrf("blue", 20)
|
|
r1.net.attach_iface_to_l3vrf("r1-eth1", "blue")
|
|
|
|
# Check that everything is restored
|
|
check_kernel(r1, "198.51.100.1", ["192.0.2.2", "192.0.2.130"], None)
|
|
check_kernel(r1, "198.51.100.2", ["r1-eth0", "r1-eth1"], None)
|
|
check_kernel(r1, "203.0.113.1", ["192.0.2.130"], "red")
|
|
check_kernel(r1, "203.0.113.2", ["r1-eth1"], "red")
|
|
check_kernel(r1, "203.0.113.129", ["192.0.2.2"], "blue")
|
|
check_kernel(r1, "203.0.113.130", ["r1-eth0"], "blue")
|