From 487f1e07ff6c636fe8a0f4e929c359e69bba9a9b Mon Sep 17 00:00:00 2001
From: Guangwen Feng <fenggw-fnst@cn.fujitsu.com>
Date: Mon, 29 Jun 2020 20:16:52 +0800
Subject: [PATCH] Add unit test case for func State in alerting.go (#7476)

Signed-off-by: Guangwen Feng <fenggw-fnst@cn.fujitsu.com>
---
 rules/alerting_test.go | 40 ++++++++++++++++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/rules/alerting_test.go b/rules/alerting_test.go
index e0b21cd99..2cca9ee7a 100644
--- a/rules/alerting_test.go
+++ b/rules/alerting_test.go
@@ -45,6 +45,46 @@ annotations:
 	testutil.Assert(t, want == got, "incorrect HTML snippet; want:\n\n|%v|\n\ngot:\n\n|%v|", want, got)
 }
 
+func TestAlertingRuleState(t *testing.T) {
+	tests := []struct {
+		name   string
+		active map[uint64]*Alert
+		want   AlertState
+	}{
+		{
+			name: "MaxStateFiring",
+			active: map[uint64]*Alert{
+				0: {State: StatePending},
+				1: {State: StateFiring},
+			},
+			want: StateFiring,
+		},
+		{
+			name: "MaxStatePending",
+			active: map[uint64]*Alert{
+				0: {State: StateInactive},
+				1: {State: StatePending},
+			},
+			want: StatePending,
+		},
+		{
+			name: "MaxStateInactive",
+			active: map[uint64]*Alert{
+				0: {State: StateInactive},
+				1: {State: StateInactive},
+			},
+			want: StateInactive,
+		},
+	}
+
+	for i, test := range tests {
+		rule := NewAlertingRule(test.name, nil, 0, nil, nil, nil, true, nil)
+		rule.active = test.active
+		got := rule.State()
+		testutil.Assert(t, test.want == got, "test case %d unexpected AlertState, want:%d got:%d", i, test.want, got)
+	}
+}
+
 func TestAlertingRuleLabelsUpdate(t *testing.T) {
 	suite, err := promql.NewTest(t, `
 		load 1m