Function TestGetRules

pkg/ruler/ruler_test.go:647–1443

(t *testing.T)

}

func TestGetRules(t *testing.T) {
	// ruler ID -> (user ID -> list of groups).
	type expectedRulesMap map[string]map[string]rulespb.RuleGroupList
	type rulesMap map[string][]*rulespb.RuleDesc

	type testCase struct {
		sharding                   bool
		shardingStrategy           string
		shuffleShardSize           float64
		rulesRequest               RulesRequest
		expectedCount              map[string]int
		expectedClientCallCount    int
		rulerStateMap              map[string]ring.InstanceState
		rulerAZMap                 map[string]string
		expectedError              error
		replicationFactor          int
		enableZoneAwareReplication bool
	}

	ruleMap := rulesMap{
		"ruler1-user1-rule-group1": []*rulespb.RuleDesc{
			{
				Record: "rtest_user1_1",
				Expr:   "sum(rate(node_cpu_seconds_total[3h:10m]))",
				Labels: []cortexpb.LabelAdapter{
					{Name: "rulename", Value: "rtest_user1_group1_rule_1"},
				},
			},
			{
				Alert: "atest_user1_1",
				Expr:  "sum(rate(node_cpu_seconds_total[3h:10m]))",
				Labels: []cortexpb.LabelAdapter{
					{Name: "alertname", Value: "atest_user1_group1_rule_1"},
				},
			},
		},
		"ruler1-user1-rule-group2": []*rulespb.RuleDesc{
			{
				Record: "rtest_user1_1",
				Expr:   "sum(rate(node_cpu_seconds_total[3h:10m]))",
				Labels: []cortexpb.LabelAdapter{
					{Name: "rulename", Value: "rtest_user1_group2_rule_1"},
				},
			},
		},
		"ruler1-user2-rule-group1": []*rulespb.RuleDesc{
			{
				Record: "rtest_user1_1",
				Expr:   "sum(rate(node_cpu_seconds_total[3h:10m]))",
				Labels: []cortexpb.LabelAdapter{
					{Name: "rulename", Value: "rtest_user2_group1_rule_1"},
				},
			},
		},
		"ruler2-user1-rule-group3": []*rulespb.RuleDesc{
			{
				Record: "rtest_user1_1",
				Expr:   "sum(rate(node_cpu_seconds_total[3h:10m]))",