chore: record errors in Config

splunk · Mar 15, 2024 · 1d77126 · 1d77126
1 parent 0d009fc
commit 1d77126
Show file tree

Hide file tree

Showing 2 changed files with 29 additions and 3 deletions.
diff --git a/src/main/java/com/splunk/kafka/connect/SplunkSinkConnector.java b/src/main/java/com/splunk/kafka/connect/SplunkSinkConnector.java
@@ -149,6 +149,17 @@ private void validateSplunkConfigurations(final Map<String, String> configs) thr
         if (configs.containsKey(DISABLE_VALIDATION) && Boolean.parseBoolean(configs.get(DISABLE_VALIDATION))) {
             return;
         }
+        if (configs.getOrDefault(TOKEN_CONF,"").equals("")
+        || configs.getOrDefault(URI_CONF,"").equals("")) {
+            String errorMessage = String.format(
+                "Either one of '%s' or '%s' must be set for Splunk validation check.",
+                TOKEN_CONF,
+                URI_CONF
+            );
+            addErrorMessage(TOKEN_CONF, errorMessage);
+            addErrorMessage(URI_CONF, errorMessage);
+            return;
+        }
         SplunkSinkConnectorConfig connectorConfig = new SplunkSinkConnectorConfig(configs);
         String[] indexes = split(connectorConfig.indexes, ",");
         if(indexes == null || indexes.length == 0) {

diff --git a/src/test/java/com/splunk/kafka/connect/SplunkSinkConnecterTest.java b/src/test/java/com/splunk/kafka/connect/SplunkSinkConnecterTest.java
@@ -72,7 +72,19 @@ public void config() {
         ConfigDef config = connector.config();
         Assert.assertNotNull(config);
     }
-
+    @Test
+    public void testErrorWithoutUriConf() {
+        final Map<String, String> configs = new HashMap<>();
+        SinkConnector connector = new SplunkSinkConnector();
+        configs.put("topics", "b");
+        configs.put("splunk.indexes", "b");
+        MockHecClientWrapper clientInstance = new MockHecClientWrapper();
+        ((SplunkSinkConnector) connector).setHecInstance(clientInstance);
+        Config result = connector.validate(configs);
+        assertHasErrorMessage(result, URI_CONF, "Either one of 'splunk.hec.token' or 'splunk.hec.uri' must be set for Splunk validation check.", 1);
+        assertHasErrorMessage(result, TOKEN_CONF, "Either one of 'splunk.hec.token' or 'splunk.hec.uri' must be set for Splunk validation check.", 1);
+    }
+
     @Test
     public void testValidKerberosBothEmpty() {
         final Map<String, String> configs = new HashMap<>();
@@ -304,12 +316,15 @@ private void addNecessaryConfigs(Map<String, String> configs) {
         configs.put(URI_CONF, TEST_URI);
         configs.put(TOKEN_CONF, "blah");
     }
-
     private void assertHasErrorMessage(Config config, String property, String msg) {
+        assertHasErrorMessage(config, property, msg, 0);
+    }
+    private void assertHasErrorMessage(Config config, String property, String msg, int idx) {
         for (ConfigValue configValue : config.configValues()) {
             if (configValue.name().equals(property)) {
+                System.out.print(configValue.errorMessages());
                 assertFalse(configValue.errorMessages().isEmpty());
-                assertTrue(configValue.errorMessages().get(0).contains(msg));
+                assertTrue(configValue.errorMessages().get(idx).contains(msg));
             }
         }
     }