[UT] Fix OlapTableSinkTest testCreateLocationWithLocalTablet (StarRoc…

…ks#38260) Signed-off-by: wyb <wybb86@gmail.com>
willWang8023 · Jan 3, 2024 · e4f0cf3 · e4f0cf3
1 parent edab5db
commit e4f0cf3
Showing 1 changed file with 7 additions and 16 deletions.
diff --git a/fe/fe-core/src/test/java/com/starrocks/planner/OlapTableSinkTest.java b/fe/fe-core/src/test/java/com/starrocks/planner/OlapTableSinkTest.java
@@ -224,23 +224,14 @@ public void testRangeUnknownPartition(
     }
 
     @Test
-    public void testCreateLocationWithLocalTablet(@Mocked GlobalStateMgr globalStateMgr,
-                                                  @Mocked SystemInfoService systemInfoService,
-                                                  @Mocked TabletInvertedIndex invertedIndex) throws Exception {
-        new Expectations() {
+    public void testCreateLocationWithLocalTablet() throws Exception {
+        SystemInfoService systemInfoService = GlobalStateMgr.getCurrentSystemInfo();
+        new Expectations(systemInfoService) {
             {
-                GlobalStateMgr.getCurrentState();
-                result = globalStateMgr;
-                GlobalStateMgr.getCurrentSystemInfo();
-                result = systemInfoService;
-                systemInfoService.checkExceedDiskCapacityLimit((Multimap<Long, Long>) any, anyBoolean);
-                result = Status.OK;
-                globalStateMgr.getOrCreateSystemInfo(anyInt);
-                result = systemInfoService;
-                systemInfoService.checkBackendAlive(anyLong);
-                result = true;
-                GlobalStateMgr.getCurrentInvertedIndex();
-                result = invertedIndex;
+               systemInfoService.checkExceedDiskCapacityLimit((Multimap<Long, Long>) any, anyBoolean);
+               result = Status.OK;
+               systemInfoService.checkBackendAlive(anyLong);
+               result = true;
             }
         };