Function TestGetRulesFromBackup

pkg/ruler/ruler_test.go:1445–1654

(t *testing.T)

}

func TestGetRulesFromBackup(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.ACTIVE,
		"ruler2": ring.PENDING,
		"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: {
				&rulespb.RuleGroupDesc{User: "user1", Namespace: "namespace", Name: "b1", Interval: 10 * time.Minute, Limit: 10, Rules: rule},