diff --git a/coderd/database/dbfake/dbfake.go b/coderd/database/dbfake/dbfake.go index ea49c78065657..1e9a3408e8571 100644 --- a/coderd/database/dbfake/dbfake.go +++ b/coderd/database/dbfake/dbfake.go @@ -274,6 +274,7 @@ type TemplateVersionBuilder struct { t testing.TB db database.Store seed database.TemplateVersion + fileID uuid.UUID ps pubsub.Pubsub resources []*sdkproto.Resource params []database.TemplateVersionParameter @@ -296,6 +297,12 @@ func (t TemplateVersionBuilder) Seed(v database.TemplateVersion) TemplateVersion return t } +func (t TemplateVersionBuilder) FileID(fid uuid.UUID) TemplateVersionBuilder { + // nolint: revive // returns modified struct + t.fileID = fid + return t +} + func (t TemplateVersionBuilder) Pubsub(ps pubsub.Pubsub) TemplateVersionBuilder { // nolint: revive // returns modified struct t.ps = ps @@ -320,6 +327,8 @@ func (t TemplateVersionBuilder) Do() TemplateVersionResponse { t.seed.OrganizationID = takeFirst(t.seed.OrganizationID, uuid.New()) t.seed.ID = takeFirst(t.seed.ID, uuid.New()) t.seed.CreatedBy = takeFirst(t.seed.CreatedBy, uuid.New()) + // nolint: revive + t.fileID = takeFirst(t.fileID, uuid.New()) var resp TemplateVersionResponse if t.seed.TemplateID.UUID == uuid.Nil { @@ -361,6 +370,7 @@ func (t TemplateVersionBuilder) Do() TemplateVersionResponse { Time: dbtime.Now(), Valid: true, }, + FileID: t.fileID, }) t.seed.JobID = job.ID