@@ -3664,8 +3664,25 @@ func (q *FakeQuerier) GetParameterSchemasByJobID(_ context.Context, jobID uuid.U
3664
3664
return parameters ,nil
3665
3665
}
3666
3666
3667
- func (q * FakeQuerier )GetPresetByWorkspaceBuildID (ctx context.Context ,workspaceBuildID uuid.UUID ) (database.GetPresetByWorkspaceBuildIDRow ,error ) {
3668
- panic ("not implemented" )
3667
+ func (q * FakeQuerier )GetPresetByWorkspaceBuildID (_ context.Context ,workspaceBuildID uuid.UUID ) (database.GetPresetByWorkspaceBuildIDRow ,error ) {
3668
+ q .mutex .RLock ()
3669
+ defer q .mutex .RUnlock ()
3670
+
3671
+ for _ ,workspaceBuild := range q .workspaceBuilds {
3672
+ if workspaceBuild .ID == workspaceBuildID {
3673
+ for _ ,preset := range q .presets {
3674
+ if preset .TemplateVersionID == workspaceBuild .TemplateVersionID {
3675
+ return database.GetPresetByWorkspaceBuildIDRow {
3676
+ ID : uuid.NullUUID {UUID :preset .ID ,Valid :true },
3677
+ Name : sql.NullString {String :preset .Name ,Valid :true },
3678
+ CreatedAt : sql.NullTime {Time :preset .CreatedAt ,Valid :true },
3679
+ UpdatedAt :preset .UpdatedAt ,
3680
+ },nil
3681
+ }
3682
+ }
3683
+ }
3684
+ }
3685
+ return database.GetPresetByWorkspaceBuildIDRow {},sql .ErrNoRows
3669
3686
}
3670
3687
3671
3688
func (q * FakeQuerier )GetPresetParametersByPresetID (_ context.Context ,templateVersionPresetID uuid.UUID ) ([]database.GetPresetParametersByPresetIDRow ,error ) {
@@ -3685,7 +3702,7 @@ func (q *FakeQuerier) GetPresetParametersByPresetID(_ context.Context, templateV
3685
3702
return parameters ,nil
3686
3703
}
3687
3704
3688
- func (q * FakeQuerier )GetPresetsByTemplateVersionID (ctx context.Context ,templateVersionID uuid.UUID ) ([]database.GetPresetsByTemplateVersionIDRow ,error ) {
3705
+ func (q * FakeQuerier )GetPresetsByTemplateVersionID (_ context.Context ,templateVersionID uuid.UUID ) ([]database.GetPresetsByTemplateVersionIDRow ,error ) {
3689
3706
q .mutex .RLock ()
3690
3707
defer q .mutex .RUnlock ()
3691
3708