package prometheus import ( "fmt" "io/ioutil" "net/http" "net/http/httptest" "strings" "testing" "time" "github.com/influxdata/telegraf" inputs "github.com/influxdata/telegraf/plugins/inputs/prometheus" "github.com/influxdata/telegraf/testutil" "github.com/stretchr/testify/require" ) func TestMetricVersion2(t *testing.T) { Logger := testutil.Logger{Name: "outputs.prometheus_client"} tests := []struct { name string output *PrometheusClient metrics []telegraf.Metric expected []byte }{ { name: "untyped telegraf metric", output: &PrometheusClient{ Listen: ":0", MetricVersion: 2, CollectorsExclude: []string{"gocollector", "process"}, Path: "/metrics", Log: Logger, }, metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{ "host": "example.org", }, map[string]interface{}{ "time_idle": 42.0, }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{host="example.org"} 42 `), }, { name: "strings as labels", output: &PrometheusClient{ Listen: ":0", MetricVersion: 2, CollectorsExclude: []string{"gocollector", "process"}, Path: "/metrics", StringAsLabel: true, Log: Logger, }, metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{}, map[string]interface{}{ "time_idle": 42.0, "host": "example.org", }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{host="example.org"} 42 `), }, { name: "when strings as labels is false string fields are discarded", output: &PrometheusClient{ Listen: ":0", MetricVersion: 2, CollectorsExclude: []string{"gocollector", "process"}, Path: "/metrics", StringAsLabel: false, Log: Logger, }, metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{}, map[string]interface{}{ "time_idle": 42.0, "host": "example.org", }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle 42 `), }, { name: "untype prometheus metric", output: &PrometheusClient{ Listen: ":0", MetricVersion: 2, CollectorsExclude: []string{"gocollector", "process"}, Path: "/metrics", Log: Logger, }, metrics: []telegraf.Metric{ testutil.MustMetric( "prometheus", map[string]string{ "host": "example.org", }, map[string]interface{}{ "cpu_time_idle": 42.0, }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{host="example.org"} 42 `), }, { name: "telegraf histogram", output: &PrometheusClient{ Listen: ":0", MetricVersion: 2, CollectorsExclude: []string{"gocollector", "process"}, Path: "/metrics", Log: Logger, }, metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu1", }, map[string]interface{}{ "usage_idle_sum": 2000.0, "usage_idle_count": 20.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu1", "le": "0.0", }, map[string]interface{}{ "usage_idle_bucket": 0.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu1", "le": "50.0", }, map[string]interface{}{ "usage_idle_bucket": 7.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu1", "le": "100.0", }, map[string]interface{}{ "usage_idle_bucket": 20.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu1", "le": "+Inf", }, map[string]interface{}{ "usage_idle_bucket": 20.0, }, time.Unix(0, 0), telegraf.Histogram, ), }, expected: []byte(` # HELP cpu_usage_idle Telegraf collected metric # TYPE cpu_usage_idle histogram cpu_usage_idle_bucket{cpu="cpu1",le="0"} 0 cpu_usage_idle_bucket{cpu="cpu1",le="50"} 7 cpu_usage_idle_bucket{cpu="cpu1",le="100"} 20 cpu_usage_idle_bucket{cpu="cpu1",le="+Inf"} 20 cpu_usage_idle_sum{cpu="cpu1"} 2000 cpu_usage_idle_count{cpu="cpu1"} 20 `), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { err := tt.output.Init() require.NoError(t, err) err = tt.output.Connect() require.NoError(t, err) defer func() { err := tt.output.Close() require.NoError(t, err) }() err = tt.output.Write(tt.metrics) require.NoError(t, err) resp, err := http.Get(tt.output.URL()) require.NoError(t, err) require.Equal(t, http.StatusOK, resp.StatusCode) defer resp.Body.Close() body, err := ioutil.ReadAll(resp.Body) require.NoError(t, err) require.Equal(t, strings.TrimSpace(string(tt.expected)), strings.TrimSpace(string(body))) }) } } func TestRoundTripMetricVersion2(t *testing.T) { Logger := testutil.Logger{Name: "outputs.prometheus_client"} tests := []struct { name string data []byte }{ { name: "untyped", data: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{host="example.org"} 42 `), }, { name: "counter", data: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle counter cpu_time_idle{host="example.org"} 42 `), }, { name: "gauge", data: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle gauge cpu_time_idle{host="example.org"} 42 `), }, { name: "multi", data: []byte(` # HELP cpu_time_guest Telegraf collected metric # TYPE cpu_time_guest gauge cpu_time_guest{host="one.example.org"} 42 cpu_time_guest{host="two.example.org"} 42 # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle gauge cpu_time_idle{host="one.example.org"} 42 cpu_time_idle{host="two.example.org"} 42 `), }, { name: "histogram", data: []byte(` # HELP http_request_duration_seconds Telegraf collected metric # TYPE http_request_duration_seconds histogram http_request_duration_seconds_bucket{le="0.05"} 24054 http_request_duration_seconds_bucket{le="0.1"} 33444 http_request_duration_seconds_bucket{le="0.2"} 100392 http_request_duration_seconds_bucket{le="0.5"} 129389 http_request_duration_seconds_bucket{le="1"} 133988 http_request_duration_seconds_bucket{le="+Inf"} 144320 http_request_duration_seconds_sum 53423 http_request_duration_seconds_count 144320 `), }, { name: "summary", data: []byte(` # HELP rpc_duration_seconds Telegraf collected metric # TYPE rpc_duration_seconds summary rpc_duration_seconds{quantile="0.01"} 3102 rpc_duration_seconds{quantile="0.05"} 3272 rpc_duration_seconds{quantile="0.5"} 4773 rpc_duration_seconds{quantile="0.9"} 9001 rpc_duration_seconds{quantile="0.99"} 76656 rpc_duration_seconds_sum 1.7560473e+07 rpc_duration_seconds_count 2693 `), }, } ts := httptest.NewServer(http.NotFoundHandler()) defer ts.Close() url := fmt.Sprintf("http://%s", ts.Listener.Addr()) for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { ts.Config.Handler = http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) { w.WriteHeader(http.StatusOK) w.Write(tt.data) }) input := &inputs.Prometheus{ URLs: []string{url}, URLTag: "", MetricVersion: 2, } var acc testutil.Accumulator err := input.Start(&acc) require.NoError(t, err) err = input.Gather(&acc) require.NoError(t, err) input.Stop() metrics := acc.GetTelegrafMetrics() output := &PrometheusClient{ Listen: "127.0.0.1:0", Path: defaultPath, MetricVersion: 2, Log: Logger, CollectorsExclude: []string{"gocollector", "process"}, } err = output.Init() require.NoError(t, err) err = output.Connect() require.NoError(t, err) defer func() { err = output.Close() require.NoError(t, err) }() err = output.Write(metrics) require.NoError(t, err) resp, err := http.Get(output.URL()) require.NoError(t, err) actual, err := ioutil.ReadAll(resp.Body) require.NoError(t, err) require.Equal(t, strings.TrimSpace(string(tt.data)), strings.TrimSpace(string(actual))) }) } }