package wavefront import ( "github.com/influxdata/telegraf" "github.com/influxdata/telegraf/metric" "github.com/influxdata/telegraf/testutil" "reflect" "strings" "testing" "time" ) // default config used by Tests func defaultWavefront() *Wavefront { return &Wavefront{ Host: "localhost", Port: 2878, Prefix: "testWF.", SimpleFields: false, MetricSeparator: ".", ConvertPaths: true, ConvertBool: true, UseRegex: false, } } func TestBuildMetrics(t *testing.T) { w := defaultWavefront() w.Prefix = "testthis." pathReplacer = strings.NewReplacer("_", w.MetricSeparator) testMetric1, _ := metric.New( "test.simple.metric", map[string]string{"tag1": "value1", "host": "testHost"}, map[string]interface{}{"value": 123}, time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC), ) var timestamp int64 = 1257894000 var metricTests = []struct { metric telegraf.Metric metricPoints []MetricPoint }{ { testutil.TestMetric(float64(1), "testing_just*a%metric:float", "metric2"), []MetricPoint{ {Metric: w.Prefix + "testing.just-a-metric-float", Value: 1, Timestamp: timestamp, Tags: map[string]string{"tag1": "value1"}}, {Metric: w.Prefix + "testing.metric2", Value: 1, Timestamp: timestamp, Tags: map[string]string{"tag1": "value1"}}, }, }, { testMetric1, []MetricPoint{{Metric: w.Prefix + "test.simple.metric", Value: 123, Timestamp: timestamp, Source: "testHost", Tags: map[string]string{"tag1": "value1"}}}, }, } for _, mt := range metricTests { ml := buildMetrics(mt.metric, w) for i, line := range ml { if mt.metricPoints[i].Metric != line.Metric || mt.metricPoints[i].Value != line.Value { t.Errorf("\nexpected\t%+v %+v\nreceived\t%+v %+v\n", mt.metricPoints[i].Metric, mt.metricPoints[i].Value, line.Metric, line.Value) } } } } func TestBuildMetricsWithSimpleFields(t *testing.T) { w := defaultWavefront() w.Prefix = "testthis." w.SimpleFields = true pathReplacer = strings.NewReplacer("_", w.MetricSeparator) testMetric1, _ := metric.New( "test.simple.metric", map[string]string{"tag1": "value1"}, map[string]interface{}{"value": 123}, time.Date(2009, time.November, 10, 23, 0, 0, 0, time.UTC), ) var metricTests = []struct { metric telegraf.Metric metricLines []MetricPoint }{ { testutil.TestMetric(float64(1), "testing_just*a%metric:float"), []MetricPoint{{Metric: w.Prefix + "testing.just-a-metric-float.value", Value: 1}}, }, { testMetric1, []MetricPoint{{Metric: w.Prefix + "test.simple.metric.value", Value: 123}}, }, } for _, mt := range metricTests { ml := buildMetrics(mt.metric, w) for i, line := range ml { if mt.metricLines[i].Metric != line.Metric || mt.metricLines[i].Value != line.Value { t.Errorf("\nexpected\t%+v %+v\nreceived\t%+v %+v\n", mt.metricLines[i].Metric, mt.metricLines[i].Value, line.Metric, line.Value) } } } } func TestBuildTags(t *testing.T) { w := defaultWavefront() var tagtests = []struct { ptIn map[string]string outSource string outTags map[string]string }{ { map[string]string{}, "", map[string]string{}, }, { map[string]string{"one": "two", "three": "four", "host": "testHost"}, "testHost", map[string]string{"one": "two", "three": "four"}, }, { map[string]string{"aaa": "bbb", "host": "testHost"}, "testHost", map[string]string{"aaa": "bbb"}, }, { map[string]string{"bbb": "789", "aaa": "123", "host": "testHost"}, "testHost", map[string]string{"aaa": "123", "bbb": "789"}, }, { map[string]string{"host": "aaa", "dc": "bbb"}, "aaa", map[string]string{"dc": "bbb"}, }, } for _, tt := range tagtests { source, tags := buildTags(tt.ptIn, w) if source != tt.outSource { t.Errorf("\nexpected\t%+v\nreceived\t%+v\n", tt.outSource, source) } if !reflect.DeepEqual(tags, tt.outTags) { t.Errorf("\nexpected\t%+v\nreceived\t%+v\n", tt.outTags, tags) } } } func TestBuildTagsWithSource(t *testing.T) { w := defaultWavefront() w.SourceOverride = []string{"snmp_host", "hostagent"} var tagtests = []struct { ptIn map[string]string outSource string outTags map[string]string }{ { map[string]string{"host": "realHost"}, "realHost", map[string]string{}, }, { map[string]string{"tag1": "value1", "host": "realHost"}, "realHost", map[string]string{"tag1": "value1"}, }, { map[string]string{"snmp_host": "realHost", "host": "origHost"}, "realHost", map[string]string{"telegraf_host": "origHost"}, }, { map[string]string{"hostagent": "realHost", "host": "origHost"}, "realHost", map[string]string{"telegraf_host": "origHost"}, }, { map[string]string{"hostagent": "abc", "snmp_host": "realHost", "host": "origHost"}, "realHost", map[string]string{"hostagent": "abc", "telegraf_host": "origHost"}, }, { map[string]string{"something": "abc", "host": "r*@l\"Ho/st"}, "r-@l\\\"Ho/st", map[string]string{"something": "abc"}, }, } for _, tt := range tagtests { source, tags := buildTags(tt.ptIn, w) if source != tt.outSource { t.Errorf("\nexpected\t%+v\nreceived\t%+v\n", tt.outSource, source) } if !reflect.DeepEqual(tags, tt.outTags) { t.Errorf("\nexpected\t%+v\nreceived\t%+v\n", tt.outTags, tags) } } } func TestBuildValue(t *testing.T) { w := defaultWavefront() var valuetests = []struct { value interface{} name string out float64 isErr bool }{ {value: int64(123), out: 123}, {value: uint64(456), out: 456}, {value: float64(789), out: 789}, {value: true, out: 1}, {value: false, out: 0}, {value: "bad", out: 0, isErr: true}, } for _, vt := range valuetests { value, err := buildValue(vt.value, vt.name, w) if vt.isErr && err == nil { t.Errorf("\nexpected error with\t%+v\nreceived\t%+v\n", vt.out, value) } else if value != vt.out { t.Errorf("\nexpected\t%+v\nreceived\t%+v\n", vt.out, value) } } } func TestBuildValueString(t *testing.T) { w := defaultWavefront() w.StringToNumber = map[string][]map[string]float64{ "test1": {{"green": 1, "red": 10}}, "test2": {{"active": 1, "hidden": 2}}, } var valuetests = []struct { value interface{} name string out float64 isErr bool }{ {value: int64(123), name: "", out: 123}, {value: "green", name: "test1", out: 1}, {value: "red", name: "test1", out: 10}, {value: "hidden", name: "test2", out: 2}, {value: "bad", name: "test1", out: 0, isErr: true}, } for _, vt := range valuetests { value, err := buildValue(vt.value, vt.name, w) if vt.isErr && err == nil { t.Errorf("\nexpected error with\t%+v\nreceived\t%+v\n", vt.out, value) } else if value != vt.out { t.Errorf("\nexpected\t%+v\nreceived\t%+v\n", vt.out, value) } } } func TestFormatMetricPoint(t *testing.T) { w := defaultWavefront() testpoint := &MetricPoint{ Metric: "test.metric.something", Value: 123.456, Timestamp: 1257894000, Source: "testSource", Tags: map[string]string{"sp*c!@l\"-ch/rs": "sp*c!@l/ val\"ue"}, } expected := "test.metric.something 123.456000 1257894000 source=\"testSource\" sp-c--l--ch-rs=\"sp-c!@l/ val\\\"ue\"" received := formatMetricPoint(testpoint, w) if expected != received { t.Errorf("\nexpected\t%+v\nreceived\t%+v\n", expected, received) } } // Benchmarks to test performance of string replacement via Regex and Replacer var testString = "this_is*my!test/string\\for=replacement" func BenchmarkReplaceAllString(b *testing.B) { for n := 0; n < b.N; n++ { sanitizedRegex.ReplaceAllString(testString, "-") } } func BenchmarkReplaceAllLiteralString(b *testing.B) { for n := 0; n < b.N; n++ { sanitizedRegex.ReplaceAllLiteralString(testString, "-") } } func BenchmarkReplacer(b *testing.B) { for n := 0; n < b.N; n++ { sanitizedChars.Replace(testString) } }