916 lines
24 KiB
Go
916 lines
24 KiB
Go
// Copyright 2015 Prometheus Team
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package notify
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"fmt"
|
|
"io"
|
|
"reflect"
|
|
"testing"
|
|
"time"
|
|
|
|
"github.com/go-kit/log"
|
|
"github.com/prometheus/client_golang/prometheus"
|
|
"github.com/prometheus/common/model"
|
|
"github.com/stretchr/testify/require"
|
|
"gopkg.in/yaml.v2"
|
|
|
|
"github.com/prometheus/alertmanager/nflog"
|
|
"github.com/prometheus/alertmanager/nflog/nflogpb"
|
|
"github.com/prometheus/alertmanager/silence"
|
|
"github.com/prometheus/alertmanager/silence/silencepb"
|
|
"github.com/prometheus/alertmanager/timeinterval"
|
|
"github.com/prometheus/alertmanager/types"
|
|
)
|
|
|
|
type sendResolved bool
|
|
|
|
func (s sendResolved) SendResolved() bool {
|
|
return bool(s)
|
|
}
|
|
|
|
type notifierFunc func(ctx context.Context, alerts ...*types.Alert) (bool, error)
|
|
|
|
func (f notifierFunc) Notify(ctx context.Context, alerts ...*types.Alert) (bool, error) {
|
|
return f(ctx, alerts...)
|
|
}
|
|
|
|
type failStage struct{}
|
|
|
|
func (s failStage) Exec(ctx context.Context, l log.Logger, as ...*types.Alert) (context.Context, []*types.Alert, error) {
|
|
return ctx, nil, fmt.Errorf("some error")
|
|
}
|
|
|
|
type testNflog struct {
|
|
qres []*nflogpb.Entry
|
|
qerr error
|
|
|
|
logFunc func(r *nflogpb.Receiver, gkey string, firingAlerts, resolvedAlerts []uint64) error
|
|
}
|
|
|
|
func (l *testNflog) Query(p ...nflog.QueryParam) ([]*nflogpb.Entry, error) {
|
|
return l.qres, l.qerr
|
|
}
|
|
|
|
func (l *testNflog) Log(r *nflogpb.Receiver, gkey string, firingAlerts, resolvedAlerts []uint64) error {
|
|
return l.logFunc(r, gkey, firingAlerts, resolvedAlerts)
|
|
}
|
|
|
|
func (l *testNflog) GC() (int, error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func (l *testNflog) Snapshot(w io.Writer) (int, error) {
|
|
return 0, nil
|
|
}
|
|
|
|
func alertHashSet(hashes ...uint64) map[uint64]struct{} {
|
|
res := map[uint64]struct{}{}
|
|
|
|
for _, h := range hashes {
|
|
res[h] = struct{}{}
|
|
}
|
|
|
|
return res
|
|
}
|
|
|
|
func TestDedupStageNeedsUpdate(t *testing.T) {
|
|
now := utcNow()
|
|
|
|
cases := []struct {
|
|
entry *nflogpb.Entry
|
|
firingAlerts map[uint64]struct{}
|
|
resolvedAlerts map[uint64]struct{}
|
|
repeat time.Duration
|
|
resolve bool
|
|
|
|
res bool
|
|
}{
|
|
{
|
|
// No matching nflog entry should update.
|
|
entry: nil,
|
|
firingAlerts: alertHashSet(2, 3, 4),
|
|
res: true,
|
|
}, {
|
|
// No matching nflog entry shouldn't update if no alert fires.
|
|
entry: nil,
|
|
resolvedAlerts: alertHashSet(2, 3, 4),
|
|
res: false,
|
|
}, {
|
|
// Different sets of firing alerts should update.
|
|
entry: &nflogpb.Entry{FiringAlerts: []uint64{1, 2, 3}},
|
|
firingAlerts: alertHashSet(2, 3, 4),
|
|
res: true,
|
|
}, {
|
|
// Zero timestamp in the nflog entry should always update.
|
|
entry: &nflogpb.Entry{
|
|
FiringAlerts: []uint64{1, 2, 3},
|
|
Timestamp: time.Time{},
|
|
},
|
|
firingAlerts: alertHashSet(1, 2, 3),
|
|
res: true,
|
|
}, {
|
|
// Identical sets of alerts shouldn't update before repeat_interval.
|
|
entry: &nflogpb.Entry{
|
|
FiringAlerts: []uint64{1, 2, 3},
|
|
Timestamp: now.Add(-9 * time.Minute),
|
|
},
|
|
repeat: 10 * time.Minute,
|
|
firingAlerts: alertHashSet(1, 2, 3),
|
|
res: false,
|
|
}, {
|
|
// Identical sets of alerts should update after repeat_interval.
|
|
entry: &nflogpb.Entry{
|
|
FiringAlerts: []uint64{1, 2, 3},
|
|
Timestamp: now.Add(-11 * time.Minute),
|
|
},
|
|
repeat: 10 * time.Minute,
|
|
firingAlerts: alertHashSet(1, 2, 3),
|
|
res: true,
|
|
}, {
|
|
// Different sets of resolved alerts without firing alerts shouldn't update after repeat_interval.
|
|
entry: &nflogpb.Entry{
|
|
ResolvedAlerts: []uint64{1, 2, 3},
|
|
Timestamp: now.Add(-11 * time.Minute),
|
|
},
|
|
repeat: 10 * time.Minute,
|
|
resolvedAlerts: alertHashSet(3, 4, 5),
|
|
resolve: true,
|
|
res: false,
|
|
}, {
|
|
// Different sets of resolved alerts shouldn't update when resolve is false.
|
|
entry: &nflogpb.Entry{
|
|
FiringAlerts: []uint64{1, 2},
|
|
ResolvedAlerts: []uint64{3},
|
|
Timestamp: now.Add(-9 * time.Minute),
|
|
},
|
|
repeat: 10 * time.Minute,
|
|
firingAlerts: alertHashSet(1),
|
|
resolvedAlerts: alertHashSet(2, 3),
|
|
resolve: false,
|
|
res: false,
|
|
}, {
|
|
// Different sets of resolved alerts should update when resolve is true.
|
|
entry: &nflogpb.Entry{
|
|
FiringAlerts: []uint64{1, 2},
|
|
ResolvedAlerts: []uint64{3},
|
|
Timestamp: now.Add(-9 * time.Minute),
|
|
},
|
|
repeat: 10 * time.Minute,
|
|
firingAlerts: alertHashSet(1),
|
|
resolvedAlerts: alertHashSet(2, 3),
|
|
resolve: true,
|
|
res: true,
|
|
}, {
|
|
// Empty set of firing alerts should update when resolve is false.
|
|
entry: &nflogpb.Entry{
|
|
FiringAlerts: []uint64{1, 2},
|
|
ResolvedAlerts: []uint64{3},
|
|
Timestamp: now.Add(-9 * time.Minute),
|
|
},
|
|
repeat: 10 * time.Minute,
|
|
firingAlerts: alertHashSet(),
|
|
resolvedAlerts: alertHashSet(1, 2, 3),
|
|
resolve: false,
|
|
res: true,
|
|
}, {
|
|
// Empty set of firing alerts should update when resolve is true.
|
|
entry: &nflogpb.Entry{
|
|
FiringAlerts: []uint64{1, 2},
|
|
ResolvedAlerts: []uint64{3},
|
|
Timestamp: now.Add(-9 * time.Minute),
|
|
},
|
|
repeat: 10 * time.Minute,
|
|
firingAlerts: alertHashSet(),
|
|
resolvedAlerts: alertHashSet(1, 2, 3),
|
|
resolve: true,
|
|
res: true,
|
|
},
|
|
}
|
|
for i, c := range cases {
|
|
t.Log("case", i)
|
|
|
|
s := &DedupStage{
|
|
now: func() time.Time { return now },
|
|
rs: sendResolved(c.resolve),
|
|
}
|
|
res := s.needsUpdate(c.entry, c.firingAlerts, c.resolvedAlerts, c.repeat)
|
|
require.Equal(t, c.res, res)
|
|
}
|
|
}
|
|
|
|
func TestDedupStage(t *testing.T) {
|
|
i := 0
|
|
now := utcNow()
|
|
s := &DedupStage{
|
|
hash: func(a *types.Alert) uint64 {
|
|
res := uint64(i)
|
|
i++
|
|
return res
|
|
},
|
|
now: func() time.Time {
|
|
return now
|
|
},
|
|
rs: sendResolved(false),
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
_, _, err := s.Exec(ctx, log.NewNopLogger())
|
|
require.EqualError(t, err, "group key missing")
|
|
|
|
ctx = WithGroupKey(ctx, "1")
|
|
|
|
_, _, err = s.Exec(ctx, log.NewNopLogger())
|
|
require.EqualError(t, err, "repeat interval missing")
|
|
|
|
ctx = WithRepeatInterval(ctx, time.Hour)
|
|
|
|
alerts := []*types.Alert{{}, {}, {}}
|
|
|
|
// Must catch notification log query errors.
|
|
s.nflog = &testNflog{
|
|
qerr: errors.New("bad things"),
|
|
}
|
|
ctx, _, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.EqualError(t, err, "bad things")
|
|
|
|
// ... but skip ErrNotFound.
|
|
s.nflog = &testNflog{
|
|
qerr: nflog.ErrNotFound,
|
|
}
|
|
ctx, res, err := s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.NoError(t, err, "unexpected error on not found log entry")
|
|
require.Equal(t, alerts, res, "input alerts differ from result alerts")
|
|
|
|
s.nflog = &testNflog{
|
|
qerr: nil,
|
|
qres: []*nflogpb.Entry{
|
|
{FiringAlerts: []uint64{0, 1, 2}},
|
|
{FiringAlerts: []uint64{1, 2, 3}},
|
|
},
|
|
}
|
|
ctx, _, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Contains(t, err.Error(), "result size")
|
|
|
|
// Must return no error and no alerts no need to update.
|
|
i = 0
|
|
s.nflog = &testNflog{
|
|
qerr: nflog.ErrNotFound,
|
|
qres: []*nflogpb.Entry{
|
|
{
|
|
FiringAlerts: []uint64{0, 1, 2},
|
|
Timestamp: now,
|
|
},
|
|
},
|
|
}
|
|
ctx, res, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.NoError(t, err)
|
|
require.Nil(t, res, "unexpected alerts returned")
|
|
|
|
// Must return no error and all input alerts on changes.
|
|
i = 0
|
|
s.nflog = &testNflog{
|
|
qerr: nil,
|
|
qres: []*nflogpb.Entry{
|
|
{
|
|
FiringAlerts: []uint64{1, 2, 3, 4},
|
|
Timestamp: now,
|
|
},
|
|
},
|
|
}
|
|
_, res, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.NoError(t, err)
|
|
require.Equal(t, alerts, res, "unexpected alerts returned")
|
|
}
|
|
|
|
func TestMultiStage(t *testing.T) {
|
|
var (
|
|
alerts1 = []*types.Alert{{}}
|
|
alerts2 = []*types.Alert{{}, {}}
|
|
alerts3 = []*types.Alert{{}, {}, {}}
|
|
)
|
|
|
|
stage := MultiStage{
|
|
StageFunc(func(ctx context.Context, l log.Logger, alerts ...*types.Alert) (context.Context, []*types.Alert, error) {
|
|
if !reflect.DeepEqual(alerts, alerts1) {
|
|
t.Fatal("Input not equal to input of MultiStage")
|
|
}
|
|
//nolint:staticcheck // Ignore SA1029
|
|
ctx = context.WithValue(ctx, "key", "value")
|
|
return ctx, alerts2, nil
|
|
}),
|
|
StageFunc(func(ctx context.Context, l log.Logger, alerts ...*types.Alert) (context.Context, []*types.Alert, error) {
|
|
if !reflect.DeepEqual(alerts, alerts2) {
|
|
t.Fatal("Input not equal to output of previous stage")
|
|
}
|
|
v, ok := ctx.Value("key").(string)
|
|
if !ok || v != "value" {
|
|
t.Fatalf("Expected value %q for key %q but got %q", "value", "key", v)
|
|
}
|
|
return ctx, alerts3, nil
|
|
}),
|
|
}
|
|
|
|
_, alerts, err := stage.Exec(context.Background(), log.NewNopLogger(), alerts1...)
|
|
if err != nil {
|
|
t.Fatalf("Exec failed: %s", err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(alerts, alerts3) {
|
|
t.Fatal("Output of MultiStage is not equal to the output of the last stage")
|
|
}
|
|
}
|
|
|
|
func TestMultiStageFailure(t *testing.T) {
|
|
var (
|
|
ctx = context.Background()
|
|
s1 = failStage{}
|
|
stage = MultiStage{s1}
|
|
)
|
|
|
|
_, _, err := stage.Exec(ctx, log.NewNopLogger(), nil)
|
|
if err.Error() != "some error" {
|
|
t.Fatal("Errors were not propagated correctly by MultiStage")
|
|
}
|
|
}
|
|
|
|
func TestRoutingStage(t *testing.T) {
|
|
var (
|
|
alerts1 = []*types.Alert{{}}
|
|
alerts2 = []*types.Alert{{}, {}}
|
|
)
|
|
|
|
stage := RoutingStage{
|
|
"name": StageFunc(func(ctx context.Context, l log.Logger, alerts ...*types.Alert) (context.Context, []*types.Alert, error) {
|
|
if !reflect.DeepEqual(alerts, alerts1) {
|
|
t.Fatal("Input not equal to input of RoutingStage")
|
|
}
|
|
return ctx, alerts2, nil
|
|
}),
|
|
"not": failStage{},
|
|
}
|
|
|
|
ctx := WithReceiverName(context.Background(), "name")
|
|
|
|
_, alerts, err := stage.Exec(ctx, log.NewNopLogger(), alerts1...)
|
|
if err != nil {
|
|
t.Fatalf("Exec failed: %s", err)
|
|
}
|
|
|
|
if !reflect.DeepEqual(alerts, alerts2) {
|
|
t.Fatal("Output of RoutingStage is not equal to the output of the inner stage")
|
|
}
|
|
}
|
|
|
|
func TestRetryStageWithError(t *testing.T) {
|
|
fail, retry := true, true
|
|
sent := []*types.Alert{}
|
|
i := Integration{
|
|
notifier: notifierFunc(func(ctx context.Context, alerts ...*types.Alert) (bool, error) {
|
|
if fail {
|
|
fail = false
|
|
return retry, errors.New("fail to deliver notification")
|
|
}
|
|
sent = append(sent, alerts...)
|
|
return false, nil
|
|
}),
|
|
rs: sendResolved(false),
|
|
}
|
|
r := RetryStage{
|
|
integration: i,
|
|
metrics: NewMetrics(prometheus.NewRegistry()),
|
|
}
|
|
|
|
alerts := []*types.Alert{
|
|
{
|
|
Alert: model.Alert{
|
|
EndsAt: time.Now().Add(time.Hour),
|
|
},
|
|
},
|
|
}
|
|
|
|
ctx := context.Background()
|
|
ctx = WithFiringAlerts(ctx, []uint64{0})
|
|
|
|
// Notify with a recoverable error should retry and succeed.
|
|
resctx, res, err := r.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Nil(t, err)
|
|
require.Equal(t, alerts, res)
|
|
require.Equal(t, alerts, sent)
|
|
require.NotNil(t, resctx)
|
|
|
|
// Notify with an unrecoverable error should fail.
|
|
sent = sent[:0]
|
|
fail = true
|
|
retry = false
|
|
resctx, _, err = r.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.NotNil(t, err)
|
|
require.NotNil(t, resctx)
|
|
}
|
|
|
|
func TestRetryStageNoResolved(t *testing.T) {
|
|
sent := []*types.Alert{}
|
|
i := Integration{
|
|
notifier: notifierFunc(func(ctx context.Context, alerts ...*types.Alert) (bool, error) {
|
|
sent = append(sent, alerts...)
|
|
return false, nil
|
|
}),
|
|
rs: sendResolved(false),
|
|
}
|
|
r := RetryStage{
|
|
integration: i,
|
|
metrics: NewMetrics(prometheus.NewRegistry()),
|
|
}
|
|
|
|
alerts := []*types.Alert{
|
|
{
|
|
Alert: model.Alert{
|
|
EndsAt: time.Now().Add(-time.Hour),
|
|
},
|
|
},
|
|
{
|
|
Alert: model.Alert{
|
|
EndsAt: time.Now().Add(time.Hour),
|
|
},
|
|
},
|
|
}
|
|
|
|
ctx := context.Background()
|
|
|
|
resctx, res, err := r.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.EqualError(t, err, "firing alerts missing")
|
|
require.Nil(t, res)
|
|
require.NotNil(t, resctx)
|
|
|
|
ctx = WithFiringAlerts(ctx, []uint64{0})
|
|
|
|
resctx, res, err = r.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Nil(t, err)
|
|
require.Equal(t, alerts, res)
|
|
require.Equal(t, []*types.Alert{alerts[1]}, sent)
|
|
require.NotNil(t, resctx)
|
|
|
|
// All alerts are resolved.
|
|
sent = sent[:0]
|
|
ctx = WithFiringAlerts(ctx, []uint64{})
|
|
alerts[1].Alert.EndsAt = time.Now().Add(-time.Hour)
|
|
|
|
resctx, res, err = r.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Nil(t, err)
|
|
require.Equal(t, alerts, res)
|
|
require.Equal(t, []*types.Alert{}, sent)
|
|
require.NotNil(t, resctx)
|
|
}
|
|
|
|
func TestRetryStageSendResolved(t *testing.T) {
|
|
sent := []*types.Alert{}
|
|
i := Integration{
|
|
notifier: notifierFunc(func(ctx context.Context, alerts ...*types.Alert) (bool, error) {
|
|
sent = append(sent, alerts...)
|
|
return false, nil
|
|
}),
|
|
rs: sendResolved(true),
|
|
}
|
|
r := RetryStage{
|
|
integration: i,
|
|
metrics: NewMetrics(prometheus.NewRegistry()),
|
|
}
|
|
|
|
alerts := []*types.Alert{
|
|
{
|
|
Alert: model.Alert{
|
|
EndsAt: time.Now().Add(-time.Hour),
|
|
},
|
|
},
|
|
{
|
|
Alert: model.Alert{
|
|
EndsAt: time.Now().Add(time.Hour),
|
|
},
|
|
},
|
|
}
|
|
|
|
ctx := context.Background()
|
|
ctx = WithFiringAlerts(ctx, []uint64{0})
|
|
|
|
resctx, res, err := r.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Nil(t, err)
|
|
require.Equal(t, alerts, res)
|
|
require.Equal(t, alerts, sent)
|
|
require.NotNil(t, resctx)
|
|
|
|
// All alerts are resolved.
|
|
sent = sent[:0]
|
|
ctx = WithFiringAlerts(ctx, []uint64{})
|
|
alerts[1].Alert.EndsAt = time.Now().Add(-time.Hour)
|
|
|
|
resctx, res, err = r.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Nil(t, err)
|
|
require.Equal(t, alerts, res)
|
|
require.Equal(t, alerts, sent)
|
|
require.NotNil(t, resctx)
|
|
}
|
|
|
|
func TestSetNotifiesStage(t *testing.T) {
|
|
tnflog := &testNflog{}
|
|
s := &SetNotifiesStage{
|
|
recv: &nflogpb.Receiver{GroupName: "test"},
|
|
nflog: tnflog,
|
|
}
|
|
alerts := []*types.Alert{{}, {}, {}}
|
|
ctx := context.Background()
|
|
|
|
resctx, res, err := s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.EqualError(t, err, "group key missing")
|
|
require.Nil(t, res)
|
|
require.NotNil(t, resctx)
|
|
|
|
ctx = WithGroupKey(ctx, "1")
|
|
|
|
resctx, res, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.EqualError(t, err, "firing alerts missing")
|
|
require.Nil(t, res)
|
|
require.NotNil(t, resctx)
|
|
|
|
ctx = WithFiringAlerts(ctx, []uint64{0, 1, 2})
|
|
|
|
resctx, res, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.EqualError(t, err, "resolved alerts missing")
|
|
require.Nil(t, res)
|
|
require.NotNil(t, resctx)
|
|
|
|
ctx = WithResolvedAlerts(ctx, []uint64{})
|
|
|
|
tnflog.logFunc = func(r *nflogpb.Receiver, gkey string, firingAlerts, resolvedAlerts []uint64) error {
|
|
require.Equal(t, s.recv, r)
|
|
require.Equal(t, "1", gkey)
|
|
require.Equal(t, []uint64{0, 1, 2}, firingAlerts)
|
|
require.Equal(t, []uint64{}, resolvedAlerts)
|
|
return nil
|
|
}
|
|
resctx, res, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Nil(t, err)
|
|
require.Equal(t, alerts, res)
|
|
require.NotNil(t, resctx)
|
|
|
|
ctx = WithFiringAlerts(ctx, []uint64{})
|
|
ctx = WithResolvedAlerts(ctx, []uint64{0, 1, 2})
|
|
|
|
tnflog.logFunc = func(r *nflogpb.Receiver, gkey string, firingAlerts, resolvedAlerts []uint64) error {
|
|
require.Equal(t, s.recv, r)
|
|
require.Equal(t, "1", gkey)
|
|
require.Equal(t, []uint64{}, firingAlerts)
|
|
require.Equal(t, []uint64{0, 1, 2}, resolvedAlerts)
|
|
return nil
|
|
}
|
|
resctx, res, err = s.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
require.Nil(t, err)
|
|
require.Equal(t, alerts, res)
|
|
require.NotNil(t, resctx)
|
|
}
|
|
|
|
func TestMuteStage(t *testing.T) {
|
|
// Mute all label sets that have a "mute" key.
|
|
muter := types.MuteFunc(func(lset model.LabelSet) bool {
|
|
_, ok := lset["mute"]
|
|
return ok
|
|
})
|
|
|
|
stage := NewMuteStage(muter)
|
|
|
|
in := []model.LabelSet{
|
|
{},
|
|
{"test": "set"},
|
|
{"mute": "me"},
|
|
{"foo": "bar", "test": "set"},
|
|
{"foo": "bar", "mute": "me"},
|
|
{},
|
|
{"not": "muted"},
|
|
}
|
|
out := []model.LabelSet{
|
|
{},
|
|
{"test": "set"},
|
|
{"foo": "bar", "test": "set"},
|
|
{},
|
|
{"not": "muted"},
|
|
}
|
|
|
|
var inAlerts []*types.Alert
|
|
for _, lset := range in {
|
|
inAlerts = append(inAlerts, &types.Alert{
|
|
Alert: model.Alert{Labels: lset},
|
|
})
|
|
}
|
|
|
|
_, alerts, err := stage.Exec(context.Background(), log.NewNopLogger(), inAlerts...)
|
|
if err != nil {
|
|
t.Fatalf("Exec failed: %s", err)
|
|
}
|
|
|
|
var got []model.LabelSet
|
|
for _, a := range alerts {
|
|
got = append(got, a.Labels)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, out) {
|
|
t.Fatalf("Muting failed, expected: %v\ngot %v", out, got)
|
|
}
|
|
}
|
|
|
|
func TestMuteStageWithSilences(t *testing.T) {
|
|
silences, err := silence.New(silence.Options{Retention: time.Hour})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
silID, err := silences.Set(&silencepb.Silence{
|
|
EndsAt: utcNow().Add(time.Hour),
|
|
Matchers: []*silencepb.Matcher{{Name: "mute", Pattern: "me"}},
|
|
})
|
|
if err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
marker := types.NewMarker(prometheus.NewRegistry())
|
|
silencer := silence.NewSilencer(silences, marker, log.NewNopLogger())
|
|
stage := NewMuteStage(silencer)
|
|
|
|
in := []model.LabelSet{
|
|
{},
|
|
{"test": "set"},
|
|
{"mute": "me"},
|
|
{"foo": "bar", "test": "set"},
|
|
{"foo": "bar", "mute": "me"},
|
|
{},
|
|
{"not": "muted"},
|
|
}
|
|
out := []model.LabelSet{
|
|
{},
|
|
{"test": "set"},
|
|
{"foo": "bar", "test": "set"},
|
|
{},
|
|
{"not": "muted"},
|
|
}
|
|
|
|
var inAlerts []*types.Alert
|
|
for _, lset := range in {
|
|
inAlerts = append(inAlerts, &types.Alert{
|
|
Alert: model.Alert{Labels: lset},
|
|
})
|
|
}
|
|
|
|
// Set the second alert as previously silenced with an old version
|
|
// number. This is expected to get unsilenced by the stage.
|
|
marker.SetSilenced(inAlerts[1].Fingerprint(), 0, []string{"123"}, nil)
|
|
|
|
_, alerts, err := stage.Exec(context.Background(), log.NewNopLogger(), inAlerts...)
|
|
if err != nil {
|
|
t.Fatalf("Exec failed: %s", err)
|
|
}
|
|
|
|
var got []model.LabelSet
|
|
for _, a := range alerts {
|
|
got = append(got, a.Labels)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, out) {
|
|
t.Fatalf("Muting failed, expected: %v\ngot %v", out, got)
|
|
}
|
|
|
|
// Do it again to exercise the version tracking of silences.
|
|
_, alerts, err = stage.Exec(context.Background(), log.NewNopLogger(), inAlerts...)
|
|
if err != nil {
|
|
t.Fatalf("Exec failed: %s", err)
|
|
}
|
|
|
|
got = got[:0]
|
|
for _, a := range alerts {
|
|
got = append(got, a.Labels)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, out) {
|
|
t.Fatalf("Muting failed, expected: %v\ngot %v", out, got)
|
|
}
|
|
|
|
// Expire the silence and verify that no alerts are silenced now.
|
|
if err := silences.Expire(silID); err != nil {
|
|
t.Fatal(err)
|
|
}
|
|
|
|
_, alerts, err = stage.Exec(context.Background(), log.NewNopLogger(), inAlerts...)
|
|
if err != nil {
|
|
t.Fatalf("Exec failed: %s", err)
|
|
}
|
|
got = got[:0]
|
|
for _, a := range alerts {
|
|
got = append(got, a.Labels)
|
|
}
|
|
|
|
if !reflect.DeepEqual(got, in) {
|
|
t.Fatalf("Unmuting failed, expected: %v\ngot %v", in, got)
|
|
}
|
|
}
|
|
|
|
func TestTimeMuteStage(t *testing.T) {
|
|
// Route mutes alerts outside business hours if it is a mute_time_interval
|
|
muteIn := `
|
|
---
|
|
- weekdays: ['monday:friday']
|
|
times:
|
|
- start_time: '00:00'
|
|
end_time: '09:00'
|
|
- start_time: '17:00'
|
|
end_time: '24:00'
|
|
- weekdays: ['saturday', 'sunday']`
|
|
|
|
cases := []struct {
|
|
fireTime string
|
|
labels model.LabelSet
|
|
shouldMute bool
|
|
}{
|
|
{
|
|
// Friday during business hours
|
|
fireTime: "01 Jan 21 09:00 +0000",
|
|
labels: model.LabelSet{"foo": "bar"},
|
|
shouldMute: false,
|
|
},
|
|
{
|
|
// Tuesday before 5pm
|
|
fireTime: "01 Dec 20 16:59 +0000",
|
|
labels: model.LabelSet{"dont": "mute"},
|
|
shouldMute: false,
|
|
},
|
|
{
|
|
// Saturday
|
|
fireTime: "17 Oct 20 10:00 +0000",
|
|
labels: model.LabelSet{"mute": "me"},
|
|
shouldMute: true,
|
|
},
|
|
{
|
|
// Wednesday before 9am
|
|
fireTime: "14 Oct 20 05:00 +0000",
|
|
labels: model.LabelSet{"mute": "me"},
|
|
shouldMute: true,
|
|
},
|
|
{
|
|
// Ensure comparisons are UTC only. 12:00 KST should be muted (03:00 UTC)
|
|
fireTime: "14 Oct 20 12:00 +0900",
|
|
labels: model.LabelSet{"mute": "kst"},
|
|
shouldMute: true,
|
|
},
|
|
{
|
|
// Ensure comparisons are UTC only. 22:00 KST should not be muted (13:00 UTC)
|
|
fireTime: "14 Oct 20 22:00 +0900",
|
|
labels: model.LabelSet{"kst": "dont_mute"},
|
|
shouldMute: false,
|
|
},
|
|
}
|
|
var intervals []timeinterval.TimeInterval
|
|
err := yaml.Unmarshal([]byte(muteIn), &intervals)
|
|
if err != nil {
|
|
t.Fatalf("Couldn't unmarshal time interval %s", err)
|
|
}
|
|
m := map[string][]timeinterval.TimeInterval{"test": intervals}
|
|
stage := NewTimeMuteStage(m)
|
|
|
|
outAlerts := []*types.Alert{}
|
|
nonMuteCount := 0
|
|
for _, tc := range cases {
|
|
now, err := time.Parse(time.RFC822Z, tc.fireTime)
|
|
if err != nil {
|
|
t.Fatalf("Couldn't parse fire time %s %s", tc.fireTime, err)
|
|
}
|
|
// Count alerts with shouldMute == false and compare to ensure none are muted incorrectly
|
|
if !tc.shouldMute {
|
|
nonMuteCount++
|
|
}
|
|
a := model.Alert{Labels: tc.labels}
|
|
alerts := []*types.Alert{{Alert: a}}
|
|
ctx := context.Background()
|
|
ctx = WithNow(ctx, now)
|
|
ctx = WithActiveTimeIntervals(ctx, []string{})
|
|
ctx = WithMuteTimeIntervals(ctx, []string{"test"})
|
|
|
|
_, out, err := stage.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error in time mute stage %s", err)
|
|
}
|
|
outAlerts = append(outAlerts, out...)
|
|
}
|
|
for _, alert := range outAlerts {
|
|
if _, ok := alert.Alert.Labels["mute"]; ok {
|
|
t.Fatalf("Expected alert to be muted %+v", alert.Alert)
|
|
}
|
|
}
|
|
if len(outAlerts) != nonMuteCount {
|
|
t.Fatalf("Expected %d alerts after time mute stage but got %d", nonMuteCount, len(outAlerts))
|
|
}
|
|
}
|
|
|
|
func TestTimeActiveStage(t *testing.T) {
|
|
// Route mutes alerts inside business hours if it is an active time interval
|
|
muteIn := `
|
|
---
|
|
- weekdays: ['monday:friday']
|
|
times:
|
|
- start_time: '00:00'
|
|
end_time: '09:00'
|
|
- start_time: '17:00'
|
|
end_time: '24:00'
|
|
- weekdays: ['saturday', 'sunday']`
|
|
|
|
cases := []struct {
|
|
fireTime string
|
|
labels model.LabelSet
|
|
shouldMute bool
|
|
}{
|
|
{
|
|
// Friday during business hours
|
|
fireTime: "01 Jan 21 09:00 +0000",
|
|
labels: model.LabelSet{"mute": "me"},
|
|
shouldMute: true,
|
|
},
|
|
{
|
|
// Tuesday before 5pm
|
|
fireTime: "01 Dec 20 16:59 +0000",
|
|
labels: model.LabelSet{"mute": "me"},
|
|
shouldMute: true,
|
|
},
|
|
{
|
|
// Saturday
|
|
fireTime: "17 Oct 20 10:00 +0000",
|
|
labels: model.LabelSet{"foo": "bar"},
|
|
shouldMute: false,
|
|
},
|
|
{
|
|
// Wednesday before 9am
|
|
fireTime: "14 Oct 20 05:00 +0000",
|
|
labels: model.LabelSet{"dont": "mute"},
|
|
shouldMute: false,
|
|
},
|
|
}
|
|
var intervals []timeinterval.TimeInterval
|
|
err := yaml.Unmarshal([]byte(muteIn), &intervals)
|
|
if err != nil {
|
|
t.Fatalf("Couldn't unmarshal time interval %s", err)
|
|
}
|
|
m := map[string][]timeinterval.TimeInterval{"test": intervals}
|
|
stage := NewTimeActiveStage(m)
|
|
|
|
outAlerts := []*types.Alert{}
|
|
nonMuteCount := 0
|
|
for _, tc := range cases {
|
|
now, err := time.Parse(time.RFC822Z, tc.fireTime)
|
|
if err != nil {
|
|
t.Fatalf("Couldn't parse fire time %s %s", tc.fireTime, err)
|
|
}
|
|
// Count alerts with shouldMute == false and compare to ensure none are muted incorrectly
|
|
if !tc.shouldMute {
|
|
nonMuteCount++
|
|
}
|
|
a := model.Alert{Labels: tc.labels}
|
|
alerts := []*types.Alert{{Alert: a}}
|
|
ctx := context.Background()
|
|
ctx = WithNow(ctx, now)
|
|
ctx = WithActiveTimeIntervals(ctx, []string{"test"})
|
|
ctx = WithMuteTimeIntervals(ctx, []string{})
|
|
|
|
_, out, err := stage.Exec(ctx, log.NewNopLogger(), alerts...)
|
|
if err != nil {
|
|
t.Fatalf("Unexpected error in time mute stage %s", err)
|
|
}
|
|
outAlerts = append(outAlerts, out...)
|
|
}
|
|
for _, alert := range outAlerts {
|
|
if _, ok := alert.Alert.Labels["mute"]; ok {
|
|
t.Fatalf("Expected alert to be muted %+v", alert.Alert)
|
|
}
|
|
}
|
|
if len(outAlerts) != nonMuteCount {
|
|
t.Fatalf("Expected %d alerts after time mute stage but got %d", nonMuteCount, len(outAlerts))
|
|
}
|
|
}
|
|
|
|
func BenchmarkHashAlert(b *testing.B) {
|
|
alert := &types.Alert{
|
|
Alert: model.Alert{
|
|
Labels: model.LabelSet{"foo": "the_first_value", "bar": "the_second_value", "another": "value"},
|
|
},
|
|
}
|
|
for i := 0; i < b.N; i++ {
|
|
hashAlert(alert)
|
|
}
|
|
}
|