def __restart_peers_common(run_command_results, run_command_expects,
last_log_crit_message, mocked_log_crit):
cfg_mgr = MagicMock()
common_objs = {
'directory': Directory(),
'cfg_mgr': cfg_mgr,
'tf': TemplateFabric(),
'constants': global_constants,
}
m = BBRMgr(common_objs, "CONFIG_DB", "BGP_BBR")
m.bbr_enabled_pgs = {
"PEER_V4": ["ipv4", "ipv6"],
"PEER_V6": ["ipv6"],
}
def run_command_mock(cmd):
assert cmd == run_command_expects[run_command_mock.run]
res = run_command_results[run_command_mock.run]
run_command_mock.run += 1
return res
run_command_mock.run = 0
bgpcfgd.managers_bbr.run_command = run_command_mock
#lambda cmd: (0, "", "")
m._BBRMgr__restart_peers()
if last_log_crit_message is not None:
mocked_log_crit.assert_called_with(last_log_crit_message)
def __set_prepare_config_common(status,
bbr_enabled_pgs,
available_pgs,
mapping_pgs,
expected_cmds,
bbr_applied_pgs=None):
cfg_mgr = MagicMock()
common_objs = {
'directory': Directory(),
'cfg_mgr': cfg_mgr,
'tf': TemplateFabric(),
'constants': global_constants,
}
m = BBRMgr(common_objs, "CONFIG_DB", "BGP_BBR")
m.directory.data = {
"CONFIG_DB__DEVICE_METADATA": {
"localhost": {
"bgp_asn": "65500"
}
}
}
m.bbr_enabled_pgs = bbr_enabled_pgs
m._BBRMgr__get_available_peer_groups = MagicMock(
return_value=available_pgs)
m._BBRMgr__get_available_peers_per_peer_group = MagicMock(
return_value=mapping_pgs)
cmds, peer_groups = m._BBRMgr__set_prepare_config(status)
assert cmds == expected_cmds
assert set(peer_groups) == (available_pgs
if not bbr_applied_pgs else bbr_applied_pgs)
def __set_prepare_config_common(status, bbr_enabled_pgs, expected_cmds):
cfg_mgr = MagicMock()
common_objs = {
'directory': Directory(),
'cfg_mgr': cfg_mgr,
'tf': TemplateFabric(),
'constants': global_constants,
}
m = BBRMgr(common_objs, "CONFIG_DB", "BGP_BBR")
m.directory.data = {
"CONFIG_DB__DEVICE_METADATA": {
"localhost": {
"bgp_asn": "65500"
}
}
}
m.bbr_enabled_pgs = bbr_enabled_pgs
cmds = m._BBRMgr__set_prepare_config(status)
assert cmds == expected_cmds