From 3d862a729cf55a981081fbfa74a1f702d8ab9e3b Mon Sep 17 00:00:00 2001
From: Preetham <51771885+preetham-singh@users.noreply.github.com>
Date: Thu, 25 Nov 2021 11:13:39 +0530
Subject: [PATCH] Fixing subport vs test script for subport under VNET (#2048)

* Fixed subport vs test script for subport under VNET
---
 tests/test_sub_port_intf.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_sub_port_intf.py b/tests/test_sub_port_intf.py
index b0f2863aa6..748e680e2a 100644
--- a/tests/test_sub_port_intf.py
+++ b/tests/test_sub_port_intf.py
@@ -1067,7 +1067,7 @@ def _test_sub_port_intf_removal(self, dvs, sub_port_intf_name, removal_seq_test=
             }
             rif_oid = self.get_newly_created_oid(ASIC_RIF_TABLE, old_rif_oids)
             #If subintf mtu deleted, it inherits from parent 
-            if vrf_name == self.VRF_UNDER_TEST:
+            if vrf_name == self.VRF_UNDER_TEST or vrf_name == self.VNET_UNDER_TEST:
                 if parent_port.startswith(ETHERNET_PREFIX):
                     fv_dict["SAI_ROUTER_INTERFACE_ATTR_MTU"] = ETHERNET_PORT_DEFAULT_MTU 
             self.check_sub_port_intf_fvs(self.asic_db, ASIC_RIF_TABLE, rif_oid, fv_dict)