118 lines
2.9 KiB
Go
118 lines
2.9 KiB
Go
package http_test
|
|
|
|
import (
|
|
"net/http"
|
|
"net/http/httptest"
|
|
"testing"
|
|
|
|
plugin "github.com/influxdata/telegraf/plugins/inputs/http"
|
|
"github.com/influxdata/telegraf/plugins/parsers"
|
|
"github.com/influxdata/telegraf/testutil"
|
|
"github.com/stretchr/testify/require"
|
|
)
|
|
|
|
func TestHTTPwithJSONFormat(t *testing.T) {
|
|
fakeServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/endpoint" {
|
|
_, _ = w.Write([]byte(simpleJSON))
|
|
} else {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
}
|
|
}))
|
|
defer fakeServer.Close()
|
|
|
|
url := fakeServer.URL + "/endpoint"
|
|
plugin := &plugin.HTTP{
|
|
URLs: []string{url},
|
|
TagURL: true,
|
|
}
|
|
metricName := "metricName"
|
|
p, _ := parsers.NewJSONParser(metricName, nil, nil)
|
|
plugin.SetParser(p)
|
|
|
|
var acc testutil.Accumulator
|
|
require.NoError(t, acc.GatherError(plugin.Gather))
|
|
|
|
require.Len(t, acc.Metrics, 1)
|
|
|
|
// basic check to see if we got the right field, value and tag
|
|
var metric = acc.Metrics[0]
|
|
require.Equal(t, metric.Measurement, metricName)
|
|
require.Len(t, acc.Metrics[0].Fields, 1)
|
|
require.Equal(t, acc.Metrics[0].Fields["a"], 1.2)
|
|
require.Equal(t, acc.Metrics[0].Tags["url"], url)
|
|
}
|
|
|
|
func TestHTTPHeaders(t *testing.T) {
|
|
header := "X-Special-Header"
|
|
headerValue := "Special-Value"
|
|
fakeServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/endpoint" {
|
|
if r.Header.Get(header) == headerValue {
|
|
_, _ = w.Write([]byte(simpleJSON))
|
|
} else {
|
|
w.WriteHeader(http.StatusForbidden)
|
|
}
|
|
} else {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
}
|
|
}))
|
|
defer fakeServer.Close()
|
|
|
|
url := fakeServer.URL + "/endpoint"
|
|
plugin := &plugin.HTTP{
|
|
URLs: []string{url},
|
|
Headers: map[string]string{header: headerValue},
|
|
}
|
|
metricName := "metricName"
|
|
p, _ := parsers.NewJSONParser(metricName, nil, nil)
|
|
plugin.SetParser(p)
|
|
|
|
var acc testutil.Accumulator
|
|
require.NoError(t, acc.GatherError(plugin.Gather))
|
|
}
|
|
|
|
func TestInvalidStatusCode(t *testing.T) {
|
|
fakeServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
}))
|
|
defer fakeServer.Close()
|
|
|
|
url := fakeServer.URL + "/endpoint"
|
|
plugin := &plugin.HTTP{
|
|
URLs: []string{url},
|
|
}
|
|
|
|
metricName := "metricName"
|
|
p, _ := parsers.NewJSONParser(metricName, nil, nil)
|
|
plugin.SetParser(p)
|
|
|
|
var acc testutil.Accumulator
|
|
require.Error(t, acc.GatherError(plugin.Gather))
|
|
}
|
|
|
|
func TestParserNotSet(t *testing.T) {
|
|
fakeServer := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
if r.URL.Path == "/endpoint" {
|
|
_, _ = w.Write([]byte(simpleJSON))
|
|
} else {
|
|
w.WriteHeader(http.StatusNotFound)
|
|
}
|
|
}))
|
|
defer fakeServer.Close()
|
|
|
|
url := fakeServer.URL + "/endpoint"
|
|
plugin := &plugin.HTTP{
|
|
URLs: []string{url},
|
|
}
|
|
|
|
var acc testutil.Accumulator
|
|
require.Error(t, acc.GatherError(plugin.Gather))
|
|
}
|
|
|
|
const simpleJSON = `
|
|
{
|
|
"a": 1.2
|
|
}
|
|
`
|