Function TestNotifierSendExternalLabels

pkg/ruler/ruler_test.go:433–525

(t *testing.T)

}

func TestNotifierSendExternalLabels(t *testing.T) {
	ctx, cancel := context.WithCancel(context.Background())
	t.Cleanup(cancel)
	receivedLabelsCh := make(chan models.LabelSet, 1)
	ts := httptest.NewServer(http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
		alerts := models.PostableAlerts{}
		err := json.NewDecoder(r.Body).Decode(&alerts)
		if err != nil {
			http.Error(w, err.Error(), http.StatusBadRequest)
			return
		}
		if len(alerts) == 1 {
			select {
			case <-ctx.Done():
			case receivedLabelsCh <- alerts[0].Labels:
			}
		}
	}))
	t.Cleanup(ts.Close)

	cfg := defaultRulerConfig(t)
	cfg.AlertmanagerURL = ts.URL
	cfg.AlertmanagerDiscovery = false
	cfg.ExternalLabels = labels.FromStrings("region", "us-east-1")
	limits := &ruleLimits{}
	engine, queryable, pusher, logger, _, reg := testSetup(t, nil)
	metrics := NewRuleEvalMetrics(cfg, nil)
	managerFactory := DefaultTenantManagerFactory(cfg, pusher, queryable, engine, limits, metrics, nil)
	manager, err := NewDefaultMultiTenantManager(cfg, limits, managerFactory, metrics, reg, logger)
	require.NoError(t, err)
	t.Cleanup(manager.Stop)

	const userID = "n1"
	manager.SyncRuleGroups(context.Background(), map[string]rulespb.RuleGroupList{
		userID: {&rulespb.RuleGroupDesc{Name: "group", Namespace: "ns", Interval: time.Minute, User: userID}},
	})

	manager.notifiersMtx.Lock()
	n, ok := manager.notifiers[userID]
	manager.notifiersMtx.Unlock()
	require.True(t, ok)

	tests := []struct {
		name                   string
		userExternalLabels     []labels.Label
		expectedExternalLabels []labels.Label
	}{
		{
			name:                   "global labels only",
			userExternalLabels:     nil,
			expectedExternalLabels: []labels.Label{{Name: "region", Value: "us-east-1"}},
		},
		{
			name:                   "local labels without overriding",
			userExternalLabels:     []labels.Label{{Name: "mylabel", Value: "local"}},
			expectedExternalLabels: []labels.Label{{Name: "region", Value: "us-east-1"}, {Name: "mylabel", Value: "local"}},
		},
		{