mirror of
https://github.com/VictoriaMetrics/VictoriaMetrics.git
synced 2024-12-20 23:46:23 +01:00
d4ac4b7813
vmalert: support option to blackhole alerting notifications https://github.com/VictoriaMetrics/VictoriaMetrics/issues/4122 --------- Co-authored-by: Rao, B V Chalapathi <b_v_chalapathi.rao@nokia.com>
125 lines
2.8 KiB
Go
125 lines
2.8 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 TestInitBlackHole(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 notifiers; 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())
|
|
}
|
|
}
|
|
|
|
func TestInitBlackHoleWithNotifierUrl(t *testing.T) {
|
|
oldAddrs := *addrs
|
|
oldBlackHole := *blackHole
|
|
defer func() {
|
|
*addrs = oldAddrs
|
|
*blackHole = oldBlackHole
|
|
}()
|
|
|
|
*addrs = flagutil.ArrayString{"127.0.0.1", "127.0.0.2"}
|
|
*blackHole = true
|
|
|
|
_, err := Init(nil, nil, "")
|
|
if err == nil {
|
|
t.Fatalf("Expect Init to return error; instead got no error")
|
|
}
|
|
|
|
}
|
|
|
|
func TestInitBlackHoleWithNotifierConfig(t *testing.T) {
|
|
oldConfigPath := *configPath
|
|
oldBlackHole := *blackHole
|
|
defer func() {
|
|
*configPath = oldConfigPath
|
|
*blackHole = oldBlackHole
|
|
}()
|
|
|
|
*configPath = "/dummy/path"
|
|
*blackHole = true
|
|
|
|
fn, err := Init(nil, nil, "")
|
|
if err == nil {
|
|
t.Fatalf("Expect Init to return error; instead got no error")
|
|
}
|
|
|
|
if fn != nil {
|
|
t.Fatalf("expected no notifiers to be returned;but got %v instead", fn())
|
|
}
|
|
}
|
|
|
|
func TestInitWithNotifierConfigAndAddr(t *testing.T) {
|
|
oldConfigPath := *configPath
|
|
oldAddrs := *addrs
|
|
|
|
defer func() {
|
|
*configPath = oldConfigPath
|
|
*addrs = oldAddrs
|
|
}()
|
|
|
|
*addrs = flagutil.ArrayString{"127.0.0.1", "127.0.0.2"}
|
|
*configPath = "/dummy/path"
|
|
|
|
_, err := Init(nil, nil, "")
|
|
if err == nil {
|
|
t.Fatalf("Expect Init to return error; instead got no error")
|
|
}
|
|
}
|