feat(coderd/database/dbfake): allow specifying fileID in TemplateVersionBuilder (#12450)

johnstcn · web-flow · commit c8aa99a5b8d4 · 2024-03-07T12:36:11.000Z
diff --git 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
