mirror of
https://github.com/coder/coder.git
synced 2026-10-29 17:44:33 +08:00
feat(coderd/database/dbfake): allow specifying fileID in TemplateVersionBuilder (#12450)
This commit is contained in:
@@ -274,6 +274,7 @@ type TemplateVersionBuilder struct {
|
|||||||
t testing.TB
|
t testing.TB
|
||||||
db database.Store
|
db database.Store
|
||||||
seed database.TemplateVersion
|
seed database.TemplateVersion
|
||||||
|
fileID uuid.UUID
|
||||||
ps pubsub.Pubsub
|
ps pubsub.Pubsub
|
||||||
resources []*sdkproto.Resource
|
resources []*sdkproto.Resource
|
||||||
params []database.TemplateVersionParameter
|
params []database.TemplateVersionParameter
|
||||||
@@ -296,6 +297,12 @@ func (t TemplateVersionBuilder) Seed(v database.TemplateVersion) TemplateVersion
|
|||||||
return t
|
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 {
|
func (t TemplateVersionBuilder) Pubsub(ps pubsub.Pubsub) TemplateVersionBuilder {
|
||||||
// nolint: revive // returns modified struct
|
// nolint: revive // returns modified struct
|
||||||
t.ps = ps
|
t.ps = ps
|
||||||
@@ -320,6 +327,8 @@ func (t TemplateVersionBuilder) Do() TemplateVersionResponse {
|
|||||||
t.seed.OrganizationID = takeFirst(t.seed.OrganizationID, uuid.New())
|
t.seed.OrganizationID = takeFirst(t.seed.OrganizationID, uuid.New())
|
||||||
t.seed.ID = takeFirst(t.seed.ID, uuid.New())
|
t.seed.ID = takeFirst(t.seed.ID, uuid.New())
|
||||||
t.seed.CreatedBy = takeFirst(t.seed.CreatedBy, uuid.New())
|
t.seed.CreatedBy = takeFirst(t.seed.CreatedBy, uuid.New())
|
||||||
|
// nolint: revive
|
||||||
|
t.fileID = takeFirst(t.fileID, uuid.New())
|
||||||
|
|
||||||
var resp TemplateVersionResponse
|
var resp TemplateVersionResponse
|
||||||
if t.seed.TemplateID.UUID == uuid.Nil {
|
if t.seed.TemplateID.UUID == uuid.Nil {
|
||||||
@@ -361,6 +370,7 @@ func (t TemplateVersionBuilder) Do() TemplateVersionResponse {
|
|||||||
Time: dbtime.Now(),
|
Time: dbtime.Now(),
|
||||||
Valid: true,
|
Valid: true,
|
||||||
},
|
},
|
||||||
|
FileID: t.fileID,
|
||||||
})
|
})
|
||||||
|
|
||||||
t.seed.JobID = job.ID
|
t.seed.JobID = job.ID
|
||||||
|
|||||||
Reference in New Issue
Block a user