Function getRulesHATest

pkg/ruler/ruler_test.go:1661–1885

(replicationFactor int)

}

func getRulesHATest(replicationFactor int) func(t *testing.T) {
	return func(t *testing.T) {
		// ruler ID -> (user ID -> list of groups).
		type expectedRulesMap map[string]map[string]rulespb.RuleGroupList

		rule := []*rulespb.RuleDesc{
			{
				Record: "rtest_user1_1",
				Expr:   "sum(rate(node_cpu_seconds_total[3h:10m]))",
			},
			{
				Alert: "atest_user1_1",
				Expr:  "sum(rate(node_cpu_seconds_total[3h:10m]))",
			},
			{
				Record: "rtest_user1_2",
				Expr:   "sum(rate(node_cpu_seconds_total[3h:10m]))",
				Labels: []cortexpb.LabelAdapter{
					{Name: "key", Value: "val"},
				},
			},
			{
				Alert: "atest_user1_2",
				Expr:  "sum(rate(node_cpu_seconds_total[3h:10m]))",
				Labels: []cortexpb.LabelAdapter{
					{Name: "key", Value: "val"},
				},
				Annotations: []cortexpb.LabelAdapter{
					{Name: "aKey", Value: "aVal"},
				},
				For:           10 * time.Second,
				KeepFiringFor: 20 * time.Second,
			},
		}

		tenantId := "user1"

		rulerStateMapOnePending := map[string]ring.InstanceState{
			"ruler1": ring.PENDING,
			"ruler2": ring.ACTIVE,
			"ruler3": ring.ACTIVE,
		}

		rulerAZEvenSpread := map[string]string{
			"ruler1": "a",
			"ruler2": "b",
			"ruler3": "c",
		}

		expectedRules := expectedRulesMap{
			"ruler1": map[string]rulespb.RuleGroupList{
				tenantId: {
					&rulespb.RuleGroupDesc{User: "user1", Namespace: "namespace", Name: "l1", Interval: 10 * time.Minute, Limit: 10, Rules: rule},
					&rulespb.RuleGroupDesc{User: "user1", Namespace: "namespace", Name: "l2", Interval: 0, Rules: rule},
				},
			},
			"ruler2": map[string]rulespb.RuleGroupList{
				tenantId: {