@@ -3263,15 +3263,15 @@ func (s *MethodTestSuite) TestSystemFunctions() {
3263
3263
check .Args (arg ).Asserts (rbac .ResourceSystem ,policy .ActionCreate )
3264
3264
}))
3265
3265
s .Run ("GetProvisionerJobTimingsByJobID" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
3266
- j := testutil .Fake (s .T (),faker , database.ProvisionerJob {Type :database .ProvisionerJobTypeWorkspaceBuild })
3267
- b := testutil .Fake (s .T (),faker , database.WorkspaceBuild {JobID :j .ID })
3268
- ws := testutil .Fake (s .T (),faker , database.Workspace {ID :b .WorkspaceID })
3269
- dbm .EXPECT ().GetProvisionerJobByID (gomock .Any (),j .ID ).Return (j ,nil ).AnyTimes ()
3270
- dbm .EXPECT ().GetWorkspaceBuildByJobID (gomock .Any (),j .ID ).Return (b ,nil ).AnyTimes ()
3271
- dbm .EXPECT ().GetWorkspaceByID (gomock .Any (),b .WorkspaceID ).Return (ws ,nil ).AnyTimes ()
3272
- dbm .EXPECT ().GetProvisionerJobTimingsByJobID (gomock .Any (),j .ID ).Return ([]database.ProvisionerJobTiming {},nil ).AnyTimes ()
3273
- check .Args (j .ID ).Asserts (ws ,policy .ActionRead )
3274
- }))
3266
+ j := testutil .Fake (s .T (),faker , database.ProvisionerJob {Type :database .ProvisionerJobTypeWorkspaceBuild })
3267
+ b := testutil .Fake (s .T (),faker , database.WorkspaceBuild {JobID :j .ID })
3268
+ ws := testutil .Fake (s .T (),faker , database.Workspace {ID :b .WorkspaceID })
3269
+ dbm .EXPECT ().GetProvisionerJobByID (gomock .Any (),j .ID ).Return (j ,nil ).AnyTimes ()
3270
+ dbm .EXPECT ().GetWorkspaceBuildByJobID (gomock .Any (),j .ID ).Return (b ,nil ).AnyTimes ()
3271
+ dbm .EXPECT ().GetWorkspaceByID (gomock .Any (),b .WorkspaceID ).Return (ws ,nil ).AnyTimes ()
3272
+ dbm .EXPECT ().GetProvisionerJobTimingsByJobID (gomock .Any (),j .ID ).Return ([]database.ProvisionerJobTiming {},nil ).AnyTimes ()
3273
+ check .Args (j .ID ).Asserts (ws ,policy .ActionRead )
3274
+ }))
3275
3275
s .Run ("GetWorkspaceAgentScriptTimingsByBuildID" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
3276
3276
build := testutil .Fake (s .T (),faker , database.WorkspaceBuild {})
3277
3277
dbm .EXPECT ().GetWorkspaceAgentScriptTimingsByBuildID (gomock .Any (),build .ID ).Return ([]database.GetWorkspaceAgentScriptTimingsByBuildIDRow {},nil ).AnyTimes ()
@@ -3337,31 +3337,33 @@ func (s *MethodTestSuite) TestSystemFunctions() {
3337
3337
dbm .EXPECT ().GetProvisionerJobByIDForUpdate (gomock .Any (),j .ID ).Return (j ,nil ).AnyTimes ()
3338
3338
// Minimal assertion check argument
3339
3339
b := testutil .Fake (s .T (),faker , database.WorkspaceBuild {JobID :j .ID })
3340
- w := testutil .Fake (s .T (),faker , database.Workspace {ID :b .WorkspaceID })
3341
- dbm .EXPECT ().GetWorkspaceBuildByJobID (gomock .Any (),j .ID ).Return (b ,nil ).AnyTimes ()
3342
- dbm .EXPECT ().GetWorkspaceByID (gomock .Any (),b .WorkspaceID ).Return (w ,nil ).AnyTimes ()
3343
- check .Args (j .ID ).Asserts (w ,policy .ActionRead ).Returns (j )
3340
+ w := testutil .Fake (s .T (),faker , database.Workspace {ID :b .WorkspaceID })
3341
+ dbm .EXPECT ().GetWorkspaceBuildByJobID (gomock .Any (),j .ID ).Return (b ,nil ).AnyTimes ()
3342
+ dbm .EXPECT ().GetWorkspaceByID (gomock .Any (),b .WorkspaceID ).Return (w ,nil ).AnyTimes ()
3343
+ check .Args (j .ID ).Asserts (w ,policy .ActionRead ).Returns (j )
3344
3344
}))
3345
3345
s .Run ("TemplateVersion/GetProvisionerJobByIDForUpdate" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
3346
- j := testutil .Fake (s .T (),faker , database.ProvisionerJob {Type :database .ProvisionerJobTypeTemplateVersionImport })
3347
- tpl := testutil .Fake (s .T (),faker , database.Template {})
3348
- tv := testutil .Fake (s .T (),faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID :tpl .ID ,Valid :true }})
3349
- dbm .EXPECT ().GetProvisionerJobByIDForUpdate (gomock .Any (),j .ID ).Return (j ,nil ).AnyTimes ()
3350
- dbm .EXPECT ().GetTemplateVersionByJobID (gomock .Any (),j .ID ).Return (tv ,nil ).AnyTimes ()
3351
- dbm .EXPECT ().GetTemplateByID (gomock .Any (),tpl .ID ).Return (tpl ,nil ).AnyTimes ()
3352
- check .Args (j .ID ).Asserts (tv .RBACObject (tpl ),policy .ActionRead ).Returns (j )
3353
- }))
3354
- s .Run ("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
3355
- tpl := testutil .Fake (s .T (),faker , database.Template {})
3356
- tv := testutil .Fake (s .T (),faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID :tpl .ID ,Valid :true }})
3357
- j := testutil .Fake (s .T (),faker , database.ProvisionerJob {})
3358
- j .Type = database .ProvisionerJobTypeTemplateVersionDryRun
3359
- j .Input = must (json .Marshal (struct {TemplateVersionID uuid.UUID `json:"template_version_id"` }{TemplateVersionID :tv .ID }))
3360
- dbm .EXPECT ().GetProvisionerJobByIDForUpdate (gomock .Any (),j .ID ).Return (j ,nil ).AnyTimes ()
3361
- dbm .EXPECT ().GetTemplateVersionByID (gomock .Any (),tv .ID ).Return (tv ,nil ).AnyTimes ()
3362
- dbm .EXPECT ().GetTemplateByID (gomock .Any (),tpl .ID ).Return (tpl ,nil ).AnyTimes ()
3363
- check .Args (j .ID ).Asserts (tv .RBACObject (tpl ),policy .ActionRead ).Returns (j )
3364
- }))
3346
+ j := testutil .Fake (s .T (),faker , database.ProvisionerJob {Type :database .ProvisionerJobTypeTemplateVersionImport })
3347
+ tpl := testutil .Fake (s .T (),faker , database.Template {})
3348
+ tv := testutil .Fake (s .T (),faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID :tpl .ID ,Valid :true }})
3349
+ dbm .EXPECT ().GetProvisionerJobByIDForUpdate (gomock .Any (),j .ID ).Return (j ,nil ).AnyTimes ()
3350
+ dbm .EXPECT ().GetTemplateVersionByJobID (gomock .Any (),j .ID ).Return (tv ,nil ).AnyTimes ()
3351
+ dbm .EXPECT ().GetTemplateByID (gomock .Any (),tpl .ID ).Return (tpl ,nil ).AnyTimes ()
3352
+ check .Args (j .ID ).Asserts (tv .RBACObject (tpl ),policy .ActionRead ).Returns (j )
3353
+ }))
3354
+ s .Run ("TemplateVersionDryRun/GetProvisionerJobByIDForUpdate" ,s .Mocked (func (dbm * dbmock.MockStore ,faker * gofakeit.Faker ,check * expects ) {
3355
+ tpl := testutil .Fake (s .T (),faker , database.Template {})
3356
+ tv := testutil .Fake (s .T (),faker , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID :tpl .ID ,Valid :true }})
3357
+ j := testutil .Fake (s .T (),faker , database.ProvisionerJob {})
3358
+ j .Type = database .ProvisionerJobTypeTemplateVersionDryRun
3359
+ j .Input = must (json .Marshal (struct {
3360
+ TemplateVersionID uuid.UUID `json:"template_version_id"`
3361
+ }{TemplateVersionID :tv .ID }))
3362
+ dbm .EXPECT ().GetProvisionerJobByIDForUpdate (gomock .Any (),j .ID ).Return (j ,nil ).AnyTimes ()
3363
+ dbm .EXPECT ().GetTemplateVersionByID (gomock .Any (),tv .ID ).Return (tv ,nil ).AnyTimes ()
3364
+ dbm .EXPECT ().GetTemplateByID (gomock .Any (),tpl .ID ).Return (tpl ,nil ).AnyTimes ()
3365
+ check .Args (j .ID ).Asserts (tv .RBACObject (tpl ),policy .ActionRead ).Returns (j )
3366
+ }))
3365
3367
}
3366
3368
3367
3369
func (s * MethodTestSuite )TestNotifications () {