50 lines
1.5 KiB
Go
50 lines
1.5 KiB
Go
|
package main
|
||
|
|
|
||
|
|
import "testing"
|
||
|
|
|
||
|
|
func TestQueryArg(t *testing.T) {
|
||
|
|
if got := queryArg([]string{"up"}, nil); got != "up" {
|
||
|
|
t.Errorf(`queryArg(["up"]) = %q, want "up"`, got)
|
||
|
|
}
|
||
|
|
if got := queryArg([]string{"up", "--json"}, nil); got != "up" {
|
||
|
|
t.Errorf(`--json should be dropped, got %q`, got)
|
||
|
|
}
|
||
|
|
// single quoted PromQL arrives as one token
|
||
|
|
if got := queryArg([]string{"count by (node) (up)", "--json"}, nil); got != "count by (node) (up)" {
|
||
|
|
t.Errorf(`quoted query mangled: %q`, got)
|
||
|
|
}
|
||
|
|
// value-flags and their values are skipped, query survives
|
||
|
|
vf := map[string]bool{"--since": true, "--limit": true}
|
||
|
|
if got := queryArg([]string{`{app="x"}`, "--since", "1h", "--limit", "50"}, vf); got != `{app="x"}` {
|
||
|
|
t.Errorf(`value-flag skipping failed: %q`, got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestLabelStr(t *testing.T) {
|
||
|
|
got := labelStr(map[string]string{"__name__": "up", "job": "x", "instance": "y"})
|
||
|
|
if got != "up{instance=y,job=x}" { // __name__ extracted, rest sorted
|
||
|
|
t.Errorf("labelStr = %q", got)
|
||
|
|
}
|
||
|
|
if got := labelStr(map[string]string{"alertname": "Foo"}); got != "{alertname=Foo}" {
|
||
|
|
t.Errorf("labelStr (no __name__) = %q", got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestOneLineList(t *testing.T) {
|
||
|
|
if got := oneLineList(" "); got != "(none)" {
|
||
|
|
t.Errorf("empty = %q, want (none)", got)
|
||
|
|
}
|
||
|
|
if got := oneLineList("a\nb"); got != "a, b" {
|
||
|
|
t.Errorf("multi = %q, want 'a, b'", got)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
func TestHostOnly(t *testing.T) {
|
||
|
|
if got := hostOnly("foo.me/path"); got != "foo.me" {
|
||
|
|
t.Errorf("hostOnly = %q", got)
|
||
|
|
}
|
||
|
|
if got := hostOnly("foo.me"); got != "foo.me" {
|
||
|
|
t.Errorf("hostOnly = %q", got)
|
||
|
|
}
|
||
|
|
}
|