fix uts

apache · Jan 24, 2025 · cbceadf · cbceadf
1 parent 1c58362
commit cbceadf
Showing 1 changed file with 20 additions and 4 deletions.
diff --git a/client/src/test/java/org/apache/celeborn/client/ShuffleClientSuiteJ.java b/client/src/test/java/org/apache/celeborn/client/ShuffleClientSuiteJ.java
@@ -417,7 +417,11 @@ public void testUpdateReducerFileGroupInterrupted() throws InterruptedException
             t -> {
               Thread.sleep(60 * 1000);
               return GetReducerFileGroupResponse$.MODULE$.apply(
-                  StatusCode.SUCCESS, locations, new int[0], Collections.emptySet());
+                  StatusCode.SUCCESS,
+                  locations,
+                  new int[0],
+                  Collections.emptySet(),
+                  Collections.emptyMap());
             });
 
     shuffleClient =
@@ -456,7 +460,11 @@ public void testUpdateReducerFileGroupNonFetchFailureExceptions() {
         .thenAnswer(
             t -> {
               return GetReducerFileGroupResponse$.MODULE$.apply(
-                  StatusCode.SHUFFLE_NOT_REGISTERED, locations, new int[0], Collections.emptySet());
+                  StatusCode.SHUFFLE_NOT_REGISTERED,
+                  locations,
+                  new int[0],
+                  Collections.emptySet(),
+                  Collections.emptyMap());
             });
 
     shuffleClient =
@@ -473,7 +481,11 @@ public void testUpdateReducerFileGroupNonFetchFailureExceptions() {
         .thenAnswer(
             t -> {
               return GetReducerFileGroupResponse$.MODULE$.apply(
-                  StatusCode.STAGE_END_TIME_OUT, locations, new int[0], Collections.emptySet());
+                  StatusCode.STAGE_END_TIME_OUT,
+                  locations,
+                  new int[0],
+                  Collections.emptySet(),
+                  Collections.emptyMap());
             });
 
     shuffleClient =
@@ -490,7 +502,11 @@ public void testUpdateReducerFileGroupNonFetchFailureExceptions() {
         .thenAnswer(
             t -> {
               return GetReducerFileGroupResponse$.MODULE$.apply(
-                  StatusCode.SHUFFLE_DATA_LOST, locations, new int[0], Collections.emptySet());
+                  StatusCode.SHUFFLE_DATA_LOST,
+                  locations,
+                  new int[0],
+                  Collections.emptySet(),
+                  Collections.emptyMap());
             });
 
     shuffleClient =