Function TestTracker_Track_Concurrent

coderd/boundaryusage/tracker_test.go:113–149

(t *testing.T)

}

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

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

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

	const numGoroutines = 100
	const requestsPerGoroutine = 10

	var wg sync.WaitGroup
	for i := 0; i < numGoroutines; i++ {
		wg.Add(1)
		go func() {
			defer wg.Done()
			workspaceID := uuid.New()
			ownerID := uuid.New()
			for j := 0; j < requestsPerGoroutine; j++ {
				tracker.Track(workspaceID, ownerID, 1, 1)
			}
		}()
	}
	wg.Wait()

	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(numGoroutines), summary.UniqueWorkspaces)
	require.Equal(t, int64(numGoroutines), summary.UniqueUsers)
	require.Equal(t, int64(numGoroutines*requestsPerGoroutine), summary.AllowedRequests)
	require.Equal(t, int64(numGoroutines*requestsPerGoroutine), summary.DeniedRequests)
}

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