@@ -18,16 +18,18 @@ func New() database.Store {
18
18
organizationMembers : make ([]database.OrganizationMember , 0 ),
19
19
users : make ([]database.User , 0 ),
20
20
21
- parameterValue : make ([]database.ParameterValue , 0 ),
22
- project : make ([]database.Project , 0 ),
23
- projectHistory : make ([]database.ProjectHistory , 0 ),
24
- projectParameter : make ([]database.ProjectParameter , 0 ),
25
- provisionerDaemons : make ([]database.ProvisionerDaemon , 0 ),
26
- provisionerJobs : make ([]database.ProvisionerJob , 0 ),
27
- workspace : make ([]database.Workspace , 0 ),
28
- workspaceResource : make ([]database.WorkspaceResource , 0 ),
29
- workspaceHistory : make ([]database.WorkspaceHistory , 0 ),
30
- workspaceAgent : make ([]database.WorkspaceAgent , 0 ),
21
+ parameterValue : make ([]database.ParameterValue , 0 ),
22
+ project : make ([]database.Project , 0 ),
23
+ projectHistory : make ([]database.ProjectHistory , 0 ),
24
+ projectHistoryLog : make ([]database.ProjectHistoryLog , 0 ),
25
+ projectParameter : make ([]database.ProjectParameter , 0 ),
26
+ provisionerDaemons : make ([]database.ProvisionerDaemon , 0 ),
27
+ provisionerJobs : make ([]database.ProvisionerJob , 0 ),
28
+ workspace : make ([]database.Workspace , 0 ),
29
+ workspaceResource : make ([]database.WorkspaceResource , 0 ),
30
+ workspaceHistory : make ([]database.WorkspaceHistory , 0 ),
31
+ workspaceHistoryLog : make ([]database.WorkspaceHistoryLog , 0 ),
32
+ workspaceAgent : make ([]database.WorkspaceAgent , 0 ),
31
33
}
32
34
}
33
35
@@ -40,16 +42,18 @@ type fakeQuerier struct {
40
42
users []database.User
41
43
42
44
// New tables
43
- parameterValue []database.ParameterValue
44
- project []database.Project
45
- projectHistory []database.ProjectHistory
46
- projectParameter []database.ProjectParameter
47
- provisionerDaemons []database.ProvisionerDaemon
48
- provisionerJobs []database.ProvisionerJob
49
- workspace []database.Workspace
50
- workspaceResource []database.WorkspaceResource
51
- workspaceHistory []database.WorkspaceHistory
52
- workspaceAgent []database.WorkspaceAgent
45
+ parameterValue []database.ParameterValue
46
+ project []database.Project
47
+ projectHistory []database.ProjectHistory
48
+ projectHistoryLog []database.ProjectHistoryLog
49
+ projectParameter []database.ProjectParameter
50
+ provisionerDaemons []database.ProvisionerDaemon
51
+ provisionerJobs []database.ProvisionerJob
52
+ workspace []database.Workspace
53
+ workspaceAgent []database.WorkspaceAgent
54
+ workspaceHistory []database.WorkspaceHistory
55
+ workspaceHistoryLog []database.WorkspaceHistoryLog
56
+ workspaceResource []database.WorkspaceResource
53
57
}
54
58
55
59
// InTx doesn't rollback data properly for in-memory yet.
@@ -184,6 +188,23 @@ func (q *fakeQuerier) GetWorkspaceHistoryByWorkspaceIDWithoutAfter(_ context.Con
184
188
return database.WorkspaceHistory {}, sql .ErrNoRows
185
189
}
186
190
191
+ func (q * fakeQuerier ) GetWorkspaceHistoryLogsByIDBefore (_ context.Context , arg database.GetWorkspaceHistoryLogsByIDBeforeParams ) ([]database.WorkspaceHistoryLog , error ) {
192
+ logs := make ([]database.WorkspaceHistoryLog , 0 )
193
+ for _ , workspaceHistoryLog := range q .workspaceHistoryLog {
194
+ if workspaceHistoryLog .WorkspaceHistoryID .String () != arg .WorkspaceHistoryID .String () {
195
+ continue
196
+ }
197
+ if workspaceHistoryLog .CreatedAt .After (arg .CreatedAt ) {
198
+ continue
199
+ }
200
+ logs = append (logs , workspaceHistoryLog )
201
+ }
202
+ if len (logs ) == 0 {
203
+ return nil , sql .ErrNoRows
204
+ }
205
+ return logs , nil
206
+ }
207
+
187
208
func (q * fakeQuerier ) GetWorkspaceHistoryByWorkspaceID (_ context.Context , workspaceID uuid.UUID ) ([]database.WorkspaceHistory , error ) {
188
209
history := make ([]database.WorkspaceHistory , 0 )
189
210
for _ , workspaceHistory := range q .workspaceHistory {
@@ -197,6 +218,19 @@ func (q *fakeQuerier) GetWorkspaceHistoryByWorkspaceID(_ context.Context, worksp
197
218
return history , nil
198
219
}
199
220
221
+ func (q * fakeQuerier ) GetWorkspaceHistoryByWorkspaceIDAndName (_ context.Context , arg database.GetWorkspaceHistoryByWorkspaceIDAndNameParams ) (database.WorkspaceHistory , error ) {
222
+ for _ , workspaceHistory := range q .workspaceHistory {
223
+ if workspaceHistory .WorkspaceID .String () != arg .WorkspaceID .String () {
224
+ continue
225
+ }
226
+ if ! strings .EqualFold (workspaceHistory .Name , arg .Name ) {
227
+ continue
228
+ }
229
+ return workspaceHistory , nil
230
+ }
231
+ return database.WorkspaceHistory {}, sql .ErrNoRows
232
+ }
233
+
200
234
func (q * fakeQuerier ) GetWorkspacesByProjectAndUserID (_ context.Context , arg database.GetWorkspacesByProjectAndUserIDParams ) ([]database.Workspace , error ) {
201
235
workspaces := make ([]database.Workspace , 0 )
202
236
for _ , workspace := range q .workspace {
@@ -318,6 +352,36 @@ func (q *fakeQuerier) GetProjectHistoryByProjectID(_ context.Context, projectID
318
352
return history , nil
319
353
}
320
354
355
+ func (q * fakeQuerier ) GetProjectHistoryByProjectIDAndName (_ context.Context , arg database.GetProjectHistoryByProjectIDAndNameParams ) (database.ProjectHistory , error ) {
356
+ for _ , projectHistory := range q .projectHistory {
357
+ if projectHistory .ProjectID .String () != arg .ProjectID .String () {
358
+ continue
359
+ }
360
+ if ! strings .EqualFold (projectHistory .Name , arg .Name ) {
361
+ continue
362
+ }
363
+ return projectHistory , nil
364
+ }
365
+ return database.ProjectHistory {}, sql .ErrNoRows
366
+ }
367
+
368
+ func (q * fakeQuerier ) GetProjectHistoryLogsByIDBefore (_ context.Context , arg database.GetProjectHistoryLogsByIDBeforeParams ) ([]database.ProjectHistoryLog , error ) {
369
+ logs := make ([]database.ProjectHistoryLog , 0 )
370
+ for _ , projectHistoryLog := range q .projectHistoryLog {
371
+ if projectHistoryLog .ProjectHistoryID .String () != arg .ProjectHistoryID .String () {
372
+ continue
373
+ }
374
+ if projectHistoryLog .CreatedAt .After (arg .CreatedAt ) {
375
+ continue
376
+ }
377
+ logs = append (logs , projectHistoryLog )
378
+ }
379
+ if len (logs ) == 0 {
380
+ return nil , sql .ErrNoRows
381
+ }
382
+ return logs , nil
383
+ }
384
+
321
385
func (q * fakeQuerier ) GetProjectHistoryByID (_ context.Context , projectHistoryID uuid.UUID ) (database.ProjectHistory , error ) {
322
386
for _ , projectHistory := range q .projectHistory {
323
387
if projectHistory .ID .String () != projectHistoryID .String () {
@@ -371,6 +435,13 @@ func (q *fakeQuerier) GetOrganizationMemberByUserID(_ context.Context, arg datab
371
435
return database.OrganizationMember {}, sql .ErrNoRows
372
436
}
373
437
438
+ func (q * fakeQuerier ) GetProvisionerDaemons (_ context.Context ) ([]database.ProvisionerDaemon , error ) {
439
+ if len (q .provisionerDaemons ) == 0 {
440
+ return nil , sql .ErrNoRows
441
+ }
442
+ return q .provisionerDaemons , nil
443
+ }
444
+
374
445
func (q * fakeQuerier ) GetProvisionerDaemonByID (_ context.Context , id uuid.UUID ) (database.ProvisionerDaemon , error ) {
375
446
for _ , provisionerDaemon := range q .provisionerDaemons {
376
447
if provisionerDaemon .ID .String () != id .String () {
@@ -486,6 +557,22 @@ func (q *fakeQuerier) InsertProjectHistory(_ context.Context, arg database.Inser
486
557
return history , nil
487
558
}
488
559
560
+ func (q * fakeQuerier ) InsertProjectHistoryLogs (_ context.Context , arg database.InsertProjectHistoryLogsParams ) ([]database.ProjectHistoryLog , error ) {
561
+ logs := make ([]database.ProjectHistoryLog , 0 )
562
+ for index , output := range arg .Output {
563
+ logs = append (logs , database.ProjectHistoryLog {
564
+ ProjectHistoryID : arg .ProjectHistoryID ,
565
+ ID : arg .ID [index ],
566
+ CreatedAt : arg .CreatedAt [index ],
567
+ Source : arg .Source [index ],
568
+ Level : arg .Level [index ],
569
+ Output : output ,
570
+ })
571
+ }
572
+ q .projectHistoryLog = append (q .projectHistoryLog , logs ... )
573
+ return logs , nil
574
+ }
575
+
489
576
func (q * fakeQuerier ) InsertProjectParameter (_ context.Context , arg database.InsertProjectParameterParams ) (database.ProjectParameter , error ) {
490
577
//nolint:gosimple
491
578
param := database.ProjectParameter {
@@ -586,6 +673,7 @@ func (q *fakeQuerier) InsertWorkspaceHistory(_ context.Context, arg database.Ins
586
673
CreatedAt : arg .CreatedAt ,
587
674
UpdatedAt : arg .UpdatedAt ,
588
675
WorkspaceID : arg .WorkspaceID ,
676
+ Name : arg .Name ,
589
677
ProjectHistoryID : arg .ProjectHistoryID ,
590
678
BeforeID : arg .BeforeID ,
591
679
Transition : arg .Transition ,
@@ -596,6 +684,22 @@ func (q *fakeQuerier) InsertWorkspaceHistory(_ context.Context, arg database.Ins
596
684
return workspaceHistory , nil
597
685
}
598
686
687
+ func (q * fakeQuerier ) InsertWorkspaceHistoryLogs (_ context.Context , arg database.InsertWorkspaceHistoryLogsParams ) ([]database.WorkspaceHistoryLog , error ) {
688
+ logs := make ([]database.WorkspaceHistoryLog , 0 )
689
+ for index , output := range arg .Output {
690
+ logs = append (logs , database.WorkspaceHistoryLog {
691
+ WorkspaceHistoryID : arg .WorkspaceHistoryID ,
692
+ ID : arg .ID [index ],
693
+ CreatedAt : arg .CreatedAt [index ],
694
+ Source : arg .Source [index ],
695
+ Level : arg .Level [index ],
696
+ Output : output ,
697
+ })
698
+ }
699
+ q .workspaceHistoryLog = append (q .workspaceHistoryLog , logs ... )
700
+ return logs , nil
701
+ }
702
+
599
703
func (q * fakeQuerier ) InsertWorkspaceResource (_ context.Context , arg database.InsertWorkspaceResourceParams ) (database.WorkspaceResource , error ) {
600
704
workspaceResource := database.WorkspaceResource {
601
705
ID : arg .ID ,
@@ -659,7 +763,9 @@ func (q *fakeQuerier) UpdateWorkspaceHistoryByID(_ context.Context, arg database
659
763
continue
660
764
}
661
765
workspaceHistory .UpdatedAt = arg .UpdatedAt
766
+ workspaceHistory .CompletedAt = arg .CompletedAt
662
767
workspaceHistory .AfterID = arg .AfterID
768
+ workspaceHistory .ProvisionerState = arg .ProvisionerState
663
769
q .workspaceHistory [index ] = workspaceHistory
664
770
return nil
665
771
}
0 commit comments