Function TestTracker_TrackDuringFlush

coderd/boundaryusage/tracker_test.go:565–598

(t *testing.T)

}

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

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

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

	// Track some initial data.
	tracker.Track(uuid.New(), uuid.New(), 10, 5)

	trackingDB := &trackDuringUpsertDB{
		Store:       db,
		tracker:     tracker,
		workspaceID: uuid.New(),
		userID:      uuid.New(),
	}

	// Flush will call Track() during the DB operation.
	err := tracker.FlushToDB(ctx, trackingDB, replicaID)
	require.NoError(t, err)

	// Second flush captures the Track() that happened during the first flush.
	err = tracker.FlushToDB(ctx, db, replicaID)
	require.NoError(t, err)

	// Verify both flushes are in the summary.
	summary, err := db.GetAndResetBoundaryUsageSummary(boundaryCtx, 60000)
	require.NoError(t, err)
	require.Equal(t, int64(10+20), summary.AllowedRequests)
	require.Equal(t, int64(5+10), summary.DeniedRequests)
}