From 2dcfe1c7cfd93455b4f22385b20cfa126cd52fa3 Mon Sep 17 00:00:00 2001 From: Kaijie Chen Date: Fri, 17 Mar 2023 15:00:38 +0800 Subject: [PATCH] [MINOR] test: fix CoordinatorGrpcTest#shuffleServerHeartbeatTest on SSD --- .../java/org/apache/uniffle/test/CoordinatorGrpcTest.java | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/integration-test/common/src/test/java/org/apache/uniffle/test/CoordinatorGrpcTest.java b/integration-test/common/src/test/java/org/apache/uniffle/test/CoordinatorGrpcTest.java index e3091a4309..9b63df7c88 100644 --- a/integration-test/common/src/test/java/org/apache/uniffle/test/CoordinatorGrpcTest.java +++ b/integration-test/common/src/test/java/org/apache/uniffle/test/CoordinatorGrpcTest.java @@ -237,7 +237,9 @@ public void shuffleServerHeartbeatTest() throws Exception { ServerNode node = nodes.get(0); assertEquals(1, node.getStorageInfo().size()); StorageInfo infoHead = node.getStorageInfo().values().iterator().next(); - assertEquals(StorageMedia.HDD, infoHead.getType()); + final StorageInfo expectedStorageInfo = shuffleServers.get(1) + .getStorageManager().getStorageInfo().values().iterator().next(); + assertEquals(expectedStorageInfo, infoHead); assertEquals(StorageStatus.NORMAL, infoHead.getStatus()); assertTrue(node.getTags().contains(Constants.SHUFFLE_SERVER_VERSION)); assertTrue(scm.getTagToNodes().get(Constants.SHUFFLE_SERVER_VERSION).contains(node));