( t *testing.T, ctx context.Context, db *globaldb.GlobalDB, manager *taskpkg.Service, creator taskpkg.ActorContext, claimer taskpkg.ActorContext, workspaceName string, title string, claimerSessionID string, now time.Time, )
| 2366 | } |
| 2367 | |
| 2368 | func createIntegrationClaimedRun( |
| 2369 | t *testing.T, |
| 2370 | ctx context.Context, |
| 2371 | db *globaldb.GlobalDB, |
| 2372 | manager *taskpkg.Service, |
| 2373 | creator taskpkg.ActorContext, |
| 2374 | claimer taskpkg.ActorContext, |
| 2375 | workspaceName string, |
| 2376 | title string, |
| 2377 | claimerSessionID string, |
| 2378 | now time.Time, |
| 2379 | ) (*taskpkg.Task, *taskpkg.Run, *taskpkg.ClaimResult) { |
| 2380 | t.Helper() |
| 2381 | |
| 2382 | workspaceID := registerTaskManagerWorkspace(t, db, workspaceName, filepath.Join(t.TempDir(), workspaceName)) |
| 2383 | taskRecord, err := manager.CreateTask(ctx, taskpkg.CreateTask{ |
| 2384 | Scope: taskpkg.ScopeWorkspace, |
| 2385 | WorkspaceID: workspaceID, |
| 2386 | Title: title, |
| 2387 | }, creator) |
| 2388 | if err != nil { |
| 2389 | t.Fatalf("CreateTask(%q) error = %v", title, err) |
| 2390 | } |
| 2391 | run, err := manager.EnqueueRun(ctx, taskpkg.EnqueueRun{TaskID: taskRecord.ID}, creator) |
| 2392 | if err != nil { |
| 2393 | t.Fatalf("EnqueueRun(%q) error = %v", title, err) |
| 2394 | } |
| 2395 | claim, err := manager.ClaimNextRun(ctx, taskpkg.ClaimCriteria{ |
| 2396 | Scope: taskpkg.ScopeWorkspace, |
| 2397 | WorkspaceID: workspaceID, |
| 2398 | ClaimerSessionID: claimerSessionID, |
| 2399 | LeaseDuration: 10 * time.Minute, |
| 2400 | Now: now, |
| 2401 | }, claimer) |
| 2402 | if err != nil { |
| 2403 | t.Fatalf("ClaimNextRun(%q) error = %v", title, err) |
| 2404 | } |
| 2405 | if got, want := claim.Run.ID, run.ID; got != want { |
| 2406 | t.Fatalf("ClaimNextRun(%q).Run.ID = %q, want %q", title, got, want) |
| 2407 | } |
| 2408 | return taskRecord, run, claim |
| 2409 | } |
| 2410 | |
| 2411 | func findIntegrationTaskEvent( |
| 2412 | t *testing.T, |
no test coverage detected