50 lines
1.2 KiB
Go
50 lines
1.2 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
|
||
|
|
`)
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestInitPushFailure(t *testing.T) {
|
||
|
|
f := func(interval time.Duration, extraLabels string) {
|
||
|
|
t.Helper()
|
||
|
|
defer func() {
|
||
|
|
if err := recover(); err == nil {
|
||
|
|
panic("expecting non-nil error")
|
||
|
|
}
|
||
|
|
}()
|
||
|
|
InitPush("http://foobar", interval, extraLabels, false)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Non-positive interval
|
||
|
|
f(0, "")
|
||
|
|
|
||
|
|
// Invalid extraLabels
|
||
|
|
f(time.Second, "foo")
|
||
|
|
f(time.Second, "foo{bar")
|
||
|
|
f(time.Second, "foo=bar")
|
||
|
|
f(time.Second, "foo='bar'")
|
||
|
|
f(time.Second, `foo="bar",baz`)
|
||
|
|
f(time.Second, `{foo="bar"}`)
|
||
|
|
f(time.Second, `a{foo="bar"}`)
|
||
|
|
}
|