From 761e120e9a920856fcb03e326402d693cb992f45 Mon Sep 17 00:00:00 2001
From: Wei Dai <daiweix@meta.com>
Date: Wed, 18 Sep 2024 09:29:44 -0700
Subject: [PATCH] fix SaiRollbackTest.rollbackLinkUpAndDown on TH5

Summary: Similar to D62894132, fix SaiRollbackTest.rollbackLinkUpAndDown on TH5 by only using interface ports

Differential Revision: D62898566

fbshipit-source-id: e5a6196c8fbd5c76a0c56dd95c4ccb4ff64b7700
---
 fboss/agent/hw/sai/hw_test/SaiLinkStateRollbackTests.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/fboss/agent/hw/sai/hw_test/SaiLinkStateRollbackTests.cpp b/fboss/agent/hw/sai/hw_test/SaiLinkStateRollbackTests.cpp
index 93f42bf7e81a5..6e0807e5b2d74 100644
--- a/fboss/agent/hw/sai/hw_test/SaiLinkStateRollbackTests.cpp
+++ b/fboss/agent/hw/sai/hw_test/SaiLinkStateRollbackTests.cpp
@@ -26,7 +26,7 @@ TEST_F(SaiRollbackTest, rollbackLinkUpAndDown) {
     rollback(StateDelta(link1DownState, getProgrammedState()));
     // Back to square 1
     rollback(StateDelta(origState, getProgrammedState()));
-    for (auto portID : masterLogicalPortIds()) {
+    for (auto portID : masterLogicalInterfacePortIds()) {
       const auto& port = getProgrammedState()->getPorts()->getNodeIf(portID);
       EXPECT_TRUE(port->isPortUp());
     }