Remove more references to initialise

gchq · Feb 21, 2025 · 02dd4d7 · 02dd4d7
1 parent bb3f6d2
commit 02dd4d7
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/java/athena/src/test/java/sleeper/athena/TestUtils.java b/java/athena/src/test/java/sleeper/athena/TestUtils.java
@@ -55,6 +55,7 @@
 import static sleeper.core.properties.testutils.InstancePropertiesTestHelper.createTestInstanceProperties;
 import static sleeper.core.properties.testutils.TablePropertiesTestHelper.createTestTableProperties;
 import static sleeper.core.properties.validation.IngestFileWritingStrategy.ONE_FILE_PER_LEAF;
+import static sleeper.core.statestore.testutils.StateStoreUpdatesWrapper.update;
 
 public class TestUtils {
 
@@ -86,7 +87,7 @@ public static TableProperties createTable(
         S3TableProperties.createStore(instance, s3Client, dynamoDB).save(tableProperties);
 
         StateStore stateStore = new StateStoreFactory(instance, s3Client, dynamoDB, configuration).getStateStore(tableProperties);
-        stateStore.initialise(new PartitionsFromSplitPoints(schema, List.of(splitPoints)).construct());
+        update(stateStore).initialise(new PartitionsFromSplitPoints(schema, List.of(splitPoints)).construct());
 
         return tableProperties;
     }

diff --git a/...em-test-dsl/src/main/java/sleeper/systemtest/dsl/partitioning/SystemTestPartitioning.java b/...em-test-dsl/src/main/java/sleeper/systemtest/dsl/partitioning/SystemTestPartitioning.java
@@ -19,6 +19,8 @@
 import sleeper.core.partition.Partition;
 import sleeper.core.partition.PartitionTree;
 import sleeper.core.statestore.StateStore;
+import sleeper.core.statestore.transactionlog.AddTransactionRequest;
+import sleeper.core.statestore.transactionlog.transaction.impl.InitialisePartitionsTransaction;
 import sleeper.systemtest.dsl.SystemTestContext;
 import sleeper.systemtest.dsl.SystemTestDrivers;
 import sleeper.systemtest.dsl.instance.SystemTestInstanceContext;
@@ -76,6 +78,7 @@ private List<Partition> allPartitions(StateStore stateStore) {
     }
 
     public void setPartitions(PartitionTree tree) {
-        instance.getStateStore().initialise(tree.getAllPartitions());
+        instance.getStateStore().addTransaction(AddTransactionRequest.withTransaction(
+                new InitialisePartitionsTransaction(tree.getAllPartitions())).build());
     }
 }