Test enum to select list conversion

Author: misolt

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/connector/snowflake/AbstractSnowflakeConnectorTest.java

@@ -17,6 +17,7 @@
 package com.google.edwmigration.dumper.application.dumper.connector.snowflake;
 
 import static java.util.Arrays.copyOf;
+import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertThrows;
 import static org.junit.Assert.assertTrue;
 
@@ -119,6 +120,18 @@ public void validate_mixedPrivateKeyAndPassword_fail() throws IOException {
                 "Private key authentication method can't be used together with user password"));
   }
 
+  enum TestEnum {
+    SomeValue
+  }
+
+  @Test
+  public void columnOf_success() {
+
+    String columnName = AbstractSnowflakeConnector.columnOf(TestEnum.SomeValue);
+
+    assertEquals("SOME_VALUE", columnName);
+  }
+
   @Test
   public void validate_assessmentEnabledWithDatabaseFilter_throwsUsageException()
       throws IOException {

dumper/app/src/test/java/com/google/edwmigration/dumper/application/dumper/connector/snowflake/SnowflakeLogsConnectorTest.java

@@ -18,17 +18,18 @@
 
 import static com.google.common.base.Predicates.alwaysTrue;
 import static com.google.edwmigration.dumper.application.dumper.connector.snowflake.SnowflakeLogsConnector.earliestTimestamp;
+import static com.google.edwmigration.dumper.application.dumper.connector.snowflake.SnowflakeLogsConnector.formatPrefix;
 import static com.google.edwmigration.dumper.application.dumper.connector.snowflake.SnowflakeLogsConnector.overrideableQuery;
 import static org.junit.Assert.assertEquals;
 import static org.junit.Assert.assertFalse;
+import static org.junit.Assert.assertThrows;
 import static org.junit.Assert.assertTrue;
 
 import com.google.edwmigration.dumper.application.dumper.ConnectorArguments;
 import com.google.edwmigration.dumper.application.dumper.MetadataDumperUsageException;
 import com.google.edwmigration.dumper.test.TestUtils;
 import java.io.File;
 import java.io.IOException;
-import org.junit.Assert;
 import org.junit.Test;
 import org.junit.runner.RunWith;
 import org.junit.runners.JUnit4;
@@ -79,6 +80,20 @@ public void earliestTimestamp_provided_resultMatches() throws IOException {
     assertTrue(result, result.endsWith("\n"));
   }
 
+  enum TestEnum {
+    FirstValue,
+    SecondValue;
+  };
+
+  @Test
+  public void formatPrefix_success() {
+
+    String result = formatPrefix(TestEnum.class, "TASK_HISTORY");
+
+    assertEquals(
+        "SELECT FIRST_VALUE, SECOND_VALUE FROM SNOWFLAKE.ACCOUNT_USAGE.TASK_HISTORY", result);
+  }
+
   @Test
   public void overrideableQuery_overrideAbsent_defaultUsed() throws IOException {
     String defaultSql = "SELECT event_name, query_id FROM WAREHOUSE_EVENTS_HISTORY";
@@ -119,6 +134,6 @@ public void validate_unsupportedOption_throwsException() throws IOException {
             "--" + ConnectorArguments.OPT_QUERY_LOG_EARLIEST_TIMESTAMP,
             "2024");
 
-    Assert.assertThrows(MetadataDumperUsageException.class, () -> connector.validate(arguments));
+    assertThrows(MetadataDumperUsageException.class, () -> connector.validate(arguments));
   }
 }