chore: fix merge orgin/main

ssncferreira · ssncferreira · commit b8a1c14902e1 · 2025-07-23T13:32:09.000Z
diff --git a/enterprise/cli/create_test.go b/enterprise/cli/create_test.go
@@ -5,9 +5,12 @@ import (
 	"database/sql"
 	"fmt"
 	"sync"
+	"sync/atomic"
 	"testing"
 	"time"
 
+	"github.com/coder/coder/v2/coderd/wsbuilder"
+
 	"github.com/google/uuid"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/stretchr/testify/assert"
@@ -351,13 +354,20 @@ func TestEnterpriseCreateWithPreset(t *testing.T) {
 
 		// Setup Prebuild reconciler
 		cache := files.New(prometheus.NewRegistry(), &coderdtest.FakeAuthorizer{})
+		newNoopUsageCheckerPtr := func() *atomic.Pointer[wsbuilder.UsageChecker] {
+			var noopUsageChecker wsbuilder.UsageChecker = wsbuilder.NoopUsageChecker{}
+			buildUsageChecker := atomic.Pointer[wsbuilder.UsageChecker]{}
+			buildUsageChecker.Store(&noopUsageChecker)
+			return &buildUsageChecker
+		}
 		reconciler := prebuilds.NewStoreReconciler(
 			db, pb, cache,
 			codersdk.PrebuildsConfig{},
 			testutil.Logger(t),
 			quartz.NewMock(t),
 			prometheus.NewRegistry(),
 			notifications.NewNoopEnqueuer(),
+			newNoopUsageCheckerPtr(),
 		)
 		var claimer agplprebuilds.Claimer = prebuilds.NewEnterpriseClaimer(db)
 		api.AGPL.PrebuildsClaimer.Store(&claimer)
@@ -454,13 +464,20 @@ func TestEnterpriseCreateWithPreset(t *testing.T) {
 
 		// Setup Prebuild reconciler
 		cache := files.New(prometheus.NewRegistry(), &coderdtest.FakeAuthorizer{})
+		newNoopUsageCheckerPtr := func() *atomic.Pointer[wsbuilder.UsageChecker] {
+			var noopUsageChecker wsbuilder.UsageChecker = wsbuilder.NoopUsageChecker{}
+			buildUsageChecker := atomic.Pointer[wsbuilder.UsageChecker]{}
+			buildUsageChecker.Store(&noopUsageChecker)
+			return &buildUsageChecker
+		}
 		reconciler := prebuilds.NewStoreReconciler(
 			db, pb, cache,
 			codersdk.PrebuildsConfig{},
 			testutil.Logger(t),
 			quartz.NewMock(t),
 			prometheus.NewRegistry(),
 			notifications.NewNoopEnqueuer(),
+			newNoopUsageCheckerPtr(),
 		)
 		var claimer agplprebuilds.Claimer = prebuilds.NewEnterpriseClaimer(db)
 		api.AGPL.PrebuildsClaimer.Store(&claimer)
