Added Multi-ASIC support for show ip(v6) route

show/main.py

@@ -1,5 +1,6 @@
 #! /usr/bin/python -u
 
+import bgp_common
 import json
 import netaddr
 import os
@@ -786,17 +787,12 @@ def get_bgp_peer():
 
 @ip.command()
 @click.argument('args', metavar='[IPADDRESS] [vrf <vrf_name>] [...]', nargs=-1, required=False)
+@multi_asic_util.multi_asic_click_options
 @click.option('--verbose', is_flag=True, help="Enable verbose output")
-def route(args, verbose):
+def route(args, namespace, display, verbose):
     """Show IP (IPv4) routing table"""
-    cmd = 'sudo vtysh -c "show ip route'
-
-    for arg in args:
-        cmd += " " + str(arg)
-
-    cmd += '"'
-
-    run_command(cmd, display_cmd=verbose)
+    # Call common handler to handle the show ip route cmd
+    bgp_common.show_routes(args, namespace, display, verbose, "ip")
 
 #
 # 'prefix-list' subcommand ("show ip prefix-list")
@@ -909,17 +905,12 @@ def interfaces():
 
 @ipv6.command()
 @click.argument('args', metavar='[IPADDRESS] [vrf <vrf_name>] [...]', nargs=-1, required=False)
+@multi_asic_util.multi_asic_click_options
 @click.option('--verbose', is_flag=True, help="Enable verbose output")
-def route(args, verbose):
+def route(args, namespace, display, verbose):
     """Show IPv6 routing table"""
-    cmd = 'sudo vtysh -c "show ipv6 route'
-
-    for arg in args:
-        cmd += " " + str(arg)
-
-    cmd += '"'
-
-    run_command(cmd, display_cmd=verbose)
+    # Call common handler to handle the show ipv6 route cmd
+    bgp_common.show_routes(args, namespace, display, verbose, "ipv6")
 
 
 # 'protocol' command

tests/conftest.py

@@ -81,18 +81,27 @@ def setup_single_bgp_instance(request):
     import utilities_common.bgp_util as bgp_util
 
     if request.param == 'v4':
-        bgp_summary_json = os.path.join(
+        bgp_mocked_json = os.path.join(
             test_path, 'mock_tables', 'ipv4_bgp_summary.json')
     elif request.param == 'v6':
-        bgp_summary_json = os.path.join(
+        bgp_mocked_json = os.path.join(
             test_path, 'mock_tables', 'ipv6_bgp_summary.json')
+    elif request.param == 'ip_route':
+        bgp_mocked_json = os.path.join(
+            test_path, 'mock_tables', 'ip_route.json')
+    elif request.param == 'ip_specific_route':
+        bgp_mocked_json = os.path.join(
+            test_path, 'mock_tables', 'ip_specific_route.json')
+    elif request.param == 'ipv6_route':
+        bgp_mocked_json = os.path.join(
+            test_path, 'mock_tables', 'ipv6_route.json')
     else:
-        bgp_summary_json = os.path.join(
+        bgp_mocked_json = os.path.join(
             test_path, 'mock_tables', 'dummy.json')
 
     def mock_run_bgp_command(vtysh_cmd, bgp_namespace):
-        if os.path.isfile(bgp_summary_json):
-            with open(bgp_summary_json) as json_data:
+        if os.path.isfile(bgp_mocked_json):
+            with open(bgp_mocked_json) as json_data:
                 mock_frr_data = json_data.read()
             return mock_frr_data
         return ""
@@ -110,3 +119,10 @@ def setup_bgp_commands():
     show.ip.add_command(bgpv4)
     show.ipv6.add_command(bgpv6)
     return show
+
+
+@pytest.fixture
+def setup_ip_route_commands():
+    import show.main as show
+
+    return show