62 lines
1.6 KiB
Go
62 lines
1.6 KiB
Go
package metrics
|
|
|
|
import (
|
|
"testing"
|
|
"time"
|
|
)
|
|
|
|
func TestAddExtraLabels(t *testing.T) {
|
|
f := func(s, extraLabels, expectedResult string) {
|
|
t.Helper()
|
|
result := addExtraLabels(nil, []byte(s), extraLabels)
|
|
if string(result) != expectedResult {
|
|
t.Fatalf("unexpected result; got\n%s\nwant\n%s", result, expectedResult)
|
|
}
|
|
}
|
|
f("", `foo="bar"`, "")
|
|
f("a 123", `foo="bar"`, `a{foo="bar"} 123`+"\n")
|
|
f(`a{b="c"} 1.3`, `foo="bar"`, `a{foo="bar",b="c"} 1.3`+"\n")
|
|
f(`a{b="c}{"} 1.3`, `foo="bar",baz="x"`, `a{foo="bar",baz="x",b="c}{"} 1.3`+"\n")
|
|
f(`foo 1
|
|
bar{a="x"} 2
|
|
`, `foo="bar"`, `foo{foo="bar"} 1
|
|
bar{foo="bar",a="x"} 2
|
|
`)
|
|
f(`
|
|
foo 1
|
|
# some counter
|
|
# type foobar counter
|
|
foobar{a="b",c="d"} 4`, `x="y"`, `foo{x="y"} 1
|
|
# some counter
|
|
# type foobar counter
|
|
foobar{x="y",a="b",c="d"} 4
|
|
`)
|
|
}
|
|
|
|
func TestInitPushFailure(t *testing.T) {
|
|
f := func(pushURL string, interval time.Duration, extraLabels string) {
|
|
t.Helper()
|
|
if err := InitPush(pushURL, interval, extraLabels, false); err == nil {
|
|
t.Fatalf("expecting non-nil error")
|
|
}
|
|
}
|
|
|
|
// Invalid url
|
|
f("foobar", time.Second, "")
|
|
f("aaa://foobar", time.Second, "")
|
|
f("http:///bar", time.Second, "")
|
|
|
|
// Non-positive interval
|
|
f("http://foobar", 0, "")
|
|
f("http://foobar", -time.Second, "")
|
|
|
|
// Invalid extraLabels
|
|
f("http://foobar", time.Second, "foo")
|
|
f("http://foobar", time.Second, "foo{bar")
|
|
f("http://foobar", time.Second, "foo=bar")
|
|
f("http://foobar", time.Second, "foo='bar'")
|
|
f("http://foobar", time.Second, `foo="bar",baz`)
|
|
f("http://foobar", time.Second, `{foo="bar"}`)
|
|
f("http://foobar", time.Second, `a{foo="bar"}`)
|
|
}
|