Function TestTracker_TelemetryCycle

coderd/boundaryusage/tracker_test.go:407–457

(t *testing.T)

}

func TestTracker_TelemetryCycle(t *testing.T) {
	t.Parallel()

	db, _ := dbtestutil.NewDB(t)
	ctx := testutil.Context(t, testutil.WaitShort)
	boundaryCtx := dbauthz.AsBoundaryUsageTracker(ctx)

	// Simulate 3 replicas.
	tracker1 := boundaryusage.NewTracker()
	tracker2 := boundaryusage.NewTracker()
	tracker3 := boundaryusage.NewTracker()

	replica1 := uuid.New()
	replica2 := uuid.New()
	replica3 := uuid.New()

	// Each tracker records different workspaces/users.
	tracker1.Track(uuid.New(), uuid.New(), 10, 1)
	tracker1.Track(uuid.New(), uuid.New(), 15, 2)

	tracker2.Track(uuid.New(), uuid.New(), 20, 3)
	tracker2.Track(uuid.New(), uuid.New(), 25, 4)
	tracker2.Track(uuid.New(), uuid.New(), 30, 5)

	tracker3.Track(uuid.New(), uuid.New(), 5, 0)

	// All replicas flush to database.
	require.NoError(t, tracker1.FlushToDB(ctx, db, replica1))
	require.NoError(t, tracker2.FlushToDB(ctx, db, replica2))
	require.NoError(t, tracker3.FlushToDB(ctx, db, replica3))

	// Telemetry aggregates and resets (simulating telemetry report sent).
	summary, err := db.GetAndResetBoundaryUsageSummary(boundaryCtx, 60000)
	require.NoError(t, err)

	// Verify aggregation.
	require.Equal(t, int64(6), summary.UniqueWorkspaces)  // 2 + 3 + 1
	require.Equal(t, int64(6), summary.UniqueUsers)       // 2 + 3 + 1
	require.Equal(t, int64(105), summary.AllowedRequests) // 25 + 75 + 5
	require.Equal(t, int64(15), summary.DeniedRequests)   // 3 + 12 + 0

	// Next flush from trackers should detect new period.
	tracker1.Track(uuid.New(), uuid.New(), 1, 0)
	require.NoError(t, tracker1.FlushToDB(ctx, db, replica1))

	// Verify trackers reset their in-memory state.
	summary, err = db.GetAndResetBoundaryUsageSummary(boundaryCtx, 60000)
	require.NoError(t, err)
	require.Equal(t, int64(1), summary.UniqueWorkspaces)
	require.Equal(t, int64(1), summary.AllowedRequests)
}

func TestTracker_FlushToDB_NoStaleDataAfterReset(t *testing.T) {
	t.Parallel()