4 # Copyright (c) 2021 by
5 # Donatas Abraitis <donatas.abraitis@gmail.com>
7 # Permission to use, copy, modify, and/or distribute this software
8 # for any purpose with or without fee is hereby granted, provided
9 # that the above copyright notice and this permission notice appear
12 # THE SOFTWARE IS PROVIDED "AS IS" AND NETDEF DISCLAIMS ALL WARRANTIES
13 # WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
14 # MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL NETDEF BE LIABLE FOR
15 # ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY
16 # DAMAGES WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS,
17 # WHETHER IN AN ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS
18 # ACTION, ARISING OUT OF OR IN CONNECTION WITH THE USE OR PERFORMANCE
23 Test if peer-group works for numbered and unnumbered configurations.
32 CWD
= os
.path
.dirname(os
.path
.realpath(__file__
))
33 sys
.path
.append(os
.path
.join(CWD
, "../"))
35 # pylint: disable=C0413
36 from lib
import topotest
37 from lib
.topogen
import Topogen
, TopoRouter
, get_topogen
38 from lib
.topolog
import logger
39 from mininet
.topo
import Topo
42 pytestmark
= [pytest
.mark
.bgpd
]
45 class TemplateTopo(Topo
):
46 def build(self
, *_args
, **_opts
):
47 tgen
= get_topogen(self
)
49 for routern
in range(1, 4):
50 tgen
.add_router("r{}".format(routern
))
52 switch
= tgen
.add_switch("s1")
53 switch
.add_link(tgen
.gears
["r1"])
54 switch
.add_link(tgen
.gears
["r2"])
55 switch
.add_link(tgen
.gears
["r3"])
58 def setup_module(mod
):
59 tgen
= Topogen(TemplateTopo
, mod
.__name
__)
62 router_list
= tgen
.routers()
64 for i
, (rname
, router
) in enumerate(router_list
.items(), 1):
66 TopoRouter
.RD_ZEBRA
, os
.path
.join(CWD
, "{}/zebra.conf".format(rname
))
69 TopoRouter
.RD_BGP
, os
.path
.join(CWD
, "{}/bgpd.conf".format(rname
))
75 def teardown_module(mod
):
80 def test_bgp_peer_group():
83 if tgen
.routers_have_failure():
84 pytest
.skip(tgen
.errors
)
86 def _bgp_peer_group_configured():
87 output
= json
.loads(tgen
.gears
["r1"].vtysh_cmd("show ip bgp neighbor json"))
89 "r1-eth0": {"peerGroup": "PG", "bgpState": "Established"},
90 "192.168.255.3": {"peerGroup": "PG", "bgpState": "Established"},
92 return topotest
.json_cmp(output
, expected
)
94 test_func
= functools
.partial(_bgp_peer_group_configured
)
95 success
, result
= topotest
.run_and_expect(test_func
, None, count
=60, wait
=0.5)
97 assert result
is None, 'Failed bgp convergence in "{}"'.format(tgen
.gears
["r1"])
100 if __name__
== "__main__":
101 args
= ["-s"] + sys
.argv
[1:]
102 sys
.exit(pytest
.main(args
))