Function TestTracker_ConcurrentFlushAndTrack

coderd/boundaryusage/tracker_test.go:496–538

(t *testing.T)

}

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

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

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

	const numOperations = 50

	var wg sync.WaitGroup

	// Goroutine 1: Continuously track.
	wg.Add(1)
	go func() {
		defer wg.Done()
		for i := 0; i < numOperations; i++ {
			tracker.Track(uuid.New(), uuid.New(), 1, 1)
		}
	}()

	// Goroutine 2: Continuously flush.
	wg.Add(1)
	go func() {
		defer wg.Done()
		for i := 0; i < numOperations; i++ {
			_ = tracker.FlushToDB(ctx, db, replicaID)
		}
	}()

	wg.Wait()

	// Final flush to capture any remaining data.
	require.NoError(t, tracker.FlushToDB(ctx, db, replicaID))

	// Verify stats are non-negative.
	boundaryCtx := dbauthz.AsBoundaryUsageTracker(ctx)
	summary, err := db.GetAndResetBoundaryUsageSummary(boundaryCtx, 60000)
	require.NoError(t, err)
	require.GreaterOrEqual(t, summary.AllowedRequests, int64(0))
	require.GreaterOrEqual(t, summary.DeniedRequests, int64(0))
}

// trackDuringUpsertDB wraps a database.Store to call Track() during the
// UpsertBoundaryUsageStats operation, simulating a concurrent Track() call.