88from swsssdk import ConfigDBConnector
99
1010from .mock_tables import dbconnector
11-
11+ from . import show_ip_route_common
1212
1313test_path = os .path .dirname (os .path .abspath (__file__ ))
1414modules_path = os .path .dirname (test_path )
@@ -85,21 +85,6 @@ def setup_single_bgp_instance(request):
8585 elif request .param == 'v6' :
8686 bgp_mocked_json = os .path .join (
8787 test_path , 'mock_tables' , 'ipv6_bgp_summary.json' )
88- elif request .param == 'ip_route' :
89- bgp_mocked_json = os .path .join (
90- test_path , 'mock_tables' , 'ip_route.json' )
91- elif request .param == 'ip_specific_route' :
92- bgp_mocked_json = os .path .join (
93- test_path , 'mock_tables' , 'ip_specific_route.json' )
94- elif request .param == 'ip_special_route' :
95- bgp_mocked_json = os .path .join (
96- test_path , 'mock_tables' , 'ip_special_route.json' )
97- elif request .param == 'ipv6_route' :
98- bgp_mocked_json = os .path .join (
99- test_path , 'mock_tables' , 'ipv6_route.json' )
100- elif request .param == 'ipv6_specific_route' :
101- bgp_mocked_json = os .path .join (
102- test_path , 'mock_tables' , 'ipv6_specific_route.json' )
10388 else :
10489 bgp_mocked_json = os .path .join (
10590 test_path , 'mock_tables' , 'dummy.json' )
@@ -111,12 +96,28 @@ def mock_run_bgp_command(vtysh_cmd, bgp_namespace):
11196 return mock_frr_data
11297 return ""
11398
114- def mock_run_bgp_ipv6_err_command (vtysh_cmd , bgp_namespace ):
115- return "% Unknown command: show ipv6 route garbage"
99+ def mock_run_show_ip_route_commands (request ):
100+ if request .param == 'ipv6_route_err' :
101+ return show_ip_route_common .show_ipv6_route_err_expected_output
102+ elif request .param == 'ip_route' :
103+ return show_ip_route_common .show_ip_route_expected_output
104+ elif request .param == 'ip_specific_route' :
105+ return show_ip_route_common .show_specific_ip_route_expected_output
106+ elif request .param == 'ip_special_route' :
107+ return show_ip_route_common .show_special_ip_route_expected_output
108+ elif request .param == 'ipv6_route' :
109+ return show_ip_route_common .show_ipv6_route_expected_output
110+ elif request .param == 'ipv6_specific_route' :
111+ return show_ip_route_common .show_ipv6_route_single_json_expected_output
112+ else :
113+ return ""
114+
116115
117- if request .param == 'ipv6_route_err' :
116+ if any ([request .param == 'ipv6_route_err' , request .param == 'ip_route' ,\
117+ request .param == 'ip_specific_route' , request .param == 'ip_special_route' ,\
118+ request .param == 'ipv6_route' , request .param == 'ipv6_specific_route' ]):
118119 bgp_util .run_bgp_command = mock .MagicMock (
119- return_value = mock_run_bgp_ipv6_err_command ( "" , "" ))
120+ return_value = mock_run_show_ip_route_commands ( request ))
120121 else :
121122 bgp_util .run_bgp_command = mock .MagicMock (
122123 return_value = mock_run_bgp_command ("" , "" ))
@@ -134,6 +135,12 @@ def setup_multi_asic_bgp_instance(request):
134135 m_asic_json_file = 'ipv6_specific_route.json'
135136 elif request .param == 'ipv6_route' :
136137 m_asic_json_file = 'ipv6_route.json'
138+ elif request .param == 'ip_special_route' :
139+ m_asic_json_file = 'ip_special_route.json'
140+ elif request .param == 'ip_empty_route' :
141+ m_asic_json_file = 'ip_empty_route.json'
142+ elif request .param == 'ip_specific_route_on_1_asic' :
143+ m_asic_json_file = 'ip_special_route_asic0_only.json'
137144 else :
138145 bgp_mocked_json = os .path .join (
139146 test_path , 'mock_tables' , 'dummy.json' )
0 commit comments