validation added for import query.

Author: AbhishekKumar9984

database-commons/src/main/java/io/cdap/plugin/db/config/AbstractDBSpecificSourceConfig.java

@@ -149,11 +149,27 @@ public void validate(FailureCollector collector) {
       TransactionIsolationLevel.validate(getTransactionIsolationLevel(), collector);
     }
 
-    if ((!containsMacro(TABLE_NAME) && !containsMacro(IMPORT_QUERY)) &&
-            (Strings.isNullOrEmpty(tableName) && Strings.isNullOrEmpty(importQuery))) {
-        collector.addFailure(" Import Query must be specified.",
-                        " Import Query, Can not be empty.")
+    if (!containsMacro(PROPERTY_IMPORT_QUERY_TYPE)) {
+      ImportQueryType importQueryType = ImportQueryType.fromString(getImportQueryType());
+
+      boolean isImportQuerySelected = importQueryType == ImportQueryType.IMPORT_QUERY;
+      if (isImportQuerySelected && !containsMacro(IMPORT_QUERY) &&
+              Strings.isNullOrEmpty(importQuery)) {
+        collector.addFailure("Import Query cannot be empty.", null)
                 .withConfigProperty(IMPORT_QUERY);
+
+      } else if (!isImportQuerySelected && !containsMacro(TABLE_NAME) &&
+              Strings.isNullOrEmpty(tableName)) {
+        collector.addFailure("Import Query cannot be empty", null)
+                .withConfigProperty(TABLE_NAME);
+      }
+    } else {
+      if (!containsMacro(IMPORT_QUERY) && Strings.isNullOrEmpty(importQuery) &&
+              !containsMacro(TABLE_NAME) && Strings.isNullOrEmpty(tableName)) {
+        collector.addFailure("Import Query cannot be empty", null)
+                .withConfigProperty(IMPORT_QUERY)
+                .withConfigProperty(TABLE_NAME);
+      }
     }
     if (!Strings.isNullOrEmpty(importQuery) &&
             (!hasOneSplit && !containsMacro(IMPORT_QUERY) && !getImportQuery().contains("$CONDITIONS"))) {