fix: fix prebuild omissions (#17579)

dannykopping · web-flow · commit 6936a7b5a256 · 2025-04-30T14:26:30.000Z
Fixes accidental omission from #17527 --------- Signed-off-by: Danny Kopping <dannykopping@gmail.com>
diff --git a/enterprise/coderd/coderd.go b/enterprise/coderd/coderd.go
@@ -1166,5 +1166,5 @@ func (api *API) setupPrebuilds(featureEnabled bool) (agplprebuilds.Reconciliatio
 
 	reconciler := prebuilds.NewStoreReconciler(api.Database, api.Pubsub, api.DeploymentValues.Prebuilds,
 		api.Logger.Named("prebuilds"), quartz.NewReal(), api.PrometheusRegistry)
-	return reconciler, prebuilds.EnterpriseClaimer{}
+	return reconciler, prebuilds.NewEnterpriseClaimer(api.Database)
 }
diff --git a/enterprise/coderd/coderd_test.go b/enterprise/coderd/coderd_test.go
@@ -331,7 +331,7 @@ func TestEntitlements_Prebuilds(t *testing.T) {
 
 			if tc.expectedEnabled {
 				require.IsType(t, &prebuilds.StoreReconciler{}, *reconciler)
-				require.IsType(t, prebuilds.EnterpriseClaimer{}, *claimer)
+				require.IsType(t, &prebuilds.EnterpriseClaimer{}, *claimer)
 			} else {
 				require.Equal(t, &agplprebuilds.DefaultReconciler, reconciler)
 				require.Equal(t, &agplprebuilds.DefaultClaimer, claimer)

Original file line number	Diff line number	Diff line change
`@@ -1166,5 +1166,5 @@ func (api *API) setupPrebuilds(featureEnabled bool) (agplprebuilds.Reconciliatio`
`1166`	`1166`
`1167`	`1167`	`reconciler := prebuilds.NewStoreReconciler(api.Database, api.Pubsub, api.DeploymentValues.Prebuilds,`
`1168`	`1168`	`api.Logger.Named("prebuilds"), quartz.NewReal(), api.PrometheusRegistry)`
`1169`		`- return reconciler, prebuilds.EnterpriseClaimer{}`
	`1169`	`+ return reconciler, prebuilds.NewEnterpriseClaimer(api.Database)`
`1170`	`1170`	`}`