From fbbf94a2c46e53ba25ed2daa801697723cbfb45c Mon Sep 17 00:00:00 2001
From: N0zoM1z0 <161784452+N0zoM1z0@users.noreply.github.com>
Date: Wed, 11 Nov 2026 12:43:53 +0000
Subject: [PATCH] test(kuna): cover cancellation during native ownership
 preparation

Co-authored-by: gensokyo-dev <342580529+gensokyo-dev@users.noreply.github.com>
---
 .../kuna/kunaWorkspaceRollback.test.ts        | 24 ++++++++++++++++---
 1 file changed, 21 insertions(+), 3 deletions(-)

diff --git a/tests/boundary/providers/kuna/kunaWorkspaceRollback.test.ts b/tests/boundary/providers/kuna/kunaWorkspaceRollback.test.ts
index 1edef11116..e892aa2ce0 100644
--- a/tests/boundary/providers/kuna/kunaWorkspaceRollback.test.ts
+++ b/tests/boundary/providers/kuna/kunaWorkspaceRollback.test.ts
@@ -2,6 +2,8 @@ import { unlink, writeFile } from "node:fs/promises";
 import { expect, it as test, vi } from "vitest";
 import { withVerifierCleanup } from "../../../../scripts/lib/verifier-cleanup.mjs";
 import { KunaTargetCommands } from "../../../../src/kuna/KunaTargetCommands.js";
+import { systemProcessOwnershipHost } from "../../../../src/process/ProcessOwnershipObservation.js";
+import { ExecFileOutputCancelledError } from "../../../../src/process/ExecFileOutput.js";
 import { PrivateRuntimeRoot } from "../../../../src/process/PrivateRuntimeRoot.js";
 import { ProviderCleanupError } from "../../../../src/domain/providerCleanupError.js";
 import { projectAnalysisError } from "../../../../src/domain/analysisErrorProjection.js";
@@ -263,9 +265,14 @@ it("keeps caller attribution and the selected path through workspace rollback re
   }
 });
 
-it.each(["clean", "refuse"] as const)(
-  "preserves workspace cancellation and retries its rollback: %s",
-  async (mode) => {
+it.each([
+  { stage: "copy", mode: "clean" },
+  { stage: "copy", mode: "refuse" },
+  { stage: "inspection", mode: "clean" },
+  { stage: "inspection", mode: "refuse" },
+] as const)(
+  "preserves workspace $stage cancellation and retries its $mode rollback",
+  async ({ stage, mode }) => {
     const fixture = await session("rea-kuna-workspace-cancel-");
     const lifetime = new AbortController();
     const reason = { caller: "cancel target copy" };
@@ -285,6 +292,16 @@ it.each(["clean", "refuse"] as const)(
         lifetime.abort(reason);
         return root;
       });
+    const prepare = systemProcessOwnershipHost.prepare;
+    const inspection = vi
+      .spyOn(systemProcessOwnershipHost, "prepare")
+      .mockImplementation(async (signal) => {
+        if (stage === "inspection" && signal === lifetime.signal) {
+          lifetime.abort(reason);
+          throw new ExecFileOutputCancelledError(reason);
+        }
+        await prepare?.(signal);
+      });
     try {
       const result = await commands.prepare();
       if (result.ok) throw new Error("Expected cancelled workspace admission");
@@ -302,6 +319,7 @@ it.each(["clean", "refuse"] as const)(
         expect(result.error.cleanupPrimary).toBe(primary);
       }
     } finally {
+      inspection.mockRestore();
       roots.mockRestore();
       rootClose?.mockRestore();
       expect((await commands.close()).ok).toBe(true);