From 90cd10bfbfb4ad34c4302358fcf714587c24499e Mon Sep 17 00:00:00 2001
From: kr565370 <112558283+kr565370@users.noreply.github.com>
Date: Thu, 28 Sep 2023 12:22:48 +0100
Subject: [PATCH] Use S3StateStoreCreator in ReinitialiseTableIT

---
 .../sleeper/clients/status/update/ReinitialiseTableIT.java     | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/java/clients/src/test/java/sleeper/clients/status/update/ReinitialiseTableIT.java b/java/clients/src/test/java/sleeper/clients/status/update/ReinitialiseTableIT.java
index 77abf8bdc1..3d57b49625 100644
--- a/java/clients/src/test/java/sleeper/clients/status/update/ReinitialiseTableIT.java
+++ b/java/clients/src/test/java/sleeper/clients/status/update/ReinitialiseTableIT.java
@@ -60,6 +60,7 @@
 import sleeper.statestore.dynamodb.DynamoDBStateStore;
 import sleeper.statestore.dynamodb.DynamoDBStateStoreCreator;
 import sleeper.statestore.s3.S3StateStore;
+import sleeper.statestore.s3.S3StateStoreCreator;
 
 import java.io.BufferedWriter;
 import java.io.FileWriter;
@@ -518,7 +519,7 @@ private DynamoDBStateStore setupDynamoStateStore(TableProperties tableProperties
 
     private S3StateStore setupS3StateStore(TableProperties tableProperties) throws IOException, StateStoreException {
         //  - CreateS3StateStore
-        createRevisionDynamoTable(instanceProperties.get(REVISION_TABLENAME));
+        new S3StateStoreCreator(instanceProperties, dynamoDBClient).create();
         Configuration configuration = new Configuration();
         configuration.set("fs.s3a.endpoint", localStackContainer.getEndpointOverride(LocalStackContainer.Service.S3).toString());
         configuration.set("fs.s3a.aws.credentials.provider", "org.apache.hadoop.fs.s3a.SimpleAWSCredentialsProvider");