Fix other 3 offline test failures too

stargate · Jul 8, 2024 · 045fe36 · 045fe36
1 parent 791f2b2
commit 045fe36
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 3 deletions.
diff --git a/...gate/sgv2/jsonapi/service/resolver/model/impl/BeginOfflineSessionCommandResolverTest.java b/...gate/sgv2/jsonapi/service/resolver/model/impl/BeginOfflineSessionCommandResolverTest.java
@@ -4,6 +4,7 @@
 import static org.junit.jupiter.api.Assertions.assertInstanceOf;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import io.stargate.sgv2.jsonapi.api.model.command.CommandContext;
 import io.stargate.sgv2.jsonapi.api.model.command.impl.BeginOfflineSessionCommand;
 import io.stargate.sgv2.jsonapi.api.model.command.impl.CreateCollectionCommand;
 import io.stargate.sgv2.jsonapi.service.operation.model.impl.BeginOfflineSessionOperation;
@@ -23,7 +24,7 @@ public void verifyOperation() {
     assertInstanceOf(
         BeginOfflineSessionOperation.class,
         beginOfflineSessionCommandResolver.resolveCommand(
-            null,
+            CommandContext.EMPTY_COLLECTION,
             new BeginOfflineSessionCommand(
                 "namespace1", createCollectionCommand, "ssTableOutputDirectory", 20)));
     assertEquals(

diff --git a/...argate/sgv2/jsonapi/service/resolver/model/impl/EndOfflineSessionCommandResolverTest.java b/...argate/sgv2/jsonapi/service/resolver/model/impl/EndOfflineSessionCommandResolverTest.java
@@ -3,6 +3,7 @@
 import static org.junit.jupiter.api.Assertions.assertEquals;
 import static org.junit.jupiter.api.Assertions.assertInstanceOf;
 
+import io.stargate.sgv2.jsonapi.api.model.command.CommandContext;
 import io.stargate.sgv2.jsonapi.api.model.command.impl.EndOfflineSessionCommand;
 import io.stargate.sgv2.jsonapi.service.operation.model.impl.EndOfflineSessionOperation;
 import java.util.UUID;
@@ -18,7 +19,8 @@ public void verifyOperation() {
     EndOfflineSessionCommand endOfflineSessionCommand = new EndOfflineSessionCommand(sessionId);
     assertInstanceOf(
         EndOfflineSessionOperation.class,
-        endOfflineSessionCommandResolver.resolveCommand(null, endOfflineSessionCommand));
+        endOfflineSessionCommandResolver.resolveCommand(
+            CommandContext.EMPTY_COLLECTION, endOfflineSessionCommand));
     assertEquals(
         EndOfflineSessionCommand.class, endOfflineSessionCommandResolver.getCommandClass());
   }

diff --git a/...targate/sgv2/jsonapi/service/resolver/model/impl/OfflineGetStatusCommandResolverTest.java b/...targate/sgv2/jsonapi/service/resolver/model/impl/OfflineGetStatusCommandResolverTest.java
@@ -4,6 +4,7 @@
 import static org.junit.jupiter.api.Assertions.assertInstanceOf;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
+import io.stargate.sgv2.jsonapi.api.model.command.CommandContext;
 import io.stargate.sgv2.jsonapi.api.model.command.impl.OfflineGetStatusCommand;
 import io.stargate.sgv2.jsonapi.service.operation.model.impl.OfflineGetStatusOperation;
 import io.stargate.sgv2.jsonapi.service.shredding.Shredder;
@@ -22,7 +23,8 @@ public void verifyOperation() {
     OfflineGetStatusCommand offlineGetStatusCommand = new OfflineGetStatusCommand(sessionId);
     assertInstanceOf(
         OfflineGetStatusOperation.class,
-        offlineGetStatusCommandResolver.resolveCommand(null, offlineGetStatusCommand));
+        offlineGetStatusCommandResolver.resolveCommand(
+            CommandContext.EMPTY_COLLECTION, offlineGetStatusCommand));
     assertEquals(OfflineGetStatusCommand.class, offlineGetStatusCommandResolver.getCommandClass());
   }
 }