]>
git.proxmox.com Git - mirror_frr.git/blob - tests/topotests/bgp_lu_topo1/test_bgp_lu.py
5 # Part of NetDEF Topology Tests
7 # Copyright (c) 2020 by Volta Networks
9 # Permission to use, copy, modify, and/or distribute this software
10 # for any purpose with or without fee is hereby granted, provided
11 # that the above copyright notice and this permission notice appear
14 # THE SOFTWARE IS PROVIDED "AS IS" AND NETDEF DISCLAIMS ALL WARRANTIES
15 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
16 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NETDEF BE LIABLE FOR
17 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY
18 # DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
19 # WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
20 # ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
25 test_bgp_lu.py: Test BGP LU label allocation
31 from functools
import partial
32 from time
import sleep
35 # Save the Current Working Directory to find configuration files.
36 CWD
= os
.path
.dirname(os
.path
.realpath(__file__
))
37 sys
.path
.append(os
.path
.join(CWD
, "../"))
39 # pylint: disable=C0413
40 # Import topogen and topotest helpers
41 from lib
import topotest
42 from lib
.topogen
import Topogen
, TopoRouter
, get_topogen
43 from lib
.topolog
import logger
45 # Required to instantiate the topology builder class.
46 from mininet
.topo
import Topo
49 pytestmark
= [pytest
.mark
.bgpd
]
52 # Basic scenario for BGP-LU. Nodes are directly connected.
53 # Node 3 is advertising many routes to 2, which advertises them
54 # as BGP-LU to 1; this way we get routes with actual labels, as
55 # opposed to implicit-null routes in the 2-node case.
57 # AS1 BGP-LU AS2 iBGP AS2
58 # +-----+ +-----+ +-----+
60 # | 1 +----------------+ 2 +-----------------+ 3 |
61 # | | 10.0.0.0/24 | | 10.0.1.0/24 | |
62 # +-----+ +-----+ +-----+
65 class TemplateTopo(Topo
):
66 "Test topology builder"
68 def build(self
, *_args
, **_opts
):
70 tgen
= get_topogen(self
)
72 # This function only purpose is to define allocation and relationship
73 # between routers, switches and hosts.
82 switch
= tgen
.add_switch("s1")
83 switch
.add_link(tgen
.gears
["R1"])
84 switch
.add_link(tgen
.gears
["R2"])
87 switch
= tgen
.add_switch("s2")
88 switch
.add_link(tgen
.gears
["R2"])
89 switch
.add_link(tgen
.gears
["R3"])
92 def setup_module(mod
):
93 "Sets up the pytest environment"
94 # This function initiates the topology build with Topogen...
95 tgen
= Topogen(TemplateTopo
, mod
.__name
__)
96 # ... and here it calls Mininet initialization functions.
99 # This is a sample of configuration loading.
100 router_list
= tgen
.routers()
102 # For all registred routers, load the zebra configuration file
103 for rname
, router
in router_list
.items():
105 TopoRouter
.RD_ZEBRA
, os
.path
.join(CWD
, "{}/zebra.conf".format(rname
))
108 TopoRouter
.RD_BGP
, os
.path
.join(CWD
, "{}/bgpd.conf".format(rname
))
111 # After loading the configurations, this function loads configured daemons.
115 def teardown_module(mod
):
116 "Teardown the pytest environment"
119 # This function tears down the whole topology.
123 def check_labelpool(router
):
124 json_file
= "{}/{}/labelpool.summ.json".format(CWD
, router
.name
)
125 expected
= json
.loads(open(json_file
).read())
128 topotest
.router_json_cmp
, router
, "show bgp labelpool summary json", expected
130 _
, result
= topotest
.run_and_expect(test_func
, None, count
=20, wait
=1)
131 assertmsg
= '"{}" JSON output mismatches - Did not converge'.format(router
.name
)
132 assert result
is None, assertmsg
135 def test_converge_bgplu():
136 "Wait for protocol convergence"
139 # Don't run this test if we have any failure.
140 if tgen
.routers_have_failure():
141 pytest
.skip(tgen
.errors
)
143 # tgen.mininet_cli();
144 r1
= tgen
.gears
["R1"]
145 r2
= tgen
.gears
["R2"]
151 def test_clear_bgplu():
152 "Wait for protocol convergence"
155 # Don't run this test if we have any failure.
156 if tgen
.routers_have_failure():
157 pytest
.skip(tgen
.errors
)
159 # tgen.mininet_cli();
160 r1
= tgen
.gears
["R1"]
161 r2
= tgen
.gears
["R2"]
163 r1
.vtysh_cmd("clear bgp 10.0.0.2")
167 r2
.vtysh_cmd("clear bgp 10.0.1.3")
171 r1
.vtysh_cmd("clear bgp 10.0.0.2")
172 r2
.vtysh_cmd("clear bgp 10.0.1.3")
177 def test_memory_leak():
178 "Run the memory leak test and report results."
180 if not tgen
.is_memleak_enabled():
181 pytest
.skip("Memory leak test/report is disabled")
183 tgen
.report_memory_leaks()
186 if __name__
== "__main__":
187 args
= ["-s"] + sys
.argv
[1:]
188 sys
.exit(pytest
.main(args
))