@@ -5,9 +5,12 @@ import (
55"database/sql"
66"fmt"
77"sync"
8+ "sync/atomic"
89"testing"
910"time"
1011
12+ "github.com/coder/coder/v2/coderd/wsbuilder"
13+
1114"github.com/google/uuid"
1215"github.com/prometheus/client_golang/prometheus"
1316"github.com/stretchr/testify/assert"
@@ -351,13 +354,20 @@ func TestEnterpriseCreateWithPreset(t *testing.T) {
351354
352355// Setup Prebuild reconciler
353356cache := files .New (prometheus .NewRegistry (),& coderdtest.FakeAuthorizer {})
357+ newNoopUsageCheckerPtr := func ()* atomic.Pointer [wsbuilder.UsageChecker ] {
358+ var noopUsageChecker wsbuilder.UsageChecker = wsbuilder.NoopUsageChecker {}
359+ buildUsageChecker := atomic.Pointer [wsbuilder.UsageChecker ]{}
360+ buildUsageChecker .Store (& noopUsageChecker )
361+ return & buildUsageChecker
362+ }
354363reconciler := prebuilds .NewStoreReconciler (
355364db ,pb ,cache ,
356365codersdk.PrebuildsConfig {},
357366testutil .Logger (t ),
358367quartz .NewMock (t ),
359368prometheus .NewRegistry (),
360369notifications .NewNoopEnqueuer (),
370+ newNoopUsageCheckerPtr (),
361371)
362372var claimer agplprebuilds.Claimer = prebuilds .NewEnterpriseClaimer (db )
363373api .AGPL .PrebuildsClaimer .Store (& claimer )
@@ -454,13 +464,20 @@ func TestEnterpriseCreateWithPreset(t *testing.T) {
454464
455465// Setup Prebuild reconciler
456466cache := files .New (prometheus .NewRegistry (),& coderdtest.FakeAuthorizer {})
467+ newNoopUsageCheckerPtr := func ()* atomic.Pointer [wsbuilder.UsageChecker ] {
468+ var noopUsageChecker wsbuilder.UsageChecker = wsbuilder.NoopUsageChecker {}
469+ buildUsageChecker := atomic.Pointer [wsbuilder.UsageChecker ]{}
470+ buildUsageChecker .Store (& noopUsageChecker )
471+ return & buildUsageChecker
472+ }
457473reconciler := prebuilds .NewStoreReconciler (
458474db ,pb ,cache ,
459475codersdk.PrebuildsConfig {},
460476testutil .Logger (t ),
461477quartz .NewMock (t ),
462478prometheus .NewRegistry (),
463479notifications .NewNoopEnqueuer (),
480+ newNoopUsageCheckerPtr (),
464481)
465482var claimer agplprebuilds.Claimer = prebuilds .NewEnterpriseClaimer (db )
466483api .AGPL .PrebuildsClaimer .Store (& claimer )