Added test

2 files changed, 72 insertions, 0 deletions

diff --git a/metrics-proxy/src/test/java/ai/vespa/metricsproxy/telegraf/TelegrafTest.java b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/telegraf/TelegrafTest.java
new file mode 100644
index 00000000000..1f6939e6327
--- /dev/null
+++ b/metrics-proxy/src/test/java/ai/vespa/metricsproxy/telegraf/TelegrafTest.java
@@ -0,0 +1,44 @@
+// Copyright 2020 Oath Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package ai.vespa.metricsproxy.telegraf;
+
+import ai.vespa.metricsproxy.TestUtil;
+import org.junit.Test;
+
+import java.io.StringWriter;
+
+import static org.junit.Assert.*;
+
+/**
+ * @author olaa
+ */
+public class TelegrafTest {
+
+
+    @Test
+    public void test_writing_correct_telegraf_plugin_config() {
+        TelegrafConfig telegrafConfig = new TelegrafConfig.Builder()
+                .cloudWatch(
+                        new TelegrafConfig.CloudWatch.Builder()
+                                .accessKeyName("accessKey1")
+                                .namespace("namespace1")
+                                .secretKeyName("secretKey1")
+                                .region("us-east-1")
+                )
+                .cloudWatch(
+                        new TelegrafConfig.CloudWatch.Builder()
+                                .namespace("namespace2")
+                                .profile("awsprofile")
+                                .region("us-east-2")
+                )
+                .intervalSeconds(300)
+                .vespa(
+                        new TelegrafConfig.Vespa.Builder()
+                                .consumer("custom-consumer")
+                )
+                .build();
+        StringWriter stringWriter = new StringWriter();
+        Telegraf.writeConfig(telegrafConfig, stringWriter);
+        String expectedConfig = TestUtil.getFileContents( "telegraf-config-with-two-cloudwatch-plugins.txt");
+        assertEquals(expectedConfig, stringWriter.toString());
+    }
+}
diff --git a/metrics-proxy/src/test/resources/telegraf-config-with-two-cloudwatch-plugins.txt b/metrics-proxy/src/test/resources/telegraf-config-with-two-cloudwatch-plugins.txt
new file mode 100644
index 00000000000..b2c2521ed59
--- /dev/null
+++ b/metrics-proxy/src/test/resources/telegraf-config-with-two-cloudwatch-plugins.txt
@@ -0,0 +1,28 @@
+# Configuration for telegraf agent
+[agent]
+  interval = "300s"
+  round_interval = true
+  metric_batch_size = 1000
+  metric_buffer_limit = 10000
+  collection_jitter = "0s"
+  flush_interval = "300s"
+  flush_jitter = "0s"
+  precision = ""
+
+# Configuration for AWS CloudWatch output.
+[[outputs.cloudwatch]]
+  region = "us-east-1"
+  namespace = "namespace1"
+  access_key = "accessKey1"
+  secret_key = "secretKey1"
+
+# Configuration for AWS CloudWatch output.
+[[outputs.cloudwatch]]
+  region = "us-east-2"
+  namespace = "namespace2"
+  profile = "awsprofile"
+
+# Configuration for Vespa input plugin
+[[inputs.vespa]]
+  url = "http://localhost:19092/metrics/v2/values?consumer=custom-consumer"
+