Skip to content

Commit f37fe1e

Browse files
shi-suderan1980
authored andcommitted
[bgpcfgd]: Fix unit test for bgp manager (sonic-net#6668)
The check for check_neig_meta was not following the value in constants. Fix it to match the value in constants.
1 parent a1d1da0 commit f37fe1e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/sonic-bgpcfgd/tests/test_bgp.py

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,11 +12,12 @@
1212

1313
def constructor():
1414
cfg_mgr = MagicMock()
15+
constants = load_constants()['constants']
1516
common_objs = {
1617
'directory': Directory(),
1718
'cfg_mgr': cfg_mgr,
1819
'tf': TemplateFabric(TEMPLATE_PATH),
19-
'constants': load_constants()['constants']
20+
'constants': constants
2021
}
2122

2223
return_value_map = {
@@ -27,7 +28,7 @@ def constructor():
2728
bgpcfgd.managers_bgp.run_command = lambda cmd: return_value_map[str(cmd)]
2829
m = bgpcfgd.managers_bgp.BGPPeerMgrBase(common_objs, "CONFIG_DB", swsscommon.CFG_BGP_NEIGHBOR_TABLE_NAME, "general", True)
2930
assert m.peer_type == "general"
30-
assert m.check_neig_meta == False # Because constants['bgp']['use_neighbors_meta'] is false in constants.yml
31+
assert m.check_neig_meta == ('bgp' in constants and 'use_neighbors_meta' in constants['bgp'] and constants['bgp']['use_neighbors_meta'])
3132

3233
m.directory.put("CONFIG_DB", swsscommon.CFG_DEVICE_METADATA_TABLE_NAME, "localhost", {"bgp_asn": "65100"})
3334
m.directory.put("CONFIG_DB", swsscommon.CFG_LOOPBACK_INTERFACE_TABLE_NAME, "Loopback0|11.11.11.11/32", {})

0 commit comments

Comments
 (0)