Add method to test DDB clients for resetting tables in between tests (#…

…189)
cashapp · Aug 7, 2024 · 64f40ef · 64f40ef
1 parent e7197de
commit 64f40ef
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 10 deletions.
diff --git a/...g-internal/src/main/kotlin/app/cash/tempest/testing/internal/DefaultTestDynamoDbClient.kt b/...g-internal/src/main/kotlin/app/cash/tempest/testing/internal/DefaultTestDynamoDbClient.kt
@@ -31,6 +31,15 @@ class DefaultTestDynamoDbClient(
   override val dynamoDbStreams = buildDynamoDbStreams(hostName, port)
 
   override fun startUp() {
+    reset()
+  }
+
+  override fun shutDown() {
+    dynamoDb.shutdown()
+    dynamoDbStreams.shutdown()
+  }
+
+  override fun reset() {
     // Cleans up the tables before each run.
     for (tableName in dynamoDb.listTables().tableNames) {
       dynamoDb.deleteTable(tableName)
@@ -39,9 +48,4 @@ class DefaultTestDynamoDbClient(
       dynamoDb.createTable(table)
     }
   }
-
-  override fun shutDown() {
-    dynamoDb.shutdown()
-    dynamoDbStreams.shutdown()
-  }
 }
diff --git a/tempest-testing/src/main/kotlin/app/cash/tempest/testing/TestDynamoDbClient.kt b/tempest-testing/src/main/kotlin/app/cash/tempest/testing/TestDynamoDbClient.kt
@@ -48,6 +48,9 @@ interface TestDynamoDbClient : Service {
   fun <DB : LogicalDb> logicalDb(type: Class<DB>, mapperConfig: DynamoDBMapperConfig): DB {
     return logicalDb(type.kotlin, mapperConfig)
   }
+
+  /** Cleans up tables in between test runs. */
+  fun reset()
 }
 
 inline fun <reified DB : LogicalDb> TestDynamoDbClient.logicalDb(): DB {

diff --git a/...-internal/src/main/kotlin/app/cash/tempest2/testing/internal/DefaultTestDynamoDbClient.kt b/...-internal/src/main/kotlin/app/cash/tempest2/testing/internal/DefaultTestDynamoDbClient.kt
@@ -34,6 +34,15 @@ class DefaultTestDynamoDbClient(
   override val asyncDynamoDbStreams = buildAsyncDynamoDbStreams(hostName, port)
 
   override fun startUp() {
+    reset()
+  }
+
+  override fun shutDown() {
+    dynamoDb.close()
+    dynamoDbStreams.close()
+  }
+
+  override fun reset() {
     // Cleans up the tables before each run.
     for (tableName in dynamoDb.listTables().tableNames()) {
       dynamoDb.deleteTable(DeleteTableRequest.builder().tableName(tableName).build())
@@ -42,9 +51,4 @@ class DefaultTestDynamoDbClient(
       dynamoDb.createTable(table)
     }
   }
-
-  override fun shutDown() {
-    dynamoDb.close()
-    dynamoDbStreams.close()
-  }
 }
diff --git a/tempest2-testing/src/main/kotlin/app/cash/tempest2/testing/TestDynamoDbClient.kt b/tempest2-testing/src/main/kotlin/app/cash/tempest2/testing/TestDynamoDbClient.kt
@@ -128,6 +128,9 @@ interface TestDynamoDbClient : Service {
   ): DB {
     return asyncLogicalDb(type.kotlin, extensions)
   }
+
+  /** Cleans up tables in between test runs. */
+  fun reset()
 }
 
 inline fun <reified DB : LogicalDb> TestDynamoDbClient.logicalDb(