enable the premium license in a prebuilds integration test

coder · SasSwart · Apr 21, 2025 · Apr 21, 2025 · Apr 23, 2025 · Apr 24, 2025
commit 9feebef22ed41a7c7f12cf19da8663cbdabc8293
diff --git a/coderd/workspaceagents.go b/coderd/workspaceagents.go
@@ -1159,7 +1159,7 @@ func (api *API) workspaceAgentPostLogSource(rw http.ResponseWriter, r *http.Requ
 // @Security CoderSessionToken
 // @Produce json
 // @Tags Agents
-// @Success 200 {object} agentsdk.ReinitializationResponse
+// @Success 200 {object} agentsdk.ReinitializationEvent
 // @Router /workspaceagents/me/reinit [get]
 func (api *API) workspaceAgentReinit(rw http.ResponseWriter, r *http.Request) {
 	// Allow us to interrupt watch via cancel.

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/workspaceagents_test.go b/enterprise/coderd/workspaceagents_test.go
@@ -82,9 +82,14 @@ func TestReinitializeAgent(t *testing.T) {
 
 	// GIVEN a live enterprise API with the prebuilds feature enabled
 	client, db, user := coderdenttest.NewWithDatabase(t, &coderdenttest.Options{
+		Options: &coderdtest.Options{
+			DeploymentValues: coderdtest.DeploymentValues(t, func(dv *codersdk.DeploymentValues) {
+				dv.Experiments.Append(string(codersdk.ExperimentWorkspacePrebuilds))
+			}),
+		},
 		LicenseOptions: &coderdenttest.LicenseOptions{
 			Features: license.Features{
-				// TODO: enable the prebuilds feature and experiment
+				codersdk.FeatureWorkspacePrebuilds: 1,
 			},
 		},
 	})