[UT] Fix unstable case for restoreJob (backport #53423) (#53446)

Co-authored-by: srlch <[email protected]>
StarRocks · Dec 3, 2024 · 938920a · 938920a
1 parent 7fcc78f
commit 938920a
Showing 1 changed file with 24 additions and 0 deletions.
diff --git a/fe/fe-core/src/test/java/com/starrocks/backup/RestoreJobTest.java b/fe/fe-core/src/test/java/com/starrocks/backup/RestoreJobTest.java
@@ -222,6 +222,14 @@ public void testRunBackupMultiSubPartitionTable() {
                 minTimes = 0;
                 result = beIds;
 
+                systemInfoService.getBackend(anyLong);
+                minTimes = 0;
+                result = null;
+
+                systemInfoService.getComputeNode(anyLong);
+                minTimes = 0;
+                result = null;
+
                 systemInfoService.checkExceedDiskCapacityLimit((Multimap<Long, Long>) any, anyBoolean);
                 minTimes = 0;
                 result = com.starrocks.common.Status.OK;
@@ -398,6 +406,14 @@ public void testRunBackupRangeTable() {
                 minTimes = 0;
                 result = beIds;
 
+                systemInfoService.getBackend(anyLong);
+                minTimes = 0;
+                result = null;
+
+                systemInfoService.getComputeNode(anyLong);
+                minTimes = 0;
+                result = null;
+
                 systemInfoService.checkExceedDiskCapacityLimit((Multimap<Long, Long>) any, anyBoolean);
                 minTimes = 0;
                 result = com.starrocks.common.Status.OK;
@@ -561,6 +577,14 @@ public void testRunBackupListTable() {
                 minTimes = 0;
                 result = beIds;
 
+                systemInfoService.getBackend(anyLong);
+                minTimes = 0;
+                result = null;
+
+                systemInfoService.getComputeNode(anyLong);
+                minTimes = 0;
+                result = null;
+
                 systemInfoService.checkExceedDiskCapacityLimit((Multimap<Long, Long>) any, anyBoolean);
                 minTimes = 0;
                 result = com.starrocks.common.Status.OK;