From e6f8961af7c5e22c3f08060e36cc980cff4bf7fa Mon Sep 17 00:00:00 2001
From: arvbaska1 <123760606+arvbaska1@users.noreply.github.com>
Date: Thu, 11 Apr 2024 23:57:55 -0400
Subject: [PATCH] parent type change to use lookup (#2808)

---
 .../telemetry_inventory_test/telemetry_inventory_test.go     | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/feature/platform/tests/telemetry_inventory_test/telemetry_inventory_test.go b/feature/platform/tests/telemetry_inventory_test/telemetry_inventory_test.go
index 7cb2a4c6a2e..0e6ef43e82f 100644
--- a/feature/platform/tests/telemetry_inventory_test/telemetry_inventory_test.go
+++ b/feature/platform/tests/telemetry_inventory_test/telemetry_inventory_test.go
@@ -745,8 +745,9 @@ func ValidateComponentState(t *testing.T, dut *ondatra.DUTDevice, cards []*oc.Co
 						t.Errorf("Component %s Parent: Chassis component NOT found in the hierarchy tree of component", cName)
 						break
 					}
-					parentType := gnmi.Get(t, dut, gnmi.OC().Component(parent).Type().State())
-					if parentType == componentType["Chassis"] {
+					pLoookup := gnmi.Lookup(t, dut, gnmi.OC().Component(parent).Type().State())
+					parentType, present := pLoookup.Val()
+					if present && parentType == componentType["Chassis"] {
 						t.Logf("Component %s Parent: Found chassis component in the hierarchy tree of component", cName)
 						break
 					}