2 # SPDX-License-Identifier: ISC
5 # bgp_aggregate-address_route-map.py
6 # Part of NetDEF Topology Tests
8 # Copyright (c) 2019 by
9 # Network Device Education Foundation, Inc. ("NetDEF")
13 bgp_aggregate-address_route-map.py:
15 Test if works the following commands:
17 address-family ipv4 unicast
18 aggregate-address 192.168.255.0/24 route-map aggr-rmap
20 route-map aggr-rmap permit 10
30 CWD
= os
.path
.dirname(os
.path
.realpath(__file__
))
31 sys
.path
.append(os
.path
.join(CWD
, "../"))
33 # pylint: disable=C0413
34 from lib
import topotest
35 from lib
.topogen
import Topogen
, TopoRouter
, get_topogen
37 pytestmark
= [pytest
.mark
.bgpd
]
41 for routern
in range(1, 3):
42 tgen
.add_router("r{}".format(routern
))
44 switch
= tgen
.add_switch("s1")
45 switch
.add_link(tgen
.gears
["r1"])
46 switch
.add_link(tgen
.gears
["r2"])
49 def setup_module(mod
):
50 tgen
= Topogen(build_topo
, mod
.__name
__)
53 router_list
= tgen
.routers()
55 for i
, (rname
, router
) in enumerate(router_list
.items(), 1):
57 TopoRouter
.RD_ZEBRA
, os
.path
.join(CWD
, "{}/zebra.conf".format(rname
))
60 TopoRouter
.RD_BGP
, os
.path
.join(CWD
, "{}/bgpd.conf".format(rname
))
66 def teardown_module(mod
):
71 def test_bgp_maximum_prefix_invalid():
74 if tgen
.routers_have_failure():
75 pytest
.skip(tgen
.errors
)
77 router
= tgen
.gears
["r2"]
79 def _bgp_converge(router
):
80 output
= json
.loads(router
.vtysh_cmd("show ip bgp neighbor 192.168.255.1 json"))
83 "bgpState": "Established",
84 "addressFamilyInfo": {"ipv4Unicast": {"acceptedPrefixCounter": 3}},
87 return topotest
.json_cmp(output
, expected
)
89 def _bgp_aggregate_address_has_metric(router
):
90 output
= json
.loads(router
.vtysh_cmd("show ip bgp 172.16.255.0/24 json"))
91 expected
= {"paths": [{"metric": 123}]}
92 return topotest
.json_cmp(output
, expected
)
94 test_func
= functools
.partial(_bgp_converge
, router
)
95 success
, result
= topotest
.run_and_expect(test_func
, None, count
=30, wait
=0.5)
97 assert result
is None, 'Failed to see bgp convergence in "{}"'.format(router
)
99 test_func
= functools
.partial(_bgp_aggregate_address_has_metric
, router
)
100 success
, result
= topotest
.run_and_expect(test_func
, None, count
=30, wait
=0.5)
104 ), 'Failed to see applied metric for aggregated prefix in "{}"'.format(router
)
107 if __name__
== "__main__":
108 args
= ["-s"] + sys
.argv
[1:]
109 sys
.exit(pytest
.main(args
))