Function TestSyncRuleGroups

pkg/ruler/manager_test.go:24–114

(t *testing.T)

)

func TestSyncRuleGroups(t *testing.T) {
	dir := t.TempDir()

	waitDurations := []time.Duration{
		1 * time.Millisecond,
		1 * time.Millisecond,
	}

	ruleManagerFactory := RuleManagerFactory(nil, waitDurations)
	limits := &ruleLimits{externalLabels: labels.FromStrings("from", "cortex")}

	m, err := NewDefaultMultiTenantManager(Config{RulePath: dir}, limits, ruleManagerFactory, nil, nil, log.NewNopLogger())
	require.NoError(t, err)

	const user = "testUser"

	userRules := map[string]rulespb.RuleGroupList{
		user: {
			&rulespb.RuleGroupDesc{
				Name:      "group1",
				Namespace: "ns",
				Interval:  1 * time.Minute,
				User:      user,
			},
		},
	}
	m.SyncRuleGroups(context.Background(), userRules)

	mgr := getManager(m, user)
	require.NotNil(t, mgr)

	test.Poll(t, 1*time.Second, true, func() any {
		return mgr.(*mockRulesManager).running.Load()
	})

	// Verify that user rule groups are now cached locally and notifiers are created.
	{
		users, err := m.mapper.users()
		_, ok := m.notifiers[user]
		require.NoError(t, err)
		require.Equal(t, []string{user}, users)
		require.True(t, ok)
		lset, ok := m.userExternalLabels.get(user)
		require.True(t, ok)
		require.Equal(t, limits.RulerExternalLabels(user), lset)
	}

	// Passing empty map / nil stops all managers.
	m.SyncRuleGroups(context.Background(), nil)
	require.Nil(t, getManager(m, user))

	// Make sure old manager was stopped.
	test.Poll(t, 1*time.Second, false, func() any {
		return mgr.(*mockRulesManager).running.Load()
	})

	// Verify that local rule groups were removed.
	{