[b/405039345] Require the assessment flag for snowflake-lite (#803)

* Require --assessment flag for snowflake-lite

* apply spotless

Author: misolt

dumper/app/src/main/java/com/google/edwmigration/dumper/application/dumper/connector/snowflake/SnowflakeLiteConnector.java

@@ -49,13 +49,8 @@ public SnowflakeLiteConnector() {
   @Nonnull
   public Handle open(ConnectorArguments arguments)
       throws MetadataDumperUsageException, SQLException {
-    if (arguments.isAssessment()) {
-      String message =
-          String.format(
-              "The %s connector supports assessment without need for extra flags."
-                  + " Try running again without the '--%s' flag",
-              NAME, ConnectorArguments.OPT_ASSESSMENT);
-      throw new MetadataDumperUsageException(message);
+    if (!arguments.isAssessment()) {
+      throw noAssessmentException();
     }
     return super.open(arguments);
   }
@@ -78,4 +73,13 @@ public final void addTasksTo(List<? super Task<?>> out, ConnectorArguments argum
     out.add(new FormatTask(FORMAT_NAME));
     out.addAll(planner.generateLiteSpecificQueries());
   }
+
+  private static MetadataDumperUsageException noAssessmentException() {
+    String message =
+        String.format(
+            "The %s connector only supports extraction for Assessment."
+                + " Provide the '--%s' flag to use this connector.",
+            NAME, ConnectorArguments.OPT_ASSESSMENT);
+    return new MetadataDumperUsageException(message);
+  }
 }

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

@@ -0,0 +1,37 @@
+/*
+ * Copyright 2022-2025 Google LLC
+ * Copyright 2013-2021 CompilerWorks
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+package com.google.edwmigration.dumper.application.dumper.connector.snowflake;
+
+import static org.junit.Assert.assertThrows;
+
+import com.google.edwmigration.dumper.application.dumper.ConnectorArguments;
+import com.google.edwmigration.dumper.application.dumper.MetadataDumperUsageException;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.junit.runners.JUnit4;
+
+@RunWith(JUnit4.class)
+public class SnowflakeLiteConnectorTest {
+
+  @Test
+  public void open_noAssessmentFlag_throwsUsageException() throws Exception {
+    ConnectorArguments noFlagArguments = new ConnectorArguments("--connector", "snowflake-lite");
+    SnowflakeLiteConnector connector = new SnowflakeLiteConnector();
+
+    assertThrows(MetadataDumperUsageException.class, () -> connector.open(noFlagArguments));
+  }
+}