1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
|
#!/usr/bin/env python
# SPDX-License-Identifier: ISC
#
# test_zebra_multiple_connected.py
#
# Copyright (c) 2022 by
# Nvidia Corporation
# Donald Sharp
#
"""
test_zebra_multiple_connected.py: Testing multiple connected
"""
import os
import re
import sys
import pytest
import json
from functools import partial
# Save the Current Working Directory to find configuration files.
CWD = os.path.dirname(os.path.realpath(__file__))
sys.path.append(os.path.join(CWD, "../"))
# pylint: disable=C0413
# Import topogen and topotest helpers
from lib import topotest
from lib.topogen import Topogen, TopoRouter, get_topogen
from lib.topolog import logger
# Required to instantiate the topology builder class.
#####################################################
##
## Network Topology Definition
##
#####################################################
def build_topo(tgen):
for routern in range(1, 4):
tgen.add_router("r{}".format(routern))
# On main router
# First switch is for a dummy interface (for local network)
switch = tgen.add_switch("sw1")
switch.add_link(tgen.gears["r1"])
# Switches for zebra
# switch 2 switch is for connection to zebra router
switch = tgen.add_switch("sw2")
switch.add_link(tgen.gears["r1"])
switch.add_link(tgen.gears["r2"])
# switch 4 is stub on remote zebra router
switch = tgen.add_switch("sw4")
switch.add_link(tgen.gears["r3"])
# switch 3 is between zebra routers
switch = tgen.add_switch("sw3")
switch.add_link(tgen.gears["r2"])
switch.add_link(tgen.gears["r3"])
#####################################################
##
## Tests starting
##
#####################################################
def setup_module(module):
"Setup topology"
tgen = Topogen(build_topo, module.__name__)
tgen.start_topology()
# This is a sample of configuration loading.
router_list = tgen.routers()
for rname, router in router_list.items():
router.load_config(
TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
)
tgen.start_router()
def teardown_module(_mod):
"Teardown the pytest environment"
tgen = get_topogen()
# This function tears down the whole topology.
tgen.stop_topology()
def test_zebra_connected_multiple():
"Test multiple connected routes that have a kernel route pointing at one"
tgen = get_topogen()
# Don't run this test if we have any failure.
if tgen.routers_have_failure():
pytest.skip(tgen.errors)
router = tgen.gears["r1"]
router.run("ip route add 192.168.1.1/32 via 10.0.1.99 dev r1-eth1")
router.run("ip link add dummy1 type dummy")
router.run("ip link set dummy1 up")
router.run("ip link set dummy1 down")
routes = "{}/{}/ip_route.json".format(CWD, router.name)
expected = json.loads(open(routes).read())
test_func = partial(
topotest.router_json_cmp, router, "show ip route json", expected
)
_, result = topotest.run_and_expect(test_func, None, count=20, wait=1)
assert result is None, "Kernel route is missing from zebra"
def test_zebra_system_recursion():
"Test a system route recursing through another system route"
tgen = get_topogen()
if tgen.routers_have_failure():
pytest.skip(tgen.errors)
router = tgen.gears["r1"]
router.run("ip route add 10.0.1.30/32 dev r1-eth1")
router.run("ip route add 10.9.9.0/24 via 10.0.1.30 dev r1-eth1")
router.run("ip link add dummy2 type dummy")
router.run("ip link set dummy2 up")
router.run("ip link set dummy2 down")
routes = "{}/{}/ip_route2.json".format(CWD, router.name)
expected = json.loads(open(routes).read())
test_func = partial(
topotest.router_json_cmp, router, "show ip route json", expected
)
_, result = topotest.run_and_expect(test_func, None, count=20, wait=1)
assert result is None, "Kernel route is missing from zebra"
if __name__ == "__main__":
args = ["-s"] + sys.argv[1:]
sys.exit(pytest.main(args))
|
