Add influx v2 output plugin (#4645)

This commit is contained in:
Greg
2018-09-12 15:48:59 -06:00
committed by Daniel Nelson
parent 71aaa844f5
commit 1fdf032db0
7 changed files with 756 additions and 0 deletions

View File

@@ -0,0 +1,59 @@
package influxdb_v2
import (
"io"
"net/url"
"testing"
"github.com/stretchr/testify/require"
)
func genURL(u string) *url.URL {
URL, _ := url.Parse(u)
return URL
}
func TestMakeWriteURL(t *testing.T) {
tests := []struct {
err bool
url *url.URL
act string
}{
{
url: genURL("http://localhost:9999"),
act: "http://localhost:9999/v2/write?bucket=telegraf&org=influx",
},
{
url: genURL("unix://var/run/influxd.sock"),
act: "http://127.0.0.1/v2/write?bucket=telegraf&org=influx",
},
{
err: true,
url: genURL("udp://localhost:9999"),
},
}
for i := range tests {
rURL, err := makeWriteURL(*tests[i].url, "influx", "telegraf")
if !tests[i].err {
require.NoError(t, err)
} else {
require.Error(t, err)
t.Log(err)
}
if err == nil {
require.Equal(t, tests[i].act, rURL)
}
}
}
func TestMakeWriteRequest(t *testing.T) {
reader, _ := io.Pipe()
cli := httpClient{
WriteURL: "http://localhost:9999/v2/write?bucket=telegraf&org=influx",
ContentEncoding: "gzip",
Headers: map[string]string{"x": "y"},
}
_, err := cli.makeWriteRequest(reader)
require.NoError(t, err)
}