mirror of
https://github.com/paperclipai/paperclip.git
synced 2026-11-12 22:56:58 +01:00
## Thinking Path > - Paperclip is the open source app people use to manage AI agents for work. > - Native recovery must keep admitted provider state and useful workspace files. > - OpenCode binds its saved session to exact executable paths and hashes. > - A replacement image can remove those files even when session data is saved. > - Retaining a manifest alone cannot restore the missing launch files. > - This pull request saves the qualified files before provider work and restores the same owned paths. > - The benefit is cold continuity without relaxing the Runner guard or replaying a prior turn. ## Linked Issues or Issue Description Refs #19558. That fix retains the selected pack for ordinary continuation. This change adds durable file retention for a cold allocation. Related #19487 adjusts the admission budget; it does not retain launch files. A search of open and closed OpenCode cold/durable artifact PRs found no duplicate. **What happened?** A saved OpenCode session could outlive the image that supplied its Node executable, proxy, and native executable. A new image supplied different bytes. The Runner then correctly refused the changed launch profile. The saved harness did not contain the old files. **Expected behavior** Core must save qualified launch files before the first turn. It must restore those exact files under the same owned account and absolute paths. Unknown ownership, missing files, and changed hashes must still stop recovery. **Steps to reproduce** Start an admitted remote OpenCode session. Save its workspace, encrypted checkpoint, and harness data. Destroy the original allocation and controller files. Restore into a new allocation with an image whose qualified proxy has different bytes. Resume the same session. The new opt-in local qualification command automates this path with actual OpenCode and a local synthetic model. ## What Changed - Store company-bound, encrypted launch objects and canonical run/backup references. Generate migration 0438 through the normal database workflow. - Stage qualified Node, proxy, and native OpenCode files under a private content-addressed OS-account home. Record the account UID and exact launch paths. Preserve the Rust profile guard. - Gate first provider work on durable capture under the current native claim. Bound Stop/deadline waits while the actual late transfer keeps its worker and scratch ownership until settlement. Restore cold caches with verified staging and atomic rename. Keep live reattachment and historical image-rooted sessions on their existing paths. - Reuse bounded canonical expiration owners. Atomically fence old checkpoint authority with a strict company/run-scoped retirement marker before external deletion. Keep its indexed cleanup anchor across failed deletes and restart. Preserve earlier resume handles and reject later header or source-consumer writes. Add no idle scan or network call inside a transaction. - Add storage, race, executor, filesystem, and actual Rust resume regressions. Add the opt-in local Linux qualification command and operating limits in `doc/opencode-cold-recovery.md`. ## Verification - RED before repair: real PostgreSQL/AES cold restore lacked the launch files; normal executor preparation chose the new image pack. Separate regressions exposed lost upload receipts, missing cache repair, last-owner retirement, an account-UID mismatch, and a resume-handle race during external deletion. Each expected failure was captured before its repair. - Local qualification before the final master integration: 1,169 executor tests, 233 storage/archive/admission/expiration tests, all seven server typecheck layers, migration safety, and the actual Rust cold-resume test passed. - The maintained Linux command passed with two cached immutable Paperclip images whose proxy hashes differ. It removed the original account/controller trees, restored the production published cwd under a new lease, resumed the same actual OpenCode session, kept a useful file and prior model context, and made one new synthetic model request. No external model call or network was available in either container. - The object-store fixture uses real encrypted StorageService streams with an independent owned local directory. It is not hosted S3 acceptance or a complete task HTTP/actor workflow. Historical tasks still require normal effects reconciliation and recovery after deployment. - Review regressions reproduced partial-cache publication, resume-versus-deletion, fresh-writer and source-consumer resurrection, and unbounded Stop/deadline waits before each repair. They now pass through real filesystem, encrypted storage, PostgreSQL, and SDK boundaries. File validation is split from database ownership (384/831 source lines). - Clean runtime head `aadab7acc69ff959678030db46ffe2dcfbefe28f`: the maintained actual OpenCode cold qualification passed with `dirty:false`. It preserved the provider session and recorded launch paths, restored useful workspace content and prior model context, and made exactly one resumed synthetic model request. The 314,560,000-byte launch capture took 23.118 seconds; the later harness was 833,536 bytes. Subsequent repairs change backup-reference retirement and governed continuity-cache classification; capture, restore, launch, and published-workspace inputs are unchanged, so that runtime receipt remains scoped to those paths. The continuity-break consumer is verified separately. - Retention repair `c14d386cecf7123257918ba6a8d47307df2da397`: 68/68 affected real PostgreSQL/AES storage tests and 15/15 focused executor authority/admission cases pass. All seven server typecheck layers pass (source layers and unchanged test shards, with the final 590-file test layer repeated after fixture edits). All review threads have evidence-backed resolutions. Its subsequent CI caught the separate lifecycle-classification gap described below. The refreshed Paperclip review reported paused work, not a completed new review. - Final head `4d1cd3e3309e328b11abc85adc6d78c6d0c7f55d`: all 19 focused root-entry/executor cases and all three source declaration layers pass. [Greptile reviewed this exact head at 5/5](https://github.com/paperclipai/paperclip/pull/19622#issuecomment-6695984874). Final-head [CI verification](https://github.com/paperclipai/paperclip/actions/runs/42308908167/job/128214100339) passes: 60 successful checks, five conditional skips, and the explicitly neutral paused Paperclip review. No check is pending or failing. - CI also caught the launch cache missing from the session-root lifecycle contract. Both the writer scan and a real governed-replacement consumer failed locally before repair: old launch bytes remained in the active root. The cache now archives with the prior session state; the consumer verifies exact bytes are preserved in the archive and absent from the active root. - The latest review found that backup retirement could delete an independent run pin while a selected session retained the checkpoint. A real PostgreSQL/AES regression failed before the repair. Backup-only ownership now uses a NULL-safe exact backup predicate; the run pin belongs to fenced checkpoint expiration. The regression verifies checkpoint decryption and launch-file restoration after pending-backup cleanup, followed by deletion only after removing the final resume handle. - An executor first-session case checks new-session durable capture through the real capture helper and receipt propagation into the actual harness manifest at the backup boundary. Real storage cases verify the run-profile receipt and canonical backup/run pins. - Sequential closure reuse and simultaneous first admission are different scopes. A deterministic two-uploader test measures both independent uploads, one verified ready winner, loser-only deletion, and usable restoration for both runs. Each capture is bounded to 512 MiB and five minutes; up to three pending receipts per run are retained. There is no claim of global single-flight capture or a company-wide egress bound. - Earlier optional warm-ownership verification encountered missing local binaries. After exact-source setup, all nine ownership assertions passed; two other suites encountered local PostgreSQL setup/teardown contention, so those interrupted attempts are not passes. Required current-head CI supplies the broad final coverage. No deployment or hosted historical recovery is claimed. ## Risks - This adds two tables and a forward migration. The file object is bounded to 512 MiB. Capture adds one initial transfer; later harness archives exclude those large files. Company-scoped ready objects are shared and retained until the last canonical owner retires. Concurrent initial admissions can duplicate bounded transfers; each keeps independent ownership and cancellation rather than waiting on another controller. A continuity break retains the controller cache with its prior state until normal session-root cleanup, adding up to one bounded launch closure per retired state. - Cold continuity requires the same qualified Linux platform, supported runtime dependencies, owned account UID/home, and published absolute cwd. A mismatch fails closed. No alias or Rust guard change is included. - Existing image-rooted profiles have no retained file object. This change cannot backfill missing historical binaries. A fresh governed session must reconcile prior effects first. - Live reattachment does not read artifact objects. Storage or provenance failures before a new turn hold admission. They do not grant new execution, model, tool, or replay authority. ## Model Used OpenAI Codex, GPT-6 family, with reasoning, repository tools, code execution, and local browser tools. The exact served model ID and context window are not exposed in this session. No claim is made for unavailable model metadata. ## 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>
991 lines
82 KiB
TypeScript
991 lines
82 KiB
TypeScript
import fs from "node:fs/promises";
|
|
import { createHash, randomUUID } from "node:crypto";
|
|
import { chmod, mkdir, readFile, rename, stat, symlink, readlink, writeFile } from "node:fs/promises";
|
|
import { join } from "node:path";
|
|
import { and, eq, inArray } from "drizzle-orm";
|
|
import { agentTaskSessions, environmentLeases, heartbeatRuns, issues, nativeHarnessBackups, nativeRunFinalizations, nativeRunResults,
|
|
nativeProviderArtifacts, nativeProviderArtifactReferences, nativeRunnerCheckpoints } from "@paperclipai/db";
|
|
import { afterEach, expect, it, vi } from "vitest";
|
|
import { describeEmbeddedPostgres, nativeRestartFixture } from "../../__tests__/helpers/native-restart.js";
|
|
import { createLocalDiskStorageProvider } from "../../storage/local-disk-provider.js";
|
|
import { createStorageService } from "../../storage/service.js";
|
|
import { nativeHarnessBackupOwner, persistNativeHarnessBackup, restoreDurableNativeHarnessBackup,
|
|
verifyDurableNativeHarnessBackup, deleteRetiredNativeHarnessBackup } from "./native-harness-backup-storage.js";
|
|
import { removeNativeHarnessBackup } from "./native-harness-backup-cleanup.js";
|
|
import { buildNativeHarnessBackupManifest } from "./native-session-executor.js";
|
|
import { createNativeHarnessBackupStamp, verifyNativeHarnessBackupStamp, verifyNativeHarnessBackupStampAsync } from "./native-harness-backup-stamp.js";
|
|
import { hasLocalNativeHarnessObligations, migrateNativeHarnessBackups, prestageNativeHarnessBackups } from "./native-harness-backup-migration.js";
|
|
import { nativeHarnessBackupExpirationService } from "./native-harness-backup-expiration.js";
|
|
import { legacyControllerBootId } from "../legacy-controller-lease.js";
|
|
import { DurablePrpControlPlane, captureLegacyDurablePrpCheckpoint } from "../../vendor/paperclip-runner/index.js";
|
|
import { loadNativeRunnerCheckpoint, nativeRunnerCheckpointWriter } from "./native-runner-checkpoint.js";
|
|
import { restoreNativeProviderArtifacts } from "./native-provider-artifacts.js";
|
|
import * as backupArchive from "./native-harness-backup-archive.js";
|
|
import { canonicalProviderPackJson } from "./remote-provider-pack-verification.js";
|
|
|
|
vi.mock("node:fs/promises", async importOriginal => {
|
|
const original = await importOriginal<typeof import("node:fs/promises")>();
|
|
const files = { ...original };
|
|
return { ...original, default: files, cp: (...args: Parameters<typeof original.cp>) => files.cp(...args) };
|
|
});
|
|
|
|
describeEmbeddedPostgres("durable native provider backups", () => {
|
|
const environment = nativeRestartFixture();
|
|
const ownedHomes: string[] = [];
|
|
function deferred() {
|
|
let resolve!: () => void;
|
|
const promise = new Promise<void>(done => { resolve = done; });
|
|
return { promise, resolve };
|
|
}
|
|
afterEach(async () => {
|
|
for (const home of ownedHomes.splice(0)) await removeNativeHarnessBackup(home);
|
|
vi.unstubAllEnvs();
|
|
});
|
|
async function fixture(provider = "codex") {
|
|
vi.stubEnv("PAPERCLIP_SECRETS_MASTER_KEY", "51".repeat(32));
|
|
const f = await environment.seedRun(randomUUID());
|
|
const { companyId, runtimeRoot, successor } = environment.current;
|
|
await f.db.update(nativeRunFinalizations).set({ controllerBootId: successor.bootId, controllerGeneration: 1,
|
|
leaseOwner: "backup-writer", leaseExpiresAt: new Date(Date.now() + 120_000) }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
const owner = await nativeHarnessBackupOwner(f.db, { companyId, issueId: f.issueId, runId: f.runId, bootId: successor.bootId });
|
|
const home = join(runtimeRoot, randomUUID()), root = join(home, "failover-backups", "current");
|
|
ownedHomes.push(home);
|
|
await mkdir(join(root, provider, "sessions"), { recursive: true });
|
|
await mkdir(join(root, "runner"));
|
|
const identity = { normalizedSessionId: randomUUID(), providerSessionId: randomUUID(), provider };
|
|
await writeFile(join(root, provider, "sessions", "history.json"), JSON.stringify(identity), { mode: 0o600 });
|
|
await writeFile(join(root, "runner", "runner-state.json"), JSON.stringify({ runId: f.runId, lifecycle: "suspended" }), { mode: 0o600 });
|
|
const manifest = JSON.stringify({ schema: "paperclip.native-harness-backup.v1", ...identity, directories: [provider, "runner"] });
|
|
await writeFile(join(root, "manifest.json"), manifest, { mode: 0o600 });
|
|
await chmod(join(root, provider, "sessions"), 0o500);
|
|
const sessionScopeId = randomUUID(), scopeDigest = createHash("sha256").update(sessionScopeId).digest("hex");
|
|
const storage = createStorageService(createLocalDiskStorageProvider(join(runtimeRoot, "durable-objects")));
|
|
return { ...f, home, root, companyId, owner, scopeDigest, sessionScopeId, storage, identity,
|
|
manifestSha256: `sha256:${createHash("sha256").update(manifest).digest("hex")}` };
|
|
}
|
|
async function offlineFixture(base: string) {
|
|
const f = await fixture("codex-home"), home = join(base, f.scopeDigest);
|
|
await rename(f.home, home);
|
|
const root = join(home, "failover-backups", "current"), runnerInstanceId = randomUUID(), providerLeaseId = randomUUID();
|
|
const execution = { provider: { kind: "codex" }, binding: { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, agentId: environment.current.agentId, executionWorkspaceId: randomUUID() },
|
|
workspace: { cwd: "/workspace", repoUrl: null, repoRef: null, branchName: null },
|
|
session: { normalizedSessionId: f.identity.normalizedSessionId, driverKind: "codex_app_server",
|
|
protocolVersion: 1, lifecyclePolicy: { mode: "per_turn", idleTimeoutMs: null } } } as never;
|
|
const manifest = buildNativeHarnessBackupManifest({ backupRoot: root, execution, runnerInstanceId,
|
|
providerSessionIdentity: { providerSessionId: f.identity.providerSessionId }, sourceProviderLeaseId: providerLeaseId });
|
|
await writeFile(join(root, "manifest.json"), JSON.stringify(manifest));
|
|
const stamp = createNativeHarnessBackupStamp({ manifestPath: join(root, "manifest.json"), sessionScopeId: f.sessionScopeId,
|
|
authorizedProviderLeaseId: providerLeaseId, runnerInstanceId, normalizedSessionId: f.identity.normalizedSessionId, completedAt: manifest.completedAt });
|
|
await f.db.update(heartbeatRuns).set({ nativeSessionId: f.identity.normalizedSessionId, runnerInstanceId }).where(eq(heartbeatRuns.id, f.runId));
|
|
await f.db.insert(environmentLeases).values({ companyId: f.companyId, issueId: f.issueId, heartbeatRunId: f.runId,
|
|
providerLeaseId, status: "expired", metadata: { nativeHarnessBackup: stamp } });
|
|
return { ...f, home, root, runnerInstanceId, providerLeaseId };
|
|
}
|
|
async function offlineFixtures(count: number) {
|
|
const base = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(base); await mkdir(base);
|
|
vi.stubEnv("PAPERCLIP_RUNNER_STATE_DIR", base);
|
|
const fixtures = [];
|
|
for (let i = 0; i < count; i++) fixtures.push(await offlineFixture(base));
|
|
const runs = await fixtures[0].db.select({ id: heartbeatRuns.id, issueId: heartbeatRuns.nativeIssueId }).from(heartbeatRuns)
|
|
.where(eq(heartbeatRuns.companyId, fixtures[0].companyId));
|
|
for (const run of runs) await settle({ ...fixtures[0], runId: run.id, issueId: run.issueId! }, new Date());
|
|
return fixtures.sort((a, b) => a.scopeDigest.localeCompare(b.scopeDigest));
|
|
}
|
|
it("overlaps verified offline backups within a four-transfer bound and restores their bytes", async () => {
|
|
const fixtures = await offlineFixtures(6), f = fixtures[0], gate = deferred();
|
|
const put = f.storage.putFile.bind(f.storage); let active = 0, peak = 0, entered = 0;
|
|
vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
active++; entered++; peak = Math.max(peak, active);
|
|
try { await gate.promise; return await put(input); } finally { active--; }
|
|
});
|
|
const migration = migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(15_000) });
|
|
try { await vi.waitFor(() => expect(active).toBe(4), { timeout: 3000 }); }
|
|
finally { gate.resolve(); await migration.catch(() => undefined); }
|
|
expect(await migration).toBe(6); expect(peak).toBe(4); expect(entered).toBe(6); expect(active).toBe(0);
|
|
for (const source of fixtures) {
|
|
await removeNativeHarnessBackup(source.home);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...source, storage: f.storage, backupRoot: replacement })).toBe(true);
|
|
expect(JSON.parse(await readFile(join(replacement, "current", "codex-home", "sessions", "history.json"), "utf8"))).toEqual(source.identity);
|
|
}
|
|
});
|
|
it("waits for cancelled peer transfers before returning failure and permits a complete retry", async () => {
|
|
const fixtures = await offlineFixtures(2), f = fixtures[0], peerEntered = deferred(), release = deferred();
|
|
const failed = deferred(), put = f.storage.putFile.bind(f.storage); let entered = 0, active = 0, settled = false;
|
|
const spy = vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
const ordinal = ++entered; active++;
|
|
try {
|
|
if (ordinal === 1) { await peerEntered.promise; failed.resolve(); throw new Error("storage unavailable"); }
|
|
peerEntered.resolve(); await release.promise; return await put(input);
|
|
} finally { active--; }
|
|
});
|
|
const migration = migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(15_000) });
|
|
const completed = migration.then(() => { settled = true; }, () => { settled = true; });
|
|
try {
|
|
await vi.waitFor(() => expect(entered).toBe(2), { timeout: 3000 }); await failed.promise;
|
|
await new Promise<void>(resolve => setImmediate(resolve)); expect(settled).toBe(false);
|
|
} finally { peerEntered.resolve(); release.resolve(); await completed; }
|
|
await expect(migration).rejects.toThrow("storage unavailable"); expect(active).toBe(0); spy.mockRestore();
|
|
for (const source of fixtures) expect(await readFile(join(source.root, "manifest.json"), "utf8")).toContain("paperclip.native-harness-backup.v1");
|
|
expect(await migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(15_000) })).toBe(2);
|
|
for (const source of fixtures) {
|
|
const rows = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, source.scopeDigest));
|
|
expect(rows.filter(row => row.status === "current")).toHaveLength(1);
|
|
}
|
|
});
|
|
it("preserves unlisted files with an exclusive scratch reservation", async () => {
|
|
const fixtures = await offlineFixtures(3), f = fixtures[0], extra = fixtures[1];
|
|
await writeFile(join(extra.root, "retained-extra.txt"), "unlisted surviving bytes");
|
|
const put = f.storage.putFile.bind(f.storage); let active = 0, peak = 0;
|
|
vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
active++; peak = Math.max(peak, active);
|
|
try { await new Promise<void>(resolve => setImmediate(resolve)); return await put(input); } finally { active--; }
|
|
});
|
|
expect(await migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(15_000) })).toBe(3);
|
|
expect(peak).toBe(1);
|
|
await removeNativeHarnessBackup(extra.home);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...extra, storage: f.storage, backupRoot: replacement })).toBe(true);
|
|
expect(await readFile(join(replacement, "current", "retained-extra.txt"), "utf8")).toBe("unlisted surviving bytes");
|
|
});
|
|
it("enforces the reserved archive size while streaming before publishing any object", async () => {
|
|
const f = await fixture(), put = vi.spyOn(f.storage, "putFile");
|
|
await expect(persistNativeHarnessBackup({ ...f, maxArchiveBytes: 512 })).rejects.toThrow("native_harness_archive_invalid");
|
|
expect(put).not.toHaveBeenCalled();
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toHaveLength(0);
|
|
expect(JSON.parse(await readFile(join(f.root, "codex", "sessions", "history.json"), "utf8"))).toEqual(f.identity);
|
|
});
|
|
it.each(["codex", "claude", "opencode", "grok", "new-provider"])("restores %s session bytes after deleting the entire controller home", async provider => {
|
|
const f = await fixture(provider);
|
|
await persistNativeHarnessBackup(f);
|
|
const [row] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
expect(row.status).toBe("current");
|
|
const object = await f.storage.getObject(f.companyId, row.reference.objectKey as string);
|
|
const parts: Buffer[] = []; for await (const part of object.stream) parts.push(part);
|
|
expect(Buffer.concat(parts).includes(Buffer.from(f.identity.providerSessionId))).toBe(false);
|
|
await removeNativeHarnessBackup(f.home);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID(), "failover-backups");
|
|
ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(JSON.parse(await readFile(join(replacement, "current", provider, "sessions", "history.json"), "utf8"))).toEqual(f.identity);
|
|
expect((await stat(join(replacement, "current", provider, "sessions"))).mode & 0o777).toBe(0o500);
|
|
expect(await verifyDurableNativeHarnessBackup(f)).toBe(true);
|
|
});
|
|
async function artifactFixture(artifactVersion = randomUUID()) {
|
|
const f = await fixture("opencode");
|
|
const sourceRoot = join(f.home, "qualified-provider-pack");
|
|
await mkdir(sourceRoot);
|
|
const original = { nodeCommand: "qualified Node executable", opencodeExecutable: "qualified OpenCode executable",
|
|
opencodeProxy: `qualified original proxy bytes ${artifactVersion}` };
|
|
const artifacts = Object.fromEntries(await Promise.all(Object.entries(original).map(async ([name, bytes]) => {
|
|
await writeFile(join(sourceRoot, name), bytes, { mode: 0o500 });
|
|
return [name, { path: name, sha256: `sha256:${createHash("sha256").update(bytes).digest("hex")}` }];
|
|
})));
|
|
const payload = { runnerSourceRevision: "a".repeat(40), target: { platform: "linux", architecture: "x64" },
|
|
pins: { opencode: "1.18.34" }, artifacts: { ...artifacts, opencodeCommand: artifacts.opencodeExecutable } };
|
|
const manifest = { schema: "paperclip-runner/remote-provider-pack/v1", payload,
|
|
digest: `sha256:${createHash("sha256").update(canonicalProviderPackJson(payload)).digest("hex")}` };
|
|
const launchRoot = `/home/daytona/.paperclip-runtime/paperclip-runner/provider-artifacts/${manifest.digest.slice(7)}`;
|
|
await f.db.update(heartbeatRuns).set({ nativeSessionId: f.identity.normalizedSessionId, driverKind: "opencode_server",
|
|
runnerProfileJson: { ...f.preparedProfile, nativeProviderPackAdmissionState: {
|
|
schema: "paperclip.native-provider-pack-admission-state.v1", state: "verified", admissionEpoch: randomUUID() },
|
|
nativeProviderPackAdmission: { schema: "paperclip.native-provider-pack-admission.v1", stage: "session_admitted",
|
|
authority: "core_verified_selection", companyId: f.companyId, agentId: environment.current.agentId,
|
|
runId: f.runId, sessionId: f.identity.normalizedSessionId, runnerInstanceId: "artifact-fixture-runner",
|
|
runnerEnvironmentLeaseId: "artifact-fixture-lease", providerPackDigest: manifest.digest,
|
|
opencodeProxySha256: artifacts.opencodeProxy.sha256, sourceRevision: payload.runnerSourceRevision } },
|
|
}).where(eq(heartbeatRuns.id, f.runId));
|
|
return { ...f, original, manifest, launchRoot, providerArtifacts: { sourceRoot, manifest, launchRoot, accountUid: 1000 } };
|
|
}
|
|
it("restores the admitted OpenCode launch bytes after the controller and sandbox image change", async () => {
|
|
const publication = await artifactFixture(), f = publication;
|
|
const { original, manifest, launchRoot } = publication;
|
|
await persistNativeHarnessBackup(publication);
|
|
await removeNativeHarnessBackup(f.home);
|
|
const newImage = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(newImage);
|
|
await mkdir(newImage);
|
|
await writeFile(join(newImage, "opencodeProxy"), "qualified newer proxy bytes");
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID(), "failover-backups"); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(JSON.parse(await readFile(join(replacement, "current/opencode/sessions/history.json"), "utf8"))).toEqual(f.identity);
|
|
for (const [name, bytes] of Object.entries(original)) {
|
|
expect(await readFile(join(replacement, "current/provider-artifacts", name), "utf8")).toBe(bytes);
|
|
}
|
|
const restored = JSON.parse(await readFile(join(replacement, "current/manifest.json"), "utf8"));
|
|
expect(restored.providerArtifacts).toMatchObject({ launchRoot, providerPackDigest: manifest.digest });
|
|
});
|
|
it("repairs a missing local launch cache without adding the binaries to each harness backup", async () => {
|
|
const f = await artifactFixture(); await persistNativeHarnessBackup(f);
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
await removeNativeHarnessBackup(join(replacement, "current/provider-artifacts"));
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(await readFile(join(replacement, "current/provider-artifacts/opencodeProxy"), "utf8")).toBe(f.original.opencodeProxy);
|
|
await persistNativeHarnessBackup({ ...f, providerArtifacts: undefined, root: join(replacement, "current") });
|
|
const [after] = await f.db.select().from(nativeHarnessBackups).where(and(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest),
|
|
eq(nativeHarnessBackups.status, "current")));
|
|
expect(after.reference.byteSize).toBe(before.reference.byteSize);
|
|
});
|
|
it("retries an interrupted artifact cache copy without publishing partial launch files", async () => {
|
|
const f = await artifactFixture(); await persistNativeHarnessBackup(f);
|
|
const receipt = JSON.parse(await readFile(join(f.root, "manifest.json"), "utf8")).providerArtifacts;
|
|
const home = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(home); await mkdir(home);
|
|
const targetRoot = join(home, "provider-artifacts"), copy = fs.cp.bind(fs);
|
|
const interrupted = vi.spyOn(fs, "cp").mockImplementation(async (source, destination, options) => {
|
|
if (String(destination).startsWith(targetRoot)) {
|
|
await mkdir(destination, { recursive: true });
|
|
await writeFile(join(String(destination), "opencodeProxy"), "partial copy");
|
|
throw Object.assign(new Error("controlled disk full during cache publication"), { code: "ENOSPC" });
|
|
}
|
|
return copy(source, destination, options);
|
|
});
|
|
try { await expect(restoreNativeProviderArtifacts({ ...f, receipt, targetRoot })).rejects.toThrow("controlled disk full"); }
|
|
finally { interrupted.mockRestore(); }
|
|
await expect(fs.stat(targetRoot)).rejects.toMatchObject({ code: "ENOENT" });
|
|
await expect(restoreNativeProviderArtifacts({ ...f, receipt, targetRoot })).resolves.toBeUndefined();
|
|
expect(await readFile(join(targetRoot, "opencodeProxy"), "utf8")).toBe(f.original.opencodeProxy);
|
|
expect(await fs.readdir(home)).toEqual(["provider-artifacts"]);
|
|
});
|
|
it("reuses a sequentially admitted closure and releases it only after the last stopped checkpoint retires", async () => {
|
|
const version = randomUUID(), first = await artifactFixture(version), other = await artifactFixture(version);
|
|
const second = { ...other, storage: first.storage };
|
|
const put = vi.spyOn(first.storage, "putFile");
|
|
await persistNativeHarnessBackup(first); await persistNativeHarnessBackup(second);
|
|
expect(put.mock.calls.filter(([input]) => input.namespace === "native-provider-artifacts")).toHaveLength(1);
|
|
const [firstBackup] = await first.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, first.scopeDigest));
|
|
const [secondBackup] = await first.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, second.scopeDigest));
|
|
const artifactId = (firstBackup.reference.providerArtifacts as { artifactId: string }).artifactId;
|
|
const [artifact] = await first.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, artifactId));
|
|
const stop = async (f: typeof first) => {
|
|
await settle(f, new Date(Date.now() - 10 * 86_400_000));
|
|
const leaseId = randomUUID(), providerLeaseId = randomUUID(), confirmedAt = new Date();
|
|
await f.db.insert(environmentLeases).values({ id: leaseId, companyId: f.companyId, heartbeatRunId: f.runId,
|
|
issueId: f.issueId, provider: "daytona", providerLeaseId, status: "expired", cleanupStatus: "success", releasedAt: confirmedAt,
|
|
metadata: { remoteExecutionTermination: { schema: "paperclip.remote-termination.v1", companyId: f.companyId,
|
|
runId: f.runId, leaseId, provider: "daytona", providerLeaseId, state: "destroyed", confirmedAt: confirmedAt.toISOString() } } });
|
|
};
|
|
await stop(first);
|
|
await first.db.update(nativeHarnessBackups).set({ status: "retired" }).where(eq(nativeHarnessBackups.id, firstBackup.id));
|
|
expect(await deleteRetiredNativeHarnessBackup(first.db, first.storage, firstBackup.id)).toBe(true);
|
|
expect(await first.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, artifact.id))).toHaveLength(1);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...second, backupRoot: replacement })).toBe(true);
|
|
await stop(second);
|
|
await second.db.update(nativeHarnessBackups).set({ status: "retired" }).where(eq(nativeHarnessBackups.id, secondBackup.id));
|
|
expect(await deleteRetiredNativeHarnessBackup(second.db, second.storage, secondBackup.id)).toBe(true);
|
|
expect(await first.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, artifact.id))).toHaveLength(1);
|
|
const runPins = await first.db.select().from(nativeProviderArtifactReferences).where(eq(nativeProviderArtifactReferences.artifactId, artifact.id));
|
|
expect(runPins).toHaveLength(2); expect(runPins.every(pin => pin.backupId === null)).toBe(true);
|
|
await first.db.insert(nativeRunnerCheckpoints).values([first, second].map(f => ({ companyId: f.companyId,
|
|
issueId: f.issueId, runId: f.runId, part: "header", material: { retained: "canonical checkpoint expiry fixture" } })));
|
|
expect((await nativeHarnessBackupExpirationService(first.db, { storage: first.storage }).tickCheckpoints()).expiredCheckpoints).toBe(2);
|
|
expect(await first.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, artifact.id))).toHaveLength(0);
|
|
expect(await first.db.select().from(nativeProviderArtifactReferences).where(eq(nativeProviderArtifactReferences.artifactId, artifact.id))).toHaveLength(0);
|
|
});
|
|
it("keeps independent overlapping uploads bounded and retains only their verified winner", async () => {
|
|
const version = randomUUID(), first = await artifactFixture(version), other = await artifactFixture(version);
|
|
const second = { ...other, storage: first.storage }, put = first.storage.putFile.bind(first.storage);
|
|
let release!: () => void, arrivals = 0;
|
|
const together = new Promise<void>(resolve => { release = resolve; });
|
|
const uploaded: string[] = [];
|
|
const calls = vi.spyOn(first.storage, "putFile").mockImplementation(async input => {
|
|
if (input.namespace === "native-provider-artifacts") {
|
|
expect(input.objectKey).toBeTypeOf("string"); uploaded.push(input.objectKey!); if (++arrivals === 2) release();
|
|
await together; expect("byteSize" in input ? input.byteSize : input.body.length).toBeLessThan(512 * 1024 * 1024 + 1024 * 1024);
|
|
}
|
|
return put(input);
|
|
});
|
|
const deleted = vi.spyOn(first.storage, "deleteObject");
|
|
const transfers = Promise.all([persistNativeHarnessBackup(first), persistNativeHarnessBackup(second)]);
|
|
void transfers.catch(() => undefined);
|
|
try {
|
|
try { await vi.waitFor(() => expect(arrivals).toBe(2), { timeout: 10_000 }); } finally { release(); }
|
|
await transfers;
|
|
} finally { calls.mockRestore(); }
|
|
expect(uploaded).toHaveLength(2);
|
|
const objects = await first.db.select().from(nativeProviderArtifacts).where(and(eq(nativeProviderArtifacts.companyId, first.companyId),
|
|
inArray(nativeProviderArtifacts.id, uploaded.map(key => key.split("/").at(-2)!))));
|
|
expect(objects).toHaveLength(1); expect(objects[0].status).toBe("ready");
|
|
expect(deleted.mock.calls.map(([, key]) => key)).toEqual(uploaded.filter(key => key !== objects[0].reference!.objectKey));
|
|
for (const f of [first, second]) {
|
|
const [run] = await f.db.select().from(heartbeatRuns).where(eq(heartbeatRuns.id, f.runId));
|
|
expect(run.runnerProfileJson?.nativeProviderArtifacts).toMatchObject({ artifactId: objects[0].id });
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(await readFile(join(replacement, "current/provider-artifacts/opencodeProxy"), "utf8")).toBe(f.original.opencodeProxy);
|
|
}
|
|
});
|
|
it("keeps a resumable run pin when an interrupted backup is retired", async () => {
|
|
const f = await artifactFixture(), put = f.storage.putFile.bind(f.storage);
|
|
const interrupted = vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
const result = await put(input);
|
|
if (input.namespace === "native-harness-backups") throw new Error("harness upload reply lost");
|
|
return result;
|
|
});
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("harness upload reply lost"); interrupted.mockRestore();
|
|
const [pending] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
expect(pending.status).toBe("pending");
|
|
const receipt = pending.reference.providerArtifacts as NonNullable<Parameters<typeof restoreNativeProviderArtifacts>[0]["receipt"]>;
|
|
const identity = { runnerInstanceId: f.native.runnerInstanceId, environmentLeaseId: f.native.environmentLeaseId,
|
|
runId: f.runId, normalizedSessionId: f.identity.normalizedSessionId, turnId: f.native.turnId, itemId: f.native.itemId };
|
|
const controller = new DurablePrpControlPlane({ stateDirectory: join(f.home, "resumable-controller"), identity,
|
|
expectedRunnerVersion: "0.0.0", expectedRunnerDigest: "sha256:" + "1".repeat(64) });
|
|
const checkpoint = captureLegacyDurablePrpCheckpoint(JSON.stringify(controller.store.state), {
|
|
runId: f.runId, runnerInstanceId: identity.runnerInstanceId, normalizedSessionId: identity.normalizedSessionId },
|
|
{ version: "0.0.0", digest: "sha256:" + "1".repeat(64) });
|
|
await nativeRunnerCheckpointWriter(f.db, { ...f.owner, leaseOwner: f.owner.leaseOwner! })(checkpoint);
|
|
const old = new Date(Date.now() - 10 * 86_400_000); await settle(f, old);
|
|
await f.db.update(nativeHarnessBackups).set({ createdAt: old, updatedAt: old }).where(eq(nativeHarnessBackups.id, pending.id));
|
|
const leaseId = randomUUID(), providerLeaseId = randomUUID(), confirmedAt = new Date();
|
|
await f.db.insert(environmentLeases).values({ id: leaseId, companyId: f.companyId, heartbeatRunId: f.runId,
|
|
issueId: f.issueId, provider: "daytona", providerLeaseId, status: "expired", cleanupStatus: "success", releasedAt: confirmedAt,
|
|
metadata: { remoteExecutionTermination: { schema: "paperclip.remote-termination.v1", companyId: f.companyId,
|
|
runId: f.runId, leaseId, provider: "daytona", providerLeaseId, state: "destroyed", confirmedAt: confirmedAt.toISOString() } } });
|
|
const [selected] = await f.db.insert(agentTaskSessions).values({ companyId: f.companyId, agentId: environment.current.agentId,
|
|
adapterType: "paperclip_runner", taskKey: f.issueId, lastRunId: f.runId,
|
|
sessionParamsJson: { sessionId: f.identity.normalizedSessionId } }).returning();
|
|
const retention = nativeHarnessBackupExpirationService(f.db, { storage: f.storage });
|
|
const first = await retention.tickCheckpoints();
|
|
expect(first.expired).toBe(1); expect(first.expiredCheckpoints).toBe(0);
|
|
expect(await f.db.select().from(nativeProviderArtifactReferences).where(eq(nativeProviderArtifactReferences.runId, f.runId)))
|
|
.toMatchObject([{ backupId: null, artifactId: receipt.artifactId }]);
|
|
expect((await loadNativeRunnerCheckpoint(f.db, { companyId: f.companyId, issueId: f.issueId, runId: f.runId }))?.recoveryEvidence.normalizedSessionId)
|
|
.toBe(f.identity.normalizedSessionId);
|
|
await removeNativeHarnessBackup(f.providerArtifacts.sourceRoot);
|
|
const restored = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(restored);
|
|
await restoreNativeProviderArtifacts({ ...f, receipt, targetRoot: restored });
|
|
expect(await readFile(join(restored, "opencodeProxy"), "utf8")).toBe(f.original.opencodeProxy);
|
|
await f.db.delete(agentTaskSessions).where(eq(agentTaskSessions.id, selected.id));
|
|
expect((await retention.tickCheckpoints()).expiredCheckpoints).toBe(1);
|
|
expect(await f.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, receipt.artifactId))).toEqual([]);
|
|
});
|
|
it("keeps the exact encrypted upload receipt when its artifact owner changes after upload", async () => {
|
|
const f = await artifactFixture(), put = f.storage.putFile.bind(f.storage);
|
|
vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
const result = await put(input);
|
|
if (input.namespace === "native-provider-artifacts") await f.db.update(nativeRunFinalizations)
|
|
.set({ controllerGeneration: 2 }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
return result;
|
|
});
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("native_runner_checkpoint_owner_lost");
|
|
const [pin] = await f.db.select().from(nativeProviderArtifactReferences).where(eq(nativeProviderArtifactReferences.runId, f.runId));
|
|
const [artifact] = await f.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, pin.artifactId));
|
|
expect(artifact.status).toBe("pending");
|
|
expect(artifact.reference).toMatchObject({ companyId: f.companyId, artifactId: artifact.id, objectKey: expect.any(String), material: expect.any(Object) });
|
|
expect(await readFile(join(f.providerArtifacts.sourceRoot, "opencodeProxy"), "utf8")).toBe(f.original.opencodeProxy);
|
|
});
|
|
it("retires an abandoned artifact upload through checkpoint expiration without losing deletion retries", async () => {
|
|
const f = await artifactFixture(), put = f.storage.putFile.bind(f.storage);
|
|
const lostReply = vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
await put(input); throw new Error("artifact upload reply lost");
|
|
});
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("artifact upload reply lost"); lostReply.mockRestore();
|
|
const [pin] = await f.db.select().from(nativeProviderArtifactReferences).where(eq(nativeProviderArtifactReferences.runId, f.runId));
|
|
const [artifact] = await f.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, pin.artifactId));
|
|
expect(artifact.reference).toMatchObject({ objectKey: expect.any(String), material: expect.any(Object) });
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toEqual([]);
|
|
await f.db.insert(nativeRunnerCheckpoints).values({ companyId: f.companyId, runId: f.runId,
|
|
issueId: f.issueId, part: "header", material: { retained: "canonical header index fixture" } });
|
|
const retention = nativeHarnessBackupExpirationService(f.db, { storage: f.storage });
|
|
expect((await retention.tickCheckpoints()).expiredCheckpoints).toBe(0);
|
|
const old = new Date(Date.now() - 10 * 86_400_000);
|
|
await settle(f, old);
|
|
const leaseId = randomUUID(), providerLeaseId = randomUUID(), confirmedAt = new Date();
|
|
await f.db.insert(environmentLeases).values({ id: leaseId, companyId: f.companyId, heartbeatRunId: f.runId,
|
|
issueId: f.issueId, provider: "daytona", providerLeaseId, status: "expired", cleanupStatus: "success", releasedAt: confirmedAt,
|
|
metadata: { remoteExecutionTermination: { schema: "paperclip.remote-termination.v1", companyId: f.companyId,
|
|
runId: f.runId, leaseId, provider: "daytona", providerLeaseId, state: "destroyed", confirmedAt: confirmedAt.toISOString() } } });
|
|
const drop = vi.spyOn(f.storage, "deleteObject").mockRejectedValueOnce(new Error("delete reply lost"));
|
|
const [earlier] = await f.db.insert(agentTaskSessions).values({ companyId: f.companyId, agentId: environment.current.agentId,
|
|
adapterType: "paperclip_runner", taskKey: f.issueId, lastRunId: f.runId }).returning();
|
|
expect((await retention.tickCheckpoints()).expiredCheckpoints).toBe(0);
|
|
expect(drop).not.toHaveBeenCalled();
|
|
expect((await f.db.select().from(nativeRunnerCheckpoints).where(eq(nativeRunnerCheckpoints.runId, f.runId)))[0].material)
|
|
.toEqual({ retained: "canonical header index fixture" });
|
|
await f.db.delete(agentTaskSessions).where(eq(agentTaskSessions.id, earlier.id));
|
|
expect((await retention.tickCheckpoints()).failed).toBe(1);
|
|
const [retired] = await f.db.select().from(nativeRunnerCheckpoints).where(eq(nativeRunnerCheckpoints.runId, f.runId));
|
|
expect(retired.material).toMatchObject({ schema: "paperclip.native-runner-checkpoint-retirement.v1", companyId: f.companyId,
|
|
runIds: [f.runId], sessionId: f.identity.normalizedSessionId });
|
|
expect(await f.db.select().from(nativeProviderArtifactReferences).where(eq(nativeProviderArtifactReferences.artifactId, artifact.id))).toHaveLength(1);
|
|
await f.db.insert(agentTaskSessions).values({ companyId: f.companyId, agentId: environment.current.agentId,
|
|
adapterType: "paperclip_runner", taskKey: f.issueId, lastRunId: f.runId });
|
|
await expect(loadNativeRunnerCheckpoint(f.db, { companyId: f.companyId, issueId: f.issueId, runId: f.runId }))
|
|
.rejects.toThrow("native_runner_checkpoint_retired");
|
|
const missingStorage = nativeHarnessBackupExpirationService(f.db);
|
|
expect((await missingStorage.tickCheckpoints()).failed).toBe(1);
|
|
expect(drop).toHaveBeenCalledTimes(1);
|
|
const [priorOwner] = await f.db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
const currentOwner = { ...f.owner, generation: priorOwner.controllerGeneration, leaseOwner: "fresh-retired-source-claim" };
|
|
await f.db.update(nativeRunFinalizations).set({ phase: "observed", leaseOwner: currentOwner.leaseOwner,
|
|
leaseExpiresAt: new Date(Date.now() + 60_000) }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
const identity = { runnerInstanceId: randomUUID(), environmentLeaseId: randomUUID(), runId: f.runId,
|
|
normalizedSessionId: f.identity.normalizedSessionId, turnId: randomUUID(), itemId: randomUUID() };
|
|
const controller = new DurablePrpControlPlane({ stateDirectory: join(f.home, "new-source-controller"), identity,
|
|
expectedRunnerVersion: "0.0.0", expectedRunnerDigest: "sha256:" + "1".repeat(64) });
|
|
const checkpoint = captureLegacyDurablePrpCheckpoint(JSON.stringify(controller.store.state), {
|
|
runId: f.runId, runnerInstanceId: identity.runnerInstanceId, normalizedSessionId: identity.normalizedSessionId },
|
|
{ version: "0.0.0", digest: "sha256:" + "1".repeat(64) });
|
|
try { await expect(nativeRunnerCheckpointWriter(f.db, currentOwner)(checkpoint)).rejects.toThrow("native_runner_checkpoint_retired"); }
|
|
finally { await f.db.update(nativeRunFinalizations).set({ phase: priorOwner.phase, leaseOwner: null,
|
|
leaseExpiresAt: null }).where(eq(nativeRunFinalizations.runId, f.runId)); }
|
|
const consumerRunId = randomUUID(), consumerLeaseId = randomUUID();
|
|
await f.db.insert(heartbeatRuns).values({ id: consumerRunId, companyId: f.companyId, agentId: environment.current.agentId,
|
|
issueId: f.issueId, nativeIssueId: f.issueId, nativeSessionId: f.identity.normalizedSessionId, status: "running" });
|
|
await f.db.insert(nativeRunFinalizations).values({ runId: consumerRunId, companyId: f.companyId, issueId: f.issueId,
|
|
phase: "observed", controllerBootId: f.owner.bootId, controllerGeneration: 1, leaseOwner: "late-source-consumer",
|
|
leaseExpiresAt: new Date(Date.now() + 60_000) });
|
|
await f.db.update(issues).set({ status: "in_progress", executionRunId: consumerRunId }).where(eq(issues.id, f.issueId));
|
|
await f.db.update(environmentLeases).set({ metadata: { remoteExecutionTermination: {
|
|
schema: "paperclip.remote-termination.v1", companyId: f.companyId, runId: f.runId, leaseId, provider: "daytona",
|
|
providerLeaseId, state: "destroyed", confirmedAt: confirmedAt.toISOString() },
|
|
runnerTaskSourceConsumer: { leaseId: consumerLeaseId, runId: consumerRunId, accountIdentity: "owned-fixture-account", identity } } })
|
|
.where(eq(environmentLeases.id, leaseId));
|
|
await f.db.insert(environmentLeases).values({ id: consumerLeaseId, companyId: f.companyId, heartbeatRunId: consumerRunId,
|
|
issueId: f.issueId, provider: "daytona", providerLeaseId: randomUUID(), status: "active", metadata: { runnerTaskSourceRecovery: {
|
|
schema: "paperclip.runner.source-recovery.v1", sourceLeaseId: leaseId, consumerLeaseId, consumerRunId,
|
|
accountIdentity: "owned-fixture-account", controllerBootId: f.owner.bootId, controllerGeneration: 1, identity } } });
|
|
const lateCheckpoint = { ...checkpoint, reset: false, changedRecords: new Map([["cursor", checkpoint.records.get("cursor")!]]), removedRecords: [] };
|
|
await expect(nativeRunnerCheckpointWriter(f.db, { ...f.owner, authorityRunId: consumerRunId, generation: 1,
|
|
leaseOwner: "late-source-consumer" })(lateCheckpoint)).rejects.toThrow("native_runner_checkpoint_retired");
|
|
const restarted = nativeHarnessBackupExpirationService(f.db, { storage: f.storage });
|
|
expect((await restarted.tickCheckpoints()).expiredCheckpoints).toBe(1);
|
|
expect(await f.db.select().from(nativeProviderArtifacts).where(eq(nativeProviderArtifacts.id, artifact.id))).toEqual([]);
|
|
expect(await f.db.select().from(nativeRunnerCheckpoints).where(eq(nativeRunnerCheckpoints.runId, f.runId))).toEqual([]);
|
|
expect((await restarted.tickCheckpoints()).expiredCheckpoints).toBe(0);
|
|
expect(drop).toHaveBeenCalledTimes(2);
|
|
});
|
|
it("retains every artifact and checkpoint when a retirement family exceeds its run bound", async () => {
|
|
const f = await artifactFixture(), put = f.storage.putFile.bind(f.storage);
|
|
const lost = vi.spyOn(f.storage, "putFile").mockImplementation(async input => { await put(input); throw new Error("upload reply lost"); });
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("upload reply lost"); lost.mockRestore();
|
|
const old = new Date(Date.now() - 10 * 86_400_000); await settle(f, old);
|
|
await f.db.insert(nativeRunnerCheckpoints).values({ companyId: f.companyId, issueId: f.issueId, runId: f.runId,
|
|
part: "header", material: { retained: "original checkpoint" } });
|
|
const relatives = Array.from({ length: 100 }, () => randomUUID());
|
|
await f.db.insert(heartbeatRuns).values(relatives.map(id => ({ id, companyId: f.companyId, agentId: environment.current.agentId,
|
|
issueId: f.issueId, nativeIssueId: f.issueId, nativeSessionId: f.identity.normalizedSessionId, status: "failed", finishedAt: old })));
|
|
await f.db.insert(nativeRunFinalizations).values(relatives.map(runId => ({ runId, companyId: f.companyId, issueId: f.issueId, phase: "terminal_failure" })));
|
|
const remove = vi.spyOn(f.storage, "deleteObject");
|
|
const result = await nativeHarnessBackupExpirationService(f.db, { storage: f.storage }).tickCheckpoints();
|
|
expect(result.expiredCheckpoints).toBe(0); expect(remove).not.toHaveBeenCalled();
|
|
expect((await f.db.select().from(nativeRunnerCheckpoints).where(eq(nativeRunnerCheckpoints.runId, f.runId)))[0].material)
|
|
.toEqual({ retained: "original checkpoint" });
|
|
expect(await f.db.select().from(nativeProviderArtifactReferences).where(eq(nativeProviderArtifactReferences.runId, f.runId))).toHaveLength(1);
|
|
});
|
|
it("keeps the last published backup when object storage fails", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const put = vi.spyOn(f.storage, "putFile").mockRejectedValue(new Error("storage unavailable"));
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("storage unavailable"); put.mockRestore();
|
|
const rows = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
expect(rows.filter(row => row.status === "current").map(row => row.id)).toEqual([before.id]);
|
|
expect(rows.filter(row => row.status === "pending")).toHaveLength(1);
|
|
expect(await readFile(join(f.root, "manifest.json"), "utf8")).toContain("native-harness-backup");
|
|
});
|
|
async function commitWarmRun(f: Awaited<ReturnType<typeof fixture>>) {
|
|
const [run] = await f.db.select().from(heartbeatRuns).where(eq(heartbeatRuns.id, f.runId));
|
|
const [result] = await f.db.insert(nativeRunResults).values({ companyId: f.companyId, issueId: f.issueId, runId: f.runId,
|
|
completionContractId: run.completionContractId!, serverFingerprint: "completed-warm-turn", schemaStatus: "accepted",
|
|
resultJson: { summary: "Retained provider history" }, canonicalSha256: "completed-warm-turn" }).returning();
|
|
await f.db.update(heartbeatRuns).set({ status: "succeeded", finishedAt: new Date() }).where(eq(heartbeatRuns.id, f.runId));
|
|
await f.db.update(nativeRunFinalizations).set({ phase: "committed", resultId: result.id, leaseOwner: null,
|
|
leaseExpiresAt: null }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
return result;
|
|
}
|
|
async function releaseWhileSavingWorkspace(f: Awaited<ReturnType<typeof fixture>>) {
|
|
const result = await commitWarmRun(f);
|
|
await f.db.update(nativeRunFinalizations).set({ phase: "workspace_finalizing" })
|
|
.where(eq(nativeRunFinalizations.runId, f.runId));
|
|
await f.db.update(heartbeatRuns).set({ status: "running", finishedAt: null })
|
|
.where(eq(heartbeatRuns.id, f.runId));
|
|
await f.db.update(issues).set({ executionRunId: f.runId }).where(eq(issues.id, f.issueId));
|
|
return result;
|
|
}
|
|
async function inactiveFixture() {
|
|
const [f] = await offlineFixtures(1);
|
|
await commitWarmRun(f);
|
|
await f.db.update(issues).set({ executionRunId: null, checkoutRunId: null }).where(eq(issues.id, f.issueId));
|
|
const owner = await nativeHarnessBackupOwner(f.db, { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, bootId: environment.current.successor.bootId });
|
|
return { ...f, owner, inactiveSource: { normalizedSessionId: f.identity.normalizedSessionId,
|
|
runnerInstanceId: f.runnerInstanceId, providerLeaseId: f.providerLeaseId } };
|
|
}
|
|
it("pre-copies released history and restores its bytes while an unrelated execution stays active", async () => {
|
|
const f = await inactiveFixture(), active = await environment.seedRun(randomUUID());
|
|
const input = { storage: f.storage, signal: AbortSignal.timeout(15_000) };
|
|
expect(await prestageNativeHarnessBackups(f.db, input)).toEqual({ inspected: 1, published: 1, current: 0, deferred: {} });
|
|
expect(await prestageNativeHarnessBackups(f.db, input)).toEqual({ inspected: 1, published: 0, current: 1, deferred: {} });
|
|
await expect(migrateNativeHarnessBackups(f.db, { ...input, writersStopped: true })).rejects.toThrow("requires_offline_writers");
|
|
expect((await active.db.select().from(heartbeatRuns).where(eq(heartbeatRuns.id, active.runId)))[0].status).toBe("running");
|
|
await removeNativeHarnessBackup(f.home);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(JSON.parse(await readFile(join(replacement, "current", "codex-home", "sessions", "history.json"), "utf8"))).toEqual(f.identity);
|
|
});
|
|
it.each(["queued", "running"] as const)("defers pre-copy when the same task has a %s execution", async status => {
|
|
const f = await inactiveFixture(), put = vi.spyOn(f.storage, "putFile");
|
|
await f.db.insert(heartbeatRuns).values({ id: randomUUID(), companyId: f.companyId, agentId: environment.current.agentId,
|
|
issueId: f.issueId, nativeIssueId: f.issueId, nativeSessionId: f.identity.normalizedSessionId, status, runtimeMode: "native" });
|
|
expect(await prestageNativeHarnessBackups(f.db, { storage: f.storage, signal: AbortSignal.timeout(15_000) }))
|
|
.toEqual({ inspected: 1, published: 0, current: 0, deferred: { source_active: 1 } });
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("native_harness_storage_prestage_source_active");
|
|
expect(put).not.toHaveBeenCalled();
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toEqual([]);
|
|
});
|
|
it("defers failed history without treating pre-copy as offline authority", async () => {
|
|
const [f] = await offlineFixtures(1);
|
|
expect(await prestageNativeHarnessBackups(f.db, { storage: f.storage, signal: AbortSignal.timeout(15_000) }))
|
|
.toEqual({ inspected: 1, published: 0, current: 0, deferred: { source_not_successful: 1 } });
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toEqual([]);
|
|
});
|
|
it("defers a source being replaced online while offline migration still refuses it", async () => {
|
|
const f = await inactiveFixture();
|
|
await writeFile(join(f.root, "runner", "runner-state.json"), JSON.stringify({ runId: f.runId, lifecycle: "running" }));
|
|
expect(await prestageNativeHarnessBackups(f.db, { storage: f.storage, signal: AbortSignal.timeout(15_000) }))
|
|
.toEqual({ inspected: 1, published: 0, current: 0, deferred: { source_unavailable: 1 } });
|
|
await expect(migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(15_000) }))
|
|
.rejects.toThrow("native_harness_migration_source_unsettled");
|
|
});
|
|
async function releasedFixtures() {
|
|
const fixtures = await offlineFixtures(2);
|
|
for (const f of fixtures) {
|
|
await commitWarmRun(f);
|
|
await f.db.update(issues).set({ executionRunId: null, checkoutRunId: null }).where(eq(issues.id, f.issueId));
|
|
}
|
|
return fixtures;
|
|
}
|
|
async function damageSource(f: Awaited<ReturnType<typeof offlineFixture>>, kind: string) {
|
|
if (kind === "unsafe") {
|
|
await rename(f.root, f.root + "-retained");
|
|
await symlink(f.root + "-retained", f.root);
|
|
} else await writeFile(join(f.root, kind === "manifest" ? "manifest.json" : "runner/runner-state.json"), "{");
|
|
}
|
|
it.each(["source", "output"])("distinguishes %s ENOENT during archive capture after source preparation", async kind => {
|
|
const [bad, good] = await releasedFixtures(), capture = backupArchive.captureNativeHarnessArchive;
|
|
const warning = vi.spyOn(console, "warn").mockImplementation(() => {});
|
|
const archive = vi.spyOn(backupArchive, "captureNativeHarnessArchive").mockImplementation(async (root, target, signal, limit) => {
|
|
if (root === bad.root) {
|
|
if (kind === "source") await rename(root, root + "-retained");
|
|
else target = join(target, "missing-parent", "archive");
|
|
}
|
|
return capture(root, target, signal, limit);
|
|
});
|
|
try {
|
|
const preparation = prestageNativeHarnessBackups(bad.db, { storage: bad.storage, signal: AbortSignal.timeout(15_000) });
|
|
if (kind === "source") {
|
|
expect(await preparation).toEqual({ inspected: 2, published: 1, current: 0, deferred: { source_changed: 1 } });
|
|
expect(JSON.parse(warning.mock.calls[0][0])).toMatchObject({ scopeDigest: bad.scopeDigest,
|
|
reason: "native_harness_migration_source_unavailable" });
|
|
expect((await good.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, good.scopeDigest)))[0].status).toBe("current");
|
|
} else {
|
|
await expect(preparation).rejects.toMatchObject({ code: "ENOENT" });
|
|
expect(warning).not.toHaveBeenCalled();
|
|
expect(await good.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, good.scopeDigest))).toHaveLength(0);
|
|
}
|
|
expect(await bad.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, bad.scopeDigest))).toHaveLength(0);
|
|
} finally { archive.mockRestore(); warning.mockRestore(); }
|
|
});
|
|
it.each(["manifest", "runner", "unsafe"])("defers a damaged %s source and still copies the next eligible backup", async kind => {
|
|
const [bad, good] = await releasedFixtures(), warning = vi.spyOn(console, "warn").mockImplementation(() => {});
|
|
try {
|
|
await damageSource(bad, kind);
|
|
expect(await prestageNativeHarnessBackups(bad.db, { storage: bad.storage, signal: AbortSignal.timeout(15_000) }))
|
|
.toEqual({ inspected: 2, published: 1, current: 0, deferred: { source_unavailable: 1 } });
|
|
expect(warning).toHaveBeenCalledWith(expect.stringContaining(bad.scopeDigest));
|
|
expect(JSON.parse(warning.mock.calls[0][0])).toMatchObject({ scopeDigest: bad.scopeDigest,
|
|
reason: kind === "unsafe" ? "native_harness_migration_unsafe_source" : "native_harness_migration_source_invalid_json" });
|
|
expect(await bad.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, bad.scopeDigest))).toHaveLength(0);
|
|
expect((await good.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, good.scopeDigest)))[0].status).toBe("current");
|
|
await expect(migrateNativeHarnessBackups(bad.db, { storage: bad.storage, writersStopped: true, signal: AbortSignal.timeout(15_000) }))
|
|
.rejects.toThrow(kind === "unsafe" ? "native_harness_migration_unsafe_source" : /JSON|json/);
|
|
} finally { warning.mockRestore(); }
|
|
});
|
|
it.each(["manifest", "runner", "unsafe"])("defers %s damage at the publication fence and copies the next backup", async kind => {
|
|
const [bad, good] = await releasedFixtures(), put = bad.storage.putFile.bind(bad.storage);
|
|
const warning = vi.spyOn(console, "warn").mockImplementation(() => {});
|
|
const upload = vi.spyOn(bad.storage, "putFile").mockImplementation(async input => {
|
|
const result = await put(input);
|
|
if (input.objectKey?.includes(bad.scopeDigest)) await damageSource(bad, kind);
|
|
return result;
|
|
});
|
|
try {
|
|
expect(await prestageNativeHarnessBackups(bad.db, { storage: bad.storage, signal: AbortSignal.timeout(15_000) }))
|
|
.toEqual({ inspected: 2, published: 1, current: 0, deferred: { source_changed: 1 } });
|
|
expect(JSON.parse(warning.mock.calls[0][0])).toMatchObject({ scopeDigest: bad.scopeDigest,
|
|
reason: kind === "unsafe" ? "native_harness_migration_unsafe_source" : "native_harness_migration_source_invalid_json" });
|
|
expect((await bad.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, bad.scopeDigest))).some(row => row.status === "current")).toBe(false);
|
|
expect((await good.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, good.scopeDigest)))[0].status).toBe("current");
|
|
} finally { upload.mockRestore(); warning.mockRestore(); }
|
|
});
|
|
it("keeps a database SyntaxError fatal instead of deferring it as source damage", async () => {
|
|
const fixtures = await releasedFixtures(), f = fixtures[0], failure = new SyntaxError("database result decoding failed");
|
|
const transaction = vi.spyOn(f.db, "transaction").mockRejectedValueOnce(failure);
|
|
try {
|
|
await expect(prestageNativeHarnessBackups(f.db, { storage: f.storage, signal: AbortSignal.timeout(15_000) })).rejects.toBe(failure);
|
|
for (const source of fixtures) expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, source.scopeDigest))).toHaveLength(0);
|
|
} finally { transaction.mockRestore(); }
|
|
});
|
|
it("keeps cancellation fatal when a source also changes during upload", async () => {
|
|
const [f] = await releasedFixtures(), cancellation = new AbortController(), failure = new Error("operator stopped");
|
|
const put = f.storage.putFile.bind(f.storage);
|
|
const upload = vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
const result = await put(input); await damageSource(f, "manifest"); cancellation.abort(failure); return result;
|
|
});
|
|
try {
|
|
await expect(prestageNativeHarnessBackups(f.db, { storage: f.storage, signal: cancellation.signal })).rejects.toBe(failure);
|
|
expect((await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).some(row => row.status === "current")).toBe(false);
|
|
} finally { upload.mockRestore(); }
|
|
});
|
|
it.each(["source_advanced", "queued_turn", "lease_revoked", "owner_changed", "run_failed"])("keeps the published source when %s happens during pre-copy upload", async change => {
|
|
const f = await inactiveFixture(); await persistNativeHarnessBackup(f);
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const put = f.storage.putFile.bind(f.storage);
|
|
vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
const result = await put(input);
|
|
if (change === "source_advanced") await writeFile(join(f.root, "manifest.json"), (await readFile(join(f.root, "manifest.json"), "utf8")) + "\n");
|
|
if (change === "queued_turn") await f.db.insert(heartbeatRuns).values({ id: randomUUID(), companyId: f.companyId,
|
|
agentId: environment.current.agentId, issueId: f.issueId, status: "queued" });
|
|
if (change === "lease_revoked") await f.db.update(environmentLeases).set({ metadata: {} })
|
|
.where(eq(environmentLeases.providerLeaseId, f.providerLeaseId));
|
|
if (change === "owner_changed") await f.db.update(nativeRunFinalizations).set({ controllerGeneration: 2 })
|
|
.where(eq(nativeRunFinalizations.runId, f.runId));
|
|
if (change === "run_failed") await f.db.update(heartbeatRuns).set({ status: "failed" }).where(eq(heartbeatRuns.id, f.runId));
|
|
return result;
|
|
});
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow(change === "owner_changed" ? "native_runner_checkpoint_owner_lost"
|
|
: change === "queued_turn" ? "native_harness_storage_prestage_source_active" : "native_harness_storage_prestage_source_changed");
|
|
const rows = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
expect(rows.filter(row => row.status === "current").map(row => row.id)).toEqual([before.id]);
|
|
expect(rows.filter(row => row.status === "pending")).toHaveLength(1);
|
|
expect(await readFile(join(f.root, "manifest.json"), "utf8")).toContain("paperclip.native-harness-backup.v1");
|
|
});
|
|
it("publishes with the captured provider owner while an accepted result is still saving its workspace", async () => {
|
|
const f = await fixture();
|
|
const result = await releaseWhileSavingWorkspace(f);
|
|
await expect(nativeHarnessBackupOwner(f.db, { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, bootId: f.owner.bootId })).rejects.toThrow("native_harness_storage_owner_missing");
|
|
await persistNativeHarnessBackup(f);
|
|
await removeNativeHarnessBackup(f.home);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID(), "restored-pending-workspace");
|
|
ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(JSON.parse(await readFile(join(replacement, "current", "codex", "sessions", "history.json"), "utf8"))).toEqual(f.identity);
|
|
const [coordinator] = await f.db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
expect(coordinator).toMatchObject({ phase: "workspace_finalizing", resultId: result.id, leaseOwner: null,
|
|
leaseExpiresAt: null, controllerBootId: f.owner.bootId, controllerGeneration: f.owner.generation });
|
|
});
|
|
it.each(["boot", "generation", "result", "successor"])("rejects a captured provider fence when its %s changes during upload", async change => {
|
|
const f = await fixture();
|
|
await persistNativeHarnessBackup(f);
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
await releaseWhileSavingWorkspace(f);
|
|
const put = f.storage.putFile.bind(f.storage);
|
|
vi.spyOn(f.storage, "putFile").mockImplementation(async input => {
|
|
const object = await put(input);
|
|
if (change === "successor") {
|
|
const nextRunId = randomUUID();
|
|
await f.db.insert(heartbeatRuns).values({ id: nextRunId, companyId: f.companyId,
|
|
agentId: environment.current.agentId, issueId: f.issueId, status: "running" });
|
|
await f.db.update(issues).set({ executionRunId: nextRunId }).where(eq(issues.id, f.issueId));
|
|
} else {
|
|
await f.db.update(nativeRunFinalizations).set(change === "boot" ? { controllerBootId: randomUUID() }
|
|
: change === "generation" ? { controllerGeneration: 2 } : { resultId: null })
|
|
.where(eq(nativeRunFinalizations.runId, f.runId));
|
|
}
|
|
return object;
|
|
});
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow(change === "successor"
|
|
? "native_harness_storage_owner_superseded" : "native_runner_checkpoint_owner_lost");
|
|
const selected = await f.db.select().from(nativeHarnessBackups).where(and(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest),
|
|
eq(nativeHarnessBackups.status, "current")));
|
|
expect(selected.map(row => row.id)).toEqual([before.id]);
|
|
});
|
|
it.each(["idle", "preparing", "native"])("preserves a released committed warm session with a %s successor", async state => {
|
|
const f = await fixture();
|
|
const result = await commitWarmRun(f);
|
|
let nextRunId: string | null = null;
|
|
if (state !== "idle") {
|
|
nextRunId = randomUUID();
|
|
await f.db.insert(heartbeatRuns).values({ id: nextRunId, companyId: f.companyId, agentId: environment.current.agentId,
|
|
issueId: f.issueId, status: "running", runtimeMode: state === "native" ? "native" : "legacy", executionStage: "preparing",
|
|
...(state === "native" ? { nativeIssueId: f.issueId } : {}),
|
|
controllerBootId: legacyControllerBootId, controllerLeaseExpiresAt: new Date(Date.now() + 120_000) });
|
|
if (state === "native") await f.db.insert(nativeRunFinalizations).values({ runId: nextRunId,
|
|
companyId: f.companyId, issueId: f.issueId, phase: "observed", controllerBootId: environment.current.successor.bootId,
|
|
controllerGeneration: 1, leaseOwner: "preparing-warm-successor", leaseExpiresAt: new Date(Date.now() + 120_000) });
|
|
}
|
|
await f.db.update(issues).set({ executionRunId: nextRunId }).where(eq(issues.id, f.issueId));
|
|
const owner = await nativeHarnessBackupOwner(f.db, { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, bootId: environment.current.successor.bootId });
|
|
await persistNativeHarnessBackup({ ...f, owner });
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID(), "restored-warm-session"); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(JSON.parse(await readFile(join(replacement, "current", "codex", "sessions", "history.json"), "utf8"))).toEqual(f.identity);
|
|
const [coordinator] = await f.db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
expect(coordinator).toMatchObject({ phase: "committed", resultId: result.id, leaseOwner: null, leaseExpiresAt: null });
|
|
const [task] = await f.db.select().from(issues).where(eq(issues.id, f.issueId));
|
|
expect(task.executionRunId).toBe(nextRunId);
|
|
});
|
|
it.each(["unaccepted", "uncommitted", "wrong_boot"])("rejects a released warm owner with %s completion evidence", async state => {
|
|
const f = await fixture(); await commitWarmRun(f);
|
|
if (state === "unaccepted") await f.db.update(nativeRunFinalizations).set({ resultId: null }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
if (state === "uncommitted") await f.db.update(nativeRunFinalizations).set({ phase: "observed" }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
await expect(nativeHarnessBackupOwner(f.db, { companyId: f.companyId, issueId: f.issueId, runId: f.runId,
|
|
bootId: state === "wrong_boot" ? randomUUID() : environment.current.successor.bootId })).rejects.toThrow("native_harness_storage_owner_missing");
|
|
});
|
|
it("rejects an active predecessor after the task moves to a successor", async () => {
|
|
const f = await fixture();
|
|
const nextRunId = randomUUID();
|
|
await f.db.insert(heartbeatRuns).values({ id: nextRunId, companyId: f.companyId, agentId: environment.current.agentId,
|
|
nativeIssueId: f.issueId, issueId: f.issueId, status: "running", runtimeMode: "native" });
|
|
await f.db.update(issues).set({ executionRunId: nextRunId }).where(eq(issues.id, f.issueId));
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("native_harness_storage_owner_superseded");
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toEqual([]);
|
|
});
|
|
it.each(["foreign_preparation", "expired_preparation", "dispatching_successor", "foreign_native"])("keeps the selected backup when a released owner has a %s", async state => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f); await commitWarmRun(f);
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const nextRunId = randomUUID();
|
|
await f.db.insert(heartbeatRuns).values({ id: nextRunId, companyId: f.companyId, agentId: environment.current.agentId,
|
|
issueId: f.issueId, status: "running", runtimeMode: state === "foreign_native" ? "native" : "legacy",
|
|
...(state === "foreign_native" ? { nativeIssueId: f.issueId } : {}),
|
|
executionStage: state === "dispatching_successor" ? "dispatching" : "preparing",
|
|
controllerBootId: state === "foreign_preparation" ? randomUUID() : legacyControllerBootId,
|
|
controllerLeaseExpiresAt: state === "expired_preparation" ? new Date(0) : new Date(Date.now() + 120_000) });
|
|
if (state === "foreign_native") await f.db.insert(nativeRunFinalizations).values({ runId: nextRunId,
|
|
companyId: f.companyId, issueId: f.issueId, phase: "observed", controllerBootId: randomUUID(),
|
|
controllerGeneration: 1, leaseOwner: "foreign-warm-successor", leaseExpiresAt: new Date(Date.now() + 120_000) });
|
|
await f.db.update(issues).set({ executionRunId: nextRunId }).where(eq(issues.id, f.issueId));
|
|
const owner = await nativeHarnessBackupOwner(f.db, { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, bootId: environment.current.successor.bootId });
|
|
await expect(persistNativeHarnessBackup({ ...f, owner })).rejects.toThrow("native_runner_checkpoint_owner_lost");
|
|
expect((await f.db.select().from(nativeHarnessBackups).where(and(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest),
|
|
eq(nativeHarnessBackups.status, "current")))).map(row => row.id)).toEqual([before.id]);
|
|
});
|
|
it.each(["generation", "result", "phase"])("rechecks a released owner's %s after object upload", async state => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f); await commitWarmRun(f);
|
|
await f.db.update(issues).set({ executionRunId: null }).where(eq(issues.id, f.issueId));
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const owner = await nativeHarnessBackupOwner(f.db, { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, bootId: environment.current.successor.bootId });
|
|
const put = f.storage.putFile.bind(f.storage);
|
|
f.storage.putFile = async input => {
|
|
const result = await put(input);
|
|
await f.db.update(nativeRunFinalizations).set(state === "generation" ? { controllerGeneration: 2 }
|
|
: state === "result" ? { resultId: null } : { phase: "observed" }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
return result;
|
|
};
|
|
await expect(persistNativeHarnessBackup({ ...f, owner })).rejects.toThrow(state === "generation"
|
|
? "native_runner_checkpoint_owner_lost" : "native_harness_storage_owner_missing");
|
|
expect((await f.db.select().from(nativeHarnessBackups).where(and(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest),
|
|
eq(nativeHarnessBackups.status, "current")))).map(row => row.id)).toEqual([before.id]);
|
|
});
|
|
it("cannot replace a newer conversation backup with a released predecessor's late upload", async () => {
|
|
const f = await fixture(); await commitWarmRun(f);
|
|
await f.db.update(issues).set({ executionRunId: null }).where(eq(issues.id, f.issueId));
|
|
const owner = await nativeHarnessBackupOwner(f.db, { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, bootId: environment.current.successor.bootId });
|
|
const next = await fixture();
|
|
const put = f.storage.putFile.bind(f.storage);
|
|
f.storage.putFile = async input => {
|
|
const result = await put(input);
|
|
await persistNativeHarnessBackup({ ...next, scopeDigest: f.scopeDigest });
|
|
return result;
|
|
};
|
|
await expect(persistNativeHarnessBackup({ ...f, owner })).rejects.toThrow("native_harness_storage_owner_superseded");
|
|
const [current] = await f.db.select().from(nativeHarnessBackups).where(and(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest),
|
|
eq(nativeHarnessBackups.status, "current")));
|
|
expect(current.runId).toBe(next.runId);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID(), "latest-conversation"); ownedHomes.push(replacement);
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).toBe(true);
|
|
expect(JSON.parse(await readFile(join(replacement, "current", "codex", "sessions", "history.json"), "utf8"))).toEqual(next.identity);
|
|
});
|
|
it("fences an old controller whose upload finishes after ownership changes", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const put = f.storage.putFile.bind(f.storage);
|
|
f.storage.putFile = async input => {
|
|
const result = await put(input);
|
|
await f.db.update(nativeRunFinalizations).set({ controllerBootId: randomUUID(), controllerGeneration: 2 })
|
|
.where(eq(nativeRunFinalizations.runId, f.runId));
|
|
return result;
|
|
};
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("native_runner_checkpoint_owner_lost");
|
|
const rows = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
expect(rows.filter(row => row.status === "current").map(row => row.id)).toEqual([before.id]);
|
|
});
|
|
it("refuses publication after a pending upload was retired without demoting the current backup", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const [current] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const put = f.storage.putFile.bind(f.storage);
|
|
f.storage.putFile = async input => {
|
|
const result = await put(input);
|
|
await f.db.update(nativeHarnessBackups).set({ status: "retired" }).where(and(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest), eq(nativeHarnessBackups.status, "pending")));
|
|
return result;
|
|
};
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("native_harness_storage_publication_retired");
|
|
expect((await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.id, current.id)))[0].status).toBe("current");
|
|
const [retired] = await f.db.select().from(nativeHarnessBackups).where(and(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest), eq(nativeHarnessBackups.status, "retired")));
|
|
expect(await deleteRetiredNativeHarnessBackup(f.db, f.storage, retired.id)).toBe(true);
|
|
});
|
|
it("rejects corrupt objects and cross-company lookups without publishing a local cache", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const replacement = join(environment.current.runtimeRoot, randomUUID(), "failover-backups");
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, companyId: randomUUID(), backupRoot: replacement })).toBe(false);
|
|
const [row] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
await f.storage.deleteObject(f.companyId, row.reference.objectKey as string);
|
|
await f.storage.putFile({ companyId: f.companyId, namespace: "native-harness-backups", objectKey: row.reference.objectKey as string,
|
|
originalFilename: null, contentType: "application/octet-stream", body: Buffer.from("corrupted") });
|
|
await expect(restoreDurableNativeHarnessBackup({ ...f, backupRoot: replacement })).rejects.toThrow("hash_mismatch");
|
|
await expect(stat(join(replacement, "current"))).rejects.toMatchObject({ code: "ENOENT" });
|
|
});
|
|
async function settle(f: { db: Awaited<ReturnType<typeof fixture>>["db"]; runId: string; issueId: string }, closedAt = new Date()) {
|
|
await f.db.update(heartbeatRuns).set({ status: "failed", finishedAt: closedAt }).where(eq(heartbeatRuns.id, f.runId));
|
|
await f.db.update(nativeRunFinalizations).set({ phase: "terminal_failure", leaseOwner: null, leaseExpiresAt: null })
|
|
.where(eq(nativeRunFinalizations.runId, f.runId));
|
|
await f.db.update(issues).set({ status: "done", executionRunId: null, completedAt: closedAt }).where(eq(issues.id, f.issueId));
|
|
}
|
|
it("migrates a lease-verified local backup once and keeps its selected session intact", async () => {
|
|
const f = await fixture("codex-home");
|
|
const base = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(base); await mkdir(base);
|
|
await rename(f.home, join(base, f.scopeDigest));
|
|
const root = join(base, f.scopeDigest, "failover-backups", "current");
|
|
vi.stubEnv("PAPERCLIP_RUNNER_STATE_DIR", base);
|
|
const runnerInstanceId = randomUUID(), providerLeaseId = randomUUID();
|
|
const execution = { provider: { kind: "codex" }, binding: { companyId: f.companyId, issueId: f.issueId,
|
|
runId: f.runId, agentId: environment.current.agentId, executionWorkspaceId: randomUUID() },
|
|
workspace: { cwd: "/workspace", repoUrl: null, repoRef: null, branchName: null },
|
|
session: { normalizedSessionId: f.identity.normalizedSessionId, driverKind: "codex_app_server",
|
|
protocolVersion: 1, lifecyclePolicy: { mode: "per_turn", idleTimeoutMs: null } } } as never;
|
|
await chmod(join(root, "codex-home", "sessions"), 0o700);
|
|
await symlink("history.json", join(root, "codex-home", "sessions", "history-link"));
|
|
await chmod(join(root, "codex-home", "sessions"), 0o500);
|
|
const manifest = buildNativeHarnessBackupManifest({ backupRoot: root, execution, runnerInstanceId,
|
|
providerSessionIdentity: { providerSessionId: f.identity.providerSessionId }, sourceProviderLeaseId: providerLeaseId });
|
|
await writeFile(join(root, "manifest.json"), JSON.stringify(manifest));
|
|
const stamp = createNativeHarnessBackupStamp({ manifestPath: join(root, "manifest.json"), sessionScopeId: f.sessionScopeId,
|
|
authorizedProviderLeaseId: providerLeaseId, runnerInstanceId, normalizedSessionId: f.identity.normalizedSessionId, completedAt: manifest.completedAt });
|
|
expect(verifyNativeHarnessBackupStamp(stamp, providerLeaseId)).toBe(true);
|
|
expect(await verifyNativeHarnessBackupStampAsync(stamp, providerLeaseId, AbortSignal.timeout(10_000))).toBe(true);
|
|
await expect(verifyNativeHarnessBackupStampAsync(stamp, providerLeaseId, AbortSignal.abort(new Error("cancelled")))).rejects.toThrow("cancelled");
|
|
await f.db.update(heartbeatRuns).set({ nativeSessionId: f.identity.normalizedSessionId, runnerInstanceId }).where(eq(heartbeatRuns.id, f.runId));
|
|
await f.db.insert(environmentLeases).values({ companyId: f.companyId, issueId: f.issueId, heartbeatRunId: f.runId,
|
|
providerLeaseId, status: "expired", metadata: { nativeHarnessBackup: stamp } });
|
|
const [selected] = await f.db.insert(agentTaskSessions).values({ companyId: f.companyId, agentId: environment.current.agentId,
|
|
adapterType: "paperclip_runner", taskKey: f.issueId, lastRunId: f.runId,
|
|
sessionParamsJson: { sessionId: f.identity.normalizedSessionId } }).returning();
|
|
expect(await hasLocalNativeHarnessObligations(f.db)).toBe(true);
|
|
await expect(migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(10_000) }))
|
|
.rejects.toThrow("native_harness_migration_requires_offline_writers");
|
|
const ownedRuns = await f.db.select({ id: heartbeatRuns.id, issueId: heartbeatRuns.nativeIssueId }).from(heartbeatRuns)
|
|
.where(eq(heartbeatRuns.companyId, f.companyId));
|
|
for (const run of ownedRuns) await settle({ ...f, runId: run.id, issueId: run.issueId! });
|
|
await f.db.update(nativeRunFinalizations).set({ recoveryState: "blocked" }).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
const [preservedOwner] = await f.db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId));
|
|
const [queued] = await f.db.insert(heartbeatRuns).values({ companyId: f.companyId, agentId: environment.current.agentId,
|
|
status: "queued", invocationSource: "on_demand", issueId: f.issueId }).returning();
|
|
await f.db.update(issues).set({ executionRunId: queued.id }).where(eq(issues.id, f.issueId));
|
|
expect(await migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(10_000) })).toBe(1);
|
|
expect(await migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(10_000) })).toBe(1);
|
|
expect(await hasLocalNativeHarnessObligations(f.db)).toBe(false);
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toHaveLength(1);
|
|
expect((await f.db.select().from(agentTaskSessions).where(eq(agentTaskSessions.id, selected.id)))[0]).toEqual(selected);
|
|
expect((await f.db.select().from(nativeRunFinalizations).where(eq(nativeRunFinalizations.runId, f.runId)))[0]).toEqual(preservedOwner);
|
|
expect((await f.db.select().from(heartbeatRuns).where(eq(heartbeatRuns.id, queued.id)))[0]).toEqual(queued);
|
|
expect((await f.db.select().from(issues).where(eq(issues.id, f.issueId)))[0].executionRunId).toBe(queued.id);
|
|
expect(JSON.parse(await readFile(join(root, "manifest.json"), "utf8"))).toEqual(manifest);
|
|
const restored = join(base, "replacement");
|
|
expect(await restoreDurableNativeHarnessBackup({ ...f, backupRoot: restored })).toBe(true);
|
|
expect(await readlink(join(restored, "current", "codex-home", "sessions", "history-link"))).toBe("history.json");
|
|
const changedManifest = JSON.stringify({ ...manifest, completedAt: new Date(Date.now() + 1_000).toISOString() });
|
|
await writeFile(join(root, "manifest.json"), changedManifest);
|
|
await expect(migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(10_000) }))
|
|
.rejects.toThrow("native_harness_migration_durable_source_conflict");
|
|
expect(await readFile(join(root, "manifest.json"), "utf8")).toBe(changedManifest);
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toHaveLength(1);
|
|
});
|
|
it("preserves orphaned local backup files without blocking other enrollment", async () => {
|
|
const f = await fixture();
|
|
const base = join(environment.current.runtimeRoot, randomUUID()); ownedHomes.push(base); await mkdir(base);
|
|
await rename(f.home, join(base, f.scopeDigest));
|
|
vi.stubEnv("PAPERCLIP_RUNNER_STATE_DIR", base);
|
|
const root = join(base, f.scopeDigest, "failover-backups", "current");
|
|
const orphan = JSON.stringify({ runId: randomUUID(), lifecycle: "suspended" });
|
|
await writeFile(join(root, "runner", "runner-state.json"), orphan);
|
|
const ownedRuns = await f.db.select({ id: heartbeatRuns.id, issueId: heartbeatRuns.nativeIssueId }).from(heartbeatRuns)
|
|
.where(eq(heartbeatRuns.companyId, f.companyId));
|
|
for (const run of ownedRuns) await settle({ ...f, runId: run.id, issueId: run.issueId! });
|
|
expect(await migrateNativeHarnessBackups(f.db, { storage: f.storage, writersStopped: true, signal: AbortSignal.timeout(10_000) })).toBe(0);
|
|
expect(await readFile(join(root, "runner", "runner-state.json"), "utf8")).toBe(orphan);
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toHaveLength(0);
|
|
});
|
|
it("retains live work and retries an uncertain object deletion through the existing startup retention pass", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const old = new Date(Date.now() - 10 * 86_400_000);
|
|
await f.db.update(nativeHarnessBackups).set({ createdAt: old }).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const retention = nativeHarnessBackupExpirationService(f.db, { storage: f.storage });
|
|
expect((await retention.tickCheckpoints()).expired).toBe(0);
|
|
await settle(f, old);
|
|
const drop = vi.spyOn(f.storage, "deleteObject").mockRejectedValueOnce(new Error("delete reply lost"));
|
|
expect((await retention.tickCheckpoints()).failed).toBe(1);
|
|
expect((await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest)))[0].status).toBe("retired");
|
|
expect((await retention.tickCheckpoints()).expired).toBe(1);
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest))).toHaveLength(0);
|
|
expect(drop).toHaveBeenCalledTimes(2);
|
|
});
|
|
it("expires obsolete objects without clearing an open conversation or its exact lease backup", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const [obsolete] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
await writeFile(join(f.root, "manifest.json"), JSON.stringify({ revision: 2 }));
|
|
await persistNativeHarnessBackup(f);
|
|
const old = new Date(Date.now() - 10 * 86_400_000);
|
|
await f.db.update(nativeHarnessBackups).set({ createdAt: old, updatedAt: old }).where(eq(nativeHarnessBackups.id, obsolete.id));
|
|
const [selected] = await f.db.insert(agentTaskSessions).values({ companyId: f.companyId, agentId: environment.current.agentId,
|
|
adapterType: "paperclip_runner", taskKey: f.issueId, lastRunId: f.runId }).returning();
|
|
const [lease] = await f.db.insert(environmentLeases).values({ companyId: f.companyId, issueId: f.issueId,
|
|
heartbeatRunId: f.runId, status: "pending_cleanup", metadata: { nativeHarnessBackup: {
|
|
sessionScopeSha256: f.scopeDigest, manifestSha256: obsolete.manifestSha256 } } }).returning();
|
|
const retention = nativeHarnessBackupExpirationService(f.db, { storage: f.storage });
|
|
expect((await retention.tickCheckpoints()).expired).toBe(0);
|
|
await f.db.update(environmentLeases).set({ status: "expired" }).where(eq(environmentLeases.id, lease.id));
|
|
expect((await retention.tickCheckpoints({ dryRun: true })).expired).toBe(0);
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.id, obsolete.id))).toHaveLength(1);
|
|
expect((await retention.tickCheckpoints()).expired).toBe(1);
|
|
const rows = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
expect(rows).toHaveLength(1); expect(rows[0].status).toBe("current");
|
|
expect((await f.db.select().from(agentTaskSessions).where(eq(agentTaskSessions.id, selected.id)))[0]).toEqual(selected);
|
|
expect((await f.db.select().from(heartbeatRuns).where(eq(heartbeatRuns.id, f.runId)))[0].status).toBe("running");
|
|
});
|
|
it("retains an in-flight pending upload and removes it once its owner settles", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const put = vi.spyOn(f.storage, "putFile").mockRejectedValueOnce(new Error("upload unavailable"));
|
|
await expect(persistNativeHarnessBackup(f)).rejects.toThrow("upload unavailable"); put.mockRestore();
|
|
const rows = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const pending = rows.find(row => row.status === "pending")!;
|
|
const old = new Date(Date.now() - 10 * 86_400_000);
|
|
await f.db.update(nativeHarnessBackups).set({ createdAt: old, updatedAt: old }).where(eq(nativeHarnessBackups.id, pending.id));
|
|
const retention = nativeHarnessBackupExpirationService(f.db, { storage: f.storage });
|
|
expect((await retention.tickCheckpoints()).expired).toBe(0);
|
|
await settle(f);
|
|
expect((await retention.tickCheckpoints()).expired).toBe(1);
|
|
const remaining = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
expect(remaining).toHaveLength(1); expect(remaining[0].status).toBe("current");
|
|
});
|
|
it("keeps deleted-source objects unchanged during a retention dry run", async () => {
|
|
const f = await fixture(); await persistNativeHarnessBackup(f);
|
|
const old = new Date(Date.now() - 10 * 86_400_000);
|
|
await f.db.update(nativeHarnessBackups).set({ createdAt: old, runId: randomUUID() })
|
|
.where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const [before] = await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.scopeDigest, f.scopeDigest));
|
|
const retention = nativeHarnessBackupExpirationService(f.db, { storage: f.storage });
|
|
expect((await retention.tickCheckpoints({ dryRun: true })).expired).toBe(0);
|
|
expect((await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.id, before.id)))[0]).toEqual(before);
|
|
expect(await verifyDurableNativeHarnessBackup(f)).toBe(true);
|
|
expect((await retention.tickCheckpoints()).expired).toBe(1);
|
|
expect(await f.db.select().from(nativeHarnessBackups).where(eq(nativeHarnessBackups.id, before.id))).toHaveLength(0);
|
|
});
|
|
});
|