fix(tests): add more dedup tests

prymitive · Oct 26, 2019 · 5591c58 · 5591c58
1 parent ff1d69f
commit 5591c58
Showing 1 changed file with 39 additions and 0 deletions.
diff --git a/internal/alertmanager/dedup_test.go b/internal/alertmanager/dedup_test.go
@@ -90,6 +90,19 @@ func TestDedupAlertsWithoutLabels(t *testing.T) {
 	}
 }
 
+func TestDedupSilences(t *testing.T) {
+	os.Setenv("ALERTMANAGER_URI", "http://localhost")
+	config.Config.Read()
+	if err := pullAlerts(); err != nil {
+		t.Error(err)
+	}
+	silences := alertmanager.DedupSilences()
+	expected := 72
+	if len(silences) != expected {
+		t.Errorf("Expected %d silences keys, got %d", expected, len(silences))
+	}
+}
+
 func TestDedupAutocomplete(t *testing.T) {
 	if err := pullAlerts(); err != nil {
 		t.Error(err)
@@ -121,6 +134,32 @@ func TestDedupColors(t *testing.T) {
 	}
 }
 
+func TestDedupKnownLabels(t *testing.T) {
+	os.Setenv("ALERTMANAGER_URI", "http://localhost")
+	config.Config.Read()
+	if err := pullAlerts(); err != nil {
+		t.Error(err)
+	}
+	labels := alertmanager.DedupKnownLabels()
+	expected := 6
+	if len(labels) != expected {
+		t.Errorf("Expected %d knownLabels keys, got %d", expected, len(labels))
+	}
+}
+
+func TestDedupKnownLabelValues(t *testing.T) {
+	os.Setenv("ALERTMANAGER_URI", "http://localhost")
+	config.Config.Read()
+	if err := pullAlerts(); err != nil {
+		t.Error(err)
+	}
+	values := alertmanager.DedupKnownLabelValues("alertname")
+	expected := 4
+	if len(values) != expected {
+		t.Errorf("Expected %d knownLabelValues keys, got %d", expected, len(values))
+	}
+}
+
 func TestStripReceivers(t *testing.T) {
 	os.Setenv("RECEIVERS_STRIP", "by-name by-cluster-service")
 	os.Setenv("ALERTMANAGER_URI", "http://localhost")