@@ -334,6 +334,16 @@ func TestAcquireJob(t *testing.T) {
334334Transition :database .WorkspaceTransitionStart ,
335335Reason :database .BuildReasonInitiator ,
336336})
337+ task := dbgen .Task (t ,db , database.TaskTable {
338+ OrganizationID :pd .OrganizationID ,
339+ OwnerID :user .ID ,
340+ WorkspaceID : uuid.NullUUID {Valid :true ,UUID :workspace .ID },
341+ TemplateVersionID :version .ID ,
342+ TemplateParameters :json .RawMessage ("{}" ),
343+ Prompt :"Build me a REST API" ,
344+ CreatedAt :dbtime .Now (),
345+ DeletedAt : sql.NullTime {},
346+ })
337347
338348var agent database.WorkspaceAgent
339349if prebuiltWorkspaceBuildStage == sdkproto .PrebuiltWorkspaceBuildStage_CLAIM {
@@ -446,6 +456,8 @@ func TestAcquireJob(t *testing.T) {
446456WorkspaceBuildId :build .ID .String (),
447457WorkspaceOwnerLoginType :string (user .LoginType ),
448458WorkspaceOwnerRbacRoles : []* sdkproto.Role {{Name :rbac .RoleOrgMember (),OrgId :pd .OrganizationID .String ()}, {Name :"member" ,OrgId :"" }, {Name :rbac .RoleOrgAuditor (),OrgId :pd .OrganizationID .String ()}},
459+ TaskId :task .ID .String (),
460+ TaskPrompt :task .Prompt ,
449461}
450462if prebuiltWorkspaceBuildStage == sdkproto .PrebuiltWorkspaceBuildStage_CLAIM {
451463// For claimed prebuilds, we expect the prebuild state to be set to CLAIM