From 02dd4d7dfd9b14c3e09fcd901742d0f5ba14b546 Mon Sep 17 00:00:00 2001
From: patchwork01 <110390516+patchwork01@users.noreply.github.com>
Date: Fri, 21 Feb 2025 13:08:35 +0000
Subject: [PATCH] Remove more references to initialise

---
 java/athena/src/test/java/sleeper/athena/TestUtils.java      | 3 ++-
 .../systemtest/dsl/partitioning/SystemTestPartitioning.java  | 5 ++++-
 2 files changed, 6 insertions(+), 2 deletions(-)

diff --git a/java/athena/src/test/java/sleeper/athena/TestUtils.java b/java/athena/src/test/java/sleeper/athena/TestUtils.java
index 414e027079..b1a3a2998c 100644
--- 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/java/system-test/system-test-dsl/src/main/java/sleeper/systemtest/dsl/partitioning/SystemTestPartitioning.java b/java/system-test/system-test-dsl/src/main/java/sleeper/systemtest/dsl/partitioning/SystemTestPartitioning.java
index 09333f86ad..062a996b36 100644
--- a/java/system-test/system-test-dsl/src/main/java/sleeper/systemtest/dsl/partitioning/SystemTestPartitioning.java
+++ b/java/system-test/system-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());
     }
 }