@@ -3259,31 +3259,28 @@ func (s *MethodTestSuite) TestExtraMethods() {
32593259}))
32603260s .Run ("GetProvisionerJobsByOrganizationAndStatusWithQueuePositionAndProvisioner" ,s .Subtest (func (db database.Store ,check * expects ) {
32613261org := dbgen .Organization (s .T (),db , database.Organization {})
3262+ user := dbgen .User (s .T (),db , database.User {})
32623263tags := database .StringMap (map [string ]string {
32633264provisionersdk .TagScope :provisionersdk .ScopeOrganization ,
32643265})
3265- t := dbgen .Template (s .T (),db , database.Template {})
3266- tv := dbgen .TemplateVersion (s .T (),db , database.TemplateVersion {TemplateID : uuid.NullUUID {UUID :t .ID ,Valid :true }})
3267- w := dbgen .Workspace (s .T (),db , database.WorkspaceTable {OrganizationID :org .ID ,TemplateID :t .ID })
3268- wb := dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {WorkspaceID :w .ID ,TemplateVersionID :tv .ID })
3269- j1 ,err := db .InsertProvisionerJob (context .Background (), database.InsertProvisionerJobParams {
3266+ t := dbgen .Template (s .T (),db , database.Template {OrganizationID :org .ID ,CreatedBy :user .ID })
3267+ tv := dbgen .TemplateVersion (s .T (),db , database.TemplateVersion {OrganizationID :org .ID ,CreatedBy :user .ID ,TemplateID : uuid.NullUUID {UUID :t .ID ,Valid :true }})
3268+ j1 := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {
32703269OrganizationID :org .ID ,
32713270Type :database .ProvisionerJobTypeTemplateVersionImport ,
32723271Input : []byte (`{"template_version_id":"` + tv .ID .String ()+ `"}` ),
32733272Tags :tags ,
3274- Provisioner :database .ProvisionerTypeEcho ,
3275- StorageMethod :database .ProvisionerStorageMethodFile ,
32763273})
3277- s .NoError (err ,"insert provisioner job" )
3278- j2 ,err := db .InsertProvisionerJob (context .Background (), database.InsertProvisionerJobParams {
3274+ w := dbgen .Workspace (s .T (),db , database.WorkspaceTable {OrganizationID :org .ID ,OwnerID :user .ID ,TemplateID :t .ID })
3275+ wbID := uuid .New ()
3276+ j2 := dbgen .ProvisionerJob (s .T (),db ,nil , database.ProvisionerJob {
32793277OrganizationID :org .ID ,
32803278Type :database .ProvisionerJobTypeWorkspaceBuild ,
3281- Input : []byte (`{"workspace_build_id":"` + wb . ID .String ()+ `"}` ),
3279+ Input : []byte (`{"workspace_build_id":"` + wbID .String ()+ `"}` ),
32823280Tags :tags ,
3283- Provisioner :database .ProvisionerTypeEcho ,
3284- StorageMethod :database .ProvisionerStorageMethodFile ,
32853281})
3286- s .NoError (err ,"insert provisioner job" )
3282+ dbgen .WorkspaceBuild (s .T (),db , database.WorkspaceBuild {ID :wbID ,WorkspaceID :w .ID ,TemplateVersionID :tv .ID ,JobID :j2 .ID })
3283+
32873284ds ,err := db .GetProvisionerJobsByOrganizationAndStatusWithQueuePositionAndProvisioner (context .Background (), database.GetProvisionerJobsByOrganizationAndStatusWithQueuePositionAndProvisionerParams {
32883285OrganizationID : uuid.NullUUID {Valid :true ,UUID :org .ID },
32893286})