Add integration tests

Author: KodaiD

core/src/integration-test/java/com/scalar/db/storage/jdbc/JdbcAdminImportTableIntegrationTest.java

@@ -6,6 +6,7 @@
 import com.scalar.db.api.TableMetadata;
 import com.scalar.db.exception.storage.ExecutionException;
 import com.scalar.db.io.DataType;
+import com.scalar.db.util.AdminTestUtils;
 import java.sql.SQLException;
 import java.util.Collections;
 import java.util.List;
@@ -68,6 +69,11 @@ protected void dropNonImportableTable(String table) throws SQLException {
     testUtils.dropTable(getNamespace(), table);
   }
 
+  @Override
+  protected AdminTestUtils getAdminTestUtils(String testName) {
+    return new JdbcAdminTestUtils(getProperties(testName));
+  }
+
   @SuppressWarnings("unused")
   private boolean isOracle() {
     return JdbcEnv.isOracle();

integration-test/src/main/java/com/scalar/db/api/DistributedStorageAdminImportTableIntegrationTestBase.java

@@ -1,12 +1,14 @@
 package com.scalar.db.api;
 
 import static org.assertj.core.api.Assertions.assertThat;
+import static org.assertj.core.api.Assertions.assertThatCode;
 import static org.assertj.core.api.Assertions.assertThatThrownBy;
 
 import com.scalar.db.exception.storage.ExecutionException;
 import com.scalar.db.io.Column;
 import com.scalar.db.io.DataType;
 import com.scalar.db.service.StorageFactory;
+import com.scalar.db.util.AdminTestUtils;
 import java.sql.SQLException;
 import java.util.ArrayList;
 import java.util.Collections;
@@ -59,7 +61,7 @@ private void dropTable() throws Exception {
       if (!testData.isImportableTable()) {
         dropNonImportableTable(testData.getTableName());
       } else {
-        admin.dropTable(getNamespace(), testData.getTableName());
+        admin.dropTable(getNamespace(), testData.getTableName(), true);
       }
     }
     if (!admin.namespaceExists(getNamespace())) {
@@ -108,6 +110,8 @@ protected abstract List<String> getFloatCompatibleColumnNamesOnExistingDatabase(
 
   protected abstract void dropNonImportableTable(String table) throws Exception;
 
+  protected abstract AdminTestUtils getAdminTestUtils(String testName);
+
   @Test
   public void importTable_ShouldWorkProperly() throws Exception {
     // Arrange
@@ -222,6 +226,43 @@ public void alterColumnType_WideningConversion_ForImportedTable_ShouldAlterPrope
     }
   }
 
+  @Test
+  public void dropNamespace_ShouldNotDropNonScalarDBTables() throws Exception {
+    AdminTestUtils adminTestUtils = getAdminTestUtils(TEST_NAME);
+    try {
+      // Arrange
+      testDataList.addAll(createExistingDatabaseWithAllDataTypes());
+      for (TestData testData : testDataList) {
+        if (testData.isImportableTable()) {
+          admin.importTable(
+              getNamespace(),
+              testData.getTableName(),
+              Collections.emptyMap(),
+              testData.getOverrideColumnsType());
+        }
+      }
+      for (TestData testData : testDataList) {
+        if (testData.isImportableTable()) {
+          admin.dropTable(getNamespace(), testData.getTableName());
+        }
+      }
+
+      // Act
+      assertThatCode(() -> admin.dropNamespace(getNamespace()))
+          .isInstanceOf(ExecutionException.class);
+
+      // Assert
+      assertThat(admin.namespaceExists(getNamespace())).isTrue();
+      for (TestData testData : testDataList) {
+        if (!testData.isImportableTable()) {
+          adminTestUtils.tableExists(getNamespace(), testData.getTableName());
+        }
+      }
+    } finally {
+      adminTestUtils.close();
+    }
+  }
+
   private void importTable_ForImportableTable_ShouldImportProperly(
       String table, Map<String, DataType> overrideColumnsType, TableMetadata metadata)
       throws ExecutionException {