Add influx v2 output plugin (#4645)
This commit is contained in:
59
plugins/outputs/influxdb_v2/http_internal_test.go
Normal file
59
plugins/outputs/influxdb_v2/http_internal_test.go
Normal 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)
|
||||
}
|
||||
Reference in New Issue
Block a user