fix: fix dbauthz

evgeniy-scherbina · evgeniy-scherbina · commit 906ceb9f4df8 · 2025-05-16T20:27:46.000Z
diff --git a/coderd/database/dbauthz/dbauthz.go b/coderd/database/dbauthz/dbauthz.go
@@ -4179,7 +4179,21 @@ func (q *querier) UpdateOrganizationDeletedByID(ctx context.Context, arg databas
 }
 
 func (q *querier) UpdatePrebuildStatus(ctx context.Context, arg database.UpdatePrebuildStatusParams) error {
-	panic("not implemented")
+	preset, err := q.db.GetPresetByID(ctx, arg.PresetID)
+	if err != nil {
+		return err
+	}
+
+	object := rbac.ResourceTemplate.
+		WithID(preset.TemplateID.UUID).
+		InOrg(preset.OrganizationID)
+
+	err = q.authorizeContext(ctx, policy.ActionUpdate, object)
+	if err != nil {
+		return err
+	}
+
+	return q.db.UpdatePrebuildStatus(ctx, arg)
 }
 
 func (q *querier) UpdateProvisionerDaemonLastSeenAt(ctx context.Context, arg database.UpdateProvisionerDaemonLastSeenAtParams) error {
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
@@ -4943,6 +4943,31 @@ func (s *MethodTestSuite) TestPrebuilds() {
 				OrganizationID:      org.ID,
 			})
 	}))
+	s.Run("UpdatePrebuildStatus", s.Subtest(func(db database.Store, check *expects) {
+		org := dbgen.Organization(s.T(), db, database.Organization{})
+		user := dbgen.User(s.T(), db, database.User{})
+		template := dbgen.Template(s.T(), db, database.Template{
+			OrganizationID: org.ID,
+			CreatedBy:      user.ID,
+		})
+		templateVersion := dbgen.TemplateVersion(s.T(), db, database.TemplateVersion{
+			TemplateID: uuid.NullUUID{
+				UUID:  template.ID,
+				Valid: true,
+			},
+			OrganizationID: org.ID,
+			CreatedBy:      user.ID,
+		})
+		preset := dbgen.Preset(s.T(), db, database.InsertPresetParams{
+			TemplateVersionID: templateVersion.ID,
+		})
+		req := database.UpdatePrebuildStatusParams{
+			PresetID: preset.ID,
+		}
+		check.Args(req).
+			Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate).
+			ErrorsWithInMemDB(dbmem.ErrUnimplemented)
+	}))
 }
 
 func (s *MethodTestSuite) TestOAuth2ProviderApps() {
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
@@ -10873,7 +10873,7 @@ func (q *FakeQuerier) UpdatePrebuildStatus(ctx context.Context, arg database.Upd
 		return err
 	}
 
-	panic("not implemented")
+	return ErrUnimplemented
 }
 
 func (q *FakeQuerier) UpdateProvisionerDaemonLastSeenAt(_ context.Context, arg database.UpdateProvisionerDaemonLastSeenAtParams) error {

Original file line number	Diff line number	Diff line change
`@@ -10873,7 +10873,7 @@ func (q *FakeQuerier) UpdatePrebuildStatus(ctx context.Context, arg database.Upd`
`10873`	`10873`	`return err`
`10874`	`10874`	`}`
`10875`	`10875`
`10876`		`- panic("not implemented")`
	`10876`	`+ return ErrUnimplemented`
`10877`	`10877`	`}`
`10878`	`10878`
`10879`	`10879`	`func (q *FakeQuerier) UpdateProvisionerDaemonLastSeenAt(_ context.Context, arg database.UpdateProvisionerDaemonLastSeenAtParams) error {`