Function TestTracker_FlushToDB_Accumulates

coderd/boundaryusage/tracker_test.go:151–184

(t *testing.T)

}

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

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

	tracker := boundaryusage.NewTracker()
	replicaID := uuid.New()
	workspaceID := uuid.New()
	ownerID := uuid.New()

	// First flush is an insert, resets unique counts (new period).
	tracker.Track(workspaceID, ownerID, 5, 3)
	err := tracker.FlushToDB(ctx, db, replicaID)
	require.NoError(t, err)

	// Track & flush more data. Same workspace/user, so unique counts stay at 1.
	tracker.Track(workspaceID, ownerID, 2, 1)
	err = tracker.FlushToDB(ctx, db, replicaID)
	require.NoError(t, err)

	// Track & flush even more data to continue accumulation.
	tracker.Track(workspaceID, ownerID, 3, 2)
	err = tracker.FlushToDB(ctx, db, replicaID)
	require.NoError(t, err)

	boundaryCtx := dbauthz.AsBoundaryUsageTracker(ctx)
	summary, err := db.GetAndResetBoundaryUsageSummary(boundaryCtx, 60000)
	require.NoError(t, err)
	require.Equal(t, int64(1), summary.UniqueWorkspaces)
	require.Equal(t, int64(1), summary.UniqueUsers)
	require.Equal(t, int64(5+2+3), summary.AllowedRequests)
	require.Equal(t, int64(3+1+2), summary.DeniedRequests)
}

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