@@ -552,10 +552,13 @@ func (s *MethodTestSuite) TestProvisionerJob() {
552
552
Asserts (v .RBACObject (tpl ), []policy.Action {policy .ActionRead ,policy .ActionUpdate }).Returns ()
553
553
}))
554
554
s .Run ("GetProvisionerJobTimingsByJobID" ,s .Subtest (func (db database.Store ,check * expects ) {
555
- jobID := uuid .New ()
556
- j := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {ID :jobID })
555
+ w := dbgen .Workspace (s .T (),db , database.Workspace {})
556
+ j := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {
557
+ Type :database .ProvisionerJobTypeWorkspaceBuild ,
558
+ })
559
+ _ = dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {JobID :j .ID ,WorkspaceID :w .ID })
557
560
t := dbgen .ProvisionerJobTimings (s .T (),db , database.InsertProvisionerJobTimingsParams {
558
- JobID :jobID ,
561
+ JobID :j . ID ,
559
562
StartedAt : []time.Time {dbtime .Now (),dbtime .Now ()},
560
563
EndedAt : []time.Time {dbtime .Now (),dbtime .Now ()},
561
564
Stage : []database.ProvisionerJobTimingStage {
@@ -566,7 +569,7 @@ func (s *MethodTestSuite) TestProvisionerJob() {
566
569
Action : []string {"action1" ,"action2" },
567
570
Resource : []string {"resource1" ,"resource2" },
568
571
})
569
- check .Args (j .ID ).Asserts ().Returns (t )
572
+ check .Args (j .ID ).Asserts (w , policy . ActionRead ).Returns (t )
570
573
}))
571
574
s .Run ("GetProvisionerJobsByIDs" ,s .Subtest (func (db database.Store ,check * expects ) {
572
575
a := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {})