2 # SPDX-License-Identifier: ISC
5 # bgp_set_local-preference_add_subtract.py
6 # Part of NetDEF Topology Tests
8 # Copyright (c) 2020 by
9 # Donatas Abraitis <donatas.abraitis@gmail.com>
13 bgp_set_local-preference_add_subtract.py:
14 Test if we can add/subtract the value to/from an existing
15 LOCAL_PREF in route-maps.
24 CWD
= os
.path
.dirname(os
.path
.realpath(__file__
))
25 sys
.path
.append(os
.path
.join(CWD
, "../"))
27 # pylint: disable=C0413
28 from lib
import topotest
29 from lib
.topogen
import Topogen
, TopoRouter
, get_topogen
31 pytestmark
= [pytest
.mark
.bgpd
]
35 for routern
in range(1, 4):
36 tgen
.add_router("r{}".format(routern
))
38 switch
= tgen
.add_switch("s1")
39 switch
.add_link(tgen
.gears
["r1"])
40 switch
.add_link(tgen
.gears
["r2"])
41 switch
.add_link(tgen
.gears
["r3"])
44 def setup_module(mod
):
45 tgen
= Topogen(build_topo
, mod
.__name
__)
48 router_list
= tgen
.routers()
50 for i
, (rname
, router
) in enumerate(router_list
.items(), 1):
52 TopoRouter
.RD_ZEBRA
, os
.path
.join(CWD
, "{}/zebra.conf".format(rname
))
55 TopoRouter
.RD_BGP
, os
.path
.join(CWD
, "{}/bgpd.conf".format(rname
))
61 def teardown_module(mod
):
66 def test_bgp_set_local_preference():
69 if tgen
.routers_have_failure():
70 pytest
.skip(tgen
.errors
)
72 router
= tgen
.gears
["r1"]
74 def _bgp_converge(router
):
75 output
= json
.loads(router
.vtysh_cmd("show ip bgp neighbor json"))
78 "bgpState": "Established",
79 "addressFamilyInfo": {"ipv4Unicast": {"acceptedPrefixCounter": 3}},
82 "bgpState": "Established",
83 "addressFamilyInfo": {"ipv4Unicast": {"acceptedPrefixCounter": 3}},
86 return topotest
.json_cmp(output
, expected
)
88 def _bgp_check_local_preference(router
):
89 output
= json
.loads(router
.vtysh_cmd("show bgp ipv4 unicast json"))
92 "10.10.10.2/32": [{"locPrf": 160}],
93 "10.10.10.3/32": [{"locPrf": 40}],
94 "172.16.255.254/32": [
95 {"locPrf": 50, "nexthops": [{"ip": "192.168.255.3"}]},
96 {"locPrf": 150, "nexthops": [{"ip": "192.168.255.2"}]},
100 return topotest
.json_cmp(output
, expected
)
102 test_func
= functools
.partial(_bgp_converge
, router
)
103 success
, result
= topotest
.run_and_expect(test_func
, None, count
=15, wait
=0.5)
105 assert result
is None, 'Failed to see BGP convergence in "{}"'.format(router
)
107 test_func
= functools
.partial(_bgp_check_local_preference
, router
)
108 success
, result
= topotest
.run_and_expect(test_func
, None, count
=15, wait
=0.5)
110 assert result
is None, 'Failed to see applied BGP local-preference in "{}"'.format(
115 if __name__
== "__main__":
116 args
= ["-s"] + sys
.argv
[1:]
117 sys
.exit(pytest
.main(args
))