@@ -111,10 +111,12 @@ type data struct {
111
111
provisionerJobs []database.ProvisionerJob
112
112
replicas []database.Replica
113
113
templateVersions []database.TemplateVersion
114
+ templateVersionParameters []database.TemplateVersionParameter
114
115
templates []database.Template
115
116
workspaceAgents []database.WorkspaceAgent
116
117
workspaceApps []database.WorkspaceApp
117
118
workspaceBuilds []database.WorkspaceBuild
119
+ workspaceBuildParameters []database.WorkspaceBuildParameter
118
120
workspaceResourceMetadata []database.WorkspaceResourceMetadatum
119
121
workspaceResources []database.WorkspaceResource
120
122
workspaces []database.Workspace
@@ -1365,6 +1367,20 @@ func (q *fakeQuerier) GetWorkspaceBuildByWorkspaceIDAndBuildNumber(_ context.Con
1365
1367
return database.WorkspaceBuild {}, sql .ErrNoRows
1366
1368
}
1367
1369
1370
+ func (q * fakeQuerier ) GetWorkspaceBuildParameters (_ context.Context , workspaceBuildID uuid.UUID ) ([]database.WorkspaceBuildParameter , error ) {
1371
+ q .mutex .RLock ()
1372
+ defer q .mutex .RUnlock ()
1373
+
1374
+ params := make ([]database.WorkspaceBuildParameter , 0 )
1375
+ for _ , param := range params {
1376
+ if param .WorkspaceBuildID != workspaceBuildID {
1377
+ continue
1378
+ }
1379
+ params = append (params , param )
1380
+ }
1381
+ return params , nil
1382
+ }
1383
+
1368
1384
func (q * fakeQuerier ) GetWorkspaceBuildsCreatedAfter (_ context.Context , after time.Time ) ([]database.WorkspaceBuild , error ) {
1369
1385
q .mutex .RLock ()
1370
1386
defer q .mutex .RUnlock ()
@@ -1656,6 +1672,20 @@ func (q *fakeQuerier) GetTemplateVersionByTemplateIDAndName(_ context.Context, a
1656
1672
return database.TemplateVersion {}, sql .ErrNoRows
1657
1673
}
1658
1674
1675
+ func (q * fakeQuerier ) GetTemplateVersionParameters (_ context.Context , templateVersionID uuid.UUID ) ([]database.TemplateVersionParameter , error ) {
1676
+ q .mutex .RLock ()
1677
+ defer q .mutex .RUnlock ()
1678
+
1679
+ parameters := make ([]database.TemplateVersionParameter , 0 )
1680
+ for _ , param := range q .templateVersionParameters {
1681
+ if param .TemplateVersionID != templateVersionID {
1682
+ continue
1683
+ }
1684
+ parameters = append (parameters , param )
1685
+ }
1686
+ return parameters , nil
1687
+ }
1688
+
1659
1689
func (q * fakeQuerier ) GetTemplateVersionByOrganizationAndName (_ context.Context , arg database.GetTemplateVersionByOrganizationAndNameParams ) (database.TemplateVersion , error ) {
1660
1690
q .mutex .RLock ()
1661
1691
defer q .mutex .RUnlock ()
@@ -2398,6 +2428,28 @@ func (q *fakeQuerier) InsertTemplateVersion(_ context.Context, arg database.Inse
2398
2428
return version , nil
2399
2429
}
2400
2430
2431
+ func (q * fakeQuerier ) InsertTemplateVersionParameter (_ context.Context , arg database.InsertTemplateVersionParameterParams ) (database.TemplateVersionParameter , error ) {
2432
+ q .mutex .Lock ()
2433
+ defer q .mutex .Unlock ()
2434
+
2435
+ //nolint:gosimple
2436
+ param := database.TemplateVersionParameter {
2437
+ TemplateVersionID : arg .TemplateVersionID ,
2438
+ Name : arg .Name ,
2439
+ Description : arg .Description ,
2440
+ Type : arg .Type ,
2441
+ Mutable : arg .Mutable ,
2442
+ DefaultValue : arg .DefaultValue ,
2443
+ Icon : arg .Icon ,
2444
+ Options : arg .Options ,
2445
+ ValidationRegex : arg .ValidationRegex ,
2446
+ ValidationMin : arg .ValidationMin ,
2447
+ ValidationMax : arg .ValidationMax ,
2448
+ }
2449
+ q .templateVersionParameters = append (q .templateVersionParameters , param )
2450
+ return param , nil
2451
+ }
2452
+
2401
2453
func (q * fakeQuerier ) InsertProvisionerJobLogs (_ context.Context , arg database.InsertProvisionerJobLogsParams ) ([]database.ProvisionerJobLog , error ) {
2402
2454
q .mutex .Lock ()
2403
2455
defer q .mutex .Unlock ()
@@ -2723,6 +2775,20 @@ func (q *fakeQuerier) InsertWorkspaceBuild(_ context.Context, arg database.Inser
2723
2775
return workspaceBuild , nil
2724
2776
}
2725
2777
2778
+ func (q * fakeQuerier ) InsertWorkspaceBuildParameters (_ context.Context , arg database.InsertWorkspaceBuildParametersParams ) error {
2779
+ q .mutex .Lock ()
2780
+ defer q .mutex .Unlock ()
2781
+
2782
+ for index , name := range arg .Name {
2783
+ q .workspaceBuildParameters = append (q .workspaceBuildParameters , database.WorkspaceBuildParameter {
2784
+ WorkspaceBuildID : arg .WorkspaceBuildID ,
2785
+ Name : name ,
2786
+ Value : arg .Value [index ],
2787
+ })
2788
+ }
2789
+ return nil
2790
+ }
2791
+
2726
2792
func (q * fakeQuerier ) InsertWorkspaceApp (_ context.Context , arg database.InsertWorkspaceAppParams ) (database.WorkspaceApp , error ) {
2727
2793
q .mutex .Lock ()
2728
2794
defer q .mutex .Unlock ()
0 commit comments