mirror of
https://github.com/paperclipai/paperclip.git
synced 2026-11-07 16:35:48 +01:00
feat(server): adopt running remote agents through durable controller checkpoints (#17439)
## Thinking Path > - Paperclip must keep admitted remote work alive while its main server is replaced. > - Runner owns recovery mechanics and exposes opaque checkpoints through #17097. > - The server owns company scope, encryption, Stop, tool authority and execution ownership. > - A successor must prove current ownership before it adopts the original execution. > - Checkpoint bursts must also leave capacity for normal requests and lease renewal. > - This PR persists and adopts that authority through the existing runtime boundaries. ## Linked Issues or Issue Description **What existing behavior does this improve?** Remote agents retain the original execution during a bounded main-server outage. **Current behavior** A server restart can lose the controller's local state and connection. An unsafe replacement could repeat a turn or an uncertain mutation. Checkpoint writes in the request pool can also delay ordinary requests during an aligned burst. **Proposed behavior** Restore encrypted control records from Postgres and adopt the original runner under a fenced lease. Recover known tool receipts through current tool authority. Give control checkpoints a shared pool capped at two connections per database owner. Depends on #17097. Successors are #17455 and #17456. Qualification and presentation remain separate review slices. ## What Changed - Add company/run-bound encrypted checkpoint records and migration 0394. Keep the SQL idempotent after renumbering against current master. - Recover startup and expired ownership through native runtime and heartbeat. Preserve the sandbox lease, admitted runner artifact, harness configuration, native session and active turn. - Keep Stop, finalization, preparation ownership and stale-writer fencing transactional. Preserve graceful detach and natural crash takeover. - Recover committed tool receipts through the existing authority. Reauthorize reads and hold uncertain mutations. - Bound startup retention with the existing coordinator and preserve zero idle scans. - Share at most two checkpoint connections per database client. Keep application and liveness capacity separate. Close the checkpoint pool during shutdown; late writes cannot reopen it. - Count SQL across both application and checkpoint clients. Prove request/liveness progress, command recovery and failure after pool shutdown with real Postgres. ## Verification Project creation recovery now uses the existing project activity receipt owner. It retains authenticated input normalization, fingerprint checks and current run authority after the receipt lock. An absent receipt returns no result before mutation. The new lost-HTTP-response and no-effect regression uses real API routes and Postgres. The seven transport cases pass locally. This Mac skips all 20 database cases. On current source `5de803c2cd`, Linux [job 122593415890](https://github.com/paperclipai/paperclip/actions/runs/40620576702/job/122593415890) executes all 20 project-tool cases successfully (9.480 seconds), including the committed-project lost-response regression. Its shard passes 1,627 cases with three unrelated skips. Server typecheck passes. The transport regression initially has two failures; a subsequent incorrect expectation about schema defaults was corrected before the final pass. Ordinary execution adds no checkpoint writes or recurring work. Current candidate `5de803c2cd` includes the project-creation receipt recovery fix and master `2cd747d1b6`. Migration 0394 retains the prior idempotent checkpoint DDL. Every master SQL file and journal entry is preserved. Snapshots 0390–0394, migration checks and server typecheck pass. The ordinary warm-turn regression fails on its second turn before `a219ab51ad`. After correction, three completed warm turns select the correct prior checkpoint and preserve the original session. All 288 provider-wiring cases pass. The control plane, external persistence callback, identity rotation and filesystem are real; the backend and database-load seam are scripted. The 13 warm admission/configuration variants also pass after final master integration. This does not qualify an underlying harness. At current source `5de803c2cd`, Linux [job 122593415907](https://github.com/paperclipai/paperclip/actions/runs/40620576702/job/122593415907) passes all 940 native-session-executor cases (61.431 seconds), including the warm external-checkpoint behavior; the shard passes 2,358 cases with one skip. Current [Runner job 122593415475](https://github.com/paperclipai/paperclip/actions/runs/40620576702/job/122593415475) passes the native recovery lane: 115 passes and five container-only skips, including the encrypted restore guard, fresh preparation (63.189 seconds), blocked claim lock (1.226 seconds) and recovery beyond 101 blocked owners (3.897 seconds). The first lane passes 2,215 cases with one skip. The SQL cost test reports zero idle and streaming SQL for both output volumes; this is component evidence, not a production capacity claim. At prior candidate `1ff7bb44f4`, Linux job 122549307215 passes all 938 native-executor cases, including the three-turn external-checkpoint regression. Job 122549306469 passes the native recovery lane with 115 passes and five container-only skips, including the encrypted-restore session guard, fresh preparation (63.664 seconds), claim-lock contention (3.961 seconds), and 101-owner recovery (3.824 seconds). This Mac skips the real-Postgres suite. The earlier full local executor attempt has 921 passes and 16 failures during embedded-Postgres initialization before assertions. The original fixture collisions and their corrections remain recorded. At earlier candidate `8fbe62b41d`, server shard job 122535673841 has 3,876 passes, one failure and one skip in an unchanged expiry-worker retry/drain timing test. The same server/database tree on qualification source `b64cac6951` passes all 20 expiry cases and all 3,877 shard cases. The single diagnostic retry, job 122541891915, passes all 20 expiry cases and all 3,877 shard cases (one skip). No assertion was weakened. The original failure remains recorded and its exact timing race is not proven. All Actions checks and legacy status passed on `8fbe62b41d`, with completed Greptile 5/5 and no open feedback. The next candidate `1ff7bb44f4` also passed all Actions checks and Greptile 5/5. Its Paperclip Review remained queued before the new master conflict required integration. The new candidate needs fresh exact-head CI and reviews; old checks do not qualify it. Migration generation is checkpoint-only, all master SQL and journal entries remain unchanged, and migration checks, server typecheck and all 13 warm-owner variants pass after integration. The first migration check was blocked by TSX sandbox IPC; its permitted repeat passes. The first warm-test command used an incorrect path and ran no tests; the corrected command supplies the 13-case receipt. All 16 database pool-lifetime/options/teardown cases pass after granting loopback permission; the first batch had two EPERM failures and 14 passes. Earlier semantic-tool/HTTP recovery, adoption, encryption-batch, zero-streaming-write and idle-SQL receipts remain source-attributed in the [qualification record](https://github.com/paperclipai/paperclip/blob/2ff66f57da0da35bac1c19a2366ea4b84d87a986/doc/plans/2026-10-22-hot-upgrade-qualification.md). That record retains earlier assertion, setup and cleanup failures and their corrections. No failed gate is waived. The historical 300-session component campaign passes four after phases: steady read p99 is 107–108 ms at 500 reads/s and 100 writes/s; aligned-burst read p99 is 103–105 ms after the bounded pool, versus 584–689 ms before. This comparison isolates the persistence corrections. It does not measure the final stack's hosted active-agent capacity. The hosted zero-agent API pair leaves SQL and CPU essentially unchanged, but both arms fail absolute latency gates and candidate peak cgroup memory fails its gate. The active-agent/viewer matrix, different-artifact upgrade and rollback remain open. Run `PAPERCLIP_TEST_DATABASE_URL=<owned-disposable-postgres> pnpm --filter @paperclipai/server exec vitest run src/services/native-runtime/native-runner-restart-recovery.integration.test.ts`. Previous head `f2db8a6568` had one failed workspace setup gate: [job 122555672441](https://github.com/paperclipai/paperclip/actions/runs/40608984238/job/122555672441) passes all 4,169 UI tests and 285 CLI cases (one skip), but the company import/export suite reaches its 30-second health deadline during doctor startup. Its output does not reach the server-start step. The CLI, server, database, shared, adapter and plugin trees are identical on qualification source `a866619d7b`, whose [job 122555886663](https://github.com/paperclipai/paperclip/actions/runs/40609057937/job/122555886663) passes that fixture and all 286 CLI cases. The one diagnostic retry, [job 122561057225](https://github.com/paperclipai/paperclip/actions/runs/40608984238/job/122561057225), passes the formerly failing fixture and all 286 CLI cases. All Actions checks and legacy status on `f2db8a6568` pass. Its exact-head Greptile completed at 5/5 with no open feedback. The current integrated head requires fresh checks and a completed Paperclip Review before merge. The original failure remains recorded; its exact timing cause is unproven. A repeated failure requires investigation and repair rather than another retry. All current-head CI, legacy status, exact-head completed Greptile 5/5, resolved feedback and ordinary GitHub merge eligibility remain required before merge. The latest integration preserves master’s published Copilot v70 profiles and every published migration. Checkpoint migration `0394_conscious_whizzer.sql` was generated with `pnpm db:generate` and retains the prior idempotent DDL. The Runner vendor rebuild and replay parity checks pass. Server typecheck and all 13 warm-owner variants pass after that rebuild. The imported report-lock patch is byte-identical to master; its intentional patch-context whitespace is retained, and the feature delta passes whitespace checks. No hosted experiment was run. ## Risks The pool adds up to two active connections per instance, shared across agents. Checkpoint delivery can wait behind that bound; it still requires its commit. Steady component-test Postgres CPU rises by about 1.7 percentage points of a two-CPU allocation. These corrections do not remove the stack's control and renewal writes. The migration and key must survive replacement. Fresh-database tests and local load do not establish different-artifact hosted upgrade or rollback continuity. ## Model Used OpenAI GPT-6 through Codex, with reasoning, repository tools and code execution. The runtime does not expose a more specific model ID or context-window size. ## 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 - [ ] 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 - [ ] All Paperclip CI gates are green - [ ] 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>
This commit is contained in:
1 parent
94a780286f
commit
09ad9adce2
62 files changed
+4671
-488
No files matched your search
@@ -1,3 +1,5 @@
|
||||
import * as runnerCheckpoint from "./native-runner-checkpoint.js";
|
||||
import { DurablePrpControlPlane } from "../../vendor/paperclip-runner/index.js";
|
||||
import * as environmentTaskRuntime from "../environment-task-runtime.js";
|
||||
import * as pluginTaskRuntime from "./plugin-runner-task.js";
|
||||
import * as completionFeedback from "./native-completion-feedback.js";
|
||||
@@ -84,6 +86,7 @@ import {
|
||||
import { nativeRuntimeContextFixture } from "./runtime-context.test-fixture.js";
|
||||
import { nativeToolContractFingerprintForTarget } from "./native-session-resume.js";
|
||||
import * as nativeSessionResume from "./native-session-resume.js";
|
||||
import * as nativeLeaseRenewal from "./native-execution-lease.js";
|
||||
import { buildNativeHeartbeatPreparationSpans, createNativeRunTrace } from "./native-run-trace.js";
|
||||
import { NativeRunnerOwnershipUnverifiedError } from "./native-runner-ownership.js";
|
||||
import type { AdapterRuntimeEvent } from "../../adapters/index.js";
|
||||
@@ -1079,6 +1082,7 @@ fs.readFileSync = function(path, ...args) {
|
||||
await vi.waitFor(() => expect(healthyProbes).toBe(1), { timeout: 5_000 });
|
||||
expect(settled).toBe(false);
|
||||
expect(handle.isExitConfirmed?.()).toBe(false);
|
||||
expect(handle.startedAt).toBe("2026-10-19T00:00:00.000Z");
|
||||
expect(() => process.kill(child!.pid!, 0)).not.toThrow();
|
||||
expect((await readFile(marker, "utf8")).split("\n")[1]).toBe(String(child!.pid));
|
||||
expect(execute.mock.calls.filter(([x]) => x.args?.[2] === "paperclip-runner-launch")).toHaveLength(1);
|
||||
@@ -6038,6 +6042,9 @@ describe("native governed waits", () => {
|
||||
});
|
||||
|
||||
type LeaseCoordinator = {
|
||||
controllerBootId?: string;
|
||||
controllerPid?: number;
|
||||
controllerGeneration?: number;
|
||||
runId: string;
|
||||
companyId: string;
|
||||
issueId: string;
|
||||
@@ -6298,6 +6305,52 @@ describe("native startup cancellation fence", () => {
|
||||
});
|
||||
|
||||
describe("native startup restart detachment", () => {
|
||||
it.each(["ordinary", "recovered", "stale-claim", "newer-detach"] as const)("preserves same-server detach fencing for %s admission", async mode => {
|
||||
const issueService = vi.spyOn(issueServiceModule, "issueService").mockReturnValue({
|
||||
update: vi.fn(async () => ({ status: "blocked", statusVersion: 1 })),
|
||||
} as unknown as ReturnType<typeof issueServiceModule.issueService>);
|
||||
const current = structuredClone(execution);
|
||||
current.binding.runId = `same-server-recovery-${mode}`;
|
||||
const { currentNativeControllerIdentity } = await import("./native-restart-recovery.js");
|
||||
const controller = await currentNativeControllerIdentity();
|
||||
await detachNativeSessionsForRestart([current.binding.runId]);
|
||||
const db = leaseDb(current, { leaseOwner: "recovered-controller", controllerBootId: controller.bootId,
|
||||
controllerPid: controller.pid, controllerGeneration: 2, attempt: 1 });
|
||||
let newerDetach: Promise<unknown> | undefined;
|
||||
const transaction = db.transaction.bind(db);
|
||||
if (mode === "newer-detach") db.transaction = (async operation => {
|
||||
const result = await transaction(operation);
|
||||
newerDetach ??= detachNativeSessionsForRestart([current.binding.runId]);
|
||||
return result;
|
||||
}) as Db["transaction"];
|
||||
const admitted = vi.fn();
|
||||
state.execute.mockReset().mockImplementationOnce(async options => {
|
||||
await options.onSession({ detachControllerForRestart: async () => undefined });
|
||||
admitted();
|
||||
throw new NativeControllerDetachedForRestartError();
|
||||
});
|
||||
try {
|
||||
const admission = executePaperclipNativeSession({ db, execution: current, runnerInstanceId: "runner",
|
||||
leaseOwner: "recovered-controller", ...(mode === "ordinary" ? {} : { restartRecovery: {
|
||||
kind: "bootstrap_incomplete" as const, runId: current.binding.runId, leaseOwner: "recovered-controller",
|
||||
controllerGeneration: mode === "stale-claim" ? 1 : 2, providerAttempt: 1,
|
||||
restartKind: "hard" as const, recoveryRequestId: null,
|
||||
} }) });
|
||||
if (mode === "recovered") {
|
||||
await expect(admission).rejects.toBeInstanceOf(NativeControllerDetachedForRestartError);
|
||||
expect(admitted).toHaveBeenCalledOnce();
|
||||
expect(state.execute).toHaveBeenCalledOnce();
|
||||
} else if (mode === "stale-claim") {
|
||||
await expect(admission).rejects.toThrow("native_restart_recovery_claim_changed");
|
||||
expect(state.execute).not.toHaveBeenCalled();
|
||||
} else {
|
||||
await expect(admission).rejects.toBeInstanceOf(NativeControllerDetachedForRestartError);
|
||||
expect(admitted).not.toHaveBeenCalled();
|
||||
if (mode === "ordinary") expect(state.execute).not.toHaveBeenCalled();
|
||||
}
|
||||
} finally { await newerDetach; issueService.mockRestore(); }
|
||||
});
|
||||
|
||||
it("uses one admission deadline for the whole restart sweep", async () => {
|
||||
const inputs = ["first", "second"].map(label => {
|
||||
const input = structuredClone(execution);
|
||||
@@ -7688,7 +7741,8 @@ describe("native session execution lease fencing", () => {
|
||||
.mockResolvedValueOnce([]);
|
||||
const where = vi.fn(() => ({ returning }));
|
||||
const set = vi.fn(() => ({ where }));
|
||||
const db = { update: vi.fn(() => ({ set })) } as unknown as Db;
|
||||
const update = vi.fn(() => ({ set }));
|
||||
const db = { update, transaction: async (run: (tx: unknown) => Promise<unknown>) => run({ update, execute: vi.fn() }) } as unknown as Db;
|
||||
const input = {
|
||||
db,
|
||||
runId: "run-lease",
|
||||
@@ -8082,7 +8136,7 @@ describe("native warm session supervision", () => {
|
||||
const { promisify } = await import("node:util");
|
||||
const { execFile } = await import("node:child_process");
|
||||
const executeCommand = promisify(execFile);
|
||||
const { startEmbeddedPostgresTestDatabase } = await import("../../__tests__/helpers/embedded-postgres.js");
|
||||
const { startRecoveryPostgres } = await import("../../__tests__/helpers/recovery-postgres.js");
|
||||
const { agentInstructionWorkingCopyService } = await import("../agent-instruction-working-copies.js");
|
||||
const { buildNativeRuntimeContext } = await import("./runtime-context.js");
|
||||
const { prepareNativeHeartbeatRun } = await import("./prepare-native-run.js");
|
||||
@@ -8090,11 +8144,7 @@ describe("native warm session supervision", () => {
|
||||
const home = await realpath(await mkdtemp(join(tmpdir(), "warm-remote-db-")));
|
||||
const previousHome = process.env.PAPERCLIP_HOME;
|
||||
process.env.PAPERCLIP_HOME = home;
|
||||
const { resolveExternalTestDatabaseUrl } = await import("../../__tests__/helpers/external-test-database.js");
|
||||
const externalUrl = resolveExternalTestDatabaseUrl("PAPERCLIP_TEST_DATABASE_URL");
|
||||
const database = externalUrl
|
||||
? { connectionString: externalUrl, cleanup: () => tables.closeRegisteredClients(externalUrl) }
|
||||
: await startEmbeddedPostgresTestDatabase("warm-remote-db-");
|
||||
const database = await startRecoveryPostgres("warm-remote-db-");
|
||||
const db = tables.createDb(database.connectionString);
|
||||
const companyId = randomUUID(), agentId = randomUUID(), userId = randomUUID(), environmentId = randomUUID(), issueId = randomUUID();
|
||||
const { resolveManagedInstructionsRoot } = await import("../agent-instructions.js");
|
||||
@@ -8330,9 +8380,6 @@ describe("native warm session supervision", () => {
|
||||
finally {
|
||||
const cleanupFailures: unknown[] = [];
|
||||
try { await closeWarmNativeSessionsForEnvironment({ environmentId, reason: "test cleanup" }); } catch (error) { cleanupFailures.push(error); }
|
||||
if (externalUrl) {
|
||||
try { await db.execute(sql`TRUNCATE TABLE ${tables.companies}, ${tables.environments}, ${tables.authUsers} CASCADE`); } catch (error) { cleanupFailures.push(error); }
|
||||
}
|
||||
try { await database.cleanup(); } catch (error) { cleanupFailures.push(error); }
|
||||
if (previousHome === undefined) delete process.env.PAPERCLIP_HOME; else process.env.PAPERCLIP_HOME = previousHome;
|
||||
try {
|
||||
@@ -10194,6 +10241,44 @@ describe("native warm session supervision", () => {
|
||||
});
|
||||
|
||||
describe("native session bounded recovery", () => {
|
||||
it("owns renewal from fresh claim commit through preparation failure", async () => {
|
||||
const fresh = { ...structuredClone(execution), provider: { kind: "acpx", agent: "claude",
|
||||
model: "claude-sonnet-4-6", profile: QUALIFIED_ACPX_PROFILES.claude } } as NativeExecutionInput;
|
||||
const db = leaseDb(fresh);
|
||||
const select = db.select.bind(db);
|
||||
const failure = new Error("accounting history unavailable");
|
||||
vi.spyOn(db, "select").mockImplementation((...args: unknown[]) => {
|
||||
const query = (select as (...args: unknown[]) => ReturnType<Db["select"]>)(...args);
|
||||
const from = query.from.bind(query);
|
||||
query.from = ((table: unknown) => {
|
||||
if (table === heartbeatRunEvents) throw failure;
|
||||
return from(table as never);
|
||||
}) as typeof query.from;
|
||||
return query;
|
||||
});
|
||||
const stop = vi.fn(async () => undefined);
|
||||
const renew = vi.spyOn(nativeLeaseRenewal, "startNativeSessionExecutionLeaseRenewal").mockReturnValue({ stop });
|
||||
let reached!: () => void, release!: () => void;
|
||||
const claiming = new Promise<void>(resolve => { reached = resolve; });
|
||||
const preparation = new Promise<void>(resolve => { release = resolve; });
|
||||
const running = executePaperclipNativeSession({ db, execution: fresh, runnerInstanceId: "fresh-runner",
|
||||
runnerExecutionTarget: { kind: "remote", transport: "sandbox", remoteCwd: "/workspace", environmentId: "fresh-environment", leaseId: "fresh-lease" },
|
||||
onEvent: async event => {
|
||||
if (event.payload?.span === "native.coordinator.claim") { reached(); await preparation; }
|
||||
},
|
||||
}).catch(error => error);
|
||||
try {
|
||||
await claiming;
|
||||
expect(renew).toHaveBeenCalledExactlyOnceWith(expect.objectContaining({
|
||||
db, runId: fresh.binding.runId, attempt: 1, remote: true,
|
||||
}));
|
||||
expect(stop).not.toHaveBeenCalled();
|
||||
release();
|
||||
expect(await running).toBe(failure);
|
||||
expect(stop).toHaveBeenCalledOnce();
|
||||
} finally { release(); await running; renew.mockRestore(); }
|
||||
});
|
||||
|
||||
it("persists closed remote recovery evidence without replacing saved run results", async () => {
|
||||
const updates: Array<{ table: unknown; values: Record<string, unknown> }> = [];
|
||||
const onEvent = vi.fn(async () => undefined);
|
||||
@@ -11219,9 +11304,9 @@ describe("runnerd provider runtime wiring", () => {
|
||||
await rm(isolatedStateDirectory, { recursive: true, force: true });
|
||||
});
|
||||
|
||||
it.each((["codex", "claude"] as const).flatMap(provider => (["unchanged", "changed", "runtime-context", "retirement-failed", "stale-owner", "active-owner", "missing-owner", "foreign-environment", "plugin-replaced", "plugin-managed-github"] as const).filter(scenario => provider === "codex" || scenario.startsWith("plugin-")).map(scenario => [provider, scenario] as const)))("rotates an owned warm %s remote runner before checking changed configuration: %s", async (provider, scenario) => {
|
||||
it.each((["codex", "claude"] as const).flatMap(provider => (["unchanged", "external-checkpoint", "changed", "runtime-context", "retirement-failed", "stale-owner", "active-owner", "missing-owner", "foreign-environment", "plugin-replaced", "plugin-managed-github"] as const).filter(scenario => provider === "codex" || scenario.startsWith("plugin-")).map(scenario => [provider, scenario] as const)))("rotates an owned warm %s remote runner before checking changed configuration: %s", async (provider, scenario) => {
|
||||
const plugin = scenario === "plugin-replaced" || scenario === "plugin-managed-github";
|
||||
const reusable = scenario === "unchanged" || scenario === "plugin-managed-github";
|
||||
const reusable = ["unchanged", "external-checkpoint", "plugin-managed-github"].includes(scenario);
|
||||
const { permissionPolicy: _permissionPolicy, ...claudeProfile } = QUALIFIED_ACPX_PROFILES.claude;
|
||||
const first = { ...scopedNativeExecution(`warm-admission-${provider}-${scenario}`),
|
||||
schema: "paperclip.native-execution-input.v5" as const,
|
||||
@@ -11230,7 +11315,7 @@ describe("runnerd provider runtime wiring", () => {
|
||||
if (provider === "claude") first.session.driverKind = "acpx_runtime";
|
||||
first.session.lifecyclePolicy = { mode: "warm", idleTimeoutMs: 60_000 };
|
||||
const next = { ...first, binding: { ...first.binding, runId: `${first.binding.runId}-next` },
|
||||
session: { ...first.session, lifecyclePolicy: { mode: "warm" as const, idleTimeoutMs: ["unchanged", "runtime-context", "stale-owner", "missing-owner", "plugin-replaced", "plugin-managed-github"].includes(scenario) ? 60_000 : 90_000 } } };
|
||||
session: { ...first.session, lifecyclePolicy: { mode: "warm" as const, idleTimeoutMs: ["unchanged", "external-checkpoint", "runtime-context", "stale-owner", "missing-owner", "plugin-replaced", "plugin-managed-github"].includes(scenario) ? 60_000 : 90_000 } } };
|
||||
if (scenario === "runtime-context") {
|
||||
next.runtimeContext = { ...first.runtimeContext,
|
||||
mcp: { ...first.runtimeContext.mcp, digest: "1".repeat(64) } };
|
||||
@@ -11239,7 +11324,7 @@ describe("runnerd provider runtime wiring", () => {
|
||||
const environmentId = `warm-admission-${scenario}`;
|
||||
const target = { kind: "remote" as const, transport: "sandbox" as const, environmentId,
|
||||
remoteCwd: "/home/daytona/workspace", runner: { execute: vi.fn() } };
|
||||
const identity = { runId: first.binding.runId, normalizedSessionId: first.session.normalizedSessionId,
|
||||
const identity = { runId: first.binding.runId, normalizedSessionId: first.session.normalizedSessionId!,
|
||||
runnerInstanceId: `runner-${scenario}`, environmentLeaseId: first.binding.executionWorkspaceId };
|
||||
let replacementPending=false;
|
||||
const acquisition={outcome:"resumed" as "resumed"|"replacement",providerLeaseId:"task-first"};
|
||||
@@ -11264,6 +11349,23 @@ describe("runnerd provider runtime wiring", () => {
|
||||
|
||||
const root = await initializeScopedRunnerRoot(first, identity.runnerInstanceId);
|
||||
await writeRunnerAuthority(root, identity, "ready");
|
||||
// Use the real control plane and persistence callback, not a handwritten marker.
|
||||
const saved = new Map<string, import("../../vendor/paperclip-runner/index.js").DurablePrpCheckpoint>();
|
||||
let persistedController: DurablePrpControlPlane | undefined;
|
||||
const restore = scenario === "external-checkpoint"
|
||||
? vi.spyOn(runnerCheckpoint, "restoreNativeRunnerCheckpoint").mockImplementation(async (_db, binding, directory) => {
|
||||
const checkpoint = saved.get(binding.runId);
|
||||
if (!checkpoint) throw new Error("missing test checkpoint");
|
||||
checkpoint.restoreSessionDirectory(directory);
|
||||
return checkpoint;
|
||||
}) : undefined;
|
||||
if (restore) {
|
||||
const completeIdentity = { ...identity, turnId: "turn-first", itemId: "item-first" };
|
||||
// The authority must begin from a complete real state envelope.
|
||||
await rm(join(root, "control-plane"), { recursive: true, force: true });
|
||||
persistedController = new DurablePrpControlPlane({ stateDirectory: join(root, "control-plane"), identity: completeIdentity,
|
||||
expectedRunnerVersion: "warm-test", expectedRunnerDigest: `sha256:${"a".repeat(64)}` });
|
||||
}
|
||||
const order: string[] = [];
|
||||
let failRetirement = scenario === "retirement-failed";
|
||||
const firstSession = { close: vi.fn(async () => {
|
||||
@@ -11277,7 +11379,7 @@ describe("runnerd provider runtime wiring", () => {
|
||||
criteria: [], remainingWork: [] }, evidence: [], verification: [], artifacts: [], attentionRequests: [] });
|
||||
if (!validated.ok) throw new Error("Invalid completion fixture");
|
||||
const result = { result: validated.result, terminal: { runTerminalState: "succeeded" },
|
||||
turnId: "turn", normalizedSessionId: first.session.normalizedSessionId,
|
||||
turnId: "turn", normalizedSessionId: first.session.normalizedSessionId!,
|
||||
providerSessionId: "provider", driverKind: "test", driverVersion: "1",
|
||||
nativeEventCount: 1, highestContiguousSourceSeq: 1, usage: null };
|
||||
const feedback = vi.spyOn(completionFeedback, "nativeCompletionFeedback").mockResolvedValue("accepted");
|
||||
@@ -11286,9 +11388,21 @@ describe("runnerd provider runtime wiring", () => {
|
||||
state.execute.mockReset().mockImplementationOnce(async options => {
|
||||
retainedFeedback = state.createBackend.mock.calls.at(-1)![1].completionFeedback;
|
||||
await options.onSession?.(firstSession);
|
||||
if (persistedController) {
|
||||
persistedController.setCheckpointPersistence(async checkpoint => { saved.set(checkpoint.runId, checkpoint); });
|
||||
await persistedController.flushCheckpoint();
|
||||
expect(JSON.parse(await readFile(join(root, "control-plane/control-plane-state.json"), "utf8")).schema)
|
||||
.toBe("paperclip.runner.external-control-plane-state.v1");
|
||||
}
|
||||
return result;
|
||||
}).mockImplementationOnce(async options => {
|
||||
order.push("execute-next");
|
||||
if (persistedController) {
|
||||
persistedController.rotateRunIdentity({ ...identity, runId: next.binding.runId, turnId: "turn-next", itemId: "item-next" });
|
||||
await persistedController.flushCheckpoint();
|
||||
await writeFile(join(root, "runner/runner-state.json"), JSON.stringify(durableRunnerState(
|
||||
{ ...identity, runId: next.binding.runId }, "ready")));
|
||||
}
|
||||
if (reusable) {
|
||||
await expect(retainedFeedback!(result.result)).resolves.toBe("accepted");
|
||||
expect(feedback).toHaveBeenLastCalledWith(firstDb, next.binding.runId, result.result);
|
||||
@@ -11343,8 +11457,8 @@ describe("runnerd provider runtime wiring", () => {
|
||||
}
|
||||
if (scenario === "plugin-managed-github") expect(rebind).toHaveBeenCalledOnce();
|
||||
expect(order).toEqual(reusable ? ["execute-next"] : ["retire", "execute-next"]);
|
||||
if (scenario === "unchanged") {
|
||||
await writeRunnerAuthority(root, { ...identity, runId: next.binding.runId }, "ready");
|
||||
if (["unchanged", "external-checkpoint"].includes(scenario)) {
|
||||
if (!persistedController) await writeRunnerAuthority(root, { ...identity, runId: next.binding.runId }, "ready");
|
||||
const third = { ...next, binding: { ...next.binding, runId: `${next.binding.runId}-third` } };
|
||||
const thirdDb = leaseDb(third);
|
||||
const thirdSelect = thirdDb.select.bind(thirdDb);
|
||||
@@ -11362,9 +11476,15 @@ describe("runnerd provider runtime wiring", () => {
|
||||
runnerExecutionTarget: pluginTarget as never,...pluginOptions })).resolves.toMatchObject({ exitCode: 0 });
|
||||
expect(firstSession.close).not.toHaveBeenCalled();
|
||||
expect(state.execute).toHaveBeenCalledTimes(3);
|
||||
if (restore) {
|
||||
expect(restore.mock.calls.map(([, binding]) => binding.runId)).toEqual([first.binding.runId, next.binding.runId]);
|
||||
expect(existsSync(root)).toBe(true);
|
||||
}
|
||||
}
|
||||
}
|
||||
} finally {
|
||||
restore?.mockRestore();
|
||||
await persistedController?.stop();
|
||||
failRetirement = false;
|
||||
resume?.mockRestore();owners?.mockRestore();feedback.mockRestore();
|
||||
await closeWarmNativeSessionsForEnvironment({ environmentId, reason: "test cleanup" });
|
||||
@@ -11985,6 +12105,49 @@ describe("runnerd provider runtime wiring", () => {
|
||||
}
|
||||
});
|
||||
|
||||
it.each(["codex", "opencode", "claude", "pi"] as const)("adopts remote %s with no old controller launch assets", async provider => {
|
||||
const base = scopedNativeExecution(`missing-launch-assets-${provider}`);
|
||||
const current = { ...base,
|
||||
provider: provider === "codex" ? base.provider : provider === "opencode"
|
||||
? { kind: "opencode", model: null }
|
||||
: { kind: "acpx", agent: provider, model: null },
|
||||
session: { ...base.session, driverKind: provider === "codex" ? "codex_app_server"
|
||||
: provider === "opencode" ? "opencode_server" : "acpx_runtime" },
|
||||
} as NativeExecutionInputV1;
|
||||
const identity = { runId: current.binding.runId, normalizedSessionId: current.session.normalizedSessionId!,
|
||||
runnerInstanceId: `admitted-${provider}`, environmentLeaseId: "admitted-lease",
|
||||
turnId: `turn-${current.binding.runId}`, itemId: `item-${current.binding.runId}` };
|
||||
const root = await initializeScopedRunnerRoot(current, identity.runnerInstanceId);
|
||||
await mkdir(join(root, "control-plane"), { recursive: true });
|
||||
await writeFile(join(root, "control-plane", "control-plane-state.json"), JSON.stringify(durableControlPlaneState(identity)));
|
||||
const missingBinary = join(root, "absent-old-controller", "runnerd");
|
||||
const missingPack = join(root, "absent-old-controller", "provider-pack");
|
||||
expect(existsSync(missingBinary)).toBe(false); expect(existsSync(missingPack)).toBe(false);
|
||||
expect(existsSync(join(root, "provider-pack-authority.json"))).toBe(false);
|
||||
const execute = vi.fn(async (request: { args: string[] }) => ({ exitCode: 0, timedOut: false, stderr: "",
|
||||
stdout: request.args[1]?.includes("base64") ? Buffer.from(JSON.stringify({ ...identity, lifecycle: "running" })).toString("base64")
|
||||
: request.args[1]?.includes("cat --") ? `nonce\n123\n2026-09-19T10:00:00.000Z\n${identity.runnerInstanceId}\nlinux:abcd-1234:100\n` : "" }));
|
||||
state.createBackend.mockClear(); state.createTransport.mockClear(); state.resolveRemoteRunnerBinary.mockClear();
|
||||
const artifact = { version: "admitted-version", digest: `sha256:${"a".repeat(64)}` };
|
||||
await createRunnerdBackend({ db: leaseDb(current), execution: current, runnerInstanceId: identity.runnerInstanceId,
|
||||
runnerRemoteBinaryPath: missingBinary, runnerRemoteProviderPackPath: missingPack,
|
||||
runnerIngressAuthorized: true,
|
||||
controllerPersistence: { bootId: "successor", runnerArtifact: artifact },
|
||||
runnerExecutionTarget: { kind: "remote", transport: "sandbox", remoteCwd: "/workspace",
|
||||
environmentId: "environment", leaseId: identity.environmentLeaseId, providerKey: "daytona", runner: { execute },
|
||||
effectiveCapabilities: { runnerWebSocketIngress: true },
|
||||
sandboxLeaseAcquisition: { outcome: "resumed", providerLeaseId: "sandbox" } } as never,
|
||||
restartRecovery: { kind: "reattach_remote_runner", runId: current.binding.runId, leaseOwner: "controller",
|
||||
controllerGeneration: 2, providerAttempt: 1, restartKind: "hard", recoveryRequestId: null,
|
||||
remote: { providerLeaseId: "sandbox", remoteCwd: "/workspace" } },
|
||||
});
|
||||
state.createBackend.mock.calls.at(-1)![1].codexTransportFactory!();
|
||||
expect(state.createTransport.mock.calls.at(-1)![0]).toMatchObject({ prpIdentity: identity,
|
||||
runnerArtifact: artifact, adoptExistingRunner: { pid: 123, startedAt: "2026-09-19T10:00:00.000Z" } });
|
||||
expect(state.resolveRemoteRunnerBinary).not.toHaveBeenCalled();
|
||||
expect(execute.mock.calls.some(([request]) => request.args.includes("paperclip-runner-launch") || request.args.includes("--build-metadata"))).toBe(false);
|
||||
});
|
||||
|
||||
it.each(["claude", "codex"].flatMap(provider => (["per_turn", "warm"] as const).flatMap(policy => ["canonical", "local-receipt", "local-receipt-missing-terminal", "local-receipt-pending-command", "quarantined", "empty-runner", "empty-runner-quarantined", "nonempty-runner", "wrong-identity", "active", "foreign-scope", "no-stop-proof", "not-suspended", "pending-outbox", "pending-command", "missing-terminal", "ambiguous", "contradictory"].map(mode => [provider, policy, mode] as const))))(
|
||||
"recovers a blocked %s %s plugin checkpoint on a replacement lease: %s", async (provider, policy, mode) => {
|
||||
const { permissionPolicy: _permissionPolicy, ...claudeProfile } = QUALIFIED_ACPX_PROFILES.claude;
|
||||
@@ -12122,6 +12285,43 @@ describe("runnerd provider runtime wiring", () => {
|
||||
} finally { resume.mockRestore(); }
|
||||
});
|
||||
|
||||
it.each(["verified", "wrong-state", "wrong-lease"] as const)("reattaches a surviving plugin task through its owner: %s", async mode => {
|
||||
const current = scopedNativeExecution(`plugin-live-${mode}`);
|
||||
const identity = { runId: current.binding.runId, normalizedSessionId: current.session.normalizedSessionId!,
|
||||
runnerInstanceId: "plugin-runner", environmentLeaseId: "same-task-lease", turnId: "admitted-turn", itemId: "admitted-item" };
|
||||
const root = await initializeScopedRunnerRoot(current, identity.runnerInstanceId);
|
||||
await mkdir(join(root, "control-plane"), { recursive: true });
|
||||
await writeFile(join(root, "control-plane/control-plane-state.json"), JSON.stringify(durableControlPlaneState(identity)));
|
||||
const task = vi.fn(async (operation: any) => {
|
||||
environmentTaskOperationSchema.parse(operation);
|
||||
if (operation.kind === "status") return { kind: "status", taskId: "same-task", phase: "running", executionStopped: false };
|
||||
if (operation.kind === "read_state") return { kind: "state", taskId: "same-task", state: {
|
||||
...durableRunnerState(identity, "ready"), ...(mode === "wrong-state" ? { itemId: "other" } : {}),
|
||||
} };
|
||||
return { kind: "description", launchPreparationVersion: 1, projectAuthCheckVersion: 1, taskId: "same-task",
|
||||
runner: { version: "new-artifact", digest: `sha256:${"b".repeat(64)}` } };
|
||||
});
|
||||
state.createBackend.mockClear(); state.createTransport.mockClear();
|
||||
const backend = createRunnerdBackend({ db: leaseDb(current), execution: current, runnerInstanceId: identity.runnerInstanceId,
|
||||
runnerIngressAuthorized: true, runnerExecutionTarget: { kind: "remote", transport: "sandbox", remoteCwd: ".",
|
||||
leaseId: mode === "wrong-lease" ? "replacement-task" : identity.environmentLeaseId, runnerTask: task },
|
||||
controllerPersistence: { bootId: "successor", runnerArtifact: { version: "admitted-artifact", digest: `sha256:${"a".repeat(64)}` } },
|
||||
restartRecovery: { kind: "reattach_remote_runner", runId: current.binding.runId, leaseOwner: "controller", controllerGeneration: 2,
|
||||
providerAttempt: 1, restartKind: "hard", recoveryRequestId: null, remote: { kind: "plugin_task", leaseId: identity.environmentLeaseId } } });
|
||||
if (mode !== "verified") {
|
||||
await expect(backend).rejects.toThrow(mode === "wrong-state" ? "adoption identity mismatch" : "recovery_lease_mismatch");
|
||||
expect(state.createBackend).not.toHaveBeenCalled();
|
||||
return;
|
||||
}
|
||||
await backend;
|
||||
state.createBackend.mock.calls.at(-1)![1].codexTransportFactory!();
|
||||
const options = state.createTransport.mock.calls.at(-1)![0];
|
||||
expect(options).toMatchObject({ prpIdentity: identity, adoptExistingRunner: { pid: null, startedAt: null },
|
||||
runnerArtifact: { version: "admitted-artifact" } });
|
||||
expect(options).toMatchObject({ runnerReconnectGraceMs: 420_000 });
|
||||
expect(task.mock.calls.some(([op]) => ["prepare", "submit", "stop"].includes(op.kind))).toBe(false);
|
||||
});
|
||||
|
||||
it.each(["codex", "opencode", "acpx"] as const)("uses a task-only %s driver without exposing Core paths or transferring the agent home", async (kind) => {
|
||||
const taskExecution = { ...execution, provider: kind === "codex" ? execution.provider
|
||||
: kind === "opencode" ? { kind, model: "openai/test", permissionMode: "deny" }
|
||||
|
||||
Reference in new issue
Block a user