@@ -3892,7 +3892,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
3892
3892
}))
3893
3893
s .Run ("GetProvisionerJobsCreatedAfter" , s .Subtest (func (db database.Store , check * expects ) {
3894
3894
_ = dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {CreatedAt : time .Now ().Add (- time .Hour )})
3895
- check .Args (time .Now ()).Asserts (rbac .ResourceSystem , policy .ActionRead )
3895
+ check .Args (time .Now ()).Asserts (rbac .ResourceProvisionerJobs , policy .ActionRead )
3896
3896
}))
3897
3897
s .Run ("GetTemplateVersionsByIDs" , s .Subtest (func (db database.Store , check * expects ) {
3898
3898
dbtestutil .DisableForeignKeysAndTriggers (s .T (), db )
@@ -3978,7 +3978,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
3978
3978
a := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {})
3979
3979
b := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {})
3980
3980
check .Args ([]uuid.UUID {a .ID , b .ID }).
3981
- Asserts (rbac .ResourceSystem , policy .ActionRead ).
3981
+ Asserts (rbac .ResourceProvisionerJobs , policy .ActionRead ).
3982
3982
Returns (slice .New (a , b ))
3983
3983
}))
3984
3984
s .Run ("InsertWorkspaceAgent" , s .Subtest (func (db database.Store , check * expects ) {
@@ -4022,26 +4022,26 @@ func (s *MethodTestSuite) TestSystemFunctions() {
4022
4022
OrganizationID : j .OrganizationID ,
4023
4023
Types : []database.ProvisionerType {j .Provisioner },
4024
4024
ProvisionerTags : must (json .Marshal (j .Tags )),
4025
- }).Asserts (rbac .ResourceSystem , policy .ActionUpdate )
4025
+ }).Asserts (rbac .ResourceProvisionerJobs , policy .ActionUpdate )
4026
4026
}))
4027
4027
s .Run ("UpdateProvisionerJobWithCompleteByID" , s .Subtest (func (db database.Store , check * expects ) {
4028
4028
j := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {})
4029
4029
check .Args (database.UpdateProvisionerJobWithCompleteByIDParams {
4030
4030
ID : j .ID ,
4031
- }).Asserts (rbac .ResourceSystem , policy .ActionUpdate )
4031
+ }).Asserts (rbac .ResourceProvisionerJobs , policy .ActionUpdate )
4032
4032
}))
4033
4033
s .Run ("UpdateProvisionerJobWithCompleteWithStartedAtByID" , s .Subtest (func (db database.Store , check * expects ) {
4034
4034
j := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {})
4035
4035
check .Args (database.UpdateProvisionerJobWithCompleteWithStartedAtByIDParams {
4036
4036
ID : j .ID ,
4037
- }).Asserts (rbac .ResourceSystem , policy .ActionUpdate )
4037
+ }).Asserts (rbac .ResourceProvisionerJobs , policy .ActionUpdate )
4038
4038
}))
4039
4039
s .Run ("UpdateProvisionerJobByID" , s .Subtest (func (db database.Store , check * expects ) {
4040
4040
j := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {})
4041
4041
check .Args (database.UpdateProvisionerJobByIDParams {
4042
4042
ID : j .ID ,
4043
4043
UpdatedAt : time .Now (),
4044
- }).Asserts (rbac .ResourceSystem , policy .ActionUpdate )
4044
+ }).Asserts (rbac .ResourceProvisionerJobs , policy .ActionUpdate )
4045
4045
}))
4046
4046
s .Run ("InsertProvisionerJob" , s .Subtest (func (db database.Store , check * expects ) {
4047
4047
dbtestutil .DisableForeignKeysAndTriggers (s .T (), db )
@@ -4051,19 +4051,19 @@ func (s *MethodTestSuite) TestSystemFunctions() {
4051
4051
StorageMethod : database .ProvisionerStorageMethodFile ,
4052
4052
Type : database .ProvisionerJobTypeWorkspaceBuild ,
4053
4053
Input : json .RawMessage ("{}" ),
4054
- }).Asserts (rbac .ResourceSystem , policy .ActionCreate )
4054
+ }).Asserts (rbac .ResourceProvisionerJobs , policy .ActionCreate )
4055
4055
}))
4056
4056
s .Run ("InsertProvisionerJobLogs" , s .Subtest (func (db database.Store , check * expects ) {
4057
4057
j := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {})
4058
4058
check .Args (database.InsertProvisionerJobLogsParams {
4059
4059
JobID : j .ID ,
4060
- }).Asserts (rbac .ResourceSystem , policy .ActionCreate )
4060
+ }).Asserts (rbac .ResourceProvisionerJobs , policy .ActionUpdate )
4061
4061
}))
4062
4062
s .Run ("InsertProvisionerJobTimings" , s .Subtest (func (db database.Store , check * expects ) {
4063
4063
j := dbgen .ProvisionerJob (s .T (), db , nil , database.ProvisionerJob {})
4064
4064
check .Args (database.InsertProvisionerJobTimingsParams {
4065
4065
JobID : j .ID ,
4066
- }).Asserts (rbac .ResourceSystem , policy .ActionCreate )
4066
+ }).Asserts (rbac .ResourceProvisionerJobs , policy .ActionUpdate )
4067
4067
}))
4068
4068
s .Run ("UpsertProvisionerDaemon" , s .Subtest (func (db database.Store , check * expects ) {
4069
4069
dbtestutil .DisableForeignKeysAndTriggers (s .T (), db )
@@ -4199,11 +4199,8 @@ func (s *MethodTestSuite) TestSystemFunctions() {
4199
4199
s .Run ("GetFileTemplates" , s .Subtest (func (db database.Store , check * expects ) {
4200
4200
check .Args (uuid .New ()).Asserts (rbac .ResourceSystem , policy .ActionRead )
4201
4201
}))
4202
- s .Run ("GetHungProvisionerJobs" , s .Subtest (func (db database.Store , check * expects ) {
4203
- check .Args (time.Time {}).Asserts ()
4204
- }))
4205
- s .Run ("GetPendingProvisionerJobs" , s .Subtest (func (db database.Store , check * expects ) {
4206
- check .Args (time.Time {}).Asserts ()
4202
+ s .Run ("GetProvisionerJobsToBeReaped" , s .Subtest (func (db database.Store , check * expects ) {
4203
+ check .Args (database.GetProvisionerJobsToBeReapedParams {}).Asserts (rbac .ResourceProvisionerJobs , policy .ActionRead )
4207
4204
}))
4208
4205
s .Run ("UpsertOAuthSigningKey" , s .Subtest (func (db database.Store , check * expects ) {
4209
4206
check .Args ("foo" ).Asserts (rbac .ResourceSystem , policy .ActionUpdate )
@@ -4282,7 +4279,7 @@ func (s *MethodTestSuite) TestSystemFunctions() {
4282
4279
check .Args ([]uuid.UUID {uuid .New ()}).Asserts (rbac .ResourceSystem , policy .ActionRead )
4283
4280
}))
4284
4281
s .Run ("GetProvisionerJobsByIDsWithQueuePosition" , s .Subtest (func (db database.Store , check * expects ) {
4285
- check .Args ([]uuid.UUID {}).Asserts ()
4282
+ check .Args ([]uuid.UUID {}).Asserts (rbac . ResourceProvisionerJobs , policy . ActionRead )
4286
4283
}))
4287
4284
s .Run ("GetReplicaByID" , s .Subtest (func (db database.Store , check * expects ) {
4288
4285
check .Args (uuid .New ()).Asserts (rbac .ResourceSystem , policy .ActionRead ).Errors (sql .ErrNoRows )
0 commit comments