test: fix dbmem

evgeniy-scherbina · evgeniy-scherbina · commit 7cd27994b16d · 2025-05-19T13:18:00.000Z
diff --git a/coderd/database/dbauthz/dbauthz_test.go b/coderd/database/dbauthz/dbauthz_test.go
@@ -4970,8 +4970,7 @@ func (s *MethodTestSuite) TestPrebuilds() {
 			PresetID: preset.ID,
 		}
 		check.Args(req).
-			Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate).
-			ErrorsWithInMemDB(dbmem.ErrUnimplemented)
+			Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate)
 	}))
 }
 
diff --git a/coderd/database/dbmem/dbmem.go b/coderd/database/dbmem/dbmem.go
@@ -4303,6 +4303,7 @@ func (q *FakeQuerier) GetPresetByID(ctx context.Context, presetID uuid.UUID) (da
 				CreatedAt:           preset.CreatedAt,
 				DesiredInstances:    preset.DesiredInstances,
 				InvalidateAfterSecs: preset.InvalidateAfterSecs,
+				PrebuildStatus:      preset.PrebuildStatus,
 				TemplateID:          tv.TemplateID,
 				OrganizationID:      tv.OrganizationID,
 			}, nil
@@ -9041,6 +9042,10 @@ func (q *FakeQuerier) InsertPreset(_ context.Context, arg database.InsertPresetP
 			Int32: 0,
 			Valid: true,
 		},
+		PrebuildStatus: database.NullPrebuildStatus{
+			PrebuildStatus: database.PrebuildStatusNormal,
+			Valid:          true,
+		},
 	}
 	q.presets = append(q.presets, preset)
 	return preset, nil
@@ -10875,7 +10880,17 @@ func (q *FakeQuerier) UpdatePrebuildStatus(ctx context.Context, arg database.Upd
 		return err
 	}
 
-	return ErrUnimplemented
+	q.mutex.RLock()
+	defer q.mutex.RUnlock()
+
+	for _, preset := range q.presets {
+		if preset.ID == arg.PresetID {
+			preset.PrebuildStatus = arg.Status
+			return nil
+		}
+	}
+
+	return xerrors.Errorf("preset %v does not exist", arg.PresetID)
 }
 
 func (q *FakeQuerier) UpdateProvisionerDaemonLastSeenAt(_ context.Context, arg database.UpdateProvisionerDaemonLastSeenAtParams) error {

Original file line number	Diff line number	Diff line change
`@@ -4970,8 +4970,7 @@ func (s *MethodTestSuite) TestPrebuilds() {`
`4970`	`4970`	`PresetID: preset.ID,`
`4971`	`4971`	`}`
`4972`	`4972`	`check.Args(req).`
`4973`		`- Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate).`
`4974`		`- ErrorsWithInMemDB(dbmem.ErrUnimplemented)`
	`4973`	`+ Asserts(rbac.ResourceTemplate.WithID(template.ID).InOrg(org.ID), policy.ActionUpdate)`
`4975`	`4974`	`}))`
`4976`	`4975`	`}`
`4977`	`4976`