@@ -310,7 +310,7 @@ def mock_run_bgp_command_for_static(vtysh_cmd, bgp_namespace="", vtysh_shell_cmd
310
310
else :
311
311
return ""
312
312
313
- def mock_run_bgp_command (vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .RVTYSH_COMMAND ):
313
+ def mock_run_bgp_command (vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .RVTYSH_COMMAND , exit_on_fail = True ):
314
314
if m_asic_json_file .startswith ('bgp_v4_network' ) or \
315
315
m_asic_json_file .startswith ('bgp_v6_network' ):
316
316
return mock_show_bgp_network_multi_asic (m_asic_json_file )
@@ -328,7 +328,8 @@ def mock_run_bgp_command(vtysh_cmd, bgp_namespace, vtysh_shell_cmd=constants.RVT
328
328
else :
329
329
return ""
330
330
331
- def mock_run_show_sum_bgp_command (vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .VTYSH_COMMAND ):
331
+ def mock_run_show_sum_bgp_command (
332
+ vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .VTYSH_COMMAND , exit_on_fail = True ):
332
333
if vtysh_cmd == "show ip bgp summary json" :
333
334
m_asic_json_file = 'no_bgp_neigh.json'
334
335
else :
@@ -343,7 +344,8 @@ def mock_run_show_sum_bgp_command(vtysh_cmd, bgp_namespace, vtysh_shell_cmd=cons
343
344
else :
344
345
return ""
345
346
346
- def mock_run_show_summ_bgp_command_no_ext_neigh_on_all_asic (vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .VTYSH_COMMAND ):
347
+ def mock_run_show_summ_bgp_command_no_ext_neigh_on_all_asic (
348
+ vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .VTYSH_COMMAND , exit_on_fail = True ):
347
349
if vtysh_cmd == "show ip bgp summary json" :
348
350
m_asic_json_file = 'no_ext_bgp_neigh.json'
349
351
else :
@@ -358,7 +360,8 @@ def mock_run_show_summ_bgp_command_no_ext_neigh_on_all_asic(vtysh_cmd, bgp_names
358
360
else :
359
361
return ""
360
362
361
- def mock_run_show_summ_bgp_command_no_ext_neigh_on_asic1 (vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .VTYSH_COMMAND ):
363
+ def mock_run_show_summ_bgp_command_no_ext_neigh_on_asic1 (
364
+ vtysh_cmd , bgp_namespace , vtysh_shell_cmd = constants .VTYSH_COMMAND , exit_on_fail = True ):
362
365
if vtysh_cmd == "show ip bgp summary json" :
363
366
if bgp_namespace == "asic1" :
364
367
m_asic_json_file = 'no_ext_bgp_neigh.json'
0 commit comments