package prometheus import ( "strings" "testing" "time" "github.com/influxdata/telegraf" "github.com/influxdata/telegraf/testutil" "github.com/stretchr/testify/require" ) func TestSerialize(t *testing.T) { tests := []struct { name string config FormatConfig metric telegraf.Metric expected []byte }{ { name: "simple", 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: "prometheus input untyped", metric: testutil.MustMetric( "prometheus", map[string]string{ "code": "400", "method": "post", }, map[string]interface{}{ "http_requests_total": 3.0, }, time.Unix(0, 0), telegraf.Untyped, ), expected: []byte(` # HELP http_requests_total Telegraf collected metric # TYPE http_requests_total untyped http_requests_total{code="400",method="post"} 3 `), }, { name: "prometheus input counter", metric: testutil.MustMetric( "prometheus", map[string]string{ "code": "400", "method": "post", }, map[string]interface{}{ "http_requests_total": 3.0, }, time.Unix(0, 0), telegraf.Counter, ), expected: []byte(` # HELP http_requests_total Telegraf collected metric # TYPE http_requests_total counter http_requests_total{code="400",method="post"} 3 `), }, { name: "prometheus input gauge", metric: testutil.MustMetric( "prometheus", map[string]string{ "code": "400", "method": "post", }, map[string]interface{}{ "http_requests_total": 3.0, }, time.Unix(0, 0), telegraf.Gauge, ), expected: []byte(` # HELP http_requests_total Telegraf collected metric # TYPE http_requests_total gauge http_requests_total{code="400",method="post"} 3 `), }, { name: "prometheus input histogram no buckets", metric: testutil.MustMetric( "prometheus", map[string]string{}, map[string]interface{}{ "http_request_duration_seconds_sum": 53423, "http_request_duration_seconds_count": 144320, }, time.Unix(0, 0), telegraf.Histogram, ), expected: []byte(` `), }, { name: "prometheus input histogram only bucket", metric: testutil.MustMetric( "prometheus", map[string]string{ "le": "0.5", }, map[string]interface{}{ "http_request_duration_seconds_bucket": 129389.0, }, time.Unix(0, 0), telegraf.Histogram, ), expected: []byte(` # HELP http_request_duration_seconds Telegraf collected metric # TYPE http_request_duration_seconds histogram http_request_duration_seconds_bucket{le="0.5"} 129389 http_request_duration_seconds_bucket{le="+Inf"} 0 http_request_duration_seconds_sum 0 http_request_duration_seconds_count 0 `), }, { name: "simple with timestamp", config: FormatConfig{ TimestampExport: ExportTimestamp, }, metric: testutil.MustMetric( "cpu", map[string]string{ "host": "example.org", }, map[string]interface{}{ "time_idle": 42.0, }, time.Unix(1574279268, 0), ), expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{host="example.org"} 42 1574279268000 `), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { s, err := NewSerializer(FormatConfig{ MetricSortOrder: SortMetrics, TimestampExport: tt.config.TimestampExport, StringHandling: tt.config.StringHandling, }) require.NoError(t, err) actual, err := s.Serialize(tt.metric) require.NoError(t, err) require.Equal(t, strings.TrimSpace(string(tt.expected)), strings.TrimSpace(string(actual))) }) } } func TestSerializeBatch(t *testing.T) { tests := []struct { name string config FormatConfig metrics []telegraf.Metric expected []byte }{ { name: "simple", metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{ "host": "one.example.org", }, map[string]interface{}{ "time_idle": 42.0, }, time.Unix(0, 0), ), testutil.MustMetric( "cpu", map[string]string{ "host": "two.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="one.example.org"} 42 cpu_time_idle{host="two.example.org"} 42 `), }, { name: "multiple metric families", metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{ "host": "one.example.org", }, map[string]interface{}{ "time_idle": 42.0, "time_guest": 42.0, }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_guest Telegraf collected metric # TYPE cpu_time_guest untyped cpu_time_guest{host="one.example.org"} 42 # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{host="one.example.org"} 42 `), }, { name: "histogram", metrics: []telegraf.Metric{ testutil.MustMetric( "prometheus", map[string]string{}, map[string]interface{}{ "http_request_duration_seconds_sum": 53423, "http_request_duration_seconds_count": 144320, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "prometheus", map[string]string{"le": "0.05"}, map[string]interface{}{ "http_request_duration_seconds_bucket": 24054.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "prometheus", map[string]string{"le": "0.1"}, map[string]interface{}{ "http_request_duration_seconds_bucket": 33444.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "prometheus", map[string]string{"le": "0.2"}, map[string]interface{}{ "http_request_duration_seconds_bucket": 100392.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "prometheus", map[string]string{"le": "0.5"}, map[string]interface{}{ "http_request_duration_seconds_bucket": 129389.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "prometheus", map[string]string{"le": "1.0"}, map[string]interface{}{ "http_request_duration_seconds_bucket": 133988.0, }, time.Unix(0, 0), telegraf.Histogram, ), testutil.MustMetric( "prometheus", map[string]string{"le": "+Inf"}, map[string]interface{}{ "http_request_duration_seconds_bucket": 144320.0, }, time.Unix(0, 0), telegraf.Histogram, ), }, expected: []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: "", metrics: []telegraf.Metric{ testutil.MustMetric( "prometheus", map[string]string{}, map[string]interface{}{ "rpc_duration_seconds_sum": 1.7560473e+07, "rpc_duration_seconds_count": 2693, }, time.Unix(0, 0), telegraf.Summary, ), testutil.MustMetric( "prometheus", map[string]string{"quantile": "0.01"}, map[string]interface{}{ "rpc_duration_seconds": 3102.0, }, time.Unix(0, 0), telegraf.Summary, ), testutil.MustMetric( "prometheus", map[string]string{"quantile": "0.05"}, map[string]interface{}{ "rpc_duration_seconds": 3272.0, }, time.Unix(0, 0), telegraf.Summary, ), testutil.MustMetric( "prometheus", map[string]string{"quantile": "0.5"}, map[string]interface{}{ "rpc_duration_seconds": 4773.0, }, time.Unix(0, 0), telegraf.Summary, ), testutil.MustMetric( "prometheus", map[string]string{"quantile": "0.9"}, map[string]interface{}{ "rpc_duration_seconds": 9001.0, }, time.Unix(0, 0), telegraf.Summary, ), testutil.MustMetric( "prometheus", map[string]string{"quantile": "0.99"}, map[string]interface{}{ "rpc_duration_seconds": 76656.0, }, time.Unix(0, 0), telegraf.Summary, ), }, expected: []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 `), }, { name: "newer sample", metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{}, map[string]interface{}{ "time_idle": 43.0, }, time.Unix(1, 0), ), testutil.MustMetric( "cpu", map[string]string{}, 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 43 `), }, { name: "invalid label", metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{ "host-name": "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_name="example.org"} 42 `), }, { name: "discard strings", metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{}, map[string]interface{}{ "time_idle": 42.0, "cpu": "cpu0", }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle 42 `), }, { name: "string as label", config: FormatConfig{ StringHandling: StringAsLabel, }, metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{}, map[string]interface{}{ "time_idle": 42.0, "cpu": "cpu0", }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{cpu="cpu0"} 42 `), }, { name: "string as label duplicate tag", config: FormatConfig{ StringHandling: StringAsLabel, }, metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu0", }, map[string]interface{}{ "time_idle": 42.0, "cpu": "cpu1", }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_idle Telegraf collected metric # TYPE cpu_time_idle untyped cpu_time_idle{cpu="cpu0"} 42 `), }, { name: "multiple fields grouping", metrics: []telegraf.Metric{ testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu0", }, map[string]interface{}{ "time_guest": 8106.04, "time_system": 26271.4, "time_user": 92904.33, }, time.Unix(0, 0), ), testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu1", }, map[string]interface{}{ "time_guest": 8181.63, "time_system": 25351.49, "time_user": 96912.57, }, time.Unix(0, 0), ), testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu2", }, map[string]interface{}{ "time_guest": 7470.04, "time_system": 24998.43, "time_user": 96034.08, }, time.Unix(0, 0), ), testutil.MustMetric( "cpu", map[string]string{ "cpu": "cpu3", }, map[string]interface{}{ "time_guest": 7517.95, "time_system": 24970.82, "time_user": 94148, }, time.Unix(0, 0), ), }, expected: []byte(` # HELP cpu_time_guest Telegraf collected metric # TYPE cpu_time_guest untyped cpu_time_guest{cpu="cpu0"} 8106.04 cpu_time_guest{cpu="cpu1"} 8181.63 cpu_time_guest{cpu="cpu2"} 7470.04 cpu_time_guest{cpu="cpu3"} 7517.95 # HELP cpu_time_system Telegraf collected metric # TYPE cpu_time_system untyped cpu_time_system{cpu="cpu0"} 26271.4 cpu_time_system{cpu="cpu1"} 25351.49 cpu_time_system{cpu="cpu2"} 24998.43 cpu_time_system{cpu="cpu3"} 24970.82 # HELP cpu_time_user Telegraf collected metric # TYPE cpu_time_user untyped cpu_time_user{cpu="cpu0"} 92904.33 cpu_time_user{cpu="cpu1"} 96912.57 cpu_time_user{cpu="cpu2"} 96034.08 cpu_time_user{cpu="cpu3"} 94148 `), }, } for _, tt := range tests { t.Run(tt.name, func(t *testing.T) { s, err := NewSerializer(FormatConfig{ MetricSortOrder: SortMetrics, TimestampExport: tt.config.TimestampExport, StringHandling: tt.config.StringHandling, }) require.NoError(t, err) actual, err := s.SerializeBatch(tt.metrics) require.NoError(t, err) require.Equal(t, strings.TrimSpace(string(tt.expected)), strings.TrimSpace(string(actual))) }) } }