Function testRulerAPIWithSharding

integration/ruler_test.go:353–589

(t *testing.T, enableRulesBackup bool)

}

func testRulerAPIWithSharding(t *testing.T, enableRulesBackup bool) {
	const numRulesGroups = 100

	random := rand.New(rand.NewSource(time.Now().UnixNano()))
	s, err := e2e.NewScenario(networkName)
	require.NoError(t, err)
	defer s.Close()

	// Generate multiple rule groups, with 1 rule each.
	ruleGroups := make([]rulefmt.RuleGroup, numRulesGroups)
	expectedNames := make([]string, numRulesGroups)
	alertCount := 0
	evalInterval, _ := model.ParseDuration("1s")
	groupLabels := map[string]string{
		"group_label_1":   "val1",
		"group_label_2":   "val2",
		"duplicate_label": "group_val",
	}
	ruleLabels := map[string]string{
		"rule_label_1":    "val1",
		"rule_label_2":    "val2",
		"duplicate_label": "rule_val",
	}
	for i := 0; i < numRulesGroups; i++ {
		num := random.Intn(100)
		ruleName := fmt.Sprintf("test_%d", i)
		expectedNames[i] = ruleName
		if num%2 == 0 {
			alertCount++
			ruleGroups[i] = rulefmt.RuleGroup{
				Name:     ruleName,
				Interval: evalInterval,
				Rules: []rulefmt.Rule{{
					Alert:  fmt.Sprintf("rule_%d", i),
					Expr:   strconv.Itoa(i),
					Labels: ruleLabels,
				}},
				Labels: groupLabels,
			}
		} else {
			ruleGroups[i] = rulefmt.RuleGroup{
				Name:     ruleName,
				Interval: evalInterval,
				Rules: []rulefmt.Rule{{
					Record: fmt.Sprintf("rule_%d", i),
					Labels: ruleLabels,
					Expr:   strconv.Itoa(i),
				}},
				Labels: groupLabels,
			}
		}
	}

	// Start dependencies.
	consul := e2edb.NewConsul()
	minio := e2edb.NewMinio(9000, rulestoreBucketName)
	require.NoError(t, s.StartAndWaitReady(consul, minio))