mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-26 20:30:10 +01:00
debe1793bb
Signed-off-by: hagen1778 <roman@victoriametrics.com>
94 lines
2.2 KiB
Go
94 lines
2.2 KiB
Go
package notifier
|
|
|
|
import (
|
|
"testing"
|
|
|
|
"github.com/VictoriaMetrics/VictoriaMetrics/lib/flagutil"
|
|
)
|
|
|
|
func TestInit(t *testing.T) {
|
|
oldAddrs := *addrs
|
|
defer func() { *addrs = oldAddrs }()
|
|
|
|
*addrs = flagutil.ArrayString{"127.0.0.1", "127.0.0.2"}
|
|
|
|
fn, err := Init(nil, nil, "")
|
|
if err != nil {
|
|
t.Fatalf("%s", err)
|
|
}
|
|
|
|
nfs := fn()
|
|
if len(nfs) != 2 {
|
|
t.Fatalf("expected to get 2 notifiers; got %d", len(nfs))
|
|
}
|
|
|
|
targets := GetTargets()
|
|
if targets == nil || targets[TargetStatic] == nil {
|
|
t.Fatalf("expected to get static targets in response")
|
|
}
|
|
|
|
nf1 := targets[TargetStatic][0]
|
|
if nf1.Addr() != "127.0.0.1/api/v2/alerts" {
|
|
t.Fatalf("expected to get \"127.0.0.1/api/v2/alerts\"; got %q instead", nf1.Addr())
|
|
}
|
|
nf2 := targets[TargetStatic][1]
|
|
if nf2.Addr() != "127.0.0.2/api/v2/alerts" {
|
|
t.Fatalf("expected to get \"127.0.0.2/api/v2/alerts\"; got %q instead", nf2.Addr())
|
|
}
|
|
}
|
|
|
|
func TestInitNegative(t *testing.T) {
|
|
oldConfigPath := *configPath
|
|
oldAddrs := *addrs
|
|
oldBlackHole := *blackHole
|
|
|
|
defer func() {
|
|
*configPath = oldConfigPath
|
|
*addrs = oldAddrs
|
|
*blackHole = oldBlackHole
|
|
}()
|
|
|
|
f := func(path, addr string, bh bool) {
|
|
*configPath = path
|
|
*addrs = flagutil.ArrayString{addr}
|
|
*blackHole = bh
|
|
if _, err := Init(nil, nil, ""); err == nil {
|
|
t.Fatalf("expected to get error; got nil instead")
|
|
}
|
|
}
|
|
|
|
// *configPath, *addrs and *blackhole are mutually exclusive
|
|
f("/dummy/path", "127.0.0.1", false)
|
|
f("/dummy/path", "", true)
|
|
f("", "127.0.0.1", true)
|
|
}
|
|
|
|
func TestBlackHole(t *testing.T) {
|
|
oldBlackHole := *blackHole
|
|
defer func() { *blackHole = oldBlackHole }()
|
|
|
|
*blackHole = true
|
|
|
|
fn, err := Init(nil, nil, "")
|
|
if err != nil {
|
|
t.Fatalf("%s", err)
|
|
}
|
|
|
|
nfs := fn()
|
|
if len(nfs) != 1 {
|
|
t.Fatalf("expected to get 1 notifier; got %d", len(nfs))
|
|
}
|
|
|
|
targets := GetTargets()
|
|
if targets == nil || targets[TargetStatic] == nil {
|
|
t.Fatalf("expected to get static targets in response")
|
|
}
|
|
if len(targets[TargetStatic]) != 1 {
|
|
t.Fatalf("expected to get 1 static targets in response; but got %d", len(targets[TargetStatic]))
|
|
}
|
|
nf1 := targets[TargetStatic][0]
|
|
if nf1.Addr() != "blackhole" {
|
|
t.Fatalf("expected to get \"blackhole\"; got %q instead", nf1.Addr())
|
|
}
|
|
}
|