From 8a7b2a50e1f9270af162e56e43cde62857c2ea5f Mon Sep 17 00:00:00 2001
From: Fabian Reinartz <fab.reinartz@gmail.com>
Date: Mon, 28 Sep 2015 14:13:18 +0200
Subject: [PATCH] Add mutedNotifier test

---
 notify_test.go | 71 ++++++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 71 insertions(+)
 create mode 100644 notify_test.go

diff --git a/notify_test.go b/notify_test.go
new file mode 100644
index 00000000..d5f6a976
--- /dev/null
+++ b/notify_test.go
@@ -0,0 +1,71 @@
+package main
+
+import (
+	"reflect"
+	"testing"
+
+	"github.com/prometheus/common/model"
+	"golang.org/x/net/context"
+
+	"github.com/prometheus/alertmanager/types"
+)
+
+type recordNotifier struct {
+	alerts []*types.Alert
+}
+
+func (n *recordNotifier) Notify(ctx context.Context, as ...*types.Alert) error {
+	n.alerts = append(n.alerts, as...)
+	return nil
+}
+
+func TestMutingNotifier(t *testing.T) {
+	// Mute all label sets that have a "mute" key.
+	muter := types.MuteFunc(func(lset model.LabelSet) bool {
+		if _, ok := lset["mute"]; ok {
+			return true
+		}
+		return false
+	})
+
+	record := &recordNotifier{}
+	muteNotifer := mutingNotifier{
+		notifier: record,
+		Muter:    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{Labels: lset})
+	}
+
+	if err := muteNotifer.Notify(nil, inAlerts...); err != nil {
+		t.Fatalf("Notifying failed: %s", err)
+	}
+
+	var got []model.LabelSet
+	for _, a := range record.alerts {
+		got = append(got, a.Labels)
+	}
+
+	if !reflect.DeepEqual(got, out) {
+		t.Fatalf("Muting failed, expected: %v\ngot %v", out, got)
+	}
+}