mirror of
https://github.com/paperclipai/paperclip.git
synced 2026-11-06 23:20:53 +01:00
## Thinking Path > - Paperclip is the open source app people use to manage AI agents for work. > - A stopped sandbox can hold workspace files that still need a durable copy. > - File recovery must reopen the exact allocation and directory recorded by the run. > - Workspace preparation can place that directory below the allocation root. > - Daytona resume returned the allocation root even when recovery needed the recorded child directory. > - This pull request lets the file-recovery owner request that exact directory. > - Ordinary resume keeps its allocation-root behavior. The recovery source checks stay strict. ## Linked Issues or Issue Description **What happened?** A stopped native run had a prepared remote workspace. The recovery service stopped and retained its allocation. The next file-recovery step failed with `source_binding_invalid` and `reopened_cwd_mismatch`. Daytona resumed the correct allocation and matched its sentinel, but returned the allocation root instead of the workspace directory recorded on the lease. **Expected behavior** The file-recovery owner must reopen its exact recorded directory without creating an allocation, an empty directory, or an agent turn. Ordinary execution resume must keep returning the allocation root for normal workspace preparation. **Steps to reproduce** 1. Prepare a task workspace below a Daytona allocation root. 2. Stop and retain the allocation with unsaved workspace files. 3. Request source-bound file recovery. 4. Before this change, the provider returns the allocation root and the strict cwd check rejects it. **Paperclip version or commit** Reproduced from `050365e09d88411960fc3bacc1c08c46c0d9b6fa`. **Deployment mode** Docker with the bundled Daytona provider. Local tests use synthetic provider boundaries and disposable PostgreSQL. Refs: #18517, #15749. Related retained-export work: #17916. That work does not change the Daytona resume handler. ## What Changed - Add an optional driver-only `requestedCwd` resume field to the SDK contract and existing host forwarders. - Set the field only after the owned file-recovery path verifies the recorded source. - Require Daytona to match the requested directory to the recorded lease directory. Require a normalized absolute path without NUL bytes. Keep the allocation and sentinel checks. - Keep ordinary resume unchanged when the field is absent. - Add provider, host, and real workspace-preparation regressions. Document the separate directory contracts. ## Verification - The original provider regression failed for reusable and ephemeral leases. The scoped regression also failed to read a real retained file from the wrong root. Ordinary resume controls passed before the fix. - Current provider handler: **345/345 passed**. This includes exact-directory, malformed, missing, foreign, sentinel, reusable, ephemeral, ordinary-root, and real-file cases. - Removing only the owned producer's cwd request makes its real PostgreSQL regression fail with the original `reopened_cwd_mismatch`; the source file was restored exactly afterward. - Current host selection: **10/10 passed** against disposable PostgreSQL 17. It covers the inspected owner, sandbox forwarding, generic plugin forwarding, normal resume without the field, same and later task workspace preparation, and ownership rejection. The same-workspace control retains saved bytes and does not append another workspace path. - Current SDK and Daytona builds, server build, module boundaries, maps, and diff checks passed. The server used its normal unchanged-Runner cache. Its canonical input digest matched the previously verified build, and the canonical assessor accepted the copied original stamp and outputs. The changed SDK and server compiled from current source. - Local package-wide validation exposed two unchanged 30-second timeouts in the 145,000-member tar-listing cases. A single-file control reproduced both timeouts. That test mocks this provider module. The failures remain recorded; no timeout changed. The earlier package run had 497 passes, two failures, and seven skips. It is historical evidence from the abandoned broader resume experiment, not final-source package acceptance. - Initial host attempts timed out during database setup before any assertion. The final host selection used the existing supported pre-migrated disposable template. No test timeout changed. - Greptile reviewed exact head `8d0e577d1234ad394887255f13707c010dee6d24`: **5/5, no actionable issues, zero review threads**. [Current review](https://github.com/paperclipai/paperclip/pull/18527#issuecomment-6555253315). - The official [Ponytail review](https://github.com/paperclipai/paperclip/pull/18527#issuecomment-6555237850) also assessed exact head `8d0e577d1234ad394887255f13707c010dee6d24`: **5/5, no findings**. Its check is terminal neutral; its completed source assessment is not a test-execution claim. - [Primary CI](https://github.com/paperclipai/paperclip/actions/runs/41286802674) initially lost its spot runner in the UI/CLI workspace lane. The log records an explicit shutdown signal and no test assertion result. UI, CLI, lane scripts and dependencies are unchanged; the adjacent identical-source baseline lane passed. The single failed-lane retry passed at the same source. The initial failure is retained. All exact-head CI gates are now green. - No live repair or image deployment is claimed by this PR. Commands: ```sh node packages/plugins/sandbox-providers/daytona/node_modules/vitest/vitest.mjs run --config packages/plugins/sandbox-providers/daytona/vitest.config.ts packages/plugins/sandbox-providers/daytona/src/plugin.test.ts node node_modules/vitest/vitest.mjs run --project @paperclipai/server server/src/services/workspace-recovery.test.ts server/src/__tests__/environment-runtime.test.ts -t 'recovers a protected retained source|owned export directory|prepares the (same|later) task workspace|does not reopen a retained source|resumes only the exact stopped sandbox lifecycle|explicit export directory' pnpm -C packages/plugins/sandbox-providers/daytona build pnpm --filter @paperclipai/server build node scripts/check-module-boundaries.mjs node scripts/check-maps.mjs git diff --check ``` ## Risks - A provider that does not support the optional field can still return a different directory. Recovery keeps failing closed on that mismatch. - Ordinary resumes omit the field. The tests cover the same workspace and a later task so that normal preparation does not gain an extra path prefix. - The field grants no source ownership, stop proof, file-persistence proof, or execution authority. Existing allocation, company, task, run, claim, and cwd checks remain in force. - No schema migration, retry-policy change, model replay, or historical-row repair is included. Existing failed attempts remain historical facts. Deployed file recovery still needs its normal admission and independent persistence and physical-stop receipts. ## Model Used - OpenAI GPT-6 through Codex. Used reasoning, source inspection, tool use, and code execution. The exact serving model identifier and context window are not exposed in this session. ## Checklist - [x] I have included a thinking path that traces from project context to this change - [x] I have specified the model used (with version and capability details) - [x] I have checked ROADMAP.md and confirmed this PR does not duplicate planned core work - [x] I have searched GitHub for duplicate or related PRs and linked them above - [x] I have either (a) linked existing issues with `Fixes: #` / `Closes #` / `Refs #` OR (b) described the issue in-PR following the relevant issue template - [x] I have not referenced internal/instance-local Paperclip issues or links (only public GitHub `#NNN` / `github.com/paperclipai/paperclip` URLs) - [x] My branch name describes the change (e.g. `docs/...`, `fix/...`) and contains no internal Paperclip ticket id or instance-derived details - [x] I have run tests locally and they pass - [x] I have added or updated tests where applicable - [x] I have updated relevant documentation to reflect my changes - [x] I have considered and documented any risks above - [x] All Paperclip CI gates are green - [x] Greptile is 5/5 with no open P2s, recommendations, or follow-ups - [x] I will address all Greptile and reviewer comments before requesting merge Co-authored-by: Paperclip <noreply@paperclip.ing>
2334 lines
187 KiB
TypeScript
2334 lines
187 KiB
TypeScript
import { createAdmissionTransactionScope, createWakeAdmissionReader, createWakeAdmissionWriter } from "../modules/wake-queue/adapters/postgres.js";
|
|
import { createAdmitWakeBehindIssueExecution } from "../modules/wake-queue/application/use-cases.js";
|
|
import { filterZombieCoalesceTarget } from "./heartbeat.js";
|
|
import { mergeCoalescedContextSnapshot } from "./heartbeat/run-preparation.js";
|
|
import { recoveryService } from "./recovery/index.js";
|
|
import { inspectWorkspaceSource } from "./workspace-source-inspection.js";
|
|
import { appendHeartbeatRunEvent } from "./heartbeat-run-events.js";
|
|
import { isPreDispatchExecutionWaitVerified } from "./pre-dispatch-execution-wait.js";
|
|
import { heartbeatRunEvents } from "@paperclipai/db";
|
|
import { createHeartbeatQueue, shouldDeferFollowupWakeForSameIssue, shouldQueueFollowupForRunningIssueWake } from "./heartbeat/queue.js";
|
|
import { createHeartbeatQueueTestDependencies } from "../__tests__/helpers/heartbeat-queue.js";
|
|
import { recordRequiredWorkspaceRestoreFailure, terminalizeLegacyExecution } from "./legacy-execution-recovery.js";
|
|
import { admitExplicitNativeContinuation } from "./explicit-native-continuation.js";
|
|
import { withQueuedCommentIdsInWakePayload } from "./issue-queued-comment-queue.js";
|
|
import * as recoverySources from "./workspace-recovery-source.js";
|
|
import { requestedRunCancellation } from "./run-cancellation.js";
|
|
import { createHeartbeatRunControl } from "./heartbeat/run-control.js";
|
|
import { callbacks } from "./heartbeat/__tests__/run-control-fixtures.js";
|
|
import { adapterExecutionControls, createAdapterExecutionControl } from "./adapter-execution-control.js";
|
|
import { resolveExternalTestDatabaseUrl } from "../__tests__/helpers/external-test-database.js";
|
|
import { DELIVERY_QUEUES, subscribeDeliveryWork } from "./delivery-work-notifications.js";
|
|
import { prepareHeartbeatWorkspace } from "./heartbeat/workspace-preparation.js";
|
|
import { createHeartbeatWorkspaceResolver, resolveExecutionWorkspaceReuseRequestForIssue } from "./heartbeat/workspaces.js";
|
|
import { environmentService } from "./environments.js";
|
|
import { classifyEnvironmentCapabilities, environmentRuntimeService } from "./environment-runtime.js";
|
|
import { environmentRunOrchestrator } from "./environment-run-orchestrator.js";
|
|
import { executionWorkspaceService } from "./execution-workspaces.js";
|
|
import { workspaceOperationService } from "./workspace-operations.js";
|
|
import { instanceSettingsService } from "./instance-settings.js";
|
|
import { projectTaskInterruption } from "./task-interruption-presentation.js";
|
|
import * as nativeWorkspaceSync from "./native-runtime/native-workspace-sync.js";
|
|
import { GitShallowMetadataError } from "@paperclipai/adapter-utils/git-workspace-sync";
|
|
import { readSystemRecovery } from "./system-recovery-state.js";
|
|
import { adoptSystemRecovery, updateSystemRecovery } from "./system-recovery.js";
|
|
import { getStorageService } from "../storage/index.js";
|
|
import { attachPreservedWorkspaceSnapshots } from "./workspace-recovery-snapshot.js";
|
|
import { adoptHistoricalWorkspaceRecoveries, historicalRecoveryCounts } from "./workspace-recovery-backfill.js";
|
|
import { createSystemRecovery, failSystemRecovery, SystemRecoveryFailure } from "./system-recovery-state.js";
|
|
import { requestTaskRetry } from "./task-retry.js";
|
|
import { readFreshWorkspaceReadiness } from "./fresh-workspace-continuation.js";
|
|
import { execFile } from "node:child_process";
|
|
import { generateKeyPairSync, randomUUID, sign } from "node:crypto";
|
|
import { cp, mkdir, mkdtemp, readFile, realpath, rm, writeFile } from "node:fs/promises";
|
|
import os from "node:os";
|
|
import path from "node:path";
|
|
import express from "express";
|
|
import request from "supertest";
|
|
import { and, eq, inArray, sql } from "drizzle-orm";
|
|
import { afterAll, afterEach, beforeAll, describe, expect, it, vi } from "vitest";
|
|
import { agents, approvals, budgetPolicies, costEvents, completionContracts, issueApprovals, issueTreeHolds, issueTreeHoldMembers, companies, createDb, applyPendingMigrations, environmentLeases, environments, heartbeatRuns, issueComments, issueRecoveryActions, issues, nativeRunFinalizations, nativeRunResults, agentWakeupRequests, workspaceOperations } from "@paperclipai/db";
|
|
import { resumePreservedResultFinalization } from "./preserved-result-finalization.js";
|
|
import type { AdapterSandboxExecutionTarget } from "@paperclipai/adapter-utils/execution-target";
|
|
import type { EnvironmentLease } from "@paperclipai/shared";
|
|
import { startEmbeddedPostgresTestDatabase } from "../__tests__/helpers/embedded-postgres.js";
|
|
import { prepareNativeWorkspaceSync, resumeNativeWorkspaceSync } from "./native-runtime/native-workspace-sync.js";
|
|
import { nativeWorkspaceFinalizationLockKey } from "./native-runtime/native-workspace-finalization-ownership.js";
|
|
import { remoteTerminationReceipt } from "./remote-execution-termination.js";
|
|
import { prepareStoppedSandboxForRecovery, prepareSandboxStopAndRetain, settleStopOnlyCleanup, readStopOnlyCleanup } from "./sandbox-stop-and-retain.js";
|
|
import { recoverRetainedWorkspace, requestWorkspaceRecovery, scheduleRecoveredWorkspaceCleanup, reconcileRetainedWorkspaces, retainedWorkspaceCandidates } from "./workspace-recovery.js";
|
|
import * as activity from "./activity-log.js";
|
|
import { createDeliveryWorkCoordinator } from "./delivery-work-coordinator.js";
|
|
import { registerWorkspaceRecoveryWork } from "./workspace-recovery-work.js";
|
|
import { heartbeatService } from "./heartbeat.js";
|
|
import { hasVerifiedWorkspaceRepair, readWorkspaceRecovery } from "./workspace-recovery-receipt.js";
|
|
import { claimNativeSessionResumptions } from "./native-runtime/native-finalization-reconciler.js";
|
|
import { executionWorkspaces } from "@paperclipai/db";
|
|
import { executionWorkspaceSnapshots } from "@paperclipai/db";
|
|
import { createSavedWorkspaceLoader, prepareRemoteWorkspace, readRemoteWorkspaceAuthority, reserveRemoteWorkspace, saveRemoteWorkspace } from "./remote-workspaces.js";
|
|
import { createReusableLeaseHandoff } from "./environment-lease-handoff.js";
|
|
import { preserveRemoteWorkspacePreparationFailure } from "./native-runtime/native-workspace-finalization-state.js";
|
|
import { getExecutionBlocker } from "./execution-blocker.js";
|
|
import { issueRoutes } from "../routes/issues.js";
|
|
import { plugins } from "@paperclipai/db";
|
|
import type { PluginWorkerManager } from "./plugin-worker-manager.js";
|
|
import { errorHandler } from "../middleware/index.js";
|
|
import * as workerProof from "./native-runtime/native-workspace-platform-proof.js";
|
|
const targets = vi.hoisted(() => new Map<string, AdapterSandboxExecutionTarget>());
|
|
vi.mock("./environment-execution-target.js", async importOriginal => {
|
|
const actual = await importOriginal<typeof import("./environment-execution-target.js")>();
|
|
return { ...actual, resolveEnvironmentExecutionTarget: async (input: Parameters<typeof actual.resolveEnvironmentExecutionTarget>[0]) =>
|
|
(input.leaseId ? targets.get(input.leaseId) : undefined) ?? actual.resolveEnvironmentExecutionTarget(input) };
|
|
});
|
|
|
|
describe("automatic retained workspace recovery", () => {
|
|
let temporary: Awaited<ReturnType<typeof startEmbeddedPostgresTestDatabase>> | undefined;
|
|
let connectionString: string;
|
|
let db: ReturnType<typeof createDb>;
|
|
let root: string;
|
|
const fixtureCompanyIds = new Set<string>();
|
|
const priorHome = process.env.PAPERCLIP_HOME, priorInstance = process.env.PAPERCLIP_INSTANCE_ID;
|
|
beforeAll(async () => {
|
|
root = await mkdtemp(path.join(os.tmpdir(), "workspace-recovery-"));
|
|
process.env.PAPERCLIP_HOME = path.join(root, "home"); process.env.PAPERCLIP_INSTANCE_ID = "recovery";
|
|
const external = resolveExternalTestDatabaseUrl("PAPERCLIP_WORKSPACE_RECOVERY_TEST_DATABASE_URL");
|
|
if (external) {
|
|
await applyPendingMigrations(external);
|
|
temporary = { connectionString: external, cleanup: async () => {} };
|
|
} else temporary = await startEmbeddedPostgresTestDatabase("workspace-recovery-");
|
|
connectionString = temporary.connectionString;
|
|
db = createDb(connectionString);
|
|
if (external) await db.execute(sql`TRUNCATE TABLE companies CASCADE`);
|
|
}, 30_000);
|
|
afterAll(async () => {
|
|
await db?.$client.end(); await temporary?.cleanup(); await rm(root, { recursive: true, force: true });
|
|
if (priorHome === undefined) delete process.env.PAPERCLIP_HOME; else process.env.PAPERCLIP_HOME = priorHome;
|
|
if (priorInstance === undefined) delete process.env.PAPERCLIP_INSTANCE_ID; else process.env.PAPERCLIP_INSTANCE_ID = priorInstance;
|
|
});
|
|
afterEach(async () => {
|
|
// Discovery is intentionally global and bounded. An unresolved source from
|
|
// an earlier test must not consume the current fixture's recovery page.
|
|
// Keep all history within each test until its assertions have completed.
|
|
const companyIds = [...fixtureCompanyIds]; fixtureCompanyIds.clear();
|
|
if (!companyIds.length) return;
|
|
const leases = await db.select({ id: environmentLeases.id }).from(environmentLeases)
|
|
.where(inArray(environmentLeases.companyId, companyIds));
|
|
for (const lease of leases) targets.delete(lease.id);
|
|
await db.delete(environmentLeases).where(inArray(environmentLeases.companyId, companyIds));
|
|
// Historical adoption can reconstruct an action from either the lease or
|
|
// the run descriptor, so retire both fixture inputs before deleting it.
|
|
await db.update(heartbeatRuns).set({ resultJson: sql`${heartbeatRuns.resultJson} - 'workspaceRestoreRecovery' - 'workspaceRestoreFailure'` })
|
|
.where(inArray(heartbeatRuns.companyId, companyIds));
|
|
await db.delete(issueRecoveryActions).where(inArray(issueRecoveryActions.companyId, companyIds));
|
|
});
|
|
async function seed(mode = "native", nativeDescriptor = true, resourceDisposition: "destroy" | "keep_running" = "destroy", reusable = false, ephemeral = false) {
|
|
const companyId = randomUUID(), issueId = randomUUID(), agentId = randomUUID(), runId = randomUUID(), actionId: string = randomUUID();
|
|
await db.insert(companies).values({ id: companyId, name: "Workspace recovery", defaultResponsibleUserId: "board", issuePrefix: `R${companyId.slice(0, 6)}` });
|
|
fixtureCompanyIds.add(companyId);
|
|
await db.insert(agents).values({ id: agentId, companyId, name: "Worker", adapterType: "paperclip_runner", status: "idle", runtimeConfig: { heartbeat: { maxConcurrentRuns: 1 } } });
|
|
await db.insert(issues).values({ id: issueId, companyId, assigneeAgentId: agentId, status: "blocked", title: "Recover existing work" });
|
|
await db.insert(heartbeatRuns).values({ id: runId, companyId, issueId, nativeIssueId: mode === "native" ? issueId : null,
|
|
agentId, runtimeMode: mode, status: "failed", finishedAt: new Date(), errorCode: "execution_interrupted", contextSnapshot: { issueId } });
|
|
if (mode === "legacy") await db.update(heartbeatRuns).set({ runnerProfileJson: { adapterDispatch: { adapterType: "codex_local" } } }).where(eq(heartbeatRuns.id, runId));
|
|
if (mode === "native") await db.insert(nativeRunFinalizations).values({ companyId, issueId, runId, phase: "terminal_failure" });
|
|
const [environment] = await db.insert(environments).values({ name: runId, driver: "sandbox" }).returning();
|
|
const providerLeaseId: string = randomUUID(), remote = path.join(root, runId, "remote"), host = path.join(root, runId, "host");
|
|
await mkdir(host, { recursive: true }); await writeFile(path.join(host, "existing.txt"), "base\n");
|
|
const leaseValues = { companyId, issueId, heartbeatRunId: runId,
|
|
environmentId: environment.id, provider: "daytona", providerLeaseId,
|
|
metadata: { remoteCwd: remote, driver: "sandbox", pluginId: randomUUID(), sandboxProviderPlugin: true, ...(reusable ? { agentId } : {}) } };
|
|
const acquired = reusable ? await environmentService(db).acquireLease({ ...leaseValues, leasePolicy: "reuse_by_environment" }) : null;
|
|
const [lease] = acquired ? await db.select().from(environmentLeases).where(eq(environmentLeases.id, acquired.id))
|
|
: await db.insert(environmentLeases).values({ ...leaseValues, leasePolicy: ephemeral ? "ephemeral" : "retain_on_failure", status: "released",
|
|
cleanupStatus: "success", releasedAt: new Date() }).returning();
|
|
const target: AdapterSandboxExecutionTarget = { kind: "remote", transport: "sandbox", providerKey: "test", leaseId: lease.id,
|
|
remoteCwd: remote, timeoutMs: 30_000, sandboxLeaseAcquisition: { outcome: "created", providerLeaseId },
|
|
runner: { execute: input => new Promise((resolve, reject) => {
|
|
const startedAt = new Date().toISOString();
|
|
const child = execFile(input.command, input.args ?? [], { cwd: input.cwd, env: { ...process.env, ...input.env }, timeout: input.timeoutMs, maxBuffer: 16 * 1024 * 1024 },
|
|
(error, stdout, stderr) => resolve({ exitCode: error ? typeof error.code === "number" ? error.code : 1 : 0,
|
|
signal: error?.signal ?? null, timedOut: error?.killed ?? false, stdout, stderr, pid: child.pid ?? null, startedAt }));
|
|
child.stdin?.on("error", (error: NodeJS.ErrnoException) => { if (error.code !== "EPIPE") reject(error); });
|
|
child.stdin?.end(input.stdin ?? "");
|
|
}) } };
|
|
targets.set(lease.id, target);
|
|
if (nativeDescriptor) {
|
|
const prepared = await prepareNativeWorkspaceSync({ db, companyId, workspaceId: issueId, workspaceLocalDir: host,
|
|
runId, lease: lease as EnvironmentLease, target, resourceDisposition });
|
|
expect(prepared).not.toBeNull();
|
|
} else {
|
|
// Provider boundary only: real temporary files, no native baseline. The
|
|
// production provider's syncOut contract validates its archive itself.
|
|
target.runner!.syncIn = async () => ({ filesTransferred: 0, bytesTransferred: 0, operations: [] });
|
|
target.runner!.syncOut = async operations => {
|
|
for (const operation of operations) for (const file of operation.files)
|
|
await cp(file.sourcePath, file.targetPath, { recursive: true, dereference: false });
|
|
return { filesTransferred: 2, bytesTransferred: 0, operations: [] };
|
|
};
|
|
await mkdir(remote, { recursive: true });
|
|
await writeFile(path.join(remote, "existing.txt"), "base\n");
|
|
}
|
|
await writeFile(path.join(remote, "saved.txt"), "finished work\n");
|
|
await writeFile(path.join(host, "newer.txt"), "newer local work\n");
|
|
const recovery = { schema: "paperclip.workspace-restore-recovery.v1", leaseIds: [lease.id] };
|
|
await db.update(heartbeatRuns).set({ resultJson: { workspaceRestoreFailure: "restore_failed", workspaceRestoreRecovery: recovery,
|
|
conversationContinuation: "continue_conversation_v1" } }).where(eq(heartbeatRuns.id, runId));
|
|
await db.update(environmentLeases).set({ metadata: { ...lease.metadata,
|
|
remoteExecutionTermination: remoteTerminationReceipt(lease, { providerLeaseId, state: "stopped" }) } }).where(eq(environmentLeases.id, lease.id));
|
|
await db.insert(issueRecoveryActions).values({ id: actionId, companyId, sourceIssueId: issueId, kind: "active_run_watchdog",
|
|
ownerType: "board", cause: mode === "native" ? "native_continuation_requires_reconciliation" : "legacy_execution_requires_reconciliation",
|
|
fingerprint: runId, nextAction: "Manual repair required", evidence: { runId, workspaceRestoreFailure: "restore_failed", workspaceRestoreRecovery: recovery } });
|
|
const runtime = { resumeRunLease: vi.fn(async (_input: { lease: EnvironmentLease }) => ({ providerLeaseId, metadata: { remoteCwd: remote } })),
|
|
retryPendingSandboxTeardown: vi.fn(async () => ({ providerLeaseId, state: "stopped" })) };
|
|
const environmentRuntime = environmentRuntimeService(db, { drivers: [{ driver: "sandbox",
|
|
acquireRunLease: vi.fn(), releaseRunLease: vi.fn(), ...runtime,
|
|
resolveCapabilities: async () => classifyEnvironmentCapabilities({ verifiedMethods: [], declared: null, narrowing: null }),
|
|
}] });
|
|
const input = { db, companyId, issueId, runId, actionId, environmentRuntime };
|
|
return { ...input, input, runtime, host, remote, lease, providerLeaseId };
|
|
}
|
|
const action = async (id: string) => (await db.select().from(issueRecoveryActions).where(eq(issueRecoveryActions.id, id)))[0]!;
|
|
const run = async (id: string) => (await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.id, id)))[0]!;
|
|
|
|
it.each([
|
|
{ status: "failed", runtimeMode: "native", uploadFailure: "none" },
|
|
{ status: "cancelled", runtimeMode: "native", uploadFailure: "none" },
|
|
{ status: "failed", runtimeMode: "legacy", uploadFailure: "final" },
|
|
{ status: "failed", runtimeMode: "legacy", uploadFailure: "initial" },
|
|
] as const)("saves the authoritative remote workspace after $runtimeMode $status without replaying a turn or certifying external effects ($uploadFailure upload failure)", async ({ status, runtimeMode, uploadFailure }) => {
|
|
const f = await seed(runtimeMode, false);
|
|
const remoteCwd = await realpath(f.remote);
|
|
targets.get(f.lease.id)!.remoteCwd = remoteCwd;
|
|
f.runtime.resumeRunLease.mockImplementation(async () => ({ providerLeaseId: f.providerLeaseId, metadata: { remoteCwd } }));
|
|
const workspaceId = randomUUID();
|
|
let authority = { schema: "paperclip.remote-workspace.v1", epoch: 1, snapshotId: null as string | null,
|
|
source: { kind: "empty" }, writerRunId: f.runId, dirty: true,
|
|
location: { environmentId: f.lease.environmentId!, leaseId: f.lease.id,
|
|
providerLeaseId: f.providerLeaseId, remoteCwd } };
|
|
await db.insert(executionWorkspaces).values({ id: workspaceId, companyId: f.companyId, sourceIssueId: f.issueId,
|
|
name: "Authoritative remote files", cwd: remoteCwd, providerRef: remoteCwd, providerType: "cloud_sandbox",
|
|
mode: "isolated_workspace", strategyType: "project_primary", status: "active", metadata: { remoteWorkspace: authority } });
|
|
await db.update(issues).set({ executionWorkspaceId: workspaceId }).where(eq(issues.id, f.issueId));
|
|
const [boundLease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id));
|
|
await db.update(environmentLeases).set({ executionWorkspaceId: workspaceId, leasePolicy: "reuse_by_execution_workspace",
|
|
metadata: { ...boundLease.metadata, remoteCwd, agentId: (await run(f.runId)).agentId } })
|
|
.where(eq(environmentLeases.id, f.lease.id));
|
|
const before = await run(f.runId);
|
|
if (uploadFailure === "final") {
|
|
// A reused allocation can fail during preparation, before native input or
|
|
// its finalization coordinator exists. Keep a real saved predecessor and
|
|
// the ordinary exact lease handoff; neither grants a new provider turn.
|
|
const [priorRun] = await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: before.agentId,
|
|
issueId: f.issueId, nativeIssueId: f.issueId, runtimeMode: "native", status: "succeeded",
|
|
createdAt: new Date(before.createdAt.getTime() - 2_000), finishedAt: new Date(before.createdAt.getTime() - 1_000),
|
|
contextSnapshot: { issueId: f.issueId } }).returning();
|
|
await db.insert(nativeRunFinalizations).values({ companyId: f.companyId, issueId: f.issueId,
|
|
runId: priorRun.id, phase: "committed" });
|
|
const [currentLease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id));
|
|
const [priorLease] = await db.insert(environmentLeases).values({ companyId: f.companyId, issueId: f.issueId,
|
|
heartbeatRunId: priorRun.id, environmentId: currentLease.environmentId, executionWorkspaceId: workspaceId,
|
|
provider: currentLease.provider, providerLeaseId: f.providerLeaseId, leasePolicy: "reuse_by_environment", status: "active",
|
|
acquiredAt: new Date(currentLease.acquiredAt.getTime() - 1_000), metadata: currentLease.metadata }).returning();
|
|
const priorAuthority = { ...authority, writerRunId: priorRun.id,
|
|
location: { ...authority.location, leaseId: priorLease.id } };
|
|
const [priorWorkspace] = await db.update(executionWorkspaces).set({ metadata: { remoteWorkspace: priorAuthority } })
|
|
.where(eq(executionWorkspaces.id, workspaceId)).returning();
|
|
const priorSaved = await saveRemoteWorkspace({ db, workspace: priorWorkspace, runId: priorRun.id,
|
|
lease: priorLease as EnvironmentLease, target: { ...targets.get(f.lease.id)!, leaseId: priorLease.id },
|
|
assertOwned: async () => undefined });
|
|
const handoff = createReusableLeaseHandoff(priorLease, currentLease);
|
|
await db.update(environmentLeases).set({ status: "expired", cleanupStatus: "success", releasedAt: currentLease.acquiredAt,
|
|
metadata: { ...priorLease.metadata, reusableLeaseSuccessor: handoff } }).where(eq(environmentLeases.id, priorLease.id));
|
|
await db.update(environmentLeases).set({ metadata: { ...currentLease.metadata, reusableLeasePredecessor: handoff } })
|
|
.where(eq(environmentLeases.id, currentLease.id));
|
|
authority = { ...authority, epoch: 2, snapshotId: priorSaved.snapshotId };
|
|
await db.update(executionWorkspaces).set({ metadata: { remoteWorkspace: authority } })
|
|
.where(eq(executionWorkspaces.id, workspaceId));
|
|
expect(before.runtimeMode).toBe("legacy");
|
|
expect(before.runnerProfileJson?.nativeExecutionInput).toBeUndefined();
|
|
expect(await db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId))).toEqual([]);
|
|
}
|
|
if (runtimeMode === "legacy") await writeFile(path.join(f.remote, "pre-native-change.txt"), "current preparation wrote these files\n");
|
|
await db.update(heartbeatRuns).set({ status, runnerProfileJson: { ...before.runnerProfileJson,
|
|
...(status === "failed" && runtimeMode === "native" ? { nativeResourceDisposition: "destroy" } : {}),
|
|
remoteWorkspace: { schema: "paperclip.remote-workspace-run.v1", workspaceId, leaseId: f.lease.id,
|
|
providerLeaseId: f.providerLeaseId, remoteCwd, epoch: authority.epoch, state: "prepared", snapshotId: authority.snapshotId } },
|
|
resultJson: { ...before.resultJson, ...(status === "cancelled" ? {
|
|
nativeCancellation: { dispatchState: "acknowledged", requestedAt: new Date().toISOString() },
|
|
} : {}) },
|
|
}).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(executionWorkspaces).set({ metadata: { remoteWorkspace: { ...authority, epoch: authority.epoch + 1 } } })
|
|
.where(eq(executionWorkspaces.id, workspaceId));
|
|
await expect(recoverySources.inspectWorkspaceRecoverySource(db, f.input)).rejects.toMatchObject({ code: "source_binding_invalid" });
|
|
expect(f.runtime.resumeRunLease).not.toHaveBeenCalled();
|
|
await db.update(executionWorkspaces).set({ metadata: { remoteWorkspace: authority } })
|
|
.where(eq(executionWorkspaces.id, workspaceId));
|
|
if (runtimeMode === "legacy") {
|
|
// Remote file ownership exists before native runtime selection.
|
|
expect((await recoverySources.inspectWorkspaceRecoverySource(db, f.input)).sources).toHaveLength(1);
|
|
const failedUpload = vi.spyOn(getStorageService(), "putFile").mockRejectedValueOnce(new Error("Synthetic saved-copy outage"));
|
|
try {
|
|
if (uploadFailure === "initial") {
|
|
// Exercise the original setup boundary, not a failed final save that
|
|
// retries the same operation. The initial ledger reservation survives.
|
|
await db.delete(issueRecoveryActions).where(eq(issueRecoveryActions.id, f.actionId));
|
|
await db.update(heartbeatRuns).set({ status: "running", finishedAt: null, error: null, errorCode: null, resultJson: null })
|
|
.where(eq(heartbeatRuns.id, f.runId));
|
|
const [originalLease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id));
|
|
const activeMetadata = { ...originalLease.metadata };
|
|
delete activeMetadata.remoteExecutionTermination;
|
|
const [activeLease] = await db.update(environmentLeases).set({ status: "active", releasedAt: null, cleanupStatus: null,
|
|
metadata: activeMetadata }).where(eq(environmentLeases.id, f.lease.id)).returning();
|
|
const [preparing] = await db.select().from(executionWorkspaces).where(eq(executionWorkspaces.id, workspaceId));
|
|
await expect(saveRemoteWorkspace({ db, workspace: preparing, runId: f.runId, lease: activeLease as EnvironmentLease,
|
|
target: targets.get(f.lease.id)!, assertOwned: async () => undefined, boundary: "initial" }))
|
|
.rejects.toThrow("Synthetic saved-copy outage");
|
|
const preserved = await preserveRemoteWorkspacePreparationFailure(db, activeLease, { assertOwned: async () => undefined });
|
|
expect(preserved?.resultJson).toMatchObject({ workspaceRestoreFailure: "restore_failed",
|
|
workspaceRestoreRecovery: { leaseIds: [f.lease.id] } });
|
|
await terminalizeLegacyExecution({ db, run: preserved!, status: "failed", patch: {
|
|
errorCode: "setup_failed", error: "Synthetic saved-copy outage", finishedAt: new Date(),
|
|
resultJson: { executionRecovery: { kind: "bootstrap", providerWorkStarted: false } },
|
|
} });
|
|
const repairs = await db.select().from(issueRecoveryActions).where(and(eq(issueRecoveryActions.companyId, f.companyId),
|
|
sql`${issueRecoveryActions.evidence}->>'runId' = ${f.runId}`));
|
|
expect(repairs).toHaveLength(1);
|
|
f.actionId = repairs[0].id;
|
|
f.input.actionId = repairs[0].id;
|
|
expect(await db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId))).toEqual([]);
|
|
expect((await run(f.runId)).runnerProfileJson).not.toHaveProperty("nativeExecutionInput");
|
|
} else await recoverRetainedWorkspace(f.input);
|
|
expect(failedUpload).toHaveBeenCalledTimes(1);
|
|
} finally { failedUpload.mockRestore(); }
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).not.toBe("complete");
|
|
const [unsaved] = await db.select().from(executionWorkspaces).where(eq(executionWorkspaces.id, workspaceId));
|
|
expect(readRemoteWorkspaceAuthority(unsaved.metadata)).toMatchObject({ dirty: true, writerRunId: f.runId,
|
|
epoch: authority.epoch, snapshotId: authority.snapshotId });
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
if (uploadFailure === "final") await makeRetryDue(f.actionId);
|
|
}
|
|
await recoverRetainedWorkspace(f.input);
|
|
const recoveredAction = await action(f.actionId);
|
|
expect(readSystemRecovery(recoveredAction.evidence)?.state).toBe("complete");
|
|
const [workspace] = await db.select().from(executionWorkspaces).where(eq(executionWorkspaces.id, workspaceId));
|
|
const saved = readRemoteWorkspaceAuthority(workspace.metadata);
|
|
expect(saved).toMatchObject({ dirty: false, writerRunId: null });
|
|
expect(saved?.snapshotId).toBeTruthy();
|
|
const snapshots = await db.select().from(executionWorkspaceSnapshots).where(eq(executionWorkspaceSnapshots.executionWorkspaceId, workspaceId));
|
|
expect(snapshots).toHaveLength(runtimeMode === "legacy" ? 2 : 1);
|
|
expect(snapshots.find(snapshot => snapshot.id === saved!.snapshotId))
|
|
.toMatchObject({ status: "committed", sourceRunId: f.runId, generation: runtimeMode === "legacy" ? 2 : 1 });
|
|
const extracted = await createSavedWorkspaceLoader(db)({ companyId: f.companyId, workspaceId, snapshotId: saved!.snapshotId! });
|
|
try {
|
|
expect(await readFile(path.join(extracted.root, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
if (runtimeMode === "legacy") expect(await readFile(path.join(extracted.root, "pre-native-change.txt"), "utf8"))
|
|
.toBe("current preparation wrote these files\n");
|
|
}
|
|
finally { await extracted.cleanup(); }
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(await run(f.runId)).toMatchObject({ status, errorCode: uploadFailure === "initial" ? "setup_failed" : "execution_interrupted",
|
|
runnerProfileJson: { remoteWorkspace: { state: "saved", snapshotId: saved!.snapshotId } } });
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.runId, f.runId))).toHaveLength(0);
|
|
expect(await db.select().from(costEvents).where(eq(costEvents.companyId, f.companyId))).toHaveLength(0);
|
|
expect(await db.select().from(workspaceOperations).where(and(eq(workspaceOperations.heartbeatRunId, f.runId),
|
|
eq(workspaceOperations.phase, "workspace_recovery_snapshot")))).toHaveLength(0);
|
|
expect(f.runtime.resumeRunLease).toHaveBeenCalledTimes(uploadFailure === "final" ? 2 : 1);
|
|
expect(f.runtime.retryPendingSandboxTeardown).toHaveBeenCalledTimes(runtimeMode === "legacy" ? 2 : 1);
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
await scheduleRecoveredWorkspaceCleanup(db, await run(f.runId));
|
|
const [settledLease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id));
|
|
if (runtimeMode === "legacy") {
|
|
expect(settledLease).toMatchObject({ status: "released",
|
|
leasePolicy: uploadFailure === "initial" ? "retain_on_failure" : "reuse_by_execution_workspace" });
|
|
expect((await run(f.runId)).runtimeMode).toBe("legacy");
|
|
expect(await db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId))).toEqual([]);
|
|
if (uploadFailure === "final") expect(snapshots.find(snapshot => snapshot.id === authority.snapshotId)).toMatchObject({ status: "superseded" });
|
|
else expect(snapshots.find(snapshot => snapshot.generation === 1)).toMatchObject({ status: "pending" });
|
|
// Explicit Retry may admit the preserved files. Saturate agent capacity
|
|
// so this tests admission without dispatching a new agent turn.
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: before.agentId, status: "running" });
|
|
const retry = await requestTaskRetry({ db, companyId: f.companyId, issueId: f.issueId, actorId: "local-board",
|
|
actorSource: "local_implicit", observedRunId: f.runId, idempotencyKey: randomUUID() });
|
|
await heartbeatService(db).resumeTaskRetry(f.companyId, retry.requestId!);
|
|
const [successor] = await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")));
|
|
expect(successor).toBeDefined();
|
|
expect(await readFreshWorkspaceReadiness(db, successor))
|
|
.toMatchObject({ mode: "resume_preserved_workspace", preservationReceiptId: expect.any(String) });
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.runId, successor.id))).toEqual([]);
|
|
expect(f.runtime.resumeRunLease).toHaveBeenCalledTimes(uploadFailure === "final" ? 2 : 1);
|
|
return;
|
|
}
|
|
if (status === "failed") {
|
|
expect(settledLease).toMatchObject({ status: "pending_cleanup", leasePolicy: "ephemeral",
|
|
metadata: { workspaceRecoveryCleanup: { runId: f.runId, snapshotId: saved!.snapshotId, epoch: authority.epoch } } });
|
|
const source = await run(f.runId);
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, status: "running" });
|
|
const retry = await requestTaskRetry({ db, companyId: f.companyId, issueId: f.issueId, actorId: "local-board",
|
|
actorSource: "local_implicit", observedRunId: source.id, idempotencyKey: randomUUID() });
|
|
await heartbeatService(db).resumeTaskRetry(f.companyId, retry.requestId!);
|
|
expect((await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, retry.requestId!)))[0])
|
|
.toMatchObject({ status: "deferred_issue_execution", payload: { executionWait: { reason: "remote_cleanup" } } });
|
|
await environmentService(db).releaseLease(settledLease.id, "released", { cleanupStatus: "success",
|
|
remoteExecutionTermination: remoteTerminationReceipt(settledLease, { providerLeaseId: settledLease.providerLeaseId, state: "destroyed" }) });
|
|
await heartbeatService(db).resumeTaskRetry(f.companyId, retry.requestId!);
|
|
const [successor] = await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")));
|
|
expect(successor).toBeDefined();
|
|
const readiness = await readFreshWorkspaceReadiness(db, successor);
|
|
expect(readiness).toMatchObject({ mode: "resume_preserved_workspace", preservationReceiptId: expect.any(String) });
|
|
} else {
|
|
expect(settledLease).toMatchObject({ status: "released", leasePolicy: "reuse_by_execution_workspace" });
|
|
expect(settledLease.metadata?.workspaceRecoveryCleanup).toBeUndefined();
|
|
}
|
|
}, 30_000);
|
|
|
|
async function warmHandoffSource(hops = 1, beforeStop?: (lease: typeof environmentLeases.$inferSelect) => Promise<void>) {
|
|
const prior = await seed("native", true, "keep_running", true), source = await run(prior.runId);
|
|
let previousTarget = targets.get(prior.lease.id)!, previousLeaseId = prior.lease.id;
|
|
let previousRunId: string = prior.runId;
|
|
let lease = prior.lease, current = source;
|
|
const leaseIds = [previousLeaseId];
|
|
for (let hop = 0; hop < hops; hop++) {
|
|
expect(await resumeNativeWorkspaceSync({ db, runId: previousRunId, target: previousTarget })).toBe(true);
|
|
await db.update(heartbeatRuns).set({ status: "succeeded", errorCode: null, resultJson: null }).where(eq(heartbeatRuns.id, previousRunId));
|
|
await db.update(nativeRunFinalizations).set({ phase: "committed" }).where(eq(nativeRunFinalizations.runId, previousRunId));
|
|
await db.delete(issueRecoveryActions).where(eq(issueRecoveryActions.id, prior.actionId));
|
|
await environmentService(db).releaseLease(previousLeaseId, "released", { cleanupStatus: "success" });
|
|
const [released] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, previousLeaseId));
|
|
[current] = await db.insert(heartbeatRuns).values({ companyId: prior.companyId, agentId: source.agentId,
|
|
issueId: prior.issueId, nativeIssueId: prior.issueId, runtimeMode: "native", status: "cancelled", finishedAt: new Date(),
|
|
contextSnapshot: { issueId: prior.issueId } }).returning();
|
|
await db.insert(nativeRunFinalizations).values({ companyId: prior.companyId, issueId: prior.issueId, runId: current.id, phase: "terminal_failure" });
|
|
const acquired = await environmentService(db).acquireLease({ companyId: prior.companyId, environmentId: prior.lease.environmentId!,
|
|
issueId: prior.issueId, heartbeatRunId: current.id, provider: "daytona", providerLeaseId: prior.providerLeaseId,
|
|
leasePolicy: "reuse_by_environment", replacesReusableLeaseId: previousLeaseId,
|
|
metadata: { ...released.metadata, sandboxLeaseAcquisition: { outcome: "resumed" } } });
|
|
[lease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, acquired.id));
|
|
const target = { ...previousTarget, leaseId: lease.id, sandboxLeaseAcquisition: { outcome: "resumed" as const, providerLeaseId: prior.providerLeaseId } };
|
|
targets.set(lease.id, target);
|
|
expect(await prepareNativeWorkspaceSync({ db, companyId: prior.companyId, workspaceId: prior.issueId,
|
|
workspaceLocalDir: prior.host, runId: current.id, lease: lease as EnvironmentLease, target, resourceDisposition: "keep_running" })).not.toBeNull();
|
|
leaseIds.push(lease.id); previousRunId = current.id; previousLeaseId = lease.id; previousTarget = target;
|
|
}
|
|
[lease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, lease.id));
|
|
await beforeStop?.(lease);
|
|
await writeFile(path.join(prior.remote, "after-warm-stop.txt"), "stopped turn work\n");
|
|
await recordRequiredWorkspaceRestoreFailure(db, await run(current.id), { workspaceRestoreFailure: "restore_failed" }, lease);
|
|
const [action] = await db.select().from(issueRecoveryActions).where(and(eq(issueRecoveryActions.companyId, prior.companyId),
|
|
eq(issueRecoveryActions.fingerprint, `native-workspace:${current.id}`)));
|
|
const [retained] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, lease.id));
|
|
const stopping = readStopOnlyCleanup(retained) ? retained : await prepareSandboxStopAndRetain(db, retained);
|
|
expect(stopping).not.toBeNull();
|
|
expect(await settleStopOnlyCleanup(db, stopping!, { attemptId: String(stopping!.metadata!.pendingCleanupAttemptId),
|
|
receipt: { providerLeaseId: prior.providerLeaseId, state: "stopped" } })).not.toBeNull();
|
|
return { ...prior, priorRunId: prior.runId, priorLeaseId: prior.lease.id, leaseIds, runId: current.id, lease, actionId: action.id,
|
|
input: { ...prior.input, runId: current.id, actionId: action.id } };
|
|
}
|
|
|
|
describe("verified warm allocation handoffs", () => {
|
|
let original: { db: typeof db; root: string; connectionString: string; home: string | undefined };
|
|
let isolated: Awaited<ReturnType<typeof startEmbeddedPostgresTestDatabase>> | undefined;
|
|
let isolatedDb: typeof db | undefined;
|
|
let isolatedRoot: string | undefined;
|
|
beforeAll(async () => {
|
|
original = { db, root, connectionString, home: process.env.PAPERCLIP_HOME };
|
|
const external = resolveExternalTestDatabaseUrl("PAPERCLIP_WORKSPACE_RECOVERY_HANDOFF_TEST_DATABASE_URL");
|
|
if (external) {
|
|
await applyPendingMigrations(external);
|
|
isolated = { connectionString: external, cleanup: async () => {} };
|
|
} else isolated = await startEmbeddedPostgresTestDatabase("workspace-warm-handoff-");
|
|
isolatedDb = createDb(isolated.connectionString);
|
|
if (external) await isolatedDb.execute(sql`TRUNCATE TABLE companies CASCADE`);
|
|
isolatedRoot = await mkdtemp(path.join(os.tmpdir(), "workspace-warm-handoff-files-"));
|
|
db = isolatedDb; root = isolatedRoot; connectionString = isolated.connectionString;
|
|
process.env.PAPERCLIP_HOME = path.join(root, "home");
|
|
}, 30_000);
|
|
afterAll(async () => {
|
|
await isolatedDb?.$client.end(); await isolated?.cleanup();
|
|
if (isolatedRoot) await rm(isolatedRoot, { recursive: true, force: true });
|
|
if (original) {
|
|
db = original.db; root = original.root; connectionString = original.connectionString;
|
|
if (original.home === undefined) delete process.env.PAPERCLIP_HOME; else process.env.PAPERCLIP_HOME = original.home;
|
|
}
|
|
});
|
|
it.each([1, 2])("recovers a stopped native source after %s verified warm predecessor handoffs", async hops => {
|
|
const f = await warmHandoffSource(hops);
|
|
expect((await run(f.priorRunId)).runnerProfileJson?.nativeWorkspaceSync).toMatchObject({ state: "finalized", leaseId: f.priorLeaseId });
|
|
const [previous] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.priorLeaseId));
|
|
expect(previous).toMatchObject({ status: "expired", cleanupStatus: "success", providerLeaseId: f.providerLeaseId });
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
expect(await readFile(path.join(f.host, "after-warm-stop.txt"), "utf8")).toBe("stopped turn work\n");
|
|
const source = await run(f.runId), freshProviderLeaseId = randomUUID();
|
|
const [fresh] = await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId,
|
|
issueId: f.issueId, nativeIssueId: f.issueId, runtimeMode: "native", status: "succeeded", finishedAt: new Date() }).returning();
|
|
const remoteCwd = path.join(root, fresh.id, "remote");
|
|
const freshLease = await environmentService(db).acquireLease({ companyId: f.companyId, environmentId: f.lease.environmentId!,
|
|
issueId: f.issueId, heartbeatRunId: fresh.id, provider: "daytona", providerLeaseId: freshProviderLeaseId,
|
|
metadata: { ...f.lease.metadata, remoteCwd } });
|
|
const freshTarget = { ...targets.get(f.lease.id)!, leaseId: freshLease.id, remoteCwd,
|
|
sandboxLeaseAcquisition: { outcome: "created" as const, providerLeaseId: freshProviderLeaseId } };
|
|
expect(await prepareNativeWorkspaceSync({ db, companyId: f.companyId, workspaceId: f.issueId,
|
|
workspaceLocalDir: f.host, runId: fresh.id, lease: freshLease, target: freshTarget, resourceDisposition: "keep_running" })).not.toBeNull();
|
|
expect(await resumeNativeWorkspaceSync({ db, runId: fresh.id, target: freshTarget })).toBe(true);
|
|
expect(await inspectWorkspaceSource(db, { companyId: f.companyId, issueId: f.issueId, runId: f.runId, actionId: f.actionId }))
|
|
.toMatchObject({ status: "verified" });
|
|
}, 30_000);
|
|
|
|
it.each(["active", "active_run", "active_control", "foreign_company", "environment", "workspace", "task", "agent", "unfinalized", "owner", "missing_proof", "conflicting_proof", "dangling_proof", "extra_history"])(
|
|
"keeps ambiguous warm source history fenced: %s", async fault => {
|
|
const f = await warmHandoffSource();
|
|
const [previous] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.priorLeaseId));
|
|
if (fault === "active") await db.update(environmentLeases).set({ status: "active" }).where(eq(environmentLeases.id, previous.id));
|
|
if (fault === "active_run") await db.update(heartbeatRuns).set({ status: "running" }).where(eq(heartbeatRuns.id, f.priorRunId));
|
|
if (fault === "active_control") adapterExecutionControls.set(f.priorRunId, createAdapterExecutionControl());
|
|
if (fault === "foreign_company") {
|
|
const [foreign] = await db.insert(companies).values({ name: "Foreign owner", issuePrefix: `F${randomUUID().slice(0, 6)}` }).returning();
|
|
await db.update(environmentLeases).set({ companyId: foreign.id }).where(eq(environmentLeases.id, previous.id));
|
|
}
|
|
if (fault === "environment") {
|
|
const [other] = await db.insert(environments).values({ name: randomUUID(), driver: "sandbox" }).returning();
|
|
await db.update(environmentLeases).set({ environmentId: other.id }).where(eq(environmentLeases.id, previous.id));
|
|
}
|
|
if (fault === "workspace") {
|
|
const [other] = await db.insert(executionWorkspaces).values({ companyId: f.companyId, mode: "isolated_workspace", strategyType: "test", name: "Other workspace" }).returning();
|
|
await db.update(environmentLeases).set({ executionWorkspaceId: other.id }).where(eq(environmentLeases.id, previous.id));
|
|
}
|
|
if (fault === "task") {
|
|
const [other] = await db.insert(issues).values({ companyId: f.companyId, title: "Other task" }).returning();
|
|
await db.update(environmentLeases).set({ issueId: other.id }).where(eq(environmentLeases.id, previous.id));
|
|
}
|
|
if (fault === "agent") {
|
|
const [other] = await db.insert(agents).values({ companyId: f.companyId, name: "Other agent", adapterType: "paperclip_runner" }).returning();
|
|
await db.update(heartbeatRuns).set({ agentId: other.id }).where(eq(heartbeatRuns.id, f.priorRunId));
|
|
}
|
|
if (fault === "unfinalized") {
|
|
const prior = await run(f.priorRunId), reference = prior.runnerProfileJson!.nativeWorkspaceSync as Record<string, unknown>;
|
|
await db.update(heartbeatRuns).set({ runnerProfileJson: { ...prior.runnerProfileJson, nativeWorkspaceSync: { ...reference, state: "prepared", finalHostSha256: null } } }).where(eq(heartbeatRuns.id, prior.id));
|
|
}
|
|
if (fault === "owner") await db.update(nativeRunFinalizations).set({ leaseOwner: randomUUID() }).where(eq(nativeRunFinalizations.runId, f.priorRunId));
|
|
if (fault === "missing_proof") {
|
|
const metadata = { ...previous.metadata }; delete metadata.reusableLeaseSuccessor;
|
|
await db.update(environmentLeases).set({ metadata }).where(eq(environmentLeases.id, previous.id));
|
|
}
|
|
if (fault === "conflicting_proof") await db.update(environmentLeases).set({ metadata: { ...previous.metadata,
|
|
reusableLeaseSuccessor: { ...(previous.metadata!.reusableLeaseSuccessor as Record<string, unknown>), successorRunId: randomUUID() } } }).where(eq(environmentLeases.id, previous.id));
|
|
if (fault === "dangling_proof") await db.update(environmentLeases).set({ metadata: { ...previous.metadata, reusableLeasePredecessor: {} } }).where(eq(environmentLeases.id, previous.id));
|
|
if (fault === "extra_history") await db.insert(environmentLeases).values({ companyId: f.companyId, environmentId: previous.environmentId,
|
|
heartbeatRunId: f.priorRunId, issueId: f.issueId, provider: "daytona", providerLeaseId: f.providerLeaseId,
|
|
status: "expired", leasePolicy: "reuse_by_environment", cleanupStatus: "success", metadata: previous.metadata });
|
|
try {
|
|
expect(await inspectWorkspaceSource(db, { companyId: f.companyId, issueId: f.issueId, runId: f.runId, actionId: f.actionId }))
|
|
.toMatchObject({ status: "blocked", failureCode: "allocation_ownership_ambiguous" });
|
|
await expect(readFile(path.join(f.host, "after-warm-stop.txt"))).rejects.toMatchObject({ code: "ENOENT" });
|
|
} finally { adapterExecutionControls.get(f.priorRunId)?.finish(); adapterExecutionControls.delete(f.priorRunId); }
|
|
}, 30_000);
|
|
|
|
it("keeps server handoff receipts across metadata updates and same-run reacquisition", async () => {
|
|
const f = await warmHandoffSource(1, async lease => {
|
|
const proof = lease.metadata!.reusableLeasePredecessor;
|
|
expect(proof).toMatchObject({ schema: "paperclip.reusable-lease-handoff.v1", successorLeaseId: lease.id });
|
|
const forged = { schema: "forged-provider-metadata" };
|
|
const updated = await environmentService(db).updateLeaseMetadata(lease.id, { ...lease.metadata,
|
|
reusableLeasePredecessor: forged, reusableLeaseSuccessor: forged, marker: "updated" });
|
|
expect(updated!.metadata?.reusableLeasePredecessor).toEqual(proof);
|
|
expect(updated!.metadata?.reusableLeaseSuccessor).toBeUndefined();
|
|
await environmentService(db).releaseLease(lease.id, "released", { cleanupStatus: "success" });
|
|
const reacquired = await environmentService(db).acquireLease({ companyId: lease.companyId, environmentId: lease.environmentId!,
|
|
issueId: lease.issueId, heartbeatRunId: lease.heartbeatRunId, provider: lease.provider, providerLeaseId: lease.providerLeaseId,
|
|
leasePolicy: "reuse_by_environment", reusesReusableLeaseId: lease.id,
|
|
metadata: { ...updated!.metadata, reusableLeasePredecessor: forged, reusableLeaseSuccessor: forged } });
|
|
expect(reacquired.id).toBe(lease.id);
|
|
expect(reacquired.metadata?.reusableLeasePredecessor).toEqual(proof);
|
|
expect(reacquired.metadata?.reusableLeaseSuccessor).toBeUndefined();
|
|
});
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
}, 30_000);
|
|
|
|
it("does not manufacture handoff authority from injected fresh or orphan lease metadata", async () => {
|
|
const f = await warmHandoffSource(), source = await run(f.runId);
|
|
const forged = { schema: "paperclip.reusable-lease-handoff.v1", successorLeaseId: f.lease.id };
|
|
const values = { companyId: f.companyId, environmentId: f.lease.environmentId!, issueId: f.issueId,
|
|
heartbeatRunId: source.id, provider: "daytona", providerLeaseId: randomUUID(),
|
|
metadata: { agentId: source.agentId, reusableLeasePredecessor: forged, reusableLeaseSuccessor: forged } };
|
|
const fresh = await environmentService(db).acquireLease(values);
|
|
const orphan = await environmentService(db).insertPendingCleanupLease({ ...values, providerLeaseId: randomUUID(), failureReason: "test" });
|
|
for (const lease of [fresh, orphan]) {
|
|
expect(lease.metadata?.reusableLeasePredecessor).toBeUndefined();
|
|
expect(lease.metadata?.reusableLeaseSuccessor).toBeUndefined();
|
|
}
|
|
}, 30_000);
|
|
|
|
it("rolls back both handoff links and keeps provider mismatches or retained sources fenced", async () => {
|
|
const f = await warmHandoffSource(), source = await run(f.runId), service = environmentService(db);
|
|
const values = { companyId: f.companyId, environmentId: f.lease.environmentId!, issueId: f.issueId,
|
|
heartbeatRunId: f.priorRunId, provider: "daytona", providerLeaseId: randomUUID(),
|
|
leasePolicy: "reuse_by_environment" as const, metadata: { agentId: source.agentId } };
|
|
const prior = await service.acquireLease(values);
|
|
await service.releaseLease(prior.id, "released", { cleanupStatus: "success" });
|
|
const [before] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, prior.id));
|
|
const rollback = new Error("rollback_handoff");
|
|
await expect(db.transaction(async tx => {
|
|
const acquired = await environmentService(tx as unknown as typeof db).acquireLease({ ...values, heartbeatRunId: f.runId, replacesReusableLeaseId: prior.id });
|
|
expect(acquired.metadata?.reusableLeasePredecessor).toMatchObject({ previousLeaseId: prior.id, successorLeaseId: acquired.id });
|
|
throw rollback;
|
|
})).rejects.toBe(rollback);
|
|
expect((await db.select().from(environmentLeases).where(eq(environmentLeases.id, prior.id)))[0]).toEqual(before);
|
|
expect(await db.select().from(environmentLeases).where(eq(environmentLeases.providerLeaseId, values.providerLeaseId))).toHaveLength(1);
|
|
await expect(service.acquireLease({ ...values, heartbeatRunId: f.runId, provider: "foreign-provider", replacesReusableLeaseId: prior.id })).rejects.toThrow("ownership changed");
|
|
await expect(service.acquireLease({ ...values, providerLeaseId: f.providerLeaseId, replacesReusableLeaseId: f.lease.id })).rejects.toThrow("ownership changed");
|
|
const [later] = await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, issueId: f.issueId,
|
|
nativeIssueId: f.issueId, runtimeMode: "native", status: "succeeded", finishedAt: new Date() }).returning();
|
|
await service.acquireLease({ ...values, heartbeatRunId: later.id, providerLeaseId: f.providerLeaseId });
|
|
expect(await inspectWorkspaceSource(db, { companyId: f.companyId, issueId: f.issueId, runId: f.runId, actionId: f.actionId }))
|
|
.toMatchObject({ status: "blocked", failureCode: "allocation_ownership_ambiguous" });
|
|
}, 30_000);
|
|
|
|
});
|
|
|
|
it("unblocks a pending workspace selection only after a superseded retry's real files are preserved", async () => {
|
|
const f = await seed();
|
|
await db.delete(issueRecoveryActions).where(eq(issueRecoveryActions.id, f.actionId));
|
|
const source = await run(f.runId);
|
|
await db.update(heartbeatRuns).set({ resultJson: null, nativePhase: "retryable_failure",
|
|
startedAt: new Date(1), finishedAt: new Date(2) }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(nativeRunFinalizations).set({ phase: "retryable_failure", attempt: 1,
|
|
recoveryState: "resuming_session", nextAttemptAt: new Date(0) }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
await db.update(environmentLeases).set({ status: "active", leasePolicy: "ephemeral",
|
|
releasedAt: null, cleanupStatus: null, metadata: f.lease.metadata }).where(eq(environmentLeases.id, f.lease.id));
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId,
|
|
issueId: f.issueId, nativeIssueId: f.issueId, runtimeMode: "native", status: "succeeded", nativePhase: "committed",
|
|
startedAt: new Date(3), finishedAt: new Date(4), contextSnapshot: { previousRunId: f.runId, taskRetryRequestId: randomUUID() } });
|
|
await db.insert(executionWorkspaces).values({ id: f.issueId, companyId: f.companyId, sourceIssueId: f.issueId,
|
|
name: "Shared files", mode: "shared_workspace", strategyType: "project_primary", cwd: f.host });
|
|
const actor = { type: "board" as const, source: "local_implicit" as const, isInstanceAdmin: true };
|
|
const workspaces = executionWorkspaceService(db);
|
|
await workspaces.selectTaskWorkspace({ companyId: f.companyId, issueId: f.issueId, actor,
|
|
selection: { kind: "existing", workspaceId: f.issueId }, expectedBindingRevision: 0, requestKey: randomUUID() });
|
|
const admission = { companyId: f.companyId, issueId: f.issueId, actor, runId: randomUUID() };
|
|
await expect(workspaces.applyPendingTaskWorkspaceSelection(admission)).rejects.toThrow("Previous task run still owns workspace recovery");
|
|
expect(await claimNativeSessionResumptions({ db, runnerInstanceId: "recovery", runIds: [f.runId] })).toEqual([]);
|
|
await expect(workspaces.applyPendingTaskWorkspaceSelection(admission)).rejects.toThrow("Previous task run still owns workspace recovery");
|
|
const [preservation] = await db.select().from(issueRecoveryActions).where(eq(issueRecoveryActions.sourceIssueId, f.issueId));
|
|
await recoverRetainedWorkspace({ ...f.input, actionId: preservation.id });
|
|
expect(readSystemRecovery((await action(preservation.id)).evidence)?.state).toBe("complete");
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(await workspaces.applyPendingTaskWorkspaceSelection(admission)).toBe(true);
|
|
const [task] = await db.select().from(issues).where(eq(issues.id, f.issueId));
|
|
expect(task).toMatchObject({ status: "blocked", assigneeAgentId: source.agentId, executionWorkspaceId: f.issueId,
|
|
workspacePendingSelection: null, executionWorkspaceSettings: { mode: "shared_workspace" } });
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(2);
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.companyId, f.companyId))).toHaveLength(0);
|
|
}, 30_000);
|
|
|
|
it.each(["HTTP Retry", "new comment", "automatic cleanup"] as const)("repairs an unregistered historical deadline failure through %s without admitting work before verified stop", async entry => {
|
|
const f = await seed("native", true, "keep_running");
|
|
const source = await run(f.runId);
|
|
await db.update(heartbeatRuns).set({ resultJson: null, errorCode: "execution_finalization_deadline_exceeded" })
|
|
.where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(environmentLeases).set({ status: "active", leasePolicy: "reuse_by_environment",
|
|
releasedAt: null, cleanupStatus: null, metadata: f.lease.metadata }).where(eq(environmentLeases.id, f.lease.id));
|
|
await db.update(issueRecoveryActions).set({ status: "resolved", outcome: "blocked",
|
|
cause: "execution_finalization_deadline_exceeded", evidence: { runId: f.runId, providerOwnership: "unverified",
|
|
automaticRecovery: { replay: "blocked", actionOutcome: "unknown" } } }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
// Leave admission observable without substituting a fake model turn.
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, status: "running" });
|
|
let releaseStop!: () => void;
|
|
const stopped = new Promise<void>(resolve => { releaseStop = resolve; });
|
|
f.runtime.retryPendingSandboxTeardown.mockImplementation(async () => {
|
|
await stopped;
|
|
return { providerLeaseId: f.providerLeaseId, state: "stopped" };
|
|
});
|
|
const errors = vi.fn(), heartbeat = heartbeatService(db, { environmentRuntime: f.environmentRuntime });
|
|
const coordinator = createDeliveryWorkCoordinator({ owner: db, canRun: () => true, canReconcile: () => true, onError: errors });
|
|
const worker = registerWorkspaceRecoveryWork(db, coordinator, f.environmentRuntime, heartbeat.resumeRemoteStopComments);
|
|
const successors = () => db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")));
|
|
try {
|
|
await worker.ready;
|
|
const app = express();
|
|
app.use(express.json());
|
|
app.use((req, _res, next) => { req.actor = { type: "board", source: "local_implicit", userId: "board" }; next(); });
|
|
app.use("/api", issueRoutes(db, {} as Parameters<typeof issueRoutes>[1]));
|
|
app.use(errorHandler);
|
|
let requestId: string | undefined;
|
|
let commentId: string | undefined;
|
|
if (entry === "HTTP Retry") {
|
|
const body = { idempotencyKey: randomUUID(), observedRunId: f.runId };
|
|
const retry = await request(app).post(`/api/issues/${f.issueId}/retry`).send(body).expect(202);
|
|
requestId = retry.body.requestId;
|
|
expect((await request(app).post(`/api/issues/${f.issueId}/retry`).send(body).expect(202)).body.requestId).toBe(requestId);
|
|
} else if (entry === "new comment") {
|
|
const response = await request(app).post(`/api/issues/${f.issueId}/comments`)
|
|
.send({ body: "Is this fixed?" }).expect(201);
|
|
commentId = response.body.id;
|
|
expect(await successors()).toHaveLength(0);
|
|
} else {
|
|
const [comment] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "board", body: "Is this fixed?" }).returning();
|
|
commentId = comment.id;
|
|
const wake = { source: "automation" as const, triggerDetail: "system" as const,
|
|
reason: "issue_reopened_via_comment", requestedByActorType: "user" as const, requestedByActorId: "board",
|
|
payload: { issueId: f.issueId, commentId: comment.id, mutation: "comment" },
|
|
contextSnapshot: { source: "issue.comment.reopen", issueId: f.issueId, wakeCommentId: comment.id },
|
|
};
|
|
await db.insert(agentWakeupRequests).values({ companyId: f.companyId, agentId: source.agentId,
|
|
source: wake.source, triggerDetail: wake.triggerDetail, reason: wake.reason,
|
|
requestedByActorType: "user", requestedByActorId: "board", status: "deferred_issue_execution",
|
|
payload: { ...wake.payload, _paperclipWakeContext: { ...wake.contextSnapshot, wakeReason: wake.reason,
|
|
wakeSource: wake.source, wakeTriggerDetail: wake.triggerDetail, wakeCommentIds: [comment.id] } },
|
|
});
|
|
expect(await successors()).toHaveLength(0);
|
|
if (entry === "automatic cleanup") await heartbeat.sweepOrphanedActiveLeases({ backoffMs: 0 });
|
|
}
|
|
await expect.poll(() => f.runtime.retryPendingSandboxTeardown.mock.calls.length, { timeout: 20_000 }).toBe(1);
|
|
if (requestId) await heartbeat.resumeTaskRetry(f.companyId, requestId);
|
|
else await heartbeat.resumeQueuedRuns();
|
|
expect(await successors()).toHaveLength(0);
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect((await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id)))[0].metadata)
|
|
.not.toHaveProperty("remoteExecutionTermination");
|
|
await expect(readFile(path.join(f.host, "saved.txt"), "utf8")).rejects.toMatchObject({ code: "ENOENT" });
|
|
releaseStop();
|
|
await expect.poll(async () => (await successors()).length, { timeout: 20_000 }).toBe(1);
|
|
const [successor] = await successors();
|
|
if (commentId) expect(successor.contextSnapshot?.wakeCommentIds).toContain(commentId);
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
if (requestId) expect(await readFreshWorkspaceReadiness(db, successor)).toMatchObject({ mode: "resume_preserved_workspace" });
|
|
else expect(successor.contextSnapshot?.previousRunId).toBe(f.runId);
|
|
await Promise.all([requestId ? heartbeat.resumeTaskRetry(f.companyId, requestId) : heartbeat.resumeQueuedRuns(),
|
|
heartbeat.resumeRemoteStopComments(await run(f.runId))]);
|
|
expect(await successors()).toHaveLength(1);
|
|
expect(f.runtime.resumeRunLease).toHaveBeenCalledOnce();
|
|
expect((await run(f.runId)).status).toBe("failed");
|
|
const reloaded = await request(app).get(`/api/issues/${f.issueId}/recovery-actions`).expect(200);
|
|
expect(reloaded.body.active).toBeNull();
|
|
const records = await db.select().from(issueRecoveryActions).where(eq(issueRecoveryActions.sourceIssueId, f.issueId));
|
|
expect(records.some(record => readSystemRecovery(record.evidence)?.state === "complete")).toBe(true);
|
|
expect(errors).not.toHaveBeenCalled();
|
|
} finally { releaseStop(); await coordinator.stop(); await worker.stop(); }
|
|
}, 40_000);
|
|
|
|
it("wakes physical preservation from HTTP Retry before passive history without replaying the provider", async () => {
|
|
for (let index = 0; index < 5; index++) {
|
|
const old = await seed();
|
|
await recoverRetainedWorkspace(old.input);
|
|
await db.update(issueRecoveryActions).set({
|
|
id: `00000000-000${index}-4000-8000-${randomUUID().slice(-12)}`,
|
|
}).where(eq(issueRecoveryActions.id, old.actionId));
|
|
}
|
|
const f = await seed();
|
|
await adoptSystemRecovery(db, f);
|
|
const before = await action(f.actionId);
|
|
const exhausted = failSystemRecovery(readSystemRecovery(before.evidence)!,
|
|
new SystemRecoveryFailure("source_binding_invalid", "service_required"));
|
|
await db.update(issueRecoveryActions).set({ cause: "native_workspace_sync_out_retry_exhausted",
|
|
evidence: { ...before.evidence, systemRecovery: exhausted } })
|
|
.where(eq(issueRecoveryActions.id, f.actionId));
|
|
const errors = vi.fn();
|
|
const competingCleanup = await retainedWorkspaceCandidates(db, undefined, false, true);
|
|
expect(competingCleanup).toHaveLength(5);
|
|
expect(competingCleanup.every(row => row.action.id.startsWith("00000000"))).toBe(true);
|
|
const coordinator = createDeliveryWorkCoordinator({ owner: db, canRun: () => true, canReconcile: () => true, onError: errors });
|
|
const recovered = vi.fn(async () => {});
|
|
const worker = registerWorkspaceRecoveryWork(db, coordinator, f.environmentRuntime, recovered);
|
|
try {
|
|
await worker.ready;
|
|
await db.update(agents).set({ status: "paused" }).where(eq(agents.companyId, f.companyId));
|
|
const app = express();
|
|
app.use(express.json());
|
|
app.use((req, _res, next) => { req.actor = { type: "board", source: "local_implicit", userId: "board" }; next(); });
|
|
app.use("/api", issueRoutes(db, {} as Parameters<typeof issueRoutes>[1]));
|
|
app.use(errorHandler);
|
|
const body = { idempotencyKey: randomUUID(), observedRunId: f.runId };
|
|
const retry = await request(app).post(`/api/issues/${f.issueId}/retry`).send(body).expect(202);
|
|
const repeated = await request(app).post(`/api/issues/${f.issueId}/retry`).send(body).expect(202);
|
|
const coalesced = await request(app).post(`/api/issues/${f.issueId}/retry`)
|
|
.send({ ...body, idempotencyKey: randomUUID() }).expect(202);
|
|
expect(repeated.body.requestId).toBe(retry.body.requestId);
|
|
expect(coalesced.body.requestId).toBe(retry.body.requestId);
|
|
await expect.poll(async () => readSystemRecovery((await action(f.actionId)).evidence)?.state,
|
|
{ timeout: 20_000 }).toBe("complete");
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1);
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.companyId, f.companyId))).toHaveLength(0);
|
|
const reloaded = await request(app).get(`/api/issues/${f.issueId}/recovery-actions`).expect(200);
|
|
expect(JSON.stringify(reloaded.body)).toContain('"state":"complete"');
|
|
expect(errors).not.toHaveBeenCalled();
|
|
} finally { await coordinator.stop(); await worker.stop(); }
|
|
}, 30_000);
|
|
|
|
it("reconstructs a resolved files-only preservation obligation at startup without replay", async () => {
|
|
const f = await seed("legacy");
|
|
await adoptSystemRecovery(db, f);
|
|
const before = await action(f.actionId);
|
|
const automaticRecovery = { policy: "preserve_without_replay_v1", runId: f.runId, replay: "blocked" };
|
|
await db.update(issueRecoveryActions).set({ status: "resolved", outcome: "blocked",
|
|
evidence: { ...before.evidence, automaticRecovery } }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
const errors = vi.fn();
|
|
const coordinator = createDeliveryWorkCoordinator({ owner: db, canRun: () => true, canReconcile: () => true, onError: errors });
|
|
const worker = registerWorkspaceRecoveryWork(db, coordinator, f.environmentRuntime, async () => {});
|
|
try {
|
|
await worker.ready;
|
|
await expect.poll(async () => readSystemRecovery((await action(f.actionId)).evidence)?.state,
|
|
{ timeout: 20_000 }).toBe("complete");
|
|
expect((await action(f.actionId)).evidence.automaticRecovery).toEqual(automaticRecovery);
|
|
expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1);
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
expect(errors).not.toHaveBeenCalled();
|
|
} finally { await coordinator.stop(); await worker.stop(); }
|
|
}, 30_000);
|
|
|
|
it("fills a free export slot across the candidate cursor while another export is held and joins both at shutdown", async () => {
|
|
const orderedSeed = async (position: number) => {
|
|
const fixture = await seed();
|
|
const id = `00000002-000${position}-4000-8000-${randomUUID().slice(-12)}` as const;
|
|
await db.update(issueRecoveryActions).set({ id }).where(eq(issueRecoveryActions.id, fixture.actionId));
|
|
fixture.actionId = id; fixture.input.actionId = id;
|
|
return fixture;
|
|
};
|
|
const held = await orderedSeed(2);
|
|
const fixtures = [held];
|
|
const runtimeFor = (runId: string) => fixtures.find(fixture => fixture.runId === runId)!.environmentRuntime;
|
|
const runtime = { ...held.environmentRuntime,
|
|
resumeWorkspaceRecoveryLease: (input: Parameters<typeof held.environmentRuntime.resumeWorkspaceRecoveryLease>[0]) =>
|
|
runtimeFor(input.source.runId).resumeWorkspaceRecoveryLease(input),
|
|
retryPendingSandboxTeardown: (input: Parameters<typeof held.environmentRuntime.retryPendingSandboxTeardown>[0]) =>
|
|
runtimeFor(input.lease.heartbeatRunId!).retryPendingSandboxTeardown(input),
|
|
};
|
|
let active = 0, maximum = 0, release!: () => void;
|
|
const gate = new Promise<void>(resolve => { release = resolve; });
|
|
const countExport = (fixture: typeof held, pause = false) => {
|
|
const resume = fixture.runtime.resumeRunLease.getMockImplementation()!;
|
|
fixture.runtime.resumeRunLease.mockImplementation(async input => {
|
|
active += 1; maximum = Math.max(maximum, active);
|
|
try { if (pause) await gate; return await resume(input); } finally { active -= 1; }
|
|
});
|
|
};
|
|
countExport(held, true);
|
|
const errors = vi.fn();
|
|
const coordinator = createDeliveryWorkCoordinator({ owner: db, canRun: () => true, onError: errors });
|
|
const worker = registerWorkspaceRecoveryWork(db, coordinator, runtime, async () => {});
|
|
try {
|
|
await expect.poll(() => held.runtime.resumeRunLease.mock.calls.length, { timeout: 20_000 }).toBe(1);
|
|
// Start the middle candidate only after the highest candidate occupies
|
|
// one slot. The next producer then sorts before that saved cursor.
|
|
const fast = await orderedSeed(1); fixtures.push(fast); countExport(fast);
|
|
await requestTaskRetry({ db, companyId: fast.companyId, issueId: fast.issueId, actorId: "board",
|
|
observedRunId: fast.runId, idempotencyKey: randomUUID() });
|
|
await expect.poll(async () => readSystemRecovery((await action(fast.actionId)).evidence)?.state,
|
|
{ timeout: 20_000 }).toBe("complete");
|
|
const third = await orderedSeed(0); fixtures.push(third); countExport(third);
|
|
await requestTaskRetry({ db, companyId: third.companyId, issueId: third.issueId, actorId: "board",
|
|
observedRunId: third.runId, idempotencyKey: randomUUID() });
|
|
await expect.poll(async () => readSystemRecovery((await action(third.actionId)).evidence)?.state,
|
|
{ timeout: 20_000 }).toBe("complete");
|
|
expect(readSystemRecovery((await action(held.actionId)).evidence)?.state).toBe("running");
|
|
expect(maximum).toBeLessThanOrEqual(2);
|
|
expect(await readFile(path.join(third.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, third.companyId))).toHaveLength(1);
|
|
let stopped = false;
|
|
const stopping = Promise.all([coordinator.stop(), worker.stop()]).then(() => { stopped = true; });
|
|
await Promise.resolve(); expect(stopped).toBe(false);
|
|
release(); await stopping;
|
|
expect(readSystemRecovery((await action(held.actionId)).evidence)?.state).toBe("complete");
|
|
expect(errors).not.toHaveBeenCalled();
|
|
} finally { release(); await coordinator.stop(); await worker.stop(); }
|
|
}, 60_000);
|
|
|
|
it("rotates past due sources whose physical locks belong to other workers", async () => {
|
|
const blocked = [await seed(), await seed()], available = await seed();
|
|
for (const [index, fixture] of [...blocked, available].entries()) {
|
|
await adoptSystemRecovery(db, fixture);
|
|
const id = `00000001-000${index}-4000-8000-${randomUUID().slice(-12)}` as const;
|
|
await db.update(issueRecoveryActions).set({ id }).where(eq(issueRecoveryActions.id, fixture.actionId));
|
|
fixture.actionId = id; fixture.input.actionId = id;
|
|
}
|
|
let unlock!: () => void;
|
|
const gate = new Promise<void>(resolve => { unlock = resolve; });
|
|
const held = blocked.map(fixture => {
|
|
let ready!: () => void;
|
|
const acquired = new Promise<void>(resolve => { ready = resolve; });
|
|
const lock = db.transaction(async tx => {
|
|
await tx.execute(sql`select pg_advisory_xact_lock(hashtextextended(${nativeWorkspaceFinalizationLockKey(fixture.companyId, fixture.runId)}, 0))`);
|
|
ready(); await gate;
|
|
});
|
|
return { acquired, lock };
|
|
});
|
|
try {
|
|
await Promise.all(held.map(item => item.acquired));
|
|
await Promise.all(await reconcileRetainedWorkspaces(db, available.environmentRuntime, async () => {}, true) ?? []);
|
|
expect(blocked.every(fixture => fixture.runtime.resumeRunLease.mock.calls.length === 0)).toBe(true);
|
|
await Promise.all(await reconcileRetainedWorkspaces(db, available.environmentRuntime, async () => {}, true) ?? []);
|
|
expect(readSystemRecovery((await action(available.actionId)).evidence)?.state).toBe("complete");
|
|
expect(await readFile(path.join(available.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(blocked.every(fixture => fixture.runtime.resumeRunLease.mock.calls.length === 0)).toBe(true);
|
|
} finally { unlock(); await Promise.all(held.map(item => item.lock)); }
|
|
}, 30_000);
|
|
|
|
it("coalesces Retry while preservation owns its physical operation", async () => {
|
|
const f = await seed();
|
|
await adoptSystemRecovery(db, f);
|
|
const resume = f.runtime.resumeRunLease.getMockImplementation()!;
|
|
let release!: () => void;
|
|
const gate = new Promise<void>(resolve => { release = resolve; });
|
|
f.runtime.resumeRunLease.mockImplementation(async input => { await gate; return resume(input); });
|
|
const preserving = recoverRetainedWorkspace(f.input);
|
|
let requestId: string | null | undefined;
|
|
try {
|
|
await expect.poll(() => f.runtime.resumeRunLease.mock.calls.length).toBe(1);
|
|
const request = { db, companyId: f.companyId, issueId: f.issueId, actorId: "board",
|
|
idempotencyKey: randomUUID(), observedRunId: f.runId };
|
|
const first = await requestTaskRetry(request);
|
|
requestId = first.requestId;
|
|
const second = await requestTaskRetry({ ...request, idempotencyKey: randomUUID() });
|
|
expect(second.requestId).toBe(first.requestId);
|
|
} finally { release(); await preserving; }
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.intentId).toBe(requestId);
|
|
expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
}, 30_000);
|
|
|
|
it("adopts only the exact historical infrastructure notice, preserving prose and user-action notices", async () => {
|
|
const f = await seed("legacy", false);
|
|
const metadata = { version: 1 as const, sourceRunId: f.runId,
|
|
sections: [{ rows: [{ type: "key_value" as const, label: "Recovery action", value: f.actionId }] }] };
|
|
const base = { companyId: f.companyId, issueId: f.issueId, body: "Original technical history",
|
|
authorType: "system" as const, presentation: { kind: "system_notice" as const, tone: "danger" as const,
|
|
title: "Recovery needs attention", detailsDefaultOpen: false }, metadata };
|
|
const comments = await db.insert(issueComments).values([
|
|
base,
|
|
{ ...base, authorType: "user", authorUserId: "board" },
|
|
{ ...base, metadata: { ...metadata, sourceRunId: randomUUID() } },
|
|
{ ...base, metadata: { ...metadata, recoveryResponsibility: "user" as const } },
|
|
]).returning();
|
|
await adoptSystemRecovery(db, f);
|
|
await adoptSystemRecovery(db, f);
|
|
const rows = await db.select().from(issueComments).where(eq(issueComments.companyId, f.companyId));
|
|
expect(rows.find(row => row.id === comments[0].id)?.metadata?.recoveryResponsibility).toBe("system");
|
|
for (const original of comments.slice(1)) {
|
|
expect(rows.find(row => row.id === original.id)?.metadata).toEqual(original.metadata);
|
|
}
|
|
expect(rows.every(row => row.body === base.body)).toBe(true);
|
|
});
|
|
|
|
async function makeRetryDue(actionId: string) {
|
|
const current = await action(actionId);
|
|
const state = readSystemRecovery(current.evidence)!;
|
|
await db.update(issueRecoveryActions).set({ evidence: { ...current.evidence,
|
|
systemRecovery: { ...state, nextAttemptAt: 0 } } }).where(eq(issueRecoveryActions.id, actionId));
|
|
}
|
|
|
|
it("publishes the committed finalization barrier after real retained workspace recovery", async () => {
|
|
const f = await seed();
|
|
const observed: Array<Promise<Array<{ status: string }>>> = [];
|
|
const stop = subscribeDeliveryWork(db, DELIVERY_QUEUES.nativeStatusReassessment, () => {
|
|
observed.push(db.select({ status: workspaceOperations.status }).from(workspaceOperations).where(and(
|
|
eq(workspaceOperations.companyId, f.companyId), eq(workspaceOperations.heartbeatRunId, f.runId),
|
|
eq(workspaceOperations.phase, "workspace_finalize"),
|
|
)).then(rows => rows));
|
|
});
|
|
try { await recoverRetainedWorkspace(f.input); } finally { stop(); }
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
expect(observed.length).toBeGreaterThan(0);
|
|
expect((await Promise.all(observed)).some(rows => rows.some(row => row.status === "succeeded"))).toBe(true);
|
|
});
|
|
|
|
it.each(["retain_on_failure", "reuse_by_environment", "reuse_by_execution_workspace"] as const)("re-enters only the accepted result finalizer for %s, preserving files and failure history", async leasePolicy => {
|
|
const f = await seed("native", true, "keep_running");
|
|
await db.update(environmentLeases).set({ leasePolicy }).where(eq(environmentLeases.id, f.lease.id));
|
|
await db.update(issueRecoveryActions).set({ cause: "native_workspace_sync_out_retry_exhausted" }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
await db.update(heartbeatRuns).set({ resultJson: {} }).where(eq(heartbeatRuns.id, f.runId));
|
|
const contractId = randomUUID(), resultId = randomUUID();
|
|
await db.insert(completionContracts).values({ id: contractId, companyId: f.companyId, issueId: f.issueId,
|
|
revision: 1, schemaVersion: "paperclip.completion-contract.v1", policyVersion: "test", risk: "standard",
|
|
completionAuthority: "server_arbiter", incompleteCriteriaPolicy: "preserve_non_terminal",
|
|
contractJson: {}, canonicalSha256: "contract", createdByActorType: "system", createdByActorId: "test" });
|
|
await db.update(heartbeatRuns).set({ completionContractId: contractId }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.insert(nativeRunResults).values({ id: resultId, companyId: f.companyId, issueId: f.issueId, runId: f.runId,
|
|
completionContractId: contractId, schemaStatus: "accepted", serverFingerprint: "result", canonicalSha256: "result",
|
|
resultJson: { terminal: { runTerminalState: "succeeded" } } });
|
|
await db.update(nativeRunFinalizations).set({ resultId, attempt: 3, failureCode: "native_workspace_sync_out_retry_exhausted" })
|
|
.where(eq(nativeRunFinalizations.runId, f.runId));
|
|
expect(await resumePreservedResultFinalization(db, f.companyId, f.runId)).toBe(false);
|
|
const priorAction = await action(f.actionId);
|
|
await db.update(issueRecoveryActions).set({ ownerType: "system", evidence: { ...priorAction.evidence,
|
|
systemRecovery: failSystemRecovery(createSystemRecovery(f.runId),
|
|
new SystemRecoveryFailure("source_binding_invalid", "service_required")),
|
|
workspaceRecovery: { schema: "paperclip.workspace-recovery.v1", runId: f.runId, status: "blocked",
|
|
stage: "verifying_stop", attempt: 1, attemptLimit: 6, requestIds: [] },
|
|
} }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
const request = { ...f.input, actorId: "board", idempotencyKey: randomUUID() };
|
|
await requestWorkspaceRecovery(request);
|
|
const admitted = await action(f.actionId);
|
|
await requestWorkspaceRecovery(request);
|
|
expect(await action(f.actionId)).toEqual(admitted);
|
|
expect(admitted.ownerType).toBe("system");
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
const attempts = await Promise.all([resumePreservedResultFinalization(db, f.companyId, f.runId),
|
|
resumePreservedResultFinalization(db, f.companyId, f.runId)]);
|
|
expect(attempts.filter(Boolean)).toHaveLength(1);
|
|
const [coordinator] = await db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
expect(coordinator).toMatchObject({ phase: "workspace_finalizing", resultId, attempt: 3 });
|
|
expect(coordinator.recoveryHistory).toEqual([expect.objectContaining({
|
|
kind: "preserved_result_finalization", previousFailureCode: "native_workspace_sync_out_retry_exhausted",
|
|
})]);
|
|
expect(await run(f.runId)).toMatchObject({ status: "failed", errorCode: "execution_interrupted" });
|
|
await db.update(nativeRunFinalizations).set({ phase: "terminal_failure" }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
expect(await resumePreservedResultFinalization(db, f.companyId, f.runId)).toBe(false);
|
|
const retry = await requestTaskRetry({ db, companyId: f.companyId, issueId: f.issueId, actorId: "board",
|
|
idempotencyKey: randomUUID(), observedRunId: f.runId });
|
|
expect(await resumePreservedResultFinalization(db, f.companyId, f.runId)).toBe(true);
|
|
expect((await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, retry.requestId!)))[0])
|
|
.toMatchObject({ status: "coalesced", runId: f.runId, payload: { taskRetry: { strategy: "finalize_result" } } });
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.companyId, f.companyId))).toHaveLength(1);
|
|
}, 30_000);
|
|
|
|
it("preserves a historical legacy source without manufacturing a native descriptor or overwriting current files", async () => {
|
|
const f = await seed("legacy", false);
|
|
await recoverRetainedWorkspace(f.input);
|
|
const operations = await db.select().from(workspaceOperations).where(and(
|
|
eq(workspaceOperations.heartbeatRunId, f.runId), eq(workspaceOperations.phase, "workspace_recovery_snapshot"),
|
|
eq(workspaceOperations.status, "succeeded"),
|
|
));
|
|
expect(operations).toHaveLength(1);
|
|
expect(operations[0].metadata).toMatchObject({ leaseId: f.lease.id, providerLeaseId: f.providerLeaseId });
|
|
const fresh = path.join(root, randomUUID()); await mkdir(fresh);
|
|
const attached = await attachPreservedWorkspaceSnapshots({ db, companyId: f.companyId, issueId: f.issueId,
|
|
sourceRunIds: [f.runId], cwd: fresh });
|
|
expect(await readFile(path.join(attached[0].path, "files", "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(attached[0].path, "provenance.json"), "utf8")).toContain(f.runId);
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect((await run(f.runId)).runnerProfileJson?.nativeWorkspaceSync).toBeUndefined();
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
expect((await run(f.runId)).status).toBe("failed");
|
|
}, 30_000);
|
|
|
|
it("keeps earlier recovered snapshots readable across repeated fresh retries", async () => {
|
|
const f = await seed("legacy", false);
|
|
await recoverRetainedWorkspace(f.input);
|
|
const source = await run(f.runId);
|
|
const replacementId = randomUUID(), laterId = randomUUID(), intentId = randomUUID();
|
|
await db.insert(heartbeatRuns).values([
|
|
{ id: replacementId, companyId: f.companyId, agentId: source.agentId,
|
|
status: "failed", contextSnapshot: { issueId: f.issueId } },
|
|
{ id: laterId, companyId: f.companyId, agentId: source.agentId,
|
|
status: "failed", retryOfRunId: replacementId, contextSnapshot: { issueId: f.issueId } },
|
|
]);
|
|
await db.insert(agentWakeupRequests).values({
|
|
id: intentId, companyId: f.companyId, agentId: source.agentId, runId: replacementId,
|
|
source: "on_demand", triggerDetail: "manual", reason: "task_retry", status: "coalesced",
|
|
requestedByActorType: "user", requestedByActorId: "owner-user",
|
|
payload: { issueId: f.issueId, taskRetry: { schema: "paperclip.task-retry.v1",
|
|
generationId: intentId, sourceRunId: f.runId, actorId: "owner-user", strategy: "fresh_isolated_workspace" } },
|
|
});
|
|
const first = path.join(root, randomUUID()), next = path.join(root, randomUUID());
|
|
await mkdir(first); await mkdir(next);
|
|
await attachPreservedWorkspaceSnapshots({ db, companyId: f.companyId, issueId: f.issueId,
|
|
sourceRunIds: [f.runId], cwd: first });
|
|
const attached = await attachPreservedWorkspaceSnapshots({ db, companyId: f.companyId, issueId: f.issueId,
|
|
sourceRunIds: [laterId], cwd: next });
|
|
expect(attached).toHaveLength(1);
|
|
expect(attached[0].sourceRunId).toBe(f.runId);
|
|
expect(await readFile(path.join(attached[0].path, "files", "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(attached[0].path, "provenance.json"), "utf8")).toContain(f.runId);
|
|
expect(await attachPreservedWorkspaceSnapshots({ db, companyId: f.companyId, issueId: randomUUID(),
|
|
sourceRunIds: [laterId], cwd: next })).toEqual([]);
|
|
}, 30_000);
|
|
|
|
it("retains files and the recovery barrier when native export rejects Git metadata", async () => {
|
|
const f = await seed();
|
|
const restore = vi.spyOn(nativeWorkspaceSync, "resumeNativeWorkspaceSync")
|
|
.mockRejectedValueOnce(new GitShallowMetadataError("Git shallow metadata exceeds its byte limit"));
|
|
try { await recoverRetainedWorkspace(f.input); } finally { restore.mockRestore(); }
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)).toMatchObject({ state: "exhausted",
|
|
failure: { code: "snapshot_verification_failed", retryClass: "service_required" }, preservationOperationIds: [] });
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
expect((await run(f.runId)).status).toBe("failed");
|
|
}, 30_000);
|
|
|
|
it("reconciles an uploaded snapshot after a lost upload acknowledgement without exporting again", async () => {
|
|
const f = await seed("legacy", false);
|
|
const target = targets.get(f.lease.id)!;
|
|
const exportFiles = vi.spyOn(target.runner!, "syncOut");
|
|
const storage = getStorageService(), put = storage.putFile.bind(storage);
|
|
let uploads = 0;
|
|
const lostReply = vi.spyOn(storage, "putFile").mockImplementation(async input => {
|
|
const receipt = await put(input);
|
|
if (++uploads === 2) throw new Error("Lost upload acknowledgement");
|
|
return receipt;
|
|
});
|
|
try { await recoverRetainedWorkspace(f.input); } finally { lostReply.mockRestore(); }
|
|
const [pending] = await db.select().from(workspaceOperations).where(and(
|
|
eq(workspaceOperations.heartbeatRunId, f.runId), eq(workspaceOperations.phase, "workspace_recovery_snapshot")));
|
|
expect(pending.status).toBe("running");
|
|
expect(uploads).toBe(2);
|
|
await makeRetryDue(f.actionId);
|
|
const restarted = createDb(connectionString);
|
|
try { await recoverRetainedWorkspace({ ...f.input, db: restarted }); }
|
|
finally { await restarted.$client.end(); }
|
|
const operations = await db.select().from(workspaceOperations).where(and(
|
|
eq(workspaceOperations.heartbeatRunId, f.runId), eq(workspaceOperations.phase, "workspace_recovery_snapshot")));
|
|
expect(operations).toHaveLength(1);
|
|
expect(operations[0]).toMatchObject({ id: pending.id, status: "succeeded", metadata: pending.metadata });
|
|
expect(exportFiles).toHaveBeenCalledTimes(1);
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
exportFiles.mockRestore();
|
|
}, 30_000);
|
|
|
|
it.each(["native", "legacy"])("recovers real files without an accepted result or another turn: %s", async mode => {
|
|
const f = await seed(mode);
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)).toMatchObject({ status: "recovered", attempt: 1 });
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
expect(await getExecutionBlocker(db, f.companyId, f.issueId)).toMatchObject({ workspaceRepairRequired: false });
|
|
expect((await run(f.runId)).status).toBe("failed");
|
|
expect((await action(f.actionId)).evidence.executionReconciliation).toBeUndefined();
|
|
expect((await db.select().from(issues).where(eq(issues.id, f.issueId)))[0].status).toBe("blocked");
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.companyId, f.companyId))).toHaveLength(0);
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1);
|
|
expect(await db.select().from(workspaceOperations).where(and(eq(workspaceOperations.companyId, f.companyId), eq(workspaceOperations.phase, "workspace_recovery")))).toHaveLength(1);
|
|
}, 30_000);
|
|
|
|
it("recovers a protected retained source through the assembled environment runtime", async () => {
|
|
const f = await seed();
|
|
await db.update(environmentLeases).set({ metadata: { ...f.lease.metadata,
|
|
workspaceRestoreRecovery: { schema: "paperclip.workspace-restore-recovery.v1", runId: f.runId },
|
|
remoteExecutionTermination: remoteTerminationReceipt(f.lease, { providerLeaseId: f.providerLeaseId, state: "stopped" }),
|
|
} }).where(eq(environmentLeases.id, f.lease.id));
|
|
const runtime = f.input.environmentRuntime;
|
|
const [lease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id));
|
|
const [environment] = await db.select().from(environments).where(eq(environments.id, lease.environmentId!));
|
|
expect(await runtime.resumeRunLease({ environment: environment as never, lease: lease as EnvironmentLease })).toBeNull();
|
|
expect(f.runtime.resumeRunLease).not.toHaveBeenCalled();
|
|
await recoverRetainedWorkspace({ ...f.input, environmentRuntime: runtime });
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)?.status).toBe("recovered");
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1);
|
|
expect(f.runtime.resumeRunLease).toHaveBeenCalledWith(expect.objectContaining({ requestedCwd: f.remote }));
|
|
await recoverRetainedWorkspace({ ...f.input, environmentRuntime: runtime });
|
|
expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1);
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
}, 30_000);
|
|
|
|
it.each(["changed_claim", "missing_stop_intent", "wrong_lease", "wrong_environment"])(
|
|
"does not reopen a retained source without exact recovery authority: %s", async kind => {
|
|
const f = await seed();
|
|
const runtime = f.input.environmentRuntime;
|
|
const resume = runtime.resumeWorkspaceRecoveryLease.bind(runtime);
|
|
vi.spyOn(runtime, "resumeWorkspaceRecoveryLease").mockImplementationOnce(async input => {
|
|
if (kind === "changed_claim") await db.update(environmentLeases).set({ metadata: sql`
|
|
${environmentLeases.metadata} || '{"pendingCleanupAttemptId":"other-owner"}'::jsonb`
|
|
}).where(eq(environmentLeases.id, f.lease.id));
|
|
if (kind === "missing_stop_intent") await db.update(environmentLeases).set({ metadata: sql`
|
|
${environmentLeases.metadata} - 'sandboxStopAndRetain'`
|
|
}).where(eq(environmentLeases.id, f.lease.id));
|
|
return resume({ ...input,
|
|
...(kind === "wrong_lease" ? { leaseId: randomUUID() } : {}),
|
|
...(kind === "wrong_environment" ? { environmentId: randomUUID() } : {}),
|
|
});
|
|
});
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.failure?.code).toBe("source_binding_invalid");
|
|
expect(f.runtime.resumeRunLease).not.toHaveBeenCalled();
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
}, 30_000);
|
|
|
|
it("forwards only the owned export directory through the sandbox provider boundary", async () => {
|
|
const driver = "sandbox";
|
|
const f = await seed();
|
|
const pluginId = f.lease.metadata!.pluginId as string;
|
|
const manifest: typeof plugins.$inferInsert.manifestJson = { id: `fixture.export-cwd.${pluginId}`, apiVersion: 1, version: "1.0.0",
|
|
displayName: "Fixture export provider", description: "Records provider boundary calls", author: "Paperclip",
|
|
categories: ["automation"], capabilities: ["environment.drivers.register"], entrypoints: { worker: "dist/worker.js" },
|
|
environmentDrivers: [{ driverKey: "daytona", kind: "sandbox_provider", displayName: "Fixture", configSchema: { type: "object" } }] };
|
|
await db.insert(plugins).values({ id: pluginId, pluginKey: manifest.id,
|
|
packageName: "@fixture/export-cwd", version: manifest.version,
|
|
apiVersion: 1, categories: manifest.categories, manifestJson: manifest, status: "ready", installOrder: 1 });
|
|
const metadata = { ...f.lease.metadata, driver, provider: "daytona", pluginKey: manifest.id,
|
|
sandboxProviderPlugin: true };
|
|
await db.update(environmentLeases).set({ metadata }).where(eq(environmentLeases.id, f.lease.id));
|
|
await db.update(environments).set({ driver, config: { provider: "daytona" } }).where(eq(environments.id, f.lease.environmentId!));
|
|
const call = vi.fn(async (_id: string, method: string, input: Record<string, unknown>) => {
|
|
if (method === "environmentResumeLease") return { providerLeaseId: f.providerLeaseId,
|
|
metadata: { remoteCwd: input.requestedCwd ?? path.dirname(f.remote) } };
|
|
if (method === "environmentStopLease") return { providerLeaseId: f.providerLeaseId, state: "stopped" };
|
|
throw new Error(`Unexpected provider operation: ${method}`);
|
|
});
|
|
const worker = { isRunning: () => true, call, getWorker: () => ({ supportedMethods: ["environmentResumeLease", "environmentStopLease"] }) } as unknown as PluginWorkerManager;
|
|
const runtime = environmentRuntimeService(db, { pluginWorkerManager: worker });
|
|
await recoverRetainedWorkspace({ ...f.input, environmentRuntime: runtime });
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)?.status).toBe("recovered");
|
|
expect(call.mock.calls.filter(([, method]) => method === "environmentResumeLease")).toEqual([
|
|
expect.arrayContaining([pluginId, "environmentResumeLease", expect.objectContaining({
|
|
companyId: f.companyId, environmentId: f.lease.environmentId, providerLeaseId: f.providerLeaseId, requestedCwd: f.remote,
|
|
})]),
|
|
]);
|
|
expect(call.mock.calls.some(([, method]) => ["environmentAcquireLease", "environmentRealizeWorkspace", "environmentDestroyLease"].includes(method))).toBe(false);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.companyId, f.companyId))).toHaveLength(1);
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
}, 30_000);
|
|
|
|
it.each(["same", "later"])("prepares the %s task workspace from an ordinary resumed allocation root", async kind => {
|
|
const f = await seed("native", false);
|
|
vi.stubEnv("DATABASE_URL", connectionString);
|
|
vi.stubEnv("PAPERCLIP_SECRETS_MASTER_KEY", "62".repeat(32));
|
|
try {
|
|
const allocationRoot = await realpath(f.remote);
|
|
const target = { ...targets.get(f.lease.id)!, remoteCwd: allocationRoot };
|
|
const storage = { ...getStorageService(), provider: "s3" as const };
|
|
const agentId = (await run(f.runId)).agentId;
|
|
await db.update(heartbeatRuns).set({ status: "running", resultJson: {} }).where(eq(heartbeatRuns.id, f.runId));
|
|
const [firstLease] = await db.update(environmentLeases).set({ status: "active", metadata: {
|
|
...f.lease.metadata, remoteCwd: allocationRoot } }).where(eq(environmentLeases.id, f.lease.id)).returning();
|
|
const reserveInput = { db, storage, companyId: f.companyId, issueId: f.issueId, runId: f.runId, agentId,
|
|
environmentId: f.lease.environmentId!, projectId: null, projectWorkspaceId: null, existingWorkspaceId: null,
|
|
source: { kind: "empty" as const } };
|
|
const first = await prepareRemoteWorkspace({ db, storage, workspace: await reserveRemoteWorkspace(reserveInput),
|
|
runId: f.runId, lease: firstLease as EnvironmentLease, target });
|
|
await writeFile(path.join(first.remoteRoot, "saved.txt"), "prior task work\n");
|
|
await saveRemoteWorkspace({ db, storage, workspace: first.workspace, runId: f.runId, lease: first.lease,
|
|
target, assertOwned: async () => undefined });
|
|
await db.update(heartbeatRuns).set({ status: "succeeded" }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(environmentLeases).set({ status: "released" }).where(eq(environmentLeases.id, f.lease.id));
|
|
const issueId = kind === "same" ? f.issueId : randomUUID();
|
|
if (kind === "later") await db.insert(issues).values({ id: issueId, companyId: f.companyId, assigneeAgentId: agentId,
|
|
title: "A later task", status: "in_progress" });
|
|
const [nextRun] = await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId, issueId, nativeIssueId: issueId,
|
|
runtimeMode: "native", status: "running", contextSnapshot: { issueId } }).returning();
|
|
const [nextLease] = await db.insert(environmentLeases).values({ companyId: f.companyId, issueId, heartbeatRunId: nextRun.id,
|
|
environmentId: f.lease.environmentId, provider: "daytona", providerLeaseId: kind === "same" ? f.providerLeaseId : randomUUID(),
|
|
status: "active", metadata: { ...f.lease.metadata, remoteCwd: first.remoteRoot } }).returning();
|
|
f.runtime.resumeRunLease.mockImplementation(async input => ({ providerLeaseId: input.lease.providerLeaseId!,
|
|
metadata: { remoteCwd: allocationRoot } }));
|
|
const [environment] = await db.select().from(environments).where(eq(environments.id, f.lease.environmentId!));
|
|
const resumed = await f.environmentRuntime.resumeRunLease({ environment: environment as never, lease: nextLease as EnvironmentLease });
|
|
expect(resumed?.metadata?.remoteCwd).toBe(allocationRoot);
|
|
expect(f.runtime.resumeRunLease.mock.calls[0]?.[0]).not.toHaveProperty("requestedCwd");
|
|
const nextWorkspace = await reserveRemoteWorkspace({ ...reserveInput, issueId, runId: nextRun.id,
|
|
existingWorkspaceId: kind === "same" ? first.workspace.id : null });
|
|
const prepared = await prepareRemoteWorkspace({ db, storage, workspace: nextWorkspace, runId: nextRun.id,
|
|
lease: nextLease as EnvironmentLease, target: { ...target, remoteCwd: resumed!.metadata!.remoteCwd as string } });
|
|
expect(prepared.remoteRoot).toBe(path.join(allocationRoot, ".paperclip-workspaces", nextWorkspace.id));
|
|
if (kind === "same") expect(await readFile(path.join(prepared.remoteRoot, "saved.txt"), "utf8")).toBe("prior task work\n");
|
|
else expect(prepared.remoteRoot).not.toBe(first.remoteRoot);
|
|
await saveRemoteWorkspace({ db, storage, workspace: prepared.workspace, runId: nextRun.id, lease: prepared.lease,
|
|
target: { ...target, remoteCwd: prepared.remoteRoot }, assertOwned: async () => undefined });
|
|
await db.update(heartbeatRuns).set({ status: "succeeded" }).where(eq(heartbeatRuns.id, nextRun.id));
|
|
} finally { vi.unstubAllEnvs(); }
|
|
}, 30_000);
|
|
|
|
it("keeps verified copyback files in the preserved workspace on ordinary Retry", async () => {
|
|
const f = await seed();
|
|
await recoverRetainedWorkspace(f.input);
|
|
const source = await run(f.runId);
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, status: "running" });
|
|
const request = await requestTaskRetry({ db, companyId: f.companyId, issueId: f.issueId, actorId: "local-board", actorSource: "local_implicit",
|
|
observedRunId: source.id, idempotencyKey: randomUUID() });
|
|
await heartbeatService(db).resumeTaskRetry(f.companyId, request.requestId!);
|
|
const [successor] = await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")));
|
|
expect(successor).toBeDefined();
|
|
const [receipt] = await db.select().from(workspaceOperations).where(and(eq(workspaceOperations.heartbeatRunId, f.runId),
|
|
eq(workspaceOperations.phase, "workspace_recovery"), eq(workspaceOperations.status, "succeeded")));
|
|
const readiness = await readFreshWorkspaceReadiness(db, successor);
|
|
expect(readiness).toEqual({ mode: "resume_preserved_workspace", preservationReceiptId: receipt.id });
|
|
const [agent] = await db.select().from(agents).where(eq(agents.id, source.agentId));
|
|
const [issue] = await db.select().from(issues).where(eq(issues.id, f.issueId));
|
|
const localEnvironment = await environmentService(db).ensureLocalEnvironment(f.companyId);
|
|
const prepared = await prepareHeartbeatWorkspace(db, {
|
|
run: successor, agent,
|
|
task: { issueRef: issue, issueId: f.issueId, context: { issueId: f.issueId }, responsibleUserId: "local-board",
|
|
previousSessionParams: null, freshWorkspace: readiness },
|
|
policy: { trustPreset: { kind: "standard", preset: "standard", boundary: null, sourcePresets: {} },
|
|
isolatedWorkspacesEnabled: false, effectiveExecutionWorkspaceMode: "shared_workspace",
|
|
requestedExecutionWorkspaceMode: "shared_workspace", useIsolatedTaskDirectory: false,
|
|
nativeChatWorkspaceScope: null, projectExecutionWorkspacePolicy: null, issueExecutionWorkspaceSettings: null,
|
|
environmentExecutionWorkspaceSettings: null, executionPolicy: { executionMode: "any" } },
|
|
environment: { selectedEnvironmentId: localEnvironment.id, localEnvironment, selectedEnvironmentForConfig: localEnvironment,
|
|
environmentResolution: { environmentId: localEnvironment.id, source: "default" },
|
|
resolvedInstanceSettings: await instanceSettingsService(db).get() },
|
|
config: { mergedConfig: { cwd: f.host }, configSnapshot: null, secretManifest: [] },
|
|
reuse: { requestedShouldReuseExisting: false, existingExecutionWorkspace: null, reusableExistingExecutionWorkspace: null,
|
|
workspaceReuseRequest: resolveExecutionWorkspaceReuseRequestForIssue({ issueExecutionWorkspaceId: null,
|
|
issueExecutionWorkspacePreference: null, existingExecutionWorkspaceStatus: null }),
|
|
nativeRecoveryExecutionWorkspaceId: null, persistedNativeExecutionInput: null, isDotRun: false, runOptions: {} },
|
|
services: { envOrchestrator: environmentRunOrchestrator(db), executionWorkspacesSvc: executionWorkspaceService(db),
|
|
workspaceOperationsSvc: workspaceOperationService(db), ...createHeartbeatWorkspaceResolver(db), appendRunEvent: vi.fn() },
|
|
controllerLease: { assertOwned: vi.fn(async () => {}) }, nativeRunnerPreparationSpans: [],
|
|
});
|
|
expect(prepared.executionWorkspace.cwd).toBe(f.host);
|
|
expect(await readFile(path.join(prepared.executionWorkspace.cwd, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
const [wake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, request.requestId!));
|
|
expect(wake.payload?.taskRetry).toMatchObject({ strategy: "continue" });
|
|
}, 30_000);
|
|
|
|
it("projects active preservation without waiting input as working, and stale observations as failed", async () => {
|
|
const f = await seed();
|
|
const [issue] = await db.select().from(issues).where(eq(issues.id, f.issueId));
|
|
const state = { ...createSystemRecovery(f.runId), state: "running" as const };
|
|
const recovery = { ...await action(f.actionId), evidence: { systemRecovery: state, systemRecoveryActiveAt: Date.now() } };
|
|
const input = { run: await run(f.runId), action: recovery, paused: false, approval: false };
|
|
expect(projectTaskInterruption(issue, input)).toMatchObject({ state: "working", intentId: null, canRetry: false });
|
|
expect(projectTaskInterruption(issue, input, Date.now() + 60_000)).toMatchObject({ state: "failed", canRetry: true });
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
});
|
|
|
|
it.each([["native_provider_model_rejected", "choose_model"], ["native_provider_approval_required", "approval"]] as const)(
|
|
"projects %s as the meaningful user action %s", async (errorCode, userAction) => {
|
|
const f = await seed();
|
|
const [issue] = await db.select().from(issues).where(eq(issues.id, f.issueId));
|
|
expect(projectTaskInterruption(issue, { run: { ...await run(f.runId), errorCode }, paused: false, approval: false }))
|
|
.toMatchObject({ state: "waiting_for_user", responsibility: "user", userAction, canRetry: false });
|
|
});
|
|
|
|
it("serializes concurrent repairs through the physical workspace lock", async () => {
|
|
const f = await seed(); let ready!: () => void, release!: () => void;
|
|
const started = new Promise<void>(resolve => { ready = resolve; }), gate = new Promise<void>(resolve => { release = resolve; });
|
|
f.runtime.resumeRunLease.mockImplementationOnce(async () => { ready(); await gate; return { providerLeaseId: f.providerLeaseId, metadata: { remoteCwd: f.remote } }; });
|
|
const first = recoverRetainedWorkspace(f.input);
|
|
await started;
|
|
try { await recoverRetainedWorkspace(f.input); expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1); }
|
|
finally { release(); await first; }
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)?.status).toBe("recovered");
|
|
}, 30_000);
|
|
|
|
it.each(["missing_stop", "wrong_stop", "another_owner", "active_writer", "invalid_descriptor"])("retains the source and starts no provider for %s", async kind => {
|
|
const f = await seed();
|
|
if (kind === "missing_stop" || kind === "wrong_stop") await db.update(environmentLeases).set({ metadata: { ...f.lease.metadata,
|
|
...(kind === "wrong_stop" ? { remoteExecutionTermination: remoteTerminationReceipt({ ...f.lease, id: randomUUID() }, { providerLeaseId: f.providerLeaseId, state: "stopped" }) } : {}) } }).where(eq(environmentLeases.id, f.lease.id));
|
|
if (kind === "missing_stop" || kind === "wrong_stop") f.runtime.retryPendingSandboxTeardown.mockResolvedValue({ providerLeaseId: randomUUID(), state: "unknown" });
|
|
if (kind === "another_owner") await db.insert(environmentLeases).values({ companyId: f.companyId, heartbeatRunId: f.runId, provider: "daytona", providerLeaseId: f.providerLeaseId });
|
|
if (kind === "active_writer") await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: f.lease.heartbeatRunId ? (await run(f.runId)).agentId : "", status: "running", nativeIssueId: f.issueId });
|
|
if (kind === "invalid_descriptor") await db.update(heartbeatRuns).set({ runnerProfileJson: { nativeWorkspaceSync: { schema: "corrupt" } } }).where(eq(heartbeatRuns.id, f.runId));
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)).toMatchObject({ state:
|
|
["missing_stop", "wrong_stop", "active_writer"].includes(kind) ? "awaiting_evidence" : "exhausted" });
|
|
expect(f.runtime.resumeRunLease).not.toHaveBeenCalled();
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
}, 30_000);
|
|
|
|
it("resumes a copied workspace after a lost stop reply without overwriting later files", async () => {
|
|
const f = await seed();
|
|
f.runtime.retryPendingSandboxTeardown.mockRejectedValueOnce(new Error("stop reply lost"));
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)).toMatchObject({ status: "queued", attempt: 1 });
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
await writeFile(path.join(f.host, "saved.txt"), "later user edit\n");
|
|
const pending = await action(f.actionId);
|
|
await db.update(issueRecoveryActions).set({ evidence: { ...pending.evidence,
|
|
systemRecovery: { ...readSystemRecovery(pending.evidence), nextAttemptAt: 0 } } }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
const request = { ...f.input, idempotencyKey: randomUUID(), actorId: "user" };
|
|
await requestWorkspaceRecovery(request); await requestWorkspaceRecovery(request);
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("later user edit\n");
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)).toMatchObject({ status: "recovered", attempt: 2, requestIds: [request.idempotencyKey] });
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
}, 30_000);
|
|
|
|
it("resumes an interrupted file export from persisted state with a new database connection", async () => {
|
|
const f = await seed(), target = targets.get(f.lease.id)!;
|
|
const execute = target.runner!.execute;
|
|
let interrupted = false, interruptExports = true;
|
|
target.runner!.execute = async input => {
|
|
const output = await execute(input);
|
|
if (interruptExports && input.args?.some(arg => arg.includes("tar -cf"))) {
|
|
interrupted = true;
|
|
throw new Error("File export transport interrupted after archive creation");
|
|
}
|
|
return output;
|
|
};
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(interrupted).toBe(true);
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)).toMatchObject({ status: "queued", stage: "restoring_files", attempt: 1 });
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
interruptExports = false;
|
|
await writeFile(path.join(f.host, "later.txt"), "work saved during the interruption\n");
|
|
const pending = await action(f.actionId);
|
|
await db.update(issueRecoveryActions).set({ evidence: { ...pending.evidence,
|
|
systemRecovery: { ...readSystemRecovery(pending.evidence), nextAttemptAt: 0 } } }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
const restarted = createDb(connectionString);
|
|
try {
|
|
await recoverRetainedWorkspace({ ...f.input, db: restarted });
|
|
expect(await hasVerifiedWorkspaceRepair(restarted, await run(f.runId))).toBe(true);
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)).toMatchObject({ status: "recovered", attempt: 2 });
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "later.txt"), "utf8")).toBe("work saved during the interruption\n");
|
|
} finally { await restarted.$client.end(); target.runner!.execute = execute; }
|
|
}, 30_000);
|
|
|
|
it("rejects a recovery request outside the task company", async () => {
|
|
const f = await seed();
|
|
await expect(requestWorkspaceRecovery({ ...f.input, companyId: randomUUID(), actorId: "user", idempotencyKey: randomUUID() })).rejects.toThrow("changed");
|
|
expect(readWorkspaceRecovery((await action(f.actionId)).evidence)).toBeNull();
|
|
});
|
|
|
|
it.each([false, true])("adopts exact retained leases without a run descriptor (existing action: %s)", async existingAction => {
|
|
const f = await seed("legacy", false);
|
|
await db.update(heartbeatRuns).set({ resultJson: null }).where(eq(heartbeatRuns.id, f.runId));
|
|
if (existingAction) await db.update(issueRecoveryActions).set({ evidence: { runId: f.runId } })
|
|
.where(eq(issueRecoveryActions.id, f.actionId));
|
|
else await db.delete(issueRecoveryActions).where(eq(issueRecoveryActions.id, f.actionId));
|
|
await adoptHistoricalWorkspaceRecoveries(db);
|
|
const [adopted] = await db.select().from(issueRecoveryActions).where(eq(issueRecoveryActions.companyId, f.companyId));
|
|
expect(adopted.evidence.historicalWorkspaceSource).toMatchObject({ runId: f.runId, leaseIds: [f.lease.id] });
|
|
const generation = readSystemRecovery(adopted.evidence)!.generationId;
|
|
expect(await recoverRetainedWorkspace({ ...f.input, actionId: adopted.id })).toBeUndefined();
|
|
expect(readSystemRecovery((await action(adopted.id)).evidence)?.state).toBe("complete");
|
|
await adoptHistoricalWorkspaceRecoveries(db);
|
|
expect(readSystemRecovery((await action(adopted.id)).evidence)?.generationId).toBe(generation);
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
});
|
|
|
|
it.each(["native", "legacy", "native_export", "legacy_snapshot", "native_snapshot"])("delivers only saved user input once through ordinary admission: %s", async mode => {
|
|
const f = await seed(mode.startsWith("legacy") ? "legacy" : "native", !mode.endsWith("_snapshot")), commentId = randomUUID(), queueId = randomUUID();
|
|
if (mode === "native_snapshot") {
|
|
await db.update(heartbeatRuns).set({ runnerProfileJson: { nativeWorkspaceSync: {
|
|
schema: "paperclip.native-workspace-sync/v3", state: "prepared", descriptorSha256: "a".repeat(64),
|
|
baselineSha256: "b".repeat(64), finalHostSha256: null, workspaceId: f.issueId,
|
|
leaseId: f.lease.id, providerLeaseId: f.providerLeaseId, remoteCwd: f.remote, resourceDisposition: "destroy",
|
|
} } }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.insert(issueRecoveryActions).values({ companyId: f.companyId, sourceIssueId: f.issueId,
|
|
kind: "active_run_watchdog", status: "resolved", ownerType: "system", outcome: "blocked",
|
|
cause: "execution_finalization_deadline_exceeded", fingerprint: randomUUID(), nextAction: "Inspect original failure",
|
|
evidence: { runId: f.runId, automaticRecovery: { replay: "blocked" } } });
|
|
}
|
|
if (mode === "native_export") {
|
|
await db.update(issueRecoveryActions).set({ cause: "native_workspace_sync_out_retry_exhausted", evidence: { runId: f.runId } }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
await db.update(heartbeatRuns).set({ resultJson: {} }).where(eq(heartbeatRuns.id, f.runId));
|
|
}
|
|
const source = await run(f.runId);
|
|
// Occupy the agent, not this workspace. This keeps the admitted turn queued
|
|
// so this test does not substitute a model mock for recovery evidence.
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, status: "running" });
|
|
await db.insert(issueComments).values({ id: commentId, companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "board", body: "Confirm the existing publication. Do not publish it again." });
|
|
await db.insert(agentWakeupRequests).values({ id: queueId, companyId: f.companyId, agentId: source.agentId,
|
|
source: "automation", reason: "issue_commented", status: "deferred_issue_execution", requestedByActorType: "user", requestedByActorId: "board",
|
|
payload: { issueId: f.issueId, commentId, _paperclipWakeContext: { issueId: f.issueId, wakeReason: "issue_commented", wakeCommentIds: [commentId], wakeCommentId: commentId } } });
|
|
await recoverRetainedWorkspace(f.input);
|
|
const service = heartbeatService(db);
|
|
await Promise.all([service.resumeRemoteStopComments(await run(f.runId)), service.resumeRemoteStopComments(await run(f.runId))]);
|
|
const turns = await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")));
|
|
expect(turns).toHaveLength(1);
|
|
expect(turns[0].contextSnapshot).toMatchObject({ forceFreshSession: true, previousRunId: f.runId, wakeCommentIds: [commentId] });
|
|
expect((await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, queueId)))[0]).toMatchObject({ status: "coalesced", runId: turns[0].id });
|
|
expect((await run(f.runId)).status).toBe("failed");
|
|
if (mode.endsWith("_snapshot")) {
|
|
expect(await readFreshWorkspaceReadiness(db, turns[0])).toMatchObject({
|
|
mode: "fresh_isolated_workspace", retryIntentId: queueId, sourceRunId: f.runId, retainedSourceIds: [f.lease.id] });
|
|
const fresh = path.join(root, randomUUID()); await mkdir(fresh);
|
|
const attached = await attachPreservedWorkspaceSnapshots({ db, companyId: f.companyId, issueId: f.issueId,
|
|
sourceRunIds: [turns[0].id], cwd: fresh });
|
|
expect(attached).toHaveLength(1);
|
|
expect(await readFile(path.join(attached[0].path, "files", "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await readFile(path.join(f.host, "newer.txt"), "utf8")).toBe("newer local work\n");
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
if (mode === "native_snapshot") expect((await run(f.runId)).runnerProfileJson?.nativeWorkspaceSync)
|
|
.toMatchObject({ state: "prepared", finalHostSha256: null });
|
|
}
|
|
}, 30_000);
|
|
|
|
it.each(["copyback_unknown", "source_unavailable", "executor_unknown"])("ordinary task Retry evaluates fresh isolation independently: %s", async failure => {
|
|
const f = await seed("legacy", false);
|
|
const source = await run(f.runId);
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, status: "running" });
|
|
if (failure === "copyback_unknown") await db.update(heartbeatRuns).set({ runnerProfileJson: {
|
|
...source.runnerProfileJson, nativeWorkspaceFinalizationOwner: {
|
|
token: randomUUID(), hostname: "previous-controller", pid: 123, processStartedAt: new Date().toISOString(),
|
|
},
|
|
} }).where(eq(heartbeatRuns.id, f.runId));
|
|
else f.runtime.resumeRunLease.mockRejectedValue(new Error("Provider source unavailable"));
|
|
await recoverRetainedWorkspace(f.input);
|
|
if (failure === "executor_unknown") await db.update(environmentLeases).set({
|
|
metadata: sql`${environmentLeases.metadata} - 'remoteExecutionTermination'`,
|
|
}).where(eq(environmentLeases.id, f.lease.id));
|
|
const commentId = randomUUID(), queueId = randomUUID();
|
|
await db.insert(issueComments).values({ id: commentId, companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "local-board", body: "Use the completed result. Do not publish it again." });
|
|
await db.insert(agentWakeupRequests).values({ id: queueId, companyId: f.companyId, agentId: source.agentId,
|
|
source: "on_demand", reason: "issue_commented", status: "deferred_issue_execution",
|
|
requestedByActorType: "user", requestedByActorId: "local-board",
|
|
payload: { issueId: f.issueId, commentId, _paperclipWakeContext: { issueId: f.issueId,
|
|
wakeReason: "issue_commented", wakeCommentIds: [commentId], wakeCommentId: commentId } } });
|
|
const request = await requestTaskRetry({ db, companyId: f.companyId, issueId: f.issueId, actorId: "local-board",
|
|
actorSource: "local_implicit", observedRunId: f.runId, idempotencyKey: randomUUID() });
|
|
const heartbeat = heartbeatService(db);
|
|
await Promise.all([heartbeat.resumeTaskRetry(f.companyId, request.requestId!), heartbeat.resumeTaskRetry(f.companyId, request.requestId!)]);
|
|
const turns = await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")));
|
|
expect(turns).toHaveLength(failure === "executor_unknown" ? 0 : 1);
|
|
if (turns[0]) {
|
|
expect(turns[0].contextSnapshot?.wakeCommentIds).toEqual([commentId]);
|
|
expect((await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, queueId)))[0])
|
|
.toMatchObject({ runId: turns[0].id, status: "coalesced" });
|
|
expect(await readFreshWorkspaceReadiness(db, turns[0])).toMatchObject({ mode: "fresh_isolated_workspace", retryIntentId: request.requestId });
|
|
// A restarted dispatch worker finds the same run and cannot create another.
|
|
await heartbeatService(db).resumeTaskRetry(f.companyId, request.requestId!);
|
|
expect(await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")))).toHaveLength(1);
|
|
}
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
}, 30_000);
|
|
|
|
it("queues cleanup only after verified persistence and keeps it separate from recovery", async () => {
|
|
const f = await seed();
|
|
await scheduleRecoveredWorkspaceCleanup(db, await run(f.runId));
|
|
expect((await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id)))[0].status).toBe("released");
|
|
await recoverRetainedWorkspace(f.input);
|
|
await scheduleRecoveredWorkspaceCleanup(db, await run(f.runId));
|
|
expect((await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id)))[0]).toMatchObject({
|
|
status: "pending_cleanup", leasePolicy: "ephemeral", metadata: { workspaceRecoveryCleanup: { runId: f.runId } } });
|
|
expect(f.runtime.retryPendingSandboxTeardown).toHaveBeenCalledOnce();
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
}, 30_000);
|
|
|
|
it.each(["single", "multiple", "interrupted"])("continues an existing Retry after signed absence evidence for every retained allocation: %s", async mode => {
|
|
const count = mode === "single" ? 1 : 2;
|
|
const f = await seed();
|
|
const leaseIds = [f.lease.id];
|
|
if (count === 2) {
|
|
const [additional] = await db.insert(environmentLeases).values({ ...f.lease, id: randomUUID(),
|
|
providerLeaseId: randomUUID(), status: "pending_cleanup", cleanupStatus: "failed", releasedAt: null }).returning();
|
|
leaseIds.push(additional.id);
|
|
const recovery = { schema: "paperclip.workspace-restore-recovery.v1", leaseIds };
|
|
await db.update(heartbeatRuns).set({ resultJson: sql`${heartbeatRuns.resultJson} || ${JSON.stringify({ workspaceRestoreRecovery: recovery })}::jsonb` })
|
|
.where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(issueRecoveryActions).set({ evidence: sql`${issueRecoveryActions.evidence} || ${JSON.stringify({ workspaceRestoreRecovery: recovery })}::jsonb` })
|
|
.where(eq(issueRecoveryActions.id, f.actionId));
|
|
}
|
|
const source = await run(f.runId);
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, status: "running" });
|
|
const [stoppedLease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id));
|
|
expect(await prepareStoppedSandboxForRecovery(db, stoppedLease)).not.toBeNull();
|
|
f.runtime.resumeRunLease.mockRejectedValue(new Error("retained allocation unavailable"));
|
|
f.runtime.retryPendingSandboxTeardown.mockRejectedValue(new Error("retained allocation unavailable"));
|
|
const request = await requestTaskRetry({ db, companyId: f.companyId, issueId: f.issueId, actorId: "local-board",
|
|
actorSource: "local_implicit", observedRunId: f.runId, idempotencyKey: randomUUID() });
|
|
const { privateKey, publicKey } = generateKeyPairSync("ed25519");
|
|
vi.stubEnv("PAPERCLIP_CLOUD_STACK_ID", "fixture-stack");
|
|
vi.stubEnv("PAPERCLIP_CLOUD_API_ORIGIN", "https://cloud.example");
|
|
vi.stubEnv("PAPERCLIP_CLOUD_TENANT_SERVER_TOKEN", "fixture-token");
|
|
vi.stubEnv("PAPERCLIP_PRIMARY_HOST", "fixture.paperclip.test");
|
|
vi.stubEnv("PAPERCLIP_AUTH_SECRET", "fixture-auth-secret");
|
|
vi.stubEnv("PAPERCLIP_CLOUD_RUNTIME_IDENTITY_JWKS", JSON.stringify({ keys: [{
|
|
...publicKey.export({ format: "jwk" }), kid: "absence-test", alg: "EdDSA", use: "sig",
|
|
}] }));
|
|
const originalFetch = globalThis.fetch;
|
|
const observed: string[] = [];
|
|
let interrupted = mode === "interrupted";
|
|
const fetchProof = vi.spyOn(globalThis, "fetch").mockImplementation(async (url, init) => {
|
|
if (!String(url).endsWith("/v1/tenant/sandbox-absence")) return originalFetch(url, init);
|
|
const binding = JSON.parse(String(init?.body));
|
|
observed.push(binding.leaseId);
|
|
if (mode === "interrupted" && binding.leaseId === f.lease.id && observed.length > 2) return new Response(null, { status: 503 });
|
|
if (interrupted && binding.leaseId !== f.lease.id) return new Response(null, { status: 503 });
|
|
const seconds = Math.floor(Date.now() / 1000);
|
|
const header = Buffer.from(JSON.stringify({ alg: "EdDSA", typ: "paperclip-sandbox-absence+jwt", kid: "absence-test" })).toString("base64url");
|
|
const payload = Buffer.from(JSON.stringify({ v: 1, iss: "paperclip-cloud", aud: "paperclip-sandbox-absence/v1",
|
|
sub: "fixture-stack", ...binding, absent: true, iat: seconds, exp: seconds + 60 })).toString("base64url");
|
|
return Response.json({ receipt: `${header}.${payload}.${sign(null, Buffer.from(`${header}.${payload}`), privateKey).toString("base64url")}` });
|
|
});
|
|
try {
|
|
await recoverRetainedWorkspace(f.input);
|
|
if (interrupted) {
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).not.toBe("exhausted");
|
|
interrupted = false;
|
|
await makeRetryDue(f.actionId);
|
|
await recoverRetainedWorkspace(f.input);
|
|
}
|
|
const leases = await db.select().from(environmentLeases).where(inArray(environmentLeases.id, leaseIds));
|
|
expect(leases).toHaveLength(count);
|
|
expect(observed).toEqual(mode === "interrupted" ? [leaseIds[0], leaseIds[1], leaseIds[1]] : leaseIds);
|
|
for (const lease of leases) expect(lease).toMatchObject({ status: "released", cleanupStatus: "success",
|
|
metadata: { remoteExecutionTermination: { state: "destroyed" } } });
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)).toMatchObject({
|
|
state: "exhausted", failure: { code: "source_unavailable", retryClass: "use_alternative_strategy" },
|
|
});
|
|
expect(f.runtime.resumeRunLease).not.toHaveBeenCalled();
|
|
expect(f.runtime.retryPendingSandboxTeardown).not.toHaveBeenCalled();
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.companyId, f.companyId))).toHaveLength(0);
|
|
const heartbeat = heartbeatService(db);
|
|
await Promise.all([heartbeat.resumeTaskRetry(f.companyId, request.requestId!), heartbeat.resumeTaskRetry(f.companyId, request.requestId!)]);
|
|
const turns = await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId), eq(heartbeatRuns.status, "queued")));
|
|
expect(turns).toHaveLength(1);
|
|
expect(await readFreshWorkspaceReadiness(db, turns[0])).toMatchObject({ mode: "fresh_isolated_workspace", retryIntentId: request.requestId });
|
|
expect((await run(f.runId)).status).toBe("failed");
|
|
expect(await readFile(path.join(f.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
} finally { fetchProof.mockRestore(); vi.unstubAllEnvs(); }
|
|
}, 30_000);
|
|
|
|
it.each(["pause", "cancelled", "approval", "budget", "reassigned"])("keeps the %s gate after files recover", async gate => {
|
|
const f = await seed(), commentId = randomUUID(), queueId = randomUUID();
|
|
const source = await run(f.runId);
|
|
const [occupied] = await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: source.agentId, status: "running" }).returning();
|
|
await db.insert(issueComments).values({ id: commentId, companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "board", body: "Confirm the existing publication. Do not publish it again." });
|
|
await db.insert(agentWakeupRequests).values({ id: queueId, companyId: f.companyId, agentId: source.agentId,
|
|
source: "automation", reason: "issue_commented", status: "deferred_issue_execution", requestedByActorType: "user", requestedByActorId: "board",
|
|
payload: { issueId: f.issueId, commentId, _paperclipWakeContext: { issueId: f.issueId, wakeReason: "issue_commented", wakeCommentIds: [commentId], wakeCommentId: commentId } } });
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
if (gate === "pause") {
|
|
const holdId = randomUUID();
|
|
await db.insert(issueTreeHolds).values({ id: holdId, companyId: f.companyId, rootIssueId: f.issueId, mode: "pause", status: "active" });
|
|
await db.insert(issueTreeHoldMembers).values({ companyId: f.companyId, holdId, issueId: f.issueId, depth: 0, issueTitle: "Recover existing work", issueStatus: "blocked" });
|
|
}
|
|
if (gate === "cancelled") await db.update(issues).set({ status: "cancelled" }).where(eq(issues.id, f.issueId));
|
|
if (gate === "approval") {
|
|
const approvalId = randomUUID();
|
|
await db.insert(approvals).values({ id: approvalId, companyId: f.companyId, type: "hire_agent", status: "pending", payload: {} });
|
|
await db.insert(issueApprovals).values({ companyId: f.companyId, issueId: f.issueId, approvalId });
|
|
}
|
|
if (gate === "budget") {
|
|
await db.insert(budgetPolicies).values({ companyId: f.companyId, scopeType: "agent", scopeId: source.agentId,
|
|
windowKind: "calendar_month_utc", amount: 100, hardStopEnabled: true });
|
|
await db.insert(costEvents).values({ companyId: f.companyId, agentId: source.agentId,
|
|
provider: "test", model: "test", costCents: 100, occurredAt: new Date(), billingType: "metered_api" });
|
|
}
|
|
if (gate === "reassigned") await db.update(issues).set({ assigneeAgentId: null, assigneeUserId: "board" }).where(eq(issues.id, f.issueId));
|
|
const admission = heartbeatService(db).resumeRemoteStopComments(await run(f.runId));
|
|
if (gate === "budget") await expect(admission).rejects.toThrow("budget hard-stop");
|
|
else await admission;
|
|
expect(await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId),
|
|
sql`${heartbeatRuns.id} not in (${f.runId}::uuid, ${occupied.id}::uuid)`, inArray(heartbeatRuns.status, ["queued", "running"])))).toHaveLength(0);
|
|
expect((await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, queueId)))[0].runId).toBeNull();
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
}, 30_000);
|
|
|
|
it("keeps an unverified old copyback owner fenced without exposing a certification form", async () => {
|
|
const f = await seed();
|
|
const prior = { token: randomUUID(), hostname: "previous-controller", pid: 123, processStartedAt: new Date().toISOString() };
|
|
const source = await run(f.runId);
|
|
await db.update(heartbeatRuns).set({ runnerProfileJson: { ...source.runnerProfileJson, nativeWorkspaceFinalizationOwner: prior } }).where(eq(heartbeatRuns.id, f.runId));
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(f.runtime.resumeRunLease).not.toHaveBeenCalled();
|
|
expect((await run(f.runId)).runnerProfileJson?.nativeWorkspaceFinalizationOwner).toEqual(prior);
|
|
expect(await getExecutionBlocker(db, f.companyId, f.issueId)).toMatchObject({ workspaceRepairRequired: true,
|
|
workspaceRecovery: { status: "queued", canRetry: false } });
|
|
}, 30_000);
|
|
|
|
it("binds the first automatic repair to authoritative old-controller stop proof without an accepted result", async () => {
|
|
const f = await seed("legacy");
|
|
const worker = { provider: "railway" as const, projectId: randomUUID(), environmentId: randomUUID(), serviceId: randomUUID(), deploymentId: randomUUID() };
|
|
const prior = { token: randomUUID(), hostname: "previous-controller", pid: 123, processStartedAt: new Date().toISOString(), worker };
|
|
const source = await run(f.runId);
|
|
await db.update(heartbeatRuns).set({ runnerProfileJson: { ...source.runnerProfileJson, nativeWorkspaceFinalizationOwner: prior } }).where(eq(heartbeatRuns.id, f.runId));
|
|
// This is the verified platform boundary. Signature and identity validation
|
|
// have independent coverage in native-workspace-platform-proof.test.ts.
|
|
const proof = vi.spyOn(workerProof, "requestWorkspaceWorkerStopProof").mockResolvedValue({ status: "verified", proof: {
|
|
v: 1, iss: "paperclip-cloud", aud: "paperclip-workspace-worker-stop/v1", sub: "test-stack", stopped: true,
|
|
iat: Math.floor(Date.now() / 1000), exp: Math.floor(Date.now() / 1000) + 30,
|
|
runId: f.runId, ownerToken: prior.token, worker,
|
|
} });
|
|
try {
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(proof).toHaveBeenCalledWith({ runId: f.runId, ownerToken: prior.token, worker });
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await db.select().from(nativeRunResults).where(eq(nativeRunResults.companyId, f.companyId))).toHaveLength(0);
|
|
} finally { proof.mockRestore(); }
|
|
}, 30_000);
|
|
|
|
it("retries persistence after rollback without repeating or discarding the copied work", async () => {
|
|
const f = await seed();
|
|
const failCommit = vi.spyOn(activity, "persistActivity").mockRejectedValueOnce(new Error("Interrupted receipt transaction"));
|
|
try { await recoverRetainedWorkspace(f.input); } finally { failCommit.mockRestore(); }
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(false);
|
|
expect(await db.select().from(workspaceOperations).where(and(eq(workspaceOperations.companyId, f.companyId), eq(workspaceOperations.phase, "workspace_recovery")))).toHaveLength(1);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
const pending = await action(f.actionId);
|
|
await db.update(issueRecoveryActions).set({ evidence: { ...pending.evidence, systemRecovery: { ...readSystemRecovery(pending.evidence), nextAttemptAt: 0 } } }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(await hasVerifiedWorkspaceRepair(db, await run(f.runId))).toBe(true);
|
|
}, 30_000);
|
|
|
|
it("persists six automatic attempts and admits one new generation per deliberate retry", async () => {
|
|
const f = await seed();
|
|
f.runtime.resumeRunLease.mockRejectedValue(new Error("Temporary provider failure"));
|
|
for (let attempt = 1; attempt <= 6; attempt++) {
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.attempts).toBe(attempt);
|
|
if (attempt < 6) await makeRetryDue(f.actionId);
|
|
}
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(f.runtime.resumeRunLease).toHaveBeenCalledTimes(6);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("exhausted");
|
|
const before = readSystemRecovery((await action(f.actionId)).evidence)!;
|
|
const request = { ...f.input, idempotencyKey: randomUUID(), actorId: "board" };
|
|
await requestWorkspaceRecovery(request); await requestWorkspaceRecovery(request);
|
|
await recoverRetainedWorkspace(f.input); await recoverRetainedWorkspace(f.input);
|
|
const after = readSystemRecovery((await action(f.actionId)).evidence)!;
|
|
expect(f.runtime.resumeRunLease).toHaveBeenCalledTimes(7);
|
|
expect(after.generationId).not.toBe(before.generationId);
|
|
expect(after.attempts).toBe(1);
|
|
expect(after.revision).toBeGreaterThan(before.revision);
|
|
}, 30_000);
|
|
|
|
it("discovers historical sources in the periodic sweep without creating a user request", async () => {
|
|
const f = await seed("legacy"), onRecovered = vi.fn(async () => {});
|
|
await expect.poll(async () => {
|
|
await reconcileRetainedWorkspaces(db, f.input.environmentRuntime, onRecovered);
|
|
return readWorkspaceRecovery((await action(f.actionId)).evidence)?.status;
|
|
}, { timeout: 20_000 }).toBe("recovered");
|
|
await expect.poll(() => onRecovered.mock.calls.length).toBeGreaterThan(0);
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
}, 30_000);
|
|
|
|
it("uses the retained-source index rather than scanning unrelated recovery history", async () => {
|
|
const f = await seed();
|
|
await db.insert(issueRecoveryActions).values(Array.from({ length: 3000 }, (_, i) => ({
|
|
companyId: f.companyId, sourceIssueId: f.issueId, status: "resolved", kind: "active_run_watchdog",
|
|
cause: "unrelated_history", fingerprint: `history-${i}`, nextAction: "", evidence: {},
|
|
})));
|
|
await db.execute(sql`analyze ${issueRecoveryActions}`);
|
|
await adoptHistoricalWorkspaceRecoveries(db);
|
|
const plan = await db.execute(sql`explain (analyze, format json) ${retainedWorkspaceCandidates(db)}`);
|
|
expect(JSON.stringify(plan)).toContain("issue_recovery_actions_system_recovery_due_idx");
|
|
expect((await retainedWorkspaceCandidates(db)).length).toBeLessThanOrEqual(5);
|
|
}, 30_000);
|
|
|
|
it.each([false, true])("discovers a committed repair with unscheduled destruction without waiting input (reconciled: %s)", async reconciled => {
|
|
const f = await seed();
|
|
await recoverRetainedWorkspace(f.input);
|
|
if (reconciled) {
|
|
const current = await action(f.actionId);
|
|
await db.update(issueRecoveryActions).set({ evidence: { ...current.evidence, executionReconciliation: { runId: f.runId } } }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
}
|
|
expect(await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId))).toHaveLength(0);
|
|
expect((await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id)))[0].status).toBe("released");
|
|
const restarted = createDb(connectionString), onRecovered = vi.fn(async (_source: { id: string }) => {});
|
|
try {
|
|
await expect.poll(async () => {
|
|
await reconcileRetainedWorkspaces(restarted, f.input.environmentRuntime, onRecovered);
|
|
return (await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id)))[0].status;
|
|
}, { timeout: 20_000 }).toBe("pending_cleanup");
|
|
await expect.poll(() => onRecovered.mock.calls.filter(([source]) => source.id === f.runId).length).toBe(1);
|
|
expect(await retainedWorkspaceCandidates(restarted)).not.toEqual(expect.arrayContaining([expect.objectContaining({ action: expect.objectContaining({ id: f.actionId }) })]));
|
|
expect(f.runtime.resumeRunLease.mock.calls.filter(([input]) => input.lease.id === f.lease.id)).toHaveLength(1);
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
} finally {
|
|
await restarted.$client.end();
|
|
}
|
|
}, 30_000);
|
|
async function queuedResetFixture(nativeDescriptor = true, ephemeral = false) {
|
|
const f = await seed("native", nativeDescriptor, "keep_running", false, ephemeral), original = await run(f.runId);
|
|
await instanceSettingsService(db).updateExperimental({ enableAgentChat: true });
|
|
await db.delete(issueRecoveryActions).where(eq(issueRecoveryActions.id, f.actionId));
|
|
await db.update(issues).set({ conversationAgentId: original.agentId, conversationUserId: "operator", conversationState: "active", status: "in_progress" }).where(eq(issues.id, f.issueId));
|
|
await db.update(heartbeatRuns).set({ status: "cancelled", nativePhase: "terminal_failure", errorCode: "cancelled", resultJson: {
|
|
cancelledByActorType: "user", cancelledByUserId: "operator",
|
|
cancellation: requestedRunCancellation({ cancelledByActorType: "user", cancelledByUserId: "operator" }, "Stop"),
|
|
executionCancellation: { state: "requested", requestedAt: new Date().toISOString() },
|
|
nativeCancellation: { schema: "paperclip.native-cancellation.v1", runId: f.runId, companyId: f.companyId, issueId: f.issueId,
|
|
scope: "run", reasonCode: "cancellation_run_only", dispatchState: "acknowledged", dispatched: true,
|
|
intentAuditId: randomUUID(), acknowledgementAuditId: randomUUID() },
|
|
} }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(environmentLeases).set({ status: "active", releasedAt: null, cleanupStatus: null, metadata: f.lease.metadata }).where(eq(environmentLeases.id, f.lease.id));
|
|
const [reset] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "operator", body: "/new" }).returning();
|
|
expect(await getExecutionBlocker(db, f.companyId, f.issueId, { conversationResetCommentId: reset.id })).toBeNull();
|
|
const context = { source: "issue.comment", issueId: f.issueId, wakeReason: "issue_commented", wakeCommentId: reset.id, wakeCommentIds: [reset.id] };
|
|
const payload = withQueuedCommentIdsInWakePayload({ issueId: f.issueId, commentId: reset.id, mutation: "comment", _paperclipWakeContext: context }, [reset.id]);
|
|
const [wake] = await db.insert(agentWakeupRequests).values({ companyId: f.companyId, agentId: original.agentId,
|
|
source: "on_demand", triggerDetail: "manual", reason: "issue_commented", status: "claimed", claimedAt: new Date(),
|
|
requestedByActorType: "user", requestedByActorId: "operator", payload }).returning();
|
|
const [queued] = await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: original.agentId,
|
|
invocationSource: "on_demand", triggerDetail: "manual", status: "queued", runtimeMode: "legacy", responsibleUserId: "operator",
|
|
wakeupRequestId: wake.id, sessionIdBefore: "saved-native-session-before-reset", contextSnapshot: context }).returning();
|
|
await db.update(agentWakeupRequests).set({ runId: queued.id }).where(eq(agentWakeupRequests.id, wake.id));
|
|
await db.update(issues).set({ executionRunId: queued.id, executionAgentNameKey: "Worker", executionLockedAt: new Date() }).where(eq(issues.id, f.issueId));
|
|
const deps = callbacks(db), control = createHeartbeatRunControl(db, deps), execution = createAdapterExecutionControl();
|
|
const queueDeps = createHeartbeatQueueTestDependencies(db);
|
|
queueDeps.activeRunExecutions = deps.activeRunExecutions;
|
|
queueDeps.getAgent.mockResolvedValue((await db.select().from(agents).where(eq(agents.id, original.agentId)))[0]);
|
|
queueDeps.instanceSettings.getExperimental.mockResolvedValue({ enableAgentChat: true, enableNativeRunner: true } as Awaited<ReturnType<typeof queueDeps.instanceSettings.getExperimental>>);
|
|
const queue = createHeartbeatQueue(db, queueDeps);
|
|
vi.mocked(deps.enqueueWakeup).mockImplementation(queue.enqueueWakeup);
|
|
await db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: original.agentId, status: "running", invocationSource: "assignment" });
|
|
deps.activeRunExecutions.add(f.runId); adapterExecutionControls.set(f.runId, execution);
|
|
return { f, original, reset, wake, queued, deps, control, execution, queueDeps, queue };
|
|
}
|
|
|
|
async function coalesceQueuedReset(t: Awaited<ReturnType<typeof queuedResetFixture>>) {
|
|
const { f, original, reset, wake, queued } = t;
|
|
const [followup] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "operator", body: "/new" }).returning();
|
|
const incoming = { source: "issue.comment", issueId: f.issueId, wakeReason: "issue_commented",
|
|
wakeCommentId: followup.id, wakeCommentIds: [followup.id] };
|
|
await db.transaction(async tx => {
|
|
const scope = createAdmissionTransactionScope(f.companyId, tx as unknown as typeof db);
|
|
const actor = { companyId: f.companyId, wakeupRequestId: wake.id, requestedByActorType: "user", requestedByActorId: "operator" };
|
|
expect(await createWakeAdmissionReader().matchesActiveWakeActor(scope, actor)).toBe(true);
|
|
expect(await createWakeAdmissionReader().matchesActiveWakeActor(scope, { ...actor, requestedByActorId: "other-user" })).toBe(false);
|
|
await createWakeAdmissionWriter().coalesceIntoActiveExecutionRun(scope, {
|
|
companyId: f.companyId, agentId: original.agentId, activeExecutionRunId: queued.id,
|
|
mergedContextSnapshot: mergeCoalescedContextSnapshot((await run(queued.id)).contextSnapshot, incoming,
|
|
{ preserveExistingInteractionContinuation: true }),
|
|
source: "on_demand", triggerDetail: "manual",
|
|
payload: withQueuedCommentIdsInWakePayload({ issueId: f.issueId, mutation: "comment", _paperclipWakeContext: incoming }, [followup.id]),
|
|
requestedByActorType: "user", requestedByActorId: "operator", idempotencyKey: null,
|
|
});
|
|
});
|
|
const [originalWake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
const [receipt] = await db.select().from(agentWakeupRequests).where(and(eq(agentWakeupRequests.runId, queued.id), eq(agentWakeupRequests.status, "coalesced")));
|
|
expect(originalWake.payload?.commentId).toBe(reset.id);
|
|
expect((await run(queued.id)).contextSnapshot).toMatchObject({ wakeCommentId: followup.id,
|
|
wakeCommentIds: [reset.id, followup.id] });
|
|
return { followup, receipt };
|
|
}
|
|
|
|
async function ordinaryMixedQueuedFixture(kind: "active" | "deferred" | "deferred-different" | "system-deferred") {
|
|
const t = await queuedResetFixture(true, kind !== "active"), { f, original, reset, wake, queued, queueDeps, queue, control } = t;
|
|
await db.update(issues).set({ conversationAgentId: null, conversationUserId: null, conversationState: null,
|
|
executionRunId: null, executionAgentNameKey: null, executionLockedAt: null }).where(eq(issues.id, f.issueId));
|
|
await db.update(heartbeatRuns).set({ wakeupRequestId: null }).where(eq(heartbeatRuns.id, queued.id));
|
|
await db.update(agentWakeupRequests).set({ runId: null }).where(eq(agentWakeupRequests.id, wake.id));
|
|
await db.delete(heartbeatRuns).where(eq(heartbeatRuns.id, queued.id));
|
|
await db.delete(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
const reader = createWakeAdmissionReader(), sameActor = vi.spyOn(reader, "matchesActiveWakeActor");
|
|
queueDeps.wakeQueue.createAdmissionTransactionScope.mockImplementation(createAdmissionTransactionScope);
|
|
queueDeps.wakeQueue.admitWakeBehindIssueExecution.mockImplementation(createAdmitWakeBehindIssueExecution({
|
|
reader, writer: createWakeAdmissionWriter(), helpers: { filterZombieCoalesceTarget, mergeCoalescedContextSnapshot,
|
|
shouldDeferFollowupWakeForSameIssue, shouldQueueFollowupForRunningIssueWake },
|
|
}));
|
|
queueDeps.liveRunExecutions.has.mockImplementation(id => t.deps.activeRunExecutions.has(id));
|
|
const stopped = await run(f.runId);
|
|
if (kind !== "active") {
|
|
await db.update(heartbeatRuns).set({ status: "running", finishedAt: null, errorCode: null,
|
|
nativePhase: "running", resultJson: null }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(nativeRunFinalizations).set({ phase: "running" }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
await db.update(issues).set({ executionRunId: f.runId, executionAgentNameKey: "worker", executionLockedAt: new Date() }).where(eq(issues.id, f.issueId));
|
|
}
|
|
const enqueue = async (comment: typeof reset) => queue.enqueueWakeup(original.agentId, {
|
|
source: "automation", triggerDetail: "system", reason: "issue_commented", requestedByActorType: "user", requestedByActorId: comment.authorUserId!,
|
|
payload: { issueId: f.issueId, commentId: comment.id, mutation: "comment", resumeIntent: true, followUpRequested: true },
|
|
contextSnapshot: { issueId: f.issueId, taskId: f.issueId, commentId: comment.id, wakeCommentId: comment.id,
|
|
wakeCommentIds: [comment.id], source: "issue.comment", wakeReason: "issue_commented", resumeIntent: true, followUpRequested: true },
|
|
});
|
|
if (kind === "system-deferred") expect(await queue.enqueueWakeup(original.agentId, {
|
|
source: "automation", triggerDetail: "system", reason: "issue_updated", requestedByActorType: "system", requestedByActorId: "second-user",
|
|
payload: { issueId: f.issueId, mutation: "update" },
|
|
contextSnapshot: { issueId: f.issueId, taskId: f.issueId, source: "issue.updated", wakeReason: "issue_updated", forceFreshSession: true },
|
|
})).toBeNull();
|
|
const first = await enqueue(reset);
|
|
const [followup] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "second-user", body: "Please also preserve the second user's input." }).returning();
|
|
const second = await enqueue(followup);
|
|
const ids = [reset.id, followup.id];
|
|
let admitted = first;
|
|
if (kind === "active") { expect(first?.status).toBe("queued"); expect(second?.id).toBe(first!.id); }
|
|
else {
|
|
expect(first).toBeNull(); expect(second).toBeNull();
|
|
const [pending] = await db.select().from(agentWakeupRequests).where(and(eq(agentWakeupRequests.companyId, f.companyId),
|
|
eq(agentWakeupRequests.agentId, original.agentId), eq(agentWakeupRequests.status, "deferred_issue_execution")));
|
|
expect(pending.payload?._paperclipWakeContext).toMatchObject({ wakeCommentIds: ids });
|
|
expect(pending.requestedByActorType).toBe(kind === "system-deferred" ? "system" : "user");
|
|
expect(pending.requestedByActorId).toBe(kind === "system-deferred" ? "second-user" : "operator");
|
|
await db.update(heartbeatRuns).set({ status: "cancelled", nativePhase: "terminal_failure", finishedAt: new Date(),
|
|
errorCode: stopped.errorCode, resultJson: stopped.resultJson }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(nativeRunFinalizations).set({ phase: "terminal_failure" }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
const [currentLease] = await db.select().from(environmentLeases).where(eq(environmentLeases.id, f.lease.id));
|
|
const stopping = await prepareSandboxStopAndRetain(db, currentLease);
|
|
expect(stopping).not.toBeNull();
|
|
expect(await settleStopOnlyCleanup(db, stopping!, { attemptId: String(stopping!.metadata!.pendingCleanupAttemptId),
|
|
receipt: { providerLeaseId: f.providerLeaseId, state: "stopped" } })).not.toBeNull();
|
|
await control.acknowledgeRemoteStop(f.runId, f.companyId);
|
|
const [last] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: kind === "deferred-different" ? "third-user" : "operator", body: "Resume after Stop and preserve all three inputs in their original order." }).returning();
|
|
expect(last.createdAt.getTime()).toBeGreaterThan((await run(f.runId)).finishedAt!.getTime());
|
|
ids.push(last.id); admitted = await enqueue(last);
|
|
}
|
|
expect(admitted?.status).toBe("queued");
|
|
const current = await run(admitted!.id);
|
|
const [claimedWake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, current.wakeupRequestId!));
|
|
expect(current.contextSnapshot?.wakeCommentIds).toEqual(ids);
|
|
expect(sameActor).not.toHaveBeenCalled();
|
|
expect(queueDeps.wakeQueue.admitWakeBehindIssueExecution.mock.calls.every(([, input]) => input.durableReceipt === undefined)).toBe(true);
|
|
expect(queueDeps.executeRun).not.toHaveBeenCalled();
|
|
return { ...t, queued: current, wake: claimedWake, followup, expectedIds: ids, latestAuthor: kind === "deferred-different" ? "third-user" : kind === "active" ? "second-user" : "operator" };
|
|
}
|
|
|
|
async function exerciseQueuedReset(nativeDescriptor: boolean, coalesced = false, ordinary?: "active" | "deferred" | "deferred-different" | "system-deferred", earlierCandidates = false) {
|
|
const mixed = ordinary ? await ordinaryMixedQueuedFixture(ordinary) : null;
|
|
const t = mixed ?? await queuedResetFixture(nativeDescriptor);
|
|
const { f, original, reset, wake, queued, deps, control, execution, queueDeps, queue } = t;
|
|
const followup = coalesced ? (await coalesceQueuedReset(t)).followup : null;
|
|
const expectedIds = mixed?.expectedIds ?? (followup ? [reset.id, followup.id] : [reset.id]);
|
|
const responsible = queued.responsibleUserId;
|
|
try {
|
|
expect(await queue.claimQueuedRun(queued)).toBeNull();
|
|
expect((await run(queued.id)).status).toBe("queued");
|
|
await recordRequiredWorkspaceRestoreFailure(db, await run(f.runId), { workspaceRestoreFailure: "restore_failed" }, f.lease);
|
|
const [preservation] = await db.select().from(issueRecoveryActions).where(and(eq(issueRecoveryActions.companyId, f.companyId),
|
|
eq(issueRecoveryActions.sourceIssueId, f.issueId), eq(issueRecoveryActions.fingerprint, `native-workspace:${f.runId}`)));
|
|
f.actionId = preservation.id; f.input.actionId = preservation.id;
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.failure?.code).toBe("agent_stop_unconfirmed");
|
|
await db.update(environmentLeases).set({ status: "released", releasedAt: new Date(), cleanupStatus: "success", metadata: { ...f.lease.metadata,
|
|
remoteExecutionTermination: remoteTerminationReceipt(f.lease, { providerLeaseId: f.providerLeaseId, state: "stopped" }) } }).where(eq(environmentLeases.id, f.lease.id));
|
|
await control.acknowledgeRemoteStop(f.runId, f.companyId);
|
|
expect((await run(f.runId)).resultJson?.executionCancellation).toMatchObject({ state: "acknowledged", proof: "provider_termination_receipt" });
|
|
deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId);
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)!.nextAttemptAt!).toBeLessThanOrEqual(Date.now());
|
|
const earlier: Awaited<ReturnType<typeof seed>>[] = [];
|
|
if (earlierCandidates) for (let index = 1; index <= 2; index++) {
|
|
const candidate = await seed("native", false, "keep_running");
|
|
const id = `00000000-0000-4000-8000-${String(index).padStart(12, "0")}`;
|
|
await db.update(issueRecoveryActions).set({ id }).where(and(eq(issueRecoveryActions.id, candidate.actionId), eq(issueRecoveryActions.companyId, candidate.companyId)));
|
|
candidate.actionId = id; candidate.input.actionId = id;
|
|
await db.update(environmentLeases).set({ status: "active", releasedAt: null, cleanupStatus: null, metadata: candidate.lease.metadata }).where(eq(environmentLeases.id, candidate.lease.id));
|
|
await requestWorkspaceRecovery({ ...candidate.input, idempotencyKey: `earlier-${index}`, actorId: "board" });
|
|
earlier.push(candidate);
|
|
}
|
|
const ownedCompanies = [f.companyId, ...earlier.map(candidate => candidate.companyId)];
|
|
const seededRuns = earlierCandidates ? await db.select({ id: heartbeatRuns.id }).from(heartbeatRuns).where(inArray(heartbeatRuns.companyId, ownedCompanies)) : [];
|
|
if (earlierCandidates) {
|
|
await adoptHistoricalWorkspaceRecoveries(db);
|
|
expect((await retainedWorkspaceCandidates(db, undefined, true)).slice(0, 2).map(row => row.action.id)).toEqual(earlier.map(candidate => candidate.actionId));
|
|
}
|
|
let ownedRecoveryObserved = false;
|
|
const onRecovered = vi.fn(async (current: Awaited<ReturnType<typeof run>>) => {
|
|
if (current.id !== f.runId) return;
|
|
if (nativeDescriptor) expect(await inspectWorkspaceSource(db, { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, actionId: f.actionId })).toMatchObject({ status: "verified", scope: "recorded_source_identity" });
|
|
await Promise.all([control.resumeRemoteStopComments(current), control.resumeRemoteStopComments(current)]);
|
|
ownedRecoveryObserved = true;
|
|
});
|
|
const sweepSizes: number[] = [];
|
|
let firstSweepOwned: boolean | undefined;
|
|
const repair = async () => {
|
|
await vi.waitFor(async () => {
|
|
const work = await reconcileRetainedWorkspaces(db, f.environmentRuntime, onRecovered) ?? [];
|
|
sweepSizes.push(work.length);
|
|
await Promise.all(work);
|
|
firstSweepOwned ??= ownedRecoveryObserved;
|
|
expect(ownedRecoveryObserved).toBe(true);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
}, { timeout: 10_000, interval: 10 });
|
|
expect(sweepSizes.every(size => size <= 2)).toBe(true);
|
|
};
|
|
if (!nativeDescriptor) await repair();
|
|
expect(await queue.startNextQueuedRunForAgent(original.agentId)).toEqual([]);
|
|
expect(await queue.claimQueuedRun(await run(queued.id))).toBeNull();
|
|
const [heldWake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
expect(await isPreDispatchExecutionWaitVerified(db, await run(queued.id))).toBe(true);
|
|
const admissionEvents = await db.select().from(heartbeatRunEvents).where(eq(heartbeatRunEvents.runId, queued.id));
|
|
expect(admissionEvents.map(event => event.eventType)).toEqual(["lifecycle"]);
|
|
expect((await run(queued.id)).nextEventSeq).toBe(2);
|
|
expect((await run(queued.id)).sessionIdBefore).toBe(queued.sessionIdBefore);
|
|
expect((await run(queued.id)).responsibleUserId).toBe(responsible);
|
|
const scheduleRecoveryRetry = vi.fn(async (_runId: string) => null);
|
|
const recoveryDependencies = { enqueueWakeup: vi.fn(async () => null), scheduleRecoveryRetry, liveRunExecutions: { has: () => false } };
|
|
const [task] = await db.select().from(issues).where(eq(issues.id, f.issueId));
|
|
const recover = (database: typeof db) => recoveryService(database, recoveryDependencies).reconcileStrandedAssignedIssues({ issueCreatedAtGte: task.createdAt });
|
|
if (nativeDescriptor) {
|
|
const rollback = new Error("restore_counterfactual_skipped_wake");
|
|
await expect(db.transaction(async tx => {
|
|
await tx.update(agentWakeupRequests).set({ status: "skipped" }).where(eq(agentWakeupRequests.id, wake.id));
|
|
await recover(tx as unknown as typeof db);
|
|
expect(scheduleRecoveryRetry).toHaveBeenCalledWith(queued.id);
|
|
throw rollback;
|
|
})).rejects.toBe(rollback);
|
|
scheduleRecoveryRetry.mockClear();
|
|
await recover(db);
|
|
expect(scheduleRecoveryRetry).not.toHaveBeenCalledWith(queued.id);
|
|
}
|
|
expect(queueDeps.executeRun).not.toHaveBeenCalled();
|
|
if (nativeDescriptor) await repair();
|
|
expect(onRecovered).toHaveBeenCalledWith(expect.objectContaining({ id: f.runId, companyId: f.companyId }));
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
if (nativeDescriptor) expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
if (earlierCandidates) {
|
|
expect(sweepSizes[0]).toBe(2);
|
|
expect(firstSweepOwned).toBe(false);
|
|
expect(sweepSizes.length).toBeGreaterThan(1);
|
|
expect((await db.select({ id: heartbeatRuns.id }).from(heartbeatRuns).where(inArray(heartbeatRuns.companyId, ownedCompanies))).map(row => row.id).sort()).toEqual(seededRuns.map(row => row.id).sort());
|
|
for (const candidate of [f, ...earlier]) expect(await readFile(path.join(candidate.remote, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
}
|
|
await Promise.all([control.resumeRemoteStopComments(await run(f.runId)),
|
|
control.resumeRemoteStopComments(await run(f.runId))]);
|
|
const [afterWake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
expect(heldWake.status).toBe(nativeDescriptor ? "deferred_issue_execution" : "skipped");
|
|
if (!nativeDescriptor) {
|
|
expect(afterWake.status).toBe("skipped");
|
|
expect(deps.enqueueWakeup).not.toHaveBeenCalled();
|
|
expect(queueDeps.executeRun).not.toHaveBeenCalled();
|
|
return;
|
|
}
|
|
expect(afterWake).toMatchObject({ status: "coalesced", runId: expect.any(String) });
|
|
expect(afterWake.runId).not.toBe(queued.id);
|
|
expect((await action(f.actionId)).evidence).toMatchObject({ automaticRecovery: { replay: "explicit_user_continuation" } });
|
|
expect(await run(queued.id)).toMatchObject({ status: "cancelled", errorCode: "execution_reconciliation_required", startedAt: null });
|
|
const successors = await db.select().from(heartbeatRuns).where(and(eq(heartbeatRuns.companyId, f.companyId),
|
|
sql`${heartbeatRuns.contextSnapshot}->>'issueId' = ${f.issueId}`, eq(heartbeatRuns.status, "queued")));
|
|
expect(successors).toHaveLength(1);
|
|
expect(successors[0].contextSnapshot?.wakeCommentIds).toEqual(expectedIds);
|
|
if (mixed) expect(deps.enqueueWakeup).toHaveBeenCalledWith(original.agentId, expect.objectContaining({ requestedByActorType: "user", requestedByActorId: mixed.latestAuthor }), wake.id);
|
|
await recover(db);
|
|
expect(scheduleRecoveryRetry).not.toHaveBeenCalledWith(queued.id);
|
|
expect(await db.select().from(heartbeatRuns).where(eq(heartbeatRuns.retryOfRunId, queued.id))).toHaveLength(0);
|
|
expect((await run(queued.id)).nextEventSeq).toBe(2);
|
|
expect(queueDeps.executeRun).not.toHaveBeenCalled();
|
|
} finally { deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId); }
|
|
}
|
|
|
|
it.each([
|
|
{ nativeDescriptor: true, title: "preserves an already queued reset until real native copyback replays its saved wake" },
|
|
{ nativeDescriptor: false, title: "does not replay a snapshot-only source without a verified native baseline" },
|
|
])("$title", async ({ nativeDescriptor }) => exerciseQueuedReset(nativeDescriptor), 30_000);
|
|
|
|
it("reaches its queued snapshot-only reset after earlier recovery candidates without starting agent work", async () => {
|
|
const previous = db;
|
|
db = createDb(connectionString);
|
|
try { await exerciseQueuedReset(false, false, undefined, true); }
|
|
finally { await db.$client.end(); db = previous; }
|
|
}, 30_000);
|
|
|
|
it("preserves coalesced queued user comments until real native copyback replays the original wake", async () => {
|
|
await exerciseQueuedReset(true, true);
|
|
}, 30_000);
|
|
|
|
it("preserves each ordinary mixed-author comment across queued Stop and native copyback", async () => {
|
|
await exerciseQueuedReset(true, false, "active");
|
|
}, 30_000);
|
|
|
|
it("preserves merged-deferred mixed-author comments across queued Stop and native copyback", async () => {
|
|
await exerciseQueuedReset(true, false, "deferred");
|
|
}, 30_000);
|
|
|
|
it("preserves mixed-author comments merged into a system-owned deferred carrier across Stop and native copyback", async () => {
|
|
await exerciseQueuedReset(true, false, "system-deferred");
|
|
}, 30_000);
|
|
|
|
it("preserves mixed-author deferred inputs when the resuming third user differs from the carrier", async () => {
|
|
await exerciseQueuedReset(true, false, "deferred-different");
|
|
}, 30_000);
|
|
|
|
it.each(["missing_primary", "agent_namespace", "system_namespace", "foreign_link", "foreign_carrier_run", "foreign_receipt_user", "array_only"] as const)("rejects mixed-author queue authority forgery: %s", async fault => {
|
|
const t = await ordinaryMixedQueuedFixture("system-deferred"), { f, wake, queued, deps, execution, queueDeps, queue, followup } = t;
|
|
try {
|
|
await recordRequiredWorkspaceRestoreFailure(db, await run(f.runId), { workspaceRestoreFailure: "restore_failed" }, f.lease);
|
|
deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId);
|
|
const sourceIds = queued.contextSnapshot!.queuedCommentSourceWakeIds as string[];
|
|
const [receipt] = await db.select().from(agentWakeupRequests).where(and(eq(agentWakeupRequests.companyId, f.companyId),
|
|
eq(agentWakeupRequests.status, "coalesced"), eq(agentWakeupRequests.requestedByActorType, "user"),
|
|
sql`${agentWakeupRequests.payload}->>'commentId' = ${followup.id}`));
|
|
expect(receipt).toBeDefined();
|
|
if (fault === "missing_primary") await db.delete(agentWakeupRequests).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "agent_namespace" || fault === "system_namespace") await db.update(agentWakeupRequests).set({ requestedByActorType: fault === "agent_namespace" ? "agent" : "system", requestedByActorId: "second-user" }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "foreign_link") await db.update(agentWakeupRequests).set({ payload: { ...receipt.payload, coalescedIntoWakeupRequestId: randomUUID() } }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "foreign_carrier_run") await db.update(agentWakeupRequests).set({ runId: f.runId }).where(eq(agentWakeupRequests.id, sourceIds[0]));
|
|
if (fault === "foreign_receipt_user") await db.update(agentWakeupRequests).set({ requestedByActorId: "operator" }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "array_only") {
|
|
const [injected] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId, authorType: "user", authorUserId: "second-user", body: "An array is not a saved admission receipt." }).returning();
|
|
await db.update(heartbeatRuns).set({ contextSnapshot: mergeCoalescedContextSnapshot(queued.contextSnapshot,
|
|
{ issueId: f.issueId, wakeReason: "issue_commented", wakeCommentId: injected.id, wakeCommentIds: [injected.id] }, { preserveExistingInteractionContinuation: true }) }).where(eq(heartbeatRuns.id, queued.id));
|
|
await db.update(agentWakeupRequests).set({ payload: withQueuedCommentIdsInWakePayload(receipt.payload!, [followup.id, injected.id]) }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
}
|
|
expect(await queue.claimQueuedRun(await run(queued.id))).toBeNull();
|
|
const [after] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
expect(after.status).toBe("skipped");
|
|
expect(queueDeps.executeRun).not.toHaveBeenCalled(); expect(deps.enqueueWakeup).not.toHaveBeenCalled();
|
|
} finally { deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId); }
|
|
}, 30_000);
|
|
|
|
it("preserves ordinary no-issue deferred merge semantics", async () => {
|
|
const f = await seed(), original = await run(f.runId), writer = createWakeAdmissionWriter();
|
|
const [carrier] = await db.insert(agentWakeupRequests).values({ companyId: f.companyId, agentId: original.agentId,
|
|
source: "automation", status: "deferred_issue_execution", payload: { text: "original" }, requestedByActorType: "system" }).returning();
|
|
await db.transaction(tx => writer.mergeIntoExistingDeferredWake(createAdmissionTransactionScope(f.companyId, tx as unknown as typeof db), {
|
|
companyId: f.companyId, existingDeferredWakeId: carrier.id, mergedPayload: { text: "merged" }, nextCoalescedCount: 1,
|
|
coalescedReceipt: { agentId: original.agentId, source: "automation", triggerDetail: null, reason: null, payload: { text: "saved" }, requestedByActorType: "system", requestedByActorId: null, idempotencyKey: null, runId: null },
|
|
}));
|
|
const [after] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, carrier.id));
|
|
expect(after.payload).toEqual({ text: "merged" });
|
|
const [receipt] = await db.select().from(agentWakeupRequests).where(and(eq(agentWakeupRequests.companyId, f.companyId), eq(agentWakeupRequests.status, "coalesced")));
|
|
expect(receipt.payload).toEqual({ text: "saved", coalescedIntoWakeupRequestId: carrier.id });
|
|
});
|
|
|
|
it("rolls back saved primary and carrier overwrite if incoming mixed-author receipt persistence fails", async () => {
|
|
const f = await seed(), original = await run(f.runId), writer = createWakeAdmissionWriter();
|
|
const [first] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId, authorType: "user", authorUserId: "operator", body: "First saved input" }).returning();
|
|
const [second] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId, authorType: "user", authorUserId: "second-user", body: "Second saved input" }).returning();
|
|
const [carrier] = await db.insert(agentWakeupRequests).values({ companyId: f.companyId, agentId: original.agentId,
|
|
source: "automation", status: "deferred_issue_execution", payload: { issueId: f.issueId, commentId: first.id }, requestedByActorType: "user", requestedByActorId: "operator" }).returning();
|
|
const merge = (tx: Parameters<Parameters<typeof db.transaction>[0]>[0], id?: string) => writer.mergeIntoExistingDeferredWake(createAdmissionTransactionScope(f.companyId, tx as unknown as typeof db), {
|
|
companyId: f.companyId, existingDeferredWakeId: carrier.id, mergedPayload: { issueId: f.issueId, commentId: second.id }, nextCoalescedCount: 1,
|
|
coalescedReceipt: { ...(id ? { id } : {}), agentId: original.agentId, source: "automation", triggerDetail: null, reason: "issue_commented", payload: { issueId: f.issueId, commentId: second.id, coalescedIntoWakeupRequestId: "untrusted" }, requestedByActorType: "user", requestedByActorId: "second-user", idempotencyKey: null, runId: null },
|
|
});
|
|
await expect(db.transaction(tx => merge(tx, carrier.id))).rejects.toThrow();
|
|
expect((await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.companyId, f.companyId)))).toEqual([carrier]);
|
|
await db.transaction(tx => merge(tx)); await db.transaction(tx => merge(tx));
|
|
const proofs = await db.select().from(agentWakeupRequests).where(and(eq(agentWakeupRequests.companyId, f.companyId), eq(agentWakeupRequests.status, "coalesced")));
|
|
expect(proofs.filter(receipt => receipt.payload?.commentId === first.id)).toHaveLength(1);
|
|
expect(proofs.every(receipt => receipt.payload?.coalescedIntoWakeupRequestId === carrier.id)).toBe(true);
|
|
expect(proofs.find(receipt => receipt.payload?.commentId === first.id)?.requestedByActorId).toBe("operator");
|
|
});
|
|
|
|
it.each(["missing_receipt", "wrong_receipt_run", "wrong_receipt_agent", "wrong_receipt_company", "wrong_receipt_user", "wrong_receipt_issue", "terminal_receipt", "foreign_comment_user", "deleted_followup", "unrecorded_comment", "unrecorded_original"] as const)("keeps coalesced Stop recovery input closed without saved receipt authority: %s", async fault => {
|
|
const t = await queuedResetFixture(), { f, wake, queued, deps, execution, queueDeps, queue } = t;
|
|
try {
|
|
const { followup, receipt } = await coalesceQueuedReset(t);
|
|
await recordRequiredWorkspaceRestoreFailure(db, await run(f.runId), { workspaceRestoreFailure: "restore_failed" }, f.lease);
|
|
deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId);
|
|
if (fault === "missing_receipt") await db.delete(agentWakeupRequests).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "wrong_receipt_run") await db.update(agentWakeupRequests).set({ runId: f.runId }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "wrong_receipt_agent") await db.update(agentWakeupRequests).set({ agentId: (await db.insert(agents).values({ companyId: f.companyId, name: "Other", role: "engineer" }).returning())[0].id }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "wrong_receipt_company") await db.update(agentWakeupRequests).set({ companyId: (await db.insert(companies).values({ name: "Other", issuePrefix: `OT${randomUUID().slice(0, 6)}` }).returning())[0].id }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "wrong_receipt_user") await db.update(agentWakeupRequests).set({ requestedByActorId: "other-user" }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "wrong_receipt_issue") await db.update(agentWakeupRequests).set({ payload: { ...receipt.payload, issueId: randomUUID() } }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "terminal_receipt") await db.update(agentWakeupRequests).set({ status: "cancelled" }).where(eq(agentWakeupRequests.id, receipt.id));
|
|
if (fault === "foreign_comment_user") await db.update(issueComments).set({ authorUserId: "other-user" }).where(eq(issueComments.id, followup.id));
|
|
if (fault === "deleted_followup") await db.update(issueComments).set({ deletedAt: new Date() }).where(eq(issueComments.id, followup.id));
|
|
if (fault === "unrecorded_comment") {
|
|
const [injected] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId, authorType: "user", authorUserId: "operator", body: "/new" }).returning();
|
|
await db.update(heartbeatRuns).set({ contextSnapshot: mergeCoalescedContextSnapshot((await run(queued.id)).contextSnapshot,
|
|
{ issueId: f.issueId, wakeReason: "issue_commented", wakeCommentId: injected.id, wakeCommentIds: [injected.id] },
|
|
{ preserveExistingInteractionContinuation: true }) }).where(eq(heartbeatRuns.id, queued.id));
|
|
}
|
|
if (fault === "unrecorded_original") await db.update(agentWakeupRequests).set({ payload: { ...wake.payload, commentId: randomUUID() } }).where(eq(agentWakeupRequests.id, wake.id));
|
|
expect(await queue.claimQueuedRun(await run(queued.id))).toBeNull();
|
|
const [afterWake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
expect(afterWake.status).toBe("skipped");
|
|
expect(queueDeps.executeRun).not.toHaveBeenCalled();
|
|
expect(deps.enqueueWakeup).not.toHaveBeenCalled();
|
|
expect((await run(queued.id)).status).toBe("cancelled");
|
|
} finally { deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId); }
|
|
});
|
|
|
|
it.each(["system_actor", "wrong_actor", "deleted_comment", "foreign_comment", "foreign_agent", "foreign_company", "foreign_run", "terminal_wake", "started", "provider_event", "provider_lease", "resolved_runtime"] as const)("keeps queued Stop recovery input closed without paired predispatch authority: %s", async fault => {
|
|
const t = await queuedResetFixture(), { f, original, reset, wake, queued, deps, execution, queueDeps, queue } = t;
|
|
try {
|
|
await recordRequiredWorkspaceRestoreFailure(db, await run(f.runId), { workspaceRestoreFailure: "restore_failed" }, f.lease);
|
|
deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId);
|
|
if (fault === "system_actor") await db.update(agentWakeupRequests).set({ requestedByActorType: "system" }).where(eq(agentWakeupRequests.id, wake.id));
|
|
if (fault === "wrong_actor") await db.update(agentWakeupRequests).set({ requestedByActorId: "other-user" }).where(eq(agentWakeupRequests.id, wake.id));
|
|
if (fault === "deleted_comment") await db.update(issueComments).set({ deletedAt: new Date() }).where(eq(issueComments.id, reset.id));
|
|
if (fault === "foreign_comment") await db.update(issueComments).set({ issueId: (await db.insert(issues).values({ companyId: f.companyId, title: "Other" }).returning())[0].id }).where(eq(issueComments.id, reset.id));
|
|
if (fault === "foreign_agent") await db.update(agentWakeupRequests).set({ agentId: (await db.insert(agents).values({ companyId: f.companyId, name: "Other", role: "engineer" }).returning())[0].id }).where(eq(agentWakeupRequests.id, wake.id));
|
|
if (fault === "foreign_company") await db.update(agentWakeupRequests).set({ companyId: (await db.insert(companies).values({ name: "Other", issuePrefix: "OT" }).returning())[0].id }).where(eq(agentWakeupRequests.id, wake.id));
|
|
if (fault === "foreign_run") await db.update(agentWakeupRequests).set({ runId: f.runId }).where(eq(agentWakeupRequests.id, wake.id));
|
|
if (fault === "terminal_wake") await db.update(agentWakeupRequests).set({ status: "cancelled", finishedAt: new Date(), error: "Operator cancelled" }).where(eq(agentWakeupRequests.id, wake.id));
|
|
if (fault === "started") await db.update(heartbeatRuns).set({ startedAt: new Date() }).where(eq(heartbeatRuns.id, queued.id));
|
|
if (fault === "provider_event") await appendHeartbeatRunEvent(db, { companyId: f.companyId, agentId: original.agentId, runId: queued.id,
|
|
eventType: "adapter.invoke", stream: "system", level: "info", message: "Controlled provider execution evidence" });
|
|
if (fault === "provider_lease") await db.insert(environmentLeases).values({ companyId: f.companyId, heartbeatRunId: queued.id, environmentId: f.lease.environmentId, provider: "daytona", providerLeaseId: randomUUID() });
|
|
if (fault === "resolved_runtime") await db.update(heartbeatRuns).set({ runtimeModeResolvedAt: new Date() }).where(eq(heartbeatRuns.id, queued.id));
|
|
const [beforeWake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
expect(await queue.claimQueuedRun(await run(queued.id))).toBeNull();
|
|
const [afterWake] = await db.select().from(agentWakeupRequests).where(eq(agentWakeupRequests.id, wake.id));
|
|
expect(afterWake.status).not.toBe("deferred_issue_execution");
|
|
if (["foreign_agent", "foreign_company", "foreign_run", "terminal_wake"].includes(fault)) expect(afterWake).toEqual(beforeWake);
|
|
else expect(afterWake.status).toBe("skipped");
|
|
expect(queueDeps.executeRun).not.toHaveBeenCalled();
|
|
expect(deps.enqueueWakeup).not.toHaveBeenCalled();
|
|
} finally { deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId); }
|
|
});
|
|
|
|
async function stoppedEvidenceFixture() {
|
|
const f = await seed();
|
|
const original = await run(f.runId);
|
|
await db.update(heartbeatRuns).set({ status: "cancelled", nativePhase: "terminal_failure", errorCode: "cancelled", resultJson: {
|
|
...original.resultJson, cancelledByActorType: "user", cancelledByUserId: "operator",
|
|
cancellation: requestedRunCancellation({ cancelledByActorType: "user", cancelledByUserId: "operator" }, "Stop"),
|
|
executionCancellation: { state: "requested", requestedAt: new Date().toISOString() },
|
|
nativeCancellation: { schema: "paperclip.native-cancellation.v1", runId: f.runId, companyId: f.companyId, issueId: f.issueId,
|
|
scope: "run", reasonCode: "cancellation_run_only", dispatchState: "acknowledged", dispatched: true,
|
|
intentAuditId: randomUUID(), acknowledgementAuditId: randomUUID() },
|
|
} }).where(eq(heartbeatRuns.id, f.runId));
|
|
await db.update(issues).set({ conversationAgentId: original.agentId, conversationUserId: "operator", conversationState: "active", status: "in_progress" }).where(eq(issues.id, f.issueId));
|
|
const [reset] = await db.insert(issueComments).values({ companyId: f.companyId, issueId: f.issueId,
|
|
authorType: "user", authorUserId: "operator", body: "/new" }).returning();
|
|
await db.delete(issueRecoveryActions).where(eq(issueRecoveryActions.id, f.actionId));
|
|
await recordRequiredWorkspaceRestoreFailure(db, await run(f.runId), { workspaceRestoreFailure: "restore_failed" }, f.lease);
|
|
const [preservation] = await db.select().from(issueRecoveryActions).where(and(eq(issueRecoveryActions.companyId, f.companyId),
|
|
eq(issueRecoveryActions.sourceIssueId, f.issueId), eq(issueRecoveryActions.fingerprint, `native-workspace:${f.runId}`)));
|
|
expect(preservation).toMatchObject({ status: "resolved", evidence: { automaticRecovery: { policy: "preserve_without_replay_v1", replay: "blocked", runId: f.runId } } });
|
|
f.actionId = preservation.id; f.input.actionId = preservation.id;
|
|
const [wake] = await db.insert(agentWakeupRequests).values({ companyId: f.companyId, agentId: original.agentId,
|
|
source: "on_demand", reason: "issue_execution_deferred", status: "deferred_issue_execution", requestedByActorType: "user", requestedByActorId: "operator",
|
|
payload: withQueuedCommentIdsInWakePayload({ issueId: f.issueId, commentId: reset.id,
|
|
_paperclipWakeContext: { issueId: f.issueId, wakeReason: "issue_commented", wakeCommentId: reset.id, wakeCommentIds: [reset.id] } }, [reset.id]) }).returning();
|
|
const deps = callbacks(db), control = createHeartbeatRunControl(db, deps);
|
|
const execution = createAdapterExecutionControl();
|
|
deps.activeRunExecutions.add(f.runId); adapterExecutionControls.set(f.runId, execution);
|
|
const cleanup = () => { deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId); };
|
|
return { f, original, reset, wake, deps, control, execution, cleanup };
|
|
}
|
|
|
|
async function heldStoppedEvidence() {
|
|
const prepared = await stoppedEvidenceFixture();
|
|
await recoverRetainedWorkspace(prepared.f.input);
|
|
const held = readSystemRecovery((await action(prepared.f.actionId)).evidence)!;
|
|
expect(held.state).toBe("awaiting_evidence");
|
|
await prepared.control.acknowledgeRemoteStop(prepared.f.runId, prepared.f.companyId);
|
|
prepared.cleanup();
|
|
return { ...prepared, held };
|
|
}
|
|
|
|
it("advances saved workspace recovery on new Stop evidence after executor cleanup", async () => {
|
|
const { f, original, reset, deps, control, execution } = await stoppedEvidenceFixture();
|
|
try {
|
|
await recoverRetainedWorkspace(f.input);
|
|
const held = readSystemRecovery((await action(f.actionId)).evidence)!;
|
|
expect(held).toMatchObject({ state: "awaiting_evidence", failure: { code: "agent_stop_unconfirmed", retryClass: "recheck_evidence" } });
|
|
expect(held.nextAttemptAt!).toBeGreaterThan(Date.now() + 20_000);
|
|
expect(await getExecutionBlocker(db, f.companyId, f.issueId, { conversationResetCommentId: reset.id })).not.toBeNull();
|
|
await control.acknowledgeRemoteStop(f.runId, f.companyId);
|
|
expect((await run(f.runId)).resultJson?.executionCancellation).toMatchObject({ state: "acknowledged", proof: "provider_termination_receipt" });
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)).toEqual(held);
|
|
deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId);
|
|
const at = Date.now(); let events = 0;
|
|
const stop = subscribeDeliveryWork(db, DELIVERY_QUEUES.workspaceRecovery, () => { events += 1; });
|
|
try { await control.resumeRemoteStopComments(await run(f.runId)); } finally { stop(); }
|
|
const advanced = readSystemRecovery((await action(f.actionId)).evidence)!;
|
|
expect(advanced.nextAttemptAt!).toBeLessThanOrEqual(Date.now());
|
|
expect(advanced).toMatchObject({ generationId: held.generationId, attempts: held.attempts, retriesUsed: held.retriesUsed,
|
|
firstFailureAt: held.firstFailureAt, deadlineAt: held.deadlineAt, providerCooldownAt: held.providerCooldownAt });
|
|
expect(events).toBeGreaterThan(0);
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
expect(await getExecutionBlocker(db, f.companyId, f.issueId)).not.toBeNull();
|
|
const successorRunId = randomUUID();
|
|
const admitted = await admitExplicitNativeContinuation({ db, companyId: f.companyId, issueId: f.issueId, agentId: original.agentId,
|
|
actorType: "user", actorId: "operator", reason: "issue_commented", commentId: reset.id, successorRunId });
|
|
expect(admitted).toMatchObject({ previousRunId: f.runId, commentId: reset.id });
|
|
expect((await action(f.actionId)).evidence).toMatchObject({ automaticRecovery: { replay: "explicit_user_continuation" } });
|
|
expect(await getExecutionBlocker(db, f.companyId, f.issueId, { conversationResetCommentId: reset.id })).toBeNull();
|
|
expect(Date.now() - at).toBeLessThan(30_000);
|
|
expect(deps.enqueueWakeup).not.toHaveBeenCalled();
|
|
} finally { deps.activeRunExecutions.delete(f.runId); execution.finish(); adapterExecutionControls.delete(f.runId); }
|
|
});
|
|
|
|
it.each(["active_adapter", "active_execution", "missing_native_ack", "wrong_native_run", "missing_physical_receipt", "wrong_allocation", "coordinator_owner", "coordinator_observed", "foreign_action_run", "expired_deadline", "exhausted", "retry_budget", "unrelated_failure"] as const)("does not advance Stop-evidence work without current authority: %s", async fault => {
|
|
const t = await heldStoppedEvidence(), { f, deps, control, execution } = t;
|
|
try {
|
|
if (fault === "active_adapter") adapterExecutionControls.set(f.runId, execution);
|
|
if (fault === "active_execution") deps.activeRunExecutions.add(f.runId);
|
|
if (fault === "missing_native_ack") await db.update(heartbeatRuns).set({ resultJson: sql`jsonb_set(${heartbeatRuns.resultJson}, '{nativeCancellation,dispatchState}', '"pending"'::jsonb)` }).where(eq(heartbeatRuns.id, f.runId));
|
|
if (fault === "wrong_native_run") await db.update(heartbeatRuns).set({ resultJson: sql`jsonb_set(${heartbeatRuns.resultJson}, '{nativeCancellation,runId}', ${JSON.stringify(randomUUID())}::jsonb)` }).where(eq(heartbeatRuns.id, f.runId));
|
|
if (fault === "missing_physical_receipt") await db.update(environmentLeases).set({ metadata: sql`${environmentLeases.metadata} - 'remoteExecutionTermination'` }).where(eq(environmentLeases.id, f.lease.id));
|
|
if (fault === "wrong_allocation") await db.update(environmentLeases).set({ metadata: sql`jsonb_set(${environmentLeases.metadata}, '{remoteExecutionTermination,providerLeaseId}', ${JSON.stringify(randomUUID())}::jsonb)` }).where(eq(environmentLeases.id, f.lease.id));
|
|
if (fault === "coordinator_owner") await db.update(nativeRunFinalizations).set({ leaseOwner: randomUUID(), leaseExpiresAt: new Date(Date.now() + 60_000) }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
if (fault === "coordinator_observed") await db.update(nativeRunFinalizations).set({ phase: "observed" }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
if (fault === "foreign_action_run") await db.update(issueRecoveryActions).set({ evidence: sql`jsonb_set(${issueRecoveryActions.evidence}, '{runId}', ${JSON.stringify(randomUUID())}::jsonb)` }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
if (fault === "expired_deadline") await updateSystemRecovery(db, f.input, t.held, { ...t.held, revision: t.held.revision + 1, deadlineAt: Date.now() - 1 });
|
|
if (fault === "exhausted") await updateSystemRecovery(db, f.input, t.held, { ...t.held, revision: t.held.revision + 1, state: "exhausted", nextAttemptAt: null });
|
|
if (fault === "retry_budget") await updateSystemRecovery(db, f.input, t.held, { ...t.held, revision: t.held.revision + 1, retriesUsed: 5 });
|
|
if (fault === "unrelated_failure") await updateSystemRecovery(db, f.input, t.held, failSystemRecovery(t.held, new SystemRecoveryFailure("provider_unavailable", "retry_transient")));
|
|
const before = (await action(f.actionId)).evidence;
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
expect((await action(f.actionId)).evidence).toEqual(before);
|
|
if (fault !== "exhausted" && fault !== "retry_budget") expect(deps.enqueueWakeup).not.toHaveBeenCalled();
|
|
} finally { t.cleanup(); }
|
|
});
|
|
|
|
it("consumes identical Stop-evidence only once across another actual failed attempt", async () => {
|
|
const t = await heldStoppedEvidence(), { f, control, execution } = t;
|
|
try {
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
expect((await action(f.actionId)).evidence.workspaceStopEvidence).toMatchObject({ consumed: true });
|
|
adapterExecutionControls.set(f.runId, execution);
|
|
await recoverRetainedWorkspace(f.input);
|
|
t.cleanup();
|
|
const before = (await action(f.actionId)).evidence;
|
|
expect(readSystemRecovery(before)?.nextAttemptAt).toBeGreaterThan(Date.now() + 20_000);
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
expect((await action(f.actionId)).evidence).toEqual(before);
|
|
} finally { t.cleanup(); }
|
|
});
|
|
|
|
it("keeps the provider cooldown and retry window when advancing new Stop-evidence", async () => {
|
|
const t = await heldStoppedEvidence(), { f, control } = t;
|
|
try {
|
|
const cooldown = Date.now() + 5_000;
|
|
const state = failSystemRecovery(t.held, new SystemRecoveryFailure("agent_stop_unconfirmed", "recheck_evidence"), { retryAfterAt: cooldown });
|
|
await updateSystemRecovery(db, f.input, t.held, state);
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
const after = readSystemRecovery((await action(f.actionId)).evidence)!;
|
|
expect(after.nextAttemptAt).toBe(cooldown);
|
|
expect(after).toMatchObject({ generationId: state.generationId, attempts: state.attempts, retriesUsed: state.retriesUsed, firstFailureAt: state.firstFailureAt, deadlineAt: state.deadlineAt, providerCooldownAt: cooldown });
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)).toEqual(after);
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)).toEqual(after);
|
|
} finally { t.cleanup(); }
|
|
});
|
|
|
|
it.each(["current", "stale_generation", "lost_receipt"] as const)("retains a post-cleanup Stop-evidence event before worker failure commits: %s", async variant => {
|
|
const t = await stoppedEvidenceFixture(), { f, control } = t;
|
|
const originalReader = recoverySources.inspectWorkspaceRecoverySource;
|
|
let observed = false, release!: () => void;
|
|
const barrier = new Promise<void>(resolve => { release = resolve; });
|
|
const reader = vi.spyOn(recoverySources, "inspectWorkspaceRecoverySource").mockImplementationOnce(async (...args) => {
|
|
try { return await originalReader(...args); }
|
|
catch (error) { observed = true; await barrier; throw error; }
|
|
});
|
|
const worker = recoverRetainedWorkspace(f.input);
|
|
try {
|
|
await expect.poll(() => observed, { timeout: 5_000 }).toBe(true);
|
|
const running = readSystemRecovery((await action(f.actionId)).evidence)!;
|
|
expect(running.state).toBe("running");
|
|
await control.acknowledgeRemoteStop(f.runId, f.companyId); t.cleanup();
|
|
await control.resumeRemoteStopComments(await run(f.runId));
|
|
const pending = await action(f.actionId);
|
|
expect(readSystemRecovery(pending.evidence)).toEqual(running);
|
|
expect(pending.evidence.workspaceStopEvidence).toMatchObject({ sourceRunId: f.runId, generationId: running.generationId, consumed: false });
|
|
if (variant === "stale_generation") await db.update(issueRecoveryActions).set({ evidence: sql`jsonb_set(${issueRecoveryActions.evidence}, '{workspaceStopEvidence,generationId}', ${JSON.stringify(randomUUID())}::jsonb)` }).where(eq(issueRecoveryActions.id, f.actionId));
|
|
if (variant === "lost_receipt") await db.update(environmentLeases).set({ metadata: sql`${environmentLeases.metadata} - 'remoteExecutionTermination'` }).where(eq(environmentLeases.id, f.lease.id));
|
|
release(); await worker;
|
|
const after = await action(f.actionId), state = readSystemRecovery(after.evidence)!;
|
|
expect(state.state).toBe("awaiting_evidence");
|
|
expect(state).toMatchObject({ attempts: running.attempts, retriesUsed: running.retriesUsed, generationId: running.generationId });
|
|
if (variant === "current") {
|
|
expect(state.nextAttemptAt!).toBeLessThanOrEqual(Date.now());
|
|
expect(after.evidence.workspaceStopEvidence).toMatchObject({ consumed: true });
|
|
await recoverRetainedWorkspace(f.input);
|
|
expect(readSystemRecovery((await action(f.actionId)).evidence)?.state).toBe("complete");
|
|
expect(await readFile(path.join(f.host, "saved.txt"), "utf8")).toBe("finished work\n");
|
|
} else {
|
|
expect(state.nextAttemptAt!).toBeGreaterThan(Date.now() + 20_000);
|
|
expect(after.evidence.workspaceStopEvidence).toMatchObject({ consumed: false });
|
|
}
|
|
} finally { release(); await worker; reader.mockRestore(); t.cleanup(); }
|
|
});
|
|
|
|
});
|