add QLDB test for create/drop tables

whummer · whummer · commit 7587ad62df96 · 2021-08-03T13:50:51.000+02:00
diff --git a/Makefile b/Makefile
@@ -1,7 +1,7 @@
 ADDITIONAL_MVN_ARGS ?= -DskipTests -q
 export AWS_DEFAULT_REGION ?= us-east-1
 export AWS_REGION ?= us-east-1
-export SERVICES ?= serverless,kinesis,sns,sqs,iam,cloudwatch
+export SERVICES ?= serverless,kinesis,sns,sqs,iam,cloudwatch,qldb
 
 usage:           ## Show this help
 	@fgrep -h "##" $(MAKEFILE_LIST) | fgrep -v fgrep | sed -e 's/\\$$//' | sed -e 's/##//'
diff --git a/src/test/java/cloud/localstack/awssdkv2/ProFeaturesSDKV2Test.java b/src/test/java/cloud/localstack/awssdkv2/ProFeaturesSDKV2Test.java
@@ -162,6 +162,28 @@ public void testUpdateQueryDataTypes() throws Exception {
         cleanUp(ledgerName);
     }
 
+    @Test
+    public void testCreateDropTable() throws Exception {
+        if (!isProEnabled()) {
+            return;
+        }
+        LOG.info("Running test testCreateDropTable() ...");
+        String ledgerName = "l123";
+        QldbDriver driver = createLedgerAndGetDriver(ledgerName);
+        driver.execute(txn -> {
+            txn.execute("CREATE TABLE A");
+        });
+
+        driver.execute(txn -> {
+            txn.execute("DROP TABLE A");
+        });
+
+        Set<String> tableNames = StreamSupport.stream(driver.getTableNames().spliterator(), false)
+                .collect(Collectors.toSet());
+
+        Assert.assertTrue(tableNames.isEmpty());
+    }
+
     // UTIL FUNCTIONS AND CLASSES BELOW
 
     public static class Wallet {
@@ -218,6 +240,12 @@ public enum WalletType {
         BUSINESS
     }
 
+
+    private QldbDriver createLedgerAndGetDriver(String ledgerName, String ... tableNames) throws Exception {
+        createLedgerAndTables(ledgerName, tableNames);
+        return getDriver(ledgerName);
+    }
+
     private void createLedgerAndTables(String ledgerName, String ... tableNames) throws Exception {
         QldbAsyncClient client = TestUtils.getClientQLDBAsyncV2();
 
