Fix bug with checking VRF's routes in route_check.py (sonic-net#2301)

* Create route_check.py compatible with VRF

Signed-off-by: Petro Bratash <[email protected]>

Author: bratashX

scripts/route_check.py

@@ -281,7 +281,10 @@ def get_routes():
 
     valid_rt = []
     for k in keys:
-        if not is_vrf(k) and not is_local(k):
+        if (is_vrf(k)):
+            k = k.split(":", 1)[1]
+
+        if not is_local(k):
             valid_rt.append(add_prefix_ifnot(k.lower()))
 
     print_message(syslog.LOG_DEBUG, json.dumps({"ROUTE_TABLE": sorted(valid_rt)}, indent=4))

tests/route_check_test.py

@@ -293,6 +293,33 @@
                 }
             }
         }
+    },
+    "7": {
+        DESCR: "Good one with VRF routes",
+        ARGS: "route_check",
+        PRE: {
+            APPL_DB: {
+                ROUTE_TABLE: {
+                    "Vrf1:0.0.0.0/0" : { "ifname": "portchannel0" },
+                    "Vrf1:10.10.196.12/31" : { "ifname": "portchannel0" },
+                    "Vrf1:10.10.196.20/31" : { "ifname": "portchannel0" }
+                },
+                INTF_TABLE: {
+                    "PortChannel1013:10.10.196.24/31": {},
+                    "PortChannel1023:2603:10b0:503:df4::5d/126": {},
+                    "PortChannel1024": {}
+                }
+            },
+            ASIC_DB: {
+                RT_ENTRY_TABLE: {
+                    RT_ENTRY_KEY_PREFIX + "10.10.196.12/31" + RT_ENTRY_KEY_SUFFIX: {},
+                    RT_ENTRY_KEY_PREFIX + "10.10.196.20/31" + RT_ENTRY_KEY_SUFFIX: {},
+                    RT_ENTRY_KEY_PREFIX + "10.10.196.24/32" + RT_ENTRY_KEY_SUFFIX: {},
+                    RT_ENTRY_KEY_PREFIX + "2603:10b0:503:df4::5d/128" + RT_ENTRY_KEY_SUFFIX: {},
+                    RT_ENTRY_KEY_PREFIX + "0.0.0.0/0" + RT_ENTRY_KEY_SUFFIX: {}
+                }
+            }
+        }
     }
 }