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
|
#!/usr/bin/env python
# SPDX-License-Identifier: ISC
#
# test_srv6_encap_src_addr.py
# Part of NetDEF Topology Tests
#
# Copyright (c) 2022 by
# University of Rome Tor Vergata, Carmine Scarpitta <carmine.scarpitta@uniroma2.it>
#
"""
test_srv6_encap_src_addr.py:
Test for SRv6 encap source address on zebra
"""
import os
import sys
import json
import pytest
import functools
CWD = os.path.dirname(os.path.realpath(__file__))
sys.path.append(os.path.join(CWD, "../"))
# pylint: disable=C0413
from lib import topotest
from lib.topogen import Topogen, TopoRouter, get_topogen
from lib.topolog import logger
pytestmark = [pytest.mark.bgpd, pytest.mark.sharpd]
def open_json_file(filename):
try:
with open(filename, "r") as f:
return json.load(f)
except IOError:
assert False, "Could not read file {}".format(filename)
def setup_module(mod):
tgen = Topogen({None: "r1"}, mod.__name__)
tgen.start_topology()
for rname, router in tgen.routers().items():
router.run("/bin/bash {}/{}/setup.sh".format(CWD, rname))
router.load_config(
TopoRouter.RD_ZEBRA, os.path.join(CWD, "{}/zebra.conf".format(rname))
)
router.load_config(
TopoRouter.RD_BGP, os.path.join(CWD, "{}/bgpd.conf".format(rname))
)
router.load_config(
TopoRouter.RD_SHARP, os.path.join(CWD, "{}/sharpd.conf".format(rname))
)
tgen.start_router()
def teardown_module(mod):
tgen = get_topogen()
tgen.stop_topology()
def test_zebra_srv6_encap_src_addr(tgen):
"Test SRv6 encapsulation source address."
logger.info(
"Test SRv6 encapsulation source address."
)
r1 = tgen.gears["r1"]
# Generate expected results
json_file = "{}/r1/expected_srv6_encap_src_addr.json".format(CWD)
expected = json.loads(open(json_file).read())
ok = topotest.router_json_cmp_retry(r1, "show segment-routing srv6 manager json", expected)
assert ok, '"r1" JSON output mismatches'
output = r1.cmd("ip sr tunsrc show")
assert output == "tunsrc addr fc00:0:1::1\n"
def test_zebra_srv6_encap_src_addr_unset(tgen):
"Test SRv6 encapsulation source address unset."
logger.info(
"Test SRv6 encapsulation source address unset."
)
r1 = tgen.gears["r1"]
# Unset SRv6 encapsulation source address
r1.vtysh_cmd(
"""
configure terminal
segment-routing
srv6
encapsulation
no source-address
"""
)
# Generate expected results
json_file = "{}/r1/expected_srv6_encap_src_addr_unset.json".format(CWD)
expected = json.loads(open(json_file).read())
ok = topotest.router_json_cmp_retry(r1, "show segment-routing srv6 manager json", expected)
assert ok, '"r1" JSON output mismatches'
output = r1.cmd("ip sr tunsrc show")
assert output == "tunsrc addr ::\n"
def test_zebra_srv6_encap_src_addr_set(tgen):
"Test SRv6 encapsulation source address set."
logger.info(
"Test SRv6 encapsulation source address set."
)
r1 = tgen.gears["r1"]
# Set SRv6 encapsulation source address
r1.vtysh_cmd(
"""
configure terminal
segment-routing
srv6
encapsulation
source-address fc00:0:1::1
"""
)
# Generate expected results
json_file = "{}/r1/expected_srv6_encap_src_addr_set.json".format(CWD)
expected = json.loads(open(json_file).read())
ok = topotest.router_json_cmp_retry(r1, "show segment-routing srv6 manager json", expected)
assert ok, '"r1" JSON output mismatches'
output = r1.cmd("ip sr tunsrc show")
assert output == "tunsrc addr fc00:0:1::1\n"
if __name__ == "__main__":
args = ["-s"] + sys.argv[1:]
sys.exit(pytest.main(args))
|
