Function Test_Ingester_AllUserStatsHandler

pkg/ingester/ingester_test.go:5452–5528

(t *testing.T)

}

func Test_Ingester_AllUserStatsHandler(t *testing.T) {
	series := []struct {
		user      string
		lbls      labels.Labels
		value     float64
		timestamp int64
	}{
		{"user-1", labels.FromStrings("__name__", "test_1_1", "route", "get_user", "status", "200"), 1, 100000},
		{"user-1", labels.FromStrings("__name__", "test_1_1", "route", "get_user", "status", "500"), 1, 110000},
		{"user-1", labels.FromStrings("__name__", "test_1_2"), 2, 200000},
		{"user-2", labels.FromStrings("__name__", "test_2_1"), 2, 200000},
		{"user-2", labels.FromStrings("__name__", "test_2_2"), 2, 200000},
	}

	// Create ingester
	i, err := prepareIngesterWithBlocksStorage(t, defaultIngesterTestConfig(t), prometheus.NewRegistry())
	require.NoError(t, err)
	require.NoError(t, services.StartAndAwaitRunning(context.Background(), i))
	defer services.StopAndAwaitTerminated(context.Background(), i) //nolint:errcheck

	// Wait until it's ACTIVE
	test.Poll(t, 1*time.Second, ring.ACTIVE, func() any {
		return i.lifecycler.GetState()
	})
	for _, series := range series {
		ctx := user.InjectOrgID(context.Background(), series.user)
		req, _ := mockWriteRequest(t, series.lbls, series.value, series.timestamp)
		_, err := i.Push(ctx, req)
		require.NoError(t, err)
	}

	// Force compaction to test loaded blocks
	compactionCallbackCh := make(chan struct{})
	i.TSDBState.forceCompactTrigger <- requestWithUsersAndCallback{users: nil, callback: compactionCallbackCh}
	<-compactionCallbackCh

	// force update statistics
	for _, db := range i.TSDBState.dbs {
		db.ingestedAPISamples.Tick()
		db.ingestedRuleSamples.Tick()
	}

	// Get label names
	response := httptest.NewRecorder()
	request := httptest.NewRequest("GET", "/all_user_stats", nil)
	request.Header.Add("Accept", "application/json")
	i.AllUserStatsHandler(response, request)
	var resp UserStatsByTimeseries
	err = json.Unmarshal(response.Body.Bytes(), &resp)
	require.NoError(t, err)

	expect := UserStatsByTimeseries{
		{
			UserID: "user-1",
			UserStats: UserStats{
				IngestionRate:     0.2,
				NumSeries:         0,
				APIIngestionRate:  0.2,