Add output plugin for Warp10 (#1923)

2020-01-09 20:29:16 +01:00
parent 07b75c57fe
commit ce02bebf30
5 changed files with 422 additions and 0 deletions
--- a/plugins/outputs/warp10/warp10_test.go
+++ b/plugins/outputs/warp10/warp10_test.go
@@ -0,0 +1,108 @@
+package warp10
+
+import (
+	"fmt"
+	"testing"
+	"time"
+
+	"github.com/stretchr/testify/require"
+
+	"github.com/influxdata/telegraf/testutil"
+)
+
+type ErrorTest struct {
+	Message  string
+	Expected string
+}
+
+func TestWriteWarp10(t *testing.T) {
+	w := Warp10{
+		Prefix:  "unit.test",
+		WarpURL: "http://localhost:8090",
+		Token:   "WRITE",
+	}
+
+	var now = time.Now()
+	payload := w.GenWarp10Payload(testutil.MockMetrics(), now)
+	require.Exactly(t, fmt.Sprintf("%d// unit.testtest1.value{source=telegraf,tag1=value1} 1.000000\n", now.UnixNano()/1000), payload)
+}
+
+func TestHandleWarp10Error(t *testing.T) {
+	w := Warp10{
+		Prefix:  "unit.test",
+		WarpURL: "http://localhost:8090",
+		Token:   "WRITE",
+	}
+	tests := [...]*ErrorTest{
+		{
+			Message: `
+			<html>
+			<head>
+			<meta http-equiv="Content-Type" content="text/html;charset=utf-8"/>
+			<title>Error 500 io.warp10.script.WarpScriptException: Invalid token.</title>
+			</head>
+			<body><h2>HTTP ERROR 500</h2>
+			<p>Problem accessing /api/v0/update. Reason:
+			<pre>    io.warp10.script.WarpScriptException: Invalid token.</pre></p>
+			</body>
+			</html>
+			`,
+			Expected: fmt.Sprintf("Invalid token"),
+		},
+		{
+			Message: `
+			<html>
+			<head>
+			<meta http-equiv="Content-Type" content="text/html;charset=utf-8"/>
+			<title>Error 500 io.warp10.script.WarpScriptException: Token Expired.</title>
+			</head>
+			<body><h2>HTTP ERROR 500</h2>
+			<p>Problem accessing /api/v0/update. Reason:
+			<pre>    io.warp10.script.WarpScriptException: Token Expired.</pre></p>
+			</body>
+			</html>
+			`,
+			Expected: fmt.Sprintf("Token Expired"),
+		},
+		{
+			Message: `
+			<html>
+			<head>
+			<meta http-equiv="Content-Type" content="text/html;charset=utf-8"/>
+			<title>Error 500 io.warp10.script.WarpScriptException: Token revoked.</title>
+			</head>
+			<body><h2>HTTP ERROR 500</h2>
+			<p>Problem accessing /api/v0/update. Reason:
+			<pre>    io.warp10.script.WarpScriptException: Token revoked.</pre></p>
+			</body>
+			</html>
+			`,
+			Expected: fmt.Sprintf("Token revoked"),
+		},
+		{
+			Message: `
+			<html>
+			<head>
+			<meta http-equiv="Content-Type" content="text/html;charset=utf-8"/>
+			<title>Error 500 io.warp10.script.WarpScriptException: Write token missing.</title>
+			</head>
+			<body><h2>HTTP ERROR 500</h2>
+			<p>Problem accessing /api/v0/update. Reason:
+			<pre>    io.warp10.script.WarpScriptException: Write token missing.</pre></p>
+			</body>
+			</html>
+			`,
+			Expected: "Write token missing",
+		},
+		{
+			Message:  `<title>Error 503: server unavailable</title>`,
+			Expected: "<title>Error 503: server unavailable</title>",
+		},
+	}
+
+	for _, handledError := range tests {
+		payload := w.HandleError(handledError.Message, 511)
+		require.Exactly(t, handledError.Expected, payload)
+	}
+
+}