Update method call for wrapper

gchq · Feb 20, 2025 · fc039e3 · fc039e3
1 parent 2f2b034
commit fc039e3
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/...rc/test/java/sleeper/core/statestore/transactionlog/TransactionLogPartitionStoreTest.java b/...rc/test/java/sleeper/core/statestore/transactionlog/TransactionLogPartitionStoreTest.java
@@ -194,7 +194,7 @@ void shouldReinitialisePartitionsWhenNoFilesArePresent() {
             initialiseWithPartitions(partitionsBefore);
 
             // When
-            store.initialise(partitionsAfter.buildList());
+            update(store).initialise(partitionsAfter.buildList());
 
             // Then
             assertThat(store.getAllPartitions())
@@ -213,10 +213,10 @@ void shouldNotReinitialisePartitionsWhenAFileIsPresent() {
                     .splitToNewChildren("root", "after1", "after2", 10L);
             initialiseWithPartitions(partitionsBefore);
 
-            store.addFile(factory.partitionFile("before2", 100L));
+            update(store).addFile(factory.partitionFile("before2", 100L));
 
             // When / Then
-            assertThatThrownBy(() -> store.initialise(partitionsAfter.buildList()))
+            assertThatThrownBy(() -> update(store).initialise(partitionsAfter.buildList()))
                     .isInstanceOf(StateStoreException.class);
             assertThat(store.getAllPartitions())
                     .containsExactlyInAnyOrderElementsOf(partitionsBefore.buildList());