gsd-pi 2.44.0-dev.848dd4c → 2.44.0

package/packages/pi-coding-agent/dist/core/session-manager.test.js

@@ -1,5 +1,5 @@
 import assert from "node:assert/strict";
-import { describe, it, afterEach } from "node:test";
+import { describe, it } from "node:test";
 import { mkdtempSync, rmSync } from "node:fs";
 import { join } from "node:path";
 import { tmpdir } from "node:os";
@@ -19,39 +19,43 @@ function makeAssistantMessage(input, output, cacheRead = 0, cacheWrite = 0, cost
     };
 }
 describe("SessionManager usage totals", () => {
-    let dir;
-    afterEach(() => {
-        if (dir) {
+    it("tracks assistant usage incrementally without rescanning entries", () => {
+        const dir = mkdtempSync(join(tmpdir(), "gsd-session-manager-test-"));
+        try {
+            const manager = SessionManager.create(dir, dir);
+            manager.appendMessage({ role: "user", content: [{ type: "text", text: "hello" }] });
+            manager.appendMessage(makeAssistantMessage(10, 5, 3, 2, 0.25));
+            manager.appendMessage(makeAssistantMessage(7, 4, 1, 0, 0.1));
+            assert.deepEqual(manager.getUsageTotals(), {
+                input: 17,
+                output: 9,
+                cacheRead: 4,
+                cacheWrite: 2,
+                cost: 0.35,
+            });
+        }
+        finally {
             rmSync(dir, { recursive: true, force: true });
         }
     });
-    it("tracks assistant usage incrementally without rescanning entries", () => {
-        dir = mkdtempSync(join(tmpdir(), "gsd-session-manager-test-"));
-        const manager = SessionManager.create(dir, dir);
-        manager.appendMessage({ role: "user", content: [{ type: "text", text: "hello" }] });
-        manager.appendMessage(makeAssistantMessage(10, 5, 3, 2, 0.25));
-        manager.appendMessage(makeAssistantMessage(7, 4, 1, 0, 0.1));
-        assert.deepEqual(manager.getUsageTotals(), {
-            input: 17,
-            output: 9,
-            cacheRead: 4,
-            cacheWrite: 2,
-            cost: 0.35,
-        });
-    });
     it("resets totals when starting a new session", () => {
-        dir = mkdtempSync(join(tmpdir(), "gsd-session-manager-test-"));
-        const manager = SessionManager.create(dir, dir);
-        manager.appendMessage(makeAssistantMessage(5, 5, 0, 0, 0.05));
-        assert.equal(manager.getUsageTotals().input, 5);
-        manager.newSession();
-        assert.deepEqual(manager.getUsageTotals(), {
-            input: 0,
-            output: 0,
-            cacheRead: 0,
-            cacheWrite: 0,
-            cost: 0,
-        });
+        const dir = mkdtempSync(join(tmpdir(), "gsd-session-manager-test-"));
+        try {
+            const manager = SessionManager.create(dir, dir);
+            manager.appendMessage(makeAssistantMessage(5, 5, 0, 0, 0.05));
+            assert.equal(manager.getUsageTotals().input, 5);
+            manager.newSession();
+            assert.deepEqual(manager.getUsageTotals(), {
+                input: 0,
+                output: 0,
+                cacheRead: 0,
+                cacheWrite: 0,
+                cost: 0,
+            });
+        }
+        finally {
+            rmSync(dir, { recursive: true, force: true });
+        }
     });
 });
 //# sourceMappingURL=session-manager.test.js.map

package/packages/pi-coding-agent/dist/core/session-manager.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"session-manager.test.js","sourceRoot":"","sources":["../../src/core/session-manager.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,SAAS,oBAAoB,CAAC,KAAa,EAAE,MAAc,EAAE,SAAS,GAAG,CAAC,EAAE,UAAU,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC;IACnG,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACvC,KAAK,EAAE;YACN,KAAK;YACL,MAAM;YACN,SAAS;YACT,UAAU;YACV,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU;YAC9C,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;SACrB;KACM,CAAC;AACV,CAAC;AAED,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,IAAI,GAAW,CAAC;IAEhB,SAAS,CAAC,GAAG,EAAE;QACd,IAAI,GAAG,EAAE,CAAC;YACT,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QAC1E,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAEhD,OAAO,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAS,CAAC,CAAC;QAC3F,OAAO,CAAC,aAAa,CAAC,oBAAoB,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;QAC/D,OAAO,CAAC,aAAa,CAAC,oBAAoB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;QAE7D,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE;YAC1C,KAAK,EAAE,EAAE;YACT,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,IAAI,EAAE,IAAI;SACV,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACpD,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;QAC/D,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;QAChD,OAAO,CAAC,aAAa,CAAC,oBAAoB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;QAEhD,OAAO,CAAC,UAAU,EAAE,CAAC;QACrB,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE;YAC1C,KAAK,EAAE,CAAC;YACR,MAAM,EAAE,CAAC;YACT,SAAS,EAAE,CAAC;YACZ,UAAU,EAAE,CAAC;YACb,IAAI,EAAE,CAAC;SACP,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { describe, it, afterEach } from \"node:test\";\nimport { mkdtempSync, rmSync } from \"node:fs\";\nimport { join } from \"node:path\";\nimport { tmpdir } from \"node:os\";\n\nimport { SessionManager } from \"./session-manager.js\";\n\nfunction makeAssistantMessage(input: number, output: number, cacheRead = 0, cacheWrite = 0, cost = 0) {\n\treturn {\n\t\trole: \"assistant\",\n\t\tcontent: [{ type: \"text\", text: \"ok\" }],\n\t\tusage: {\n\t\t\tinput,\n\t\t\toutput,\n\t\t\tcacheRead,\n\t\t\tcacheWrite,\n\t\t\ttotal: input + output + cacheRead + cacheWrite,\n\t\t\tcost: { total: cost },\n\t\t},\n\t} as any;\n}\n\ndescribe(\"SessionManager usage totals\", () => {\n\tlet dir: string;\n\n\tafterEach(() => {\n\t\tif (dir) {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t}\n\t});\n\n\tit(\"tracks assistant usage incrementally without rescanning entries\", () => {\n\t\tdir = mkdtempSync(join(tmpdir(), \"gsd-session-manager-test-\"));\n\t\tconst manager = SessionManager.create(dir, dir);\n\n\t\tmanager.appendMessage({ role: \"user\", content: [{ type: \"text\", text: \"hello\" }] } as any);\n\t\tmanager.appendMessage(makeAssistantMessage(10, 5, 3, 2, 0.25));\n\t\tmanager.appendMessage(makeAssistantMessage(7, 4, 1, 0, 0.1));\n\n\t\tassert.deepEqual(manager.getUsageTotals(), {\n\t\t\tinput: 17,\n\t\t\toutput: 9,\n\t\t\tcacheRead: 4,\n\t\t\tcacheWrite: 2,\n\t\t\tcost: 0.35,\n\t\t});\n\t});\n\n\tit(\"resets totals when starting a new session\", () => {\n\t\tdir = mkdtempSync(join(tmpdir(), \"gsd-session-manager-test-\"));\n\t\tconst manager = SessionManager.create(dir, dir);\n\t\tmanager.appendMessage(makeAssistantMessage(5, 5, 0, 0, 0.05));\n\t\tassert.equal(manager.getUsageTotals().input, 5);\n\n\t\tmanager.newSession();\n\t\tassert.deepEqual(manager.getUsageTotals(), {\n\t\t\tinput: 0,\n\t\t\toutput: 0,\n\t\t\tcacheRead: 0,\n\t\t\tcacheWrite: 0,\n\t\t\tcost: 0,\n\t\t});\n\t});\n});\n"]}
+{"version":3,"file":"session-manager.test.js","sourceRoot":"","sources":["../../src/core/session-manager.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AACzC,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAEtD,SAAS,oBAAoB,CAAC,KAAa,EAAE,MAAc,EAAE,SAAS,GAAG,CAAC,EAAE,UAAU,GAAG,CAAC,EAAE,IAAI,GAAG,CAAC;IACnG,OAAO;QACN,IAAI,EAAE,WAAW;QACjB,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QACvC,KAAK,EAAE;YACN,KAAK;YACL,MAAM;YACN,SAAS;YACT,UAAU;YACV,KAAK,EAAE,KAAK,GAAG,MAAM,GAAG,SAAS,GAAG,UAAU;YAC9C,IAAI,EAAE,EAAE,KAAK,EAAE,IAAI,EAAE;SACrB;KACM,CAAC;AACV,CAAC;AAED,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC5C,EAAE,CAAC,iEAAiE,EAAE,GAAG,EAAE;QAC1E,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;QACrE,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAEhD,OAAO,CAAC,aAAa,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,OAAO,EAAE,CAAC,EAAS,CAAC,CAAC;YAC3F,OAAO,CAAC,aAAa,CAAC,oBAAoB,CAAC,EAAE,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAC/D,OAAO,CAAC,aAAa,CAAC,oBAAoB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,GAAG,CAAC,CAAC,CAAC;YAE7D,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE;gBAC1C,KAAK,EAAE,EAAE;gBACT,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,IAAI,EAAE,IAAI;aACV,CAAC,CAAC;QACJ,CAAC;gBAAS,CAAC;YACV,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,2CAA2C,EAAE,GAAG,EAAE;QACpD,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,2BAA2B,CAAC,CAAC,CAAC;QACrE,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;YAChD,OAAO,CAAC,aAAa,CAAC,oBAAoB,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,CAAC,EAAE,IAAI,CAAC,CAAC,CAAC;YAC9D,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,cAAc,EAAE,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC;YAEhD,OAAO,CAAC,UAAU,EAAE,CAAC;YACrB,MAAM,CAAC,SAAS,CAAC,OAAO,CAAC,cAAc,EAAE,EAAE;gBAC1C,KAAK,EAAE,CAAC;gBACR,MAAM,EAAE,CAAC;gBACT,SAAS,EAAE,CAAC;gBACZ,UAAU,EAAE,CAAC;gBACb,IAAI,EAAE,CAAC;aACP,CAAC,CAAC;QACJ,CAAC;gBAAS,CAAC;YACV,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { describe, it } from \"node:test\";\nimport { mkdtempSync, rmSync } from \"node:fs\";\nimport { join } from \"node:path\";\nimport { tmpdir } from \"node:os\";\n\nimport { SessionManager } from \"./session-manager.js\";\n\nfunction makeAssistantMessage(input: number, output: number, cacheRead = 0, cacheWrite = 0, cost = 0) {\n\treturn {\n\t\trole: \"assistant\",\n\t\tcontent: [{ type: \"text\", text: \"ok\" }],\n\t\tusage: {\n\t\t\tinput,\n\t\t\toutput,\n\t\t\tcacheRead,\n\t\t\tcacheWrite,\n\t\t\ttotal: input + output + cacheRead + cacheWrite,\n\t\t\tcost: { total: cost },\n\t\t},\n\t} as any;\n}\n\ndescribe(\"SessionManager usage totals\", () => {\n\tit(\"tracks assistant usage incrementally without rescanning entries\", () => {\n\t\tconst dir = mkdtempSync(join(tmpdir(), \"gsd-session-manager-test-\"));\n\t\ttry {\n\t\t\tconst manager = SessionManager.create(dir, dir);\n\n\t\t\tmanager.appendMessage({ role: \"user\", content: [{ type: \"text\", text: \"hello\" }] } as any);\n\t\t\tmanager.appendMessage(makeAssistantMessage(10, 5, 3, 2, 0.25));\n\t\t\tmanager.appendMessage(makeAssistantMessage(7, 4, 1, 0, 0.1));\n\n\t\t\tassert.deepEqual(manager.getUsageTotals(), {\n\t\t\t\tinput: 17,\n\t\t\t\toutput: 9,\n\t\t\t\tcacheRead: 4,\n\t\t\t\tcacheWrite: 2,\n\t\t\t\tcost: 0.35,\n\t\t\t});\n\t\t} finally {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t}\n\t});\n\n\tit(\"resets totals when starting a new session\", () => {\n\t\tconst dir = mkdtempSync(join(tmpdir(), \"gsd-session-manager-test-\"));\n\t\ttry {\n\t\t\tconst manager = SessionManager.create(dir, dir);\n\t\t\tmanager.appendMessage(makeAssistantMessage(5, 5, 0, 0, 0.05));\n\t\t\tassert.equal(manager.getUsageTotals().input, 5);\n\n\t\t\tmanager.newSession();\n\t\t\tassert.deepEqual(manager.getUsageTotals(), {\n\t\t\t\tinput: 0,\n\t\t\t\toutput: 0,\n\t\t\t\tcacheRead: 0,\n\t\t\t\tcacheWrite: 0,\n\t\t\t\tcost: 0,\n\t\t\t});\n\t\t} finally {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t}\n\t});\n});\n"]}

package/packages/pi-coding-agent/dist/core/tools/edit-diff.test.js

@@ -47,18 +47,20 @@ describe("edit-diff", () => {
         assert.ok(result.firstChangedLine !== undefined);
         assert.match(result.diff, /CHANGED/);
     });
-    it("computes diffs for preview without native helpers", async (t) => {
+    it("computes diffs for preview without native helpers", async () => {
         const dir = mkdtempSync(join(tmpdir(), "edit-diff-test-"));
-        t.after(() => {
+        try {
+            const file = join(dir, "sample.ts");
+            writeFileSync(file, "const title = “Hello”;\n", "utf-8");
+            const result = await computeEditDiff(file, "const title = \"Hello\";\n", "const title = \"Hi\";\n", dir);
+            assert.ok(!("error" in result), "expected a diff result");
+            if (!("error" in result)) {
+                assert.equal(result.firstChangedLine, 1);
+                assert.match(result.diff, /\+1 const title = "Hi";/);
+            }
+        }
+        finally {
             rmSync(dir, { recursive: true, force: true });
-        });
-        const file = join(dir, "sample.ts");
-        writeFileSync(file, "const title = “Hello”;\n", "utf-8");
-        const result = await computeEditDiff(file, "const title = \"Hello\";\n", "const title = \"Hi\";\n", dir);
-        assert.ok(!("error" in result), "expected a diff result");
-        if (!("error" in result)) {
-            assert.equal(result.firstChangedLine, 1);
-            assert.match(result.diff, /\+1 const title = "Hi";/);
         }
     });
 });

package/packages/pi-coding-agent/dist/core/tools/edit-diff.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"edit-diff.test.js","sourceRoot":"","sources":["../../../src/core/tools/edit-diff.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,EACN,eAAe,EACf,aAAa,EACb,kBAAkB,EAClB,sBAAsB,GACtB,MAAM,gBAAgB,CAAC;AAExB,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACrE,MAAM,KAAK,GAAG,yCAAyC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,KAAK,CAAC,EAAE,gCAAgC,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACxE,MAAM,MAAM,GAAG,aAAa,CAAC,0BAA0B,EAAE,4BAA4B,CAAC,CAAC;QACvF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,qBAAqB,EAAE,4BAA4B,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC7D,MAAM,MAAM,GAAG,kBAAkB,CAAC,0BAA0B,EAAE,4BAA4B,CAAC,CAAC;QAC5F,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC3E,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACpE,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QAC3C,MAAM,QAAQ,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;QAC5B,QAAQ,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,CAAC,SAAS;QACpC,QAAQ,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,CAAC,UAAU;QACvC,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QAE9C,MAAM,MAAM,GAAG,kBAAkB,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC7D,kEAAkE;QAClE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACpC,2DAA2D;QAC3D,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC5C,mCAAmC;QACnC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACzE,uDAAuD;QACvD,MAAM,SAAS,GAAG,IAAI,CAAC;QACvB,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1E,MAAM,QAAQ,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC/B,QAAQ,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;QAC3B,MAAM,MAAM,GAAG,kBAAkB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;QAC1F,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE;QACnE,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;QAC3D,CAAC,CAAC,KAAK,CAAC,GAAG,EAAE;YACZ,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC,CAAC,CAAC;QAEH,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;QACpC,aAAa,CAAC,IAAI,EAAE,0BAA0B,EAAE,OAAO,CAAC,CAAC;QAEzD,MAAM,MAAM,GAAG,MAAM,eAAe,CACnC,IAAI,EACJ,4BAA4B,EAC5B,yBAAyB,EACzB,GAAG,CACH,CAAC;QAEF,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,wBAAwB,CAAC,CAAC;QAC1D,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;YACzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;QACtD,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { mkdtempSync, rmSync, writeFileSync } from \"node:fs\";\nimport { tmpdir } from \"node:os\";\nimport { join } from \"node:path\";\nimport { describe, it } from \"node:test\";\n\nimport {\n\tcomputeEditDiff,\n\tfuzzyFindText,\n\tgenerateDiffString,\n\tnormalizeForFuzzyMatch,\n} from \"./edit-diff.js\";\n\ndescribe(\"edit-diff\", () => {\n\tit(\"normalizes quotes, dashes, spaces, and trailing whitespace\", () => {\n\t\tconst input = \"“hello”\\u00A0world — test  \\nnext\\t\\t\\n\";\n\t\tassert.equal(normalizeForFuzzyMatch(input), \"\\\"hello\\\" world - test\\nnext\\n\");\n\t});\n\n\tit(\"falls back to fuzzy matching when unicode punctuation differs\", () => {\n\t\tconst result = fuzzyFindText(\"const title = “Hello”;\\n\", \"const title = \\\"Hello\\\";\\n\");\n\t\tassert.equal(result.found, true);\n\t\tassert.equal(result.usedFuzzyMatch, true);\n\t\tassert.equal(result.contentForReplacement, \"const title = \\\"Hello\\\";\\n\");\n\t});\n\n\tit(\"renders numbered diffs with the first changed line\", () => {\n\t\tconst result = generateDiffString(\"line 1\\nline 2\\nline 3\\n\", \"line 1\\nline two\\nline 3\\n\");\n\t\tassert.equal(result.firstChangedLine, 2);\n\t\tassert.match(result.diff, /-2 line 2/);\n\t\tassert.match(result.diff, /\\+2 line two/);\n\t});\n\n\tit(\"respects contextLines and inserts separators for distant changes\", () => {\n\t\tconst lines = Array.from({ length: 20 }, (_, i) => `line ${i + 1}`);\n\t\tconst oldContent = lines.join(\"\\n\") + \"\\n\";\n\t\tconst modified = [...lines];\n\t\tmodified[1] = \"changed 2\"; // line 2\n\t\tmodified[17] = \"changed 18\"; // line 18\n\t\tconst newContent = modified.join(\"\\n\") + \"\\n\";\n\n\t\tconst result = generateDiffString(oldContent, newContent, 2);\n\t\t// Should contain separator between the two distant change regions\n\t\tassert.match(result.diff, /\\.\\.\\./);\n\t\t// Should NOT contain lines far from changes (e.g. line 10)\n\t\tassert.doesNotMatch(result.diff, /line 10/);\n\t\t// Should contain the changed lines\n\t\tassert.match(result.diff, /changed 2/);\n\t\tassert.match(result.diff, /changed 18/);\n\t});\n\n\tit(\"handles large files without OOM by falling back to linear diff\", () => {\n\t\t// Create files large enough to exceed the DP threshold\n\t\tconst lineCount = 3000;\n\t\tconst oldLines = Array.from({ length: lineCount }, (_, i) => `line ${i}`);\n\t\tconst newLines = [...oldLines];\n\t\tnewLines[1500] = \"CHANGED\";\n\t\tconst result = generateDiffString(oldLines.join(\"\\n\") + \"\\n\", newLines.join(\"\\n\") + \"\\n\");\n\t\tassert.ok(result.firstChangedLine !== undefined);\n\t\tassert.match(result.diff, /CHANGED/);\n\t});\n\n\tit(\"computes diffs for preview without native helpers\", async (t) => {\n\t\tconst dir = mkdtempSync(join(tmpdir(), \"edit-diff-test-\"));\n\t\tt.after(() => {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t});\n\n\t\tconst file = join(dir, \"sample.ts\");\n\t\twriteFileSync(file, \"const title = “Hello”;\\n\", \"utf-8\");\n\n\t\tconst result = await computeEditDiff(\n\t\t\tfile,\n\t\t\t\"const title = \\\"Hello\\\";\\n\",\n\t\t\t\"const title = \\\"Hi\\\";\\n\",\n\t\t\tdir,\n\t\t);\n\n\t\tassert.ok(!(\"error\" in result), \"expected a diff result\");\n\t\tif (!(\"error\" in result)) {\n\t\t\tassert.equal(result.firstChangedLine, 1);\n\t\t\tassert.match(result.diff, /\\+1 const title = \"Hi\";/);\n\t\t}\n\t});\n});\n"]}
+{"version":3,"file":"edit-diff.test.js","sourceRoot":"","sources":["../../../src/core/tools/edit-diff.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,SAAS,CAAC;AAC7D,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AACjC,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,MAAM,WAAW,CAAC;AAEzC,OAAO,EACN,eAAe,EACf,aAAa,EACb,kBAAkB,EAClB,sBAAsB,GACtB,MAAM,gBAAgB,CAAC;AAExB,QAAQ,CAAC,WAAW,EAAE,GAAG,EAAE;IAC1B,EAAE,CAAC,4DAA4D,EAAE,GAAG,EAAE;QACrE,MAAM,KAAK,GAAG,yCAAyC,CAAC;QACxD,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,KAAK,CAAC,EAAE,gCAAgC,CAAC,CAAC;IAC/E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,+DAA+D,EAAE,GAAG,EAAE;QACxE,MAAM,MAAM,GAAG,aAAa,CAAC,0BAA0B,EAAE,4BAA4B,CAAC,CAAC;QACvF,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,EAAE,IAAI,CAAC,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,cAAc,EAAE,IAAI,CAAC,CAAC;QAC1C,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,qBAAqB,EAAE,4BAA4B,CAAC,CAAC;IAC1E,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,oDAAoD,EAAE,GAAG,EAAE;QAC7D,MAAM,MAAM,GAAG,kBAAkB,CAAC,0BAA0B,EAAE,4BAA4B,CAAC,CAAC;QAC5F,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;QACzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,GAAG,EAAE;QAC3E,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACpE,MAAM,UAAU,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QAC3C,MAAM,QAAQ,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC;QAC5B,QAAQ,CAAC,CAAC,CAAC,GAAG,WAAW,CAAC,CAAC,SAAS;QACpC,QAAQ,CAAC,EAAE,CAAC,GAAG,YAAY,CAAC,CAAC,UAAU;QACvC,MAAM,UAAU,GAAG,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;QAE9C,MAAM,MAAM,GAAG,kBAAkB,CAAC,UAAU,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;QAC7D,kEAAkE;QAClE,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,QAAQ,CAAC,CAAC;QACpC,2DAA2D;QAC3D,MAAM,CAAC,YAAY,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;QAC5C,mCAAmC;QACnC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,WAAW,CAAC,CAAC;QACvC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,YAAY,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,GAAG,EAAE;QACzE,uDAAuD;QACvD,MAAM,SAAS,GAAG,IAAI,CAAC;QACvB,MAAM,QAAQ,GAAG,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QAC1E,MAAM,QAAQ,GAAG,CAAC,GAAG,QAAQ,CAAC,CAAC;QAC/B,QAAQ,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC;QAC3B,MAAM,MAAM,GAAG,kBAAkB,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC,CAAC;QAC1F,MAAM,CAAC,EAAE,CAAC,MAAM,CAAC,gBAAgB,KAAK,SAAS,CAAC,CAAC;QACjD,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IACtC,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,mDAAmD,EAAE,KAAK,IAAI,EAAE;QAClE,MAAM,GAAG,GAAG,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,iBAAiB,CAAC,CAAC,CAAC;QAC3D,IAAI,CAAC;YACJ,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YACpC,aAAa,CAAC,IAAI,EAAE,0BAA0B,EAAE,OAAO,CAAC,CAAC;YAEzD,MAAM,MAAM,GAAG,MAAM,eAAe,CACnC,IAAI,EACJ,4BAA4B,EAC5B,yBAAyB,EACzB,GAAG,CACH,CAAC;YAEF,MAAM,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,wBAAwB,CAAC,CAAC;YAC1D,IAAI,CAAC,CAAC,OAAO,IAAI,MAAM,CAAC,EAAE,CAAC;gBAC1B,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,gBAAgB,EAAE,CAAC,CAAC,CAAC;gBACzC,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,EAAE,yBAAyB,CAAC,CAAC;YACtD,CAAC;QACF,CAAC;gBAAS,CAAC;YACV,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { mkdtempSync, rmSync, writeFileSync } from \"node:fs\";\nimport { tmpdir } from \"node:os\";\nimport { join } from \"node:path\";\nimport { describe, it } from \"node:test\";\n\nimport {\n\tcomputeEditDiff,\n\tfuzzyFindText,\n\tgenerateDiffString,\n\tnormalizeForFuzzyMatch,\n} from \"./edit-diff.js\";\n\ndescribe(\"edit-diff\", () => {\n\tit(\"normalizes quotes, dashes, spaces, and trailing whitespace\", () => {\n\t\tconst input = \"“hello”\\u00A0world — test  \\nnext\\t\\t\\n\";\n\t\tassert.equal(normalizeForFuzzyMatch(input), \"\\\"hello\\\" world - test\\nnext\\n\");\n\t});\n\n\tit(\"falls back to fuzzy matching when unicode punctuation differs\", () => {\n\t\tconst result = fuzzyFindText(\"const title = “Hello”;\\n\", \"const title = \\\"Hello\\\";\\n\");\n\t\tassert.equal(result.found, true);\n\t\tassert.equal(result.usedFuzzyMatch, true);\n\t\tassert.equal(result.contentForReplacement, \"const title = \\\"Hello\\\";\\n\");\n\t});\n\n\tit(\"renders numbered diffs with the first changed line\", () => {\n\t\tconst result = generateDiffString(\"line 1\\nline 2\\nline 3\\n\", \"line 1\\nline two\\nline 3\\n\");\n\t\tassert.equal(result.firstChangedLine, 2);\n\t\tassert.match(result.diff, /-2 line 2/);\n\t\tassert.match(result.diff, /\\+2 line two/);\n\t});\n\n\tit(\"respects contextLines and inserts separators for distant changes\", () => {\n\t\tconst lines = Array.from({ length: 20 }, (_, i) => `line ${i + 1}`);\n\t\tconst oldContent = lines.join(\"\\n\") + \"\\n\";\n\t\tconst modified = [...lines];\n\t\tmodified[1] = \"changed 2\"; // line 2\n\t\tmodified[17] = \"changed 18\"; // line 18\n\t\tconst newContent = modified.join(\"\\n\") + \"\\n\";\n\n\t\tconst result = generateDiffString(oldContent, newContent, 2);\n\t\t// Should contain separator between the two distant change regions\n\t\tassert.match(result.diff, /\\.\\.\\./);\n\t\t// Should NOT contain lines far from changes (e.g. line 10)\n\t\tassert.doesNotMatch(result.diff, /line 10/);\n\t\t// Should contain the changed lines\n\t\tassert.match(result.diff, /changed 2/);\n\t\tassert.match(result.diff, /changed 18/);\n\t});\n\n\tit(\"handles large files without OOM by falling back to linear diff\", () => {\n\t\t// Create files large enough to exceed the DP threshold\n\t\tconst lineCount = 3000;\n\t\tconst oldLines = Array.from({ length: lineCount }, (_, i) => `line ${i}`);\n\t\tconst newLines = [...oldLines];\n\t\tnewLines[1500] = \"CHANGED\";\n\t\tconst result = generateDiffString(oldLines.join(\"\\n\") + \"\\n\", newLines.join(\"\\n\") + \"\\n\");\n\t\tassert.ok(result.firstChangedLine !== undefined);\n\t\tassert.match(result.diff, /CHANGED/);\n\t});\n\n\tit(\"computes diffs for preview without native helpers\", async () => {\n\t\tconst dir = mkdtempSync(join(tmpdir(), \"edit-diff-test-\"));\n\t\ttry {\n\t\t\tconst file = join(dir, \"sample.ts\");\n\t\t\twriteFileSync(file, \"const title = “Hello”;\\n\", \"utf-8\");\n\n\t\t\tconst result = await computeEditDiff(\n\t\t\t\tfile,\n\t\t\t\t\"const title = \\\"Hello\\\";\\n\",\n\t\t\t\t\"const title = \\\"Hi\\\";\\n\",\n\t\t\t\tdir,\n\t\t\t);\n\n\t\t\tassert.ok(!(\"error\" in result), \"expected a diff result\");\n\t\t\tif (!(\"error\" in result)) {\n\t\t\t\tassert.equal(result.firstChangedLine, 1);\n\t\t\t\tassert.match(result.diff, /\\+1 const title = \"Hi\";/);\n\t\t\t}\n\t\t} finally {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t}\n\t});\n});\n"]}

package/packages/pi-coding-agent/dist/resources/extensions/memory/storage.test.js

@@ -1,5 +1,5 @@
 import assert from "node:assert/strict";
-import { describe, it, afterEach } from "node:test";
+import { describe, it } from "node:test";
 import { mkdtempSync, rmSync, readFileSync } from "node:fs";
 import { join } from "node:path";
 import { tmpdir } from "node:os";
@@ -11,53 +11,57 @@ function wait(ms) {
     return new Promise((resolve) => setTimeout(resolve, ms));
 }
 describe("MemoryStorage debounced persistence", () => {
-    let dir;
-    afterEach(() => {
-        if (dir) {
-            rmSync(dir, { recursive: true, force: true });
-        }
-    });
     it("multiple rapid mutations only trigger one persist write", async () => {
-        dir = makeTmpDir();
+        const dir = makeTmpDir();
         const dbPath = join(dir, "test.db");
-        const storage = await MemoryStorage.create(dbPath);
-        const initialStat = readFileSync(dbPath);
-        const initialMtime = initialStat.length;
-        storage.upsertThreads([
-            { threadId: "t1", filePath: "/a.txt", fileSize: 100, fileMtime: 1000, cwd: "/proj" },
-        ]);
-        storage.upsertThreads([
-            { threadId: "t2", filePath: "/b.txt", fileSize: 200, fileMtime: 2000, cwd: "/proj" },
-        ]);
-        storage.upsertThreads([
-            { threadId: "t3", filePath: "/c.txt", fileSize: 300, fileMtime: 3000, cwd: "/proj" },
-        ]);
-        const afterMutationsBuf = readFileSync(dbPath);
-        assert.deepEqual(afterMutationsBuf, initialStat, "File should not have been written yet (debounce window has not elapsed)");
-        await wait(700);
-        const afterDebounceBuf = readFileSync(dbPath);
-        assert.notDeepEqual(afterDebounceBuf, initialStat, "File should have been written after debounce window elapsed");
-        const stats = storage.getStats();
-        assert.equal(stats.totalThreads, 3);
-        storage.close();
+        try {
+            const storage = await MemoryStorage.create(dbPath);
+            const initialStat = readFileSync(dbPath);
+            const initialMtime = initialStat.length;
+            storage.upsertThreads([
+                { threadId: "t1", filePath: "/a.txt", fileSize: 100, fileMtime: 1000, cwd: "/proj" },
+            ]);
+            storage.upsertThreads([
+                { threadId: "t2", filePath: "/b.txt", fileSize: 200, fileMtime: 2000, cwd: "/proj" },
+            ]);
+            storage.upsertThreads([
+                { threadId: "t3", filePath: "/c.txt", fileSize: 300, fileMtime: 3000, cwd: "/proj" },
+            ]);
+            const afterMutationsBuf = readFileSync(dbPath);
+            assert.deepEqual(afterMutationsBuf, initialStat, "File should not have been written yet (debounce window has not elapsed)");
+            await wait(700);
+            const afterDebounceBuf = readFileSync(dbPath);
+            assert.notDeepEqual(afterDebounceBuf, initialStat, "File should have been written after debounce window elapsed");
+            const stats = storage.getStats();
+            assert.equal(stats.totalThreads, 3);
+            storage.close();
+        }
+        finally {
+            rmSync(dir, { recursive: true, force: true });
+        }
     });
     it("close() flushes pending changes immediately without waiting for debounce", async () => {
-        dir = makeTmpDir();
+        const dir = makeTmpDir();
         const dbPath = join(dir, "test.db");
-        const storage = await MemoryStorage.create(dbPath);
-        const initialBuf = readFileSync(dbPath);
-        storage.upsertThreads([
-            { threadId: "t1", filePath: "/a.txt", fileSize: 100, fileMtime: 1000, cwd: "/proj" },
-        ]);
-        const beforeCloseBuf = readFileSync(dbPath);
-        assert.deepEqual(beforeCloseBuf, initialBuf, "File should not have been written yet (debounce window has not elapsed)");
-        storage.close();
-        const afterCloseBuf = readFileSync(dbPath);
-        assert.notDeepEqual(afterCloseBuf, initialBuf, "File should have been written immediately on close()");
-        const reopened = await MemoryStorage.create(dbPath);
-        const stats = reopened.getStats();
-        assert.equal(stats.totalThreads, 1, "Data should be persisted and readable after close");
-        reopened.close();
+        try {
+            const storage = await MemoryStorage.create(dbPath);
+            const initialBuf = readFileSync(dbPath);
+            storage.upsertThreads([
+                { threadId: "t1", filePath: "/a.txt", fileSize: 100, fileMtime: 1000, cwd: "/proj" },
+            ]);
+            const beforeCloseBuf = readFileSync(dbPath);
+            assert.deepEqual(beforeCloseBuf, initialBuf, "File should not have been written yet (debounce window has not elapsed)");
+            storage.close();
+            const afterCloseBuf = readFileSync(dbPath);
+            assert.notDeepEqual(afterCloseBuf, initialBuf, "File should have been written immediately on close()");
+            const reopened = await MemoryStorage.create(dbPath);
+            const stats = reopened.getStats();
+            assert.equal(stats.totalThreads, 1, "Data should be persisted and readable after close");
+            reopened.close();
+        }
+        finally {
+            rmSync(dir, { recursive: true, force: true });
+        }
     });
 });
 //# sourceMappingURL=storage.test.js.map

package/packages/pi-coding-agent/dist/resources/extensions/memory/storage.test.js.map

@@ -1 +1 @@
-{"version":3,"file":"storage.test.js","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAE,SAAS,EAAE,MAAM,WAAW,CAAC;AACpD,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAc,MAAM,SAAS,CAAC;AACxE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,SAAS,UAAU;IAClB,OAAO,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,0BAA0B,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,SAAS,IAAI,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC1D,CAAC;AAED,QAAQ,CAAC,qCAAqC,EAAE,GAAG,EAAE;IACpD,IAAI,GAAW,CAAC;IAEhB,SAAS,CAAC,GAAG,EAAE;QACd,IAAI,GAAG,EAAE,CAAC;YACT,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACxE,GAAG,GAAG,UAAU,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEnD,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QACzC,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,CAAC;QAExC,OAAO,CAAC,aAAa,CAAC;YACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;SACpF,CAAC,CAAC;QACH,OAAO,CAAC,aAAa,CAAC;YACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;SACpF,CAAC,CAAC;QACH,OAAO,CAAC,aAAa,CAAC;YACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;SACpF,CAAC,CAAC;QAEH,MAAM,iBAAiB,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAC/C,MAAM,CAAC,SAAS,CACf,iBAAiB,EACjB,WAAW,EACX,yEAAyE,CACzE,CAAC;QAEF,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;QAEhB,MAAM,gBAAgB,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAC9C,MAAM,CAAC,YAAY,CAClB,gBAAgB,EAChB,WAAW,EACX,6DAA6D,CAC7D,CAAC;QAEF,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;QACjC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;QAEpC,OAAO,CAAC,KAAK,EAAE,CAAC;IACjB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACzF,GAAG,GAAG,UAAU,EAAE,CAAC;QACnB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACpC,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAEnD,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAExC,OAAO,CAAC,aAAa,CAAC;YACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;SACpF,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAC5C,MAAM,CAAC,SAAS,CACf,cAAc,EACd,UAAU,EACV,yEAAyE,CACzE,CAAC;QAEF,OAAO,CAAC,KAAK,EAAE,CAAC;QAEhB,MAAM,aAAa,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,CAAC,YAAY,CAClB,aAAa,EACb,UAAU,EACV,sDAAsD,CACtD,CAAC;QAEF,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACpD,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC;QAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,EAAE,mDAAmD,CAAC,CAAC;QACzF,QAAQ,CAAC,KAAK,EAAE,CAAC;IAClB,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { describe, it, afterEach } from \"node:test\";\nimport { mkdtempSync, rmSync, readFileSync, existsSync } from \"node:fs\";\nimport { join } from \"node:path\";\nimport { tmpdir } from \"node:os\";\n\nimport { MemoryStorage } from \"./storage.js\";\n\nfunction makeTmpDir(): string {\n\treturn mkdtempSync(join(tmpdir(), \"gsd-memory-storage-test-\"));\n}\n\nfunction wait(ms: number): Promise<void> {\n\treturn new Promise((resolve) => setTimeout(resolve, ms));\n}\n\ndescribe(\"MemoryStorage debounced persistence\", () => {\n\tlet dir: string;\n\n\tafterEach(() => {\n\t\tif (dir) {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t}\n\t});\n\n\tit(\"multiple rapid mutations only trigger one persist write\", async () => {\n\t\tdir = makeTmpDir();\n\t\tconst dbPath = join(dir, \"test.db\");\n\t\tconst storage = await MemoryStorage.create(dbPath);\n\n\t\tconst initialStat = readFileSync(dbPath);\n\t\tconst initialMtime = initialStat.length;\n\n\t\tstorage.upsertThreads([\n\t\t\t{ threadId: \"t1\", filePath: \"/a.txt\", fileSize: 100, fileMtime: 1000, cwd: \"/proj\" },\n\t\t]);\n\t\tstorage.upsertThreads([\n\t\t\t{ threadId: \"t2\", filePath: \"/b.txt\", fileSize: 200, fileMtime: 2000, cwd: \"/proj\" },\n\t\t]);\n\t\tstorage.upsertThreads([\n\t\t\t{ threadId: \"t3\", filePath: \"/c.txt\", fileSize: 300, fileMtime: 3000, cwd: \"/proj\" },\n\t\t]);\n\n\t\tconst afterMutationsBuf = readFileSync(dbPath);\n\t\tassert.deepEqual(\n\t\t\tafterMutationsBuf,\n\t\t\tinitialStat,\n\t\t\t\"File should not have been written yet (debounce window has not elapsed)\",\n\t\t);\n\n\t\tawait wait(700);\n\n\t\tconst afterDebounceBuf = readFileSync(dbPath);\n\t\tassert.notDeepEqual(\n\t\t\tafterDebounceBuf,\n\t\t\tinitialStat,\n\t\t\t\"File should have been written after debounce window elapsed\",\n\t\t);\n\n\t\tconst stats = storage.getStats();\n\t\tassert.equal(stats.totalThreads, 3);\n\n\t\tstorage.close();\n\t});\n\n\tit(\"close() flushes pending changes immediately without waiting for debounce\", async () => {\n\t\tdir = makeTmpDir();\n\t\tconst dbPath = join(dir, \"test.db\");\n\t\tconst storage = await MemoryStorage.create(dbPath);\n\n\t\tconst initialBuf = readFileSync(dbPath);\n\n\t\tstorage.upsertThreads([\n\t\t\t{ threadId: \"t1\", filePath: \"/a.txt\", fileSize: 100, fileMtime: 1000, cwd: \"/proj\" },\n\t\t]);\n\n\t\tconst beforeCloseBuf = readFileSync(dbPath);\n\t\tassert.deepEqual(\n\t\t\tbeforeCloseBuf,\n\t\t\tinitialBuf,\n\t\t\t\"File should not have been written yet (debounce window has not elapsed)\",\n\t\t);\n\n\t\tstorage.close();\n\n\t\tconst afterCloseBuf = readFileSync(dbPath);\n\t\tassert.notDeepEqual(\n\t\t\tafterCloseBuf,\n\t\t\tinitialBuf,\n\t\t\t\"File should have been written immediately on close()\",\n\t\t);\n\n\t\tconst reopened = await MemoryStorage.create(dbPath);\n\t\tconst stats = reopened.getStats();\n\t\tassert.equal(stats.totalThreads, 1, \"Data should be persisted and readable after close\");\n\t\treopened.close();\n\t});\n});\n"]}
+{"version":3,"file":"storage.test.js","sourceRoot":"","sources":["../../../../src/resources/extensions/memory/storage.test.ts"],"names":[],"mappings":"AAAA,OAAO,MAAM,MAAM,oBAAoB,CAAC;AACxC,OAAO,EAAE,QAAQ,EAAE,EAAE,EAAQ,MAAM,WAAW,CAAC;AAC/C,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,YAAY,EAAc,MAAM,SAAS,CAAC;AACxE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AACjC,OAAO,EAAE,MAAM,EAAE,MAAM,SAAS,CAAC;AAEjC,OAAO,EAAE,aAAa,EAAE,MAAM,cAAc,CAAC;AAE7C,SAAS,UAAU;IAClB,OAAO,WAAW,CAAC,IAAI,CAAC,MAAM,EAAE,EAAE,0BAA0B,CAAC,CAAC,CAAC;AAChE,CAAC;AAED,SAAS,IAAI,CAAC,EAAU;IACvB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;AAC1D,CAAC;AAED,QAAQ,CAAC,qCAAqC,EAAE,GAAG,EAAE;IACpD,EAAE,CAAC,yDAAyD,EAAE,KAAK,IAAI,EAAE;QACxE,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACpC,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEnD,MAAM,WAAW,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;YACzC,MAAM,YAAY,GAAG,WAAW,CAAC,MAAM,CAAC;YAExC,OAAO,CAAC,aAAa,CAAC;gBACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;aACpF,CAAC,CAAC;YACH,OAAO,CAAC,aAAa,CAAC;gBACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;aACpF,CAAC,CAAC;YACH,OAAO,CAAC,aAAa,CAAC;gBACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;aACpF,CAAC,CAAC;YAEH,MAAM,iBAAiB,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;YAC/C,MAAM,CAAC,SAAS,CACf,iBAAiB,EACjB,WAAW,EACX,yEAAyE,CACzE,CAAC;YAEF,MAAM,IAAI,CAAC,GAAG,CAAC,CAAC;YAEhB,MAAM,gBAAgB,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;YAC9C,MAAM,CAAC,YAAY,CAClB,gBAAgB,EAChB,WAAW,EACX,6DAA6D,CAC7D,CAAC;YAEF,MAAM,KAAK,GAAG,OAAO,CAAC,QAAQ,EAAE,CAAC;YACjC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC,CAAC;YAEpC,OAAO,CAAC,KAAK,EAAE,CAAC;QACjB,CAAC;gBAAS,CAAC;YACV,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,0EAA0E,EAAE,KAAK,IAAI,EAAE;QACzF,MAAM,GAAG,GAAG,UAAU,EAAE,CAAC;QACzB,MAAM,MAAM,GAAG,IAAI,CAAC,GAAG,EAAE,SAAS,CAAC,CAAC;QACpC,IAAI,CAAC;YACJ,MAAM,OAAO,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YAEnD,MAAM,UAAU,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;YAExC,OAAO,CAAC,aAAa,CAAC;gBACrB,EAAE,QAAQ,EAAE,IAAI,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,EAAE,SAAS,EAAE,IAAI,EAAE,GAAG,EAAE,OAAO,EAAE;aACpF,CAAC,CAAC;YAEH,MAAM,cAAc,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,CAAC,SAAS,CACf,cAAc,EACd,UAAU,EACV,yEAAyE,CACzE,CAAC;YAEF,OAAO,CAAC,KAAK,EAAE,CAAC;YAEhB,MAAM,aAAa,GAAG,YAAY,CAAC,MAAM,CAAC,CAAC;YAC3C,MAAM,CAAC,YAAY,CAClB,aAAa,EACb,UAAU,EACV,sDAAsD,CACtD,CAAC;YAEF,MAAM,QAAQ,GAAG,MAAM,aAAa,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;YACpD,MAAM,KAAK,GAAG,QAAQ,CAAC,QAAQ,EAAE,CAAC;YAClC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,YAAY,EAAE,CAAC,EAAE,mDAAmD,CAAC,CAAC;YACzF,QAAQ,CAAC,KAAK,EAAE,CAAC;QAClB,CAAC;gBAAS,CAAC;YACV,MAAM,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;QAC/C,CAAC;IACF,CAAC,CAAC,CAAC;AACJ,CAAC,CAAC,CAAC","sourcesContent":["import assert from \"node:assert/strict\";\nimport { describe, it, mock } from \"node:test\";\nimport { mkdtempSync, rmSync, readFileSync, existsSync } from \"node:fs\";\nimport { join } from \"node:path\";\nimport { tmpdir } from \"node:os\";\n\nimport { MemoryStorage } from \"./storage.js\";\n\nfunction makeTmpDir(): string {\n\treturn mkdtempSync(join(tmpdir(), \"gsd-memory-storage-test-\"));\n}\n\nfunction wait(ms: number): Promise<void> {\n\treturn new Promise((resolve) => setTimeout(resolve, ms));\n}\n\ndescribe(\"MemoryStorage debounced persistence\", () => {\n\tit(\"multiple rapid mutations only trigger one persist write\", async () => {\n\t\tconst dir = makeTmpDir();\n\t\tconst dbPath = join(dir, \"test.db\");\n\t\ttry {\n\t\t\tconst storage = await MemoryStorage.create(dbPath);\n\n\t\t\tconst initialStat = readFileSync(dbPath);\n\t\t\tconst initialMtime = initialStat.length;\n\n\t\t\tstorage.upsertThreads([\n\t\t\t\t{ threadId: \"t1\", filePath: \"/a.txt\", fileSize: 100, fileMtime: 1000, cwd: \"/proj\" },\n\t\t\t]);\n\t\t\tstorage.upsertThreads([\n\t\t\t\t{ threadId: \"t2\", filePath: \"/b.txt\", fileSize: 200, fileMtime: 2000, cwd: \"/proj\" },\n\t\t\t]);\n\t\t\tstorage.upsertThreads([\n\t\t\t\t{ threadId: \"t3\", filePath: \"/c.txt\", fileSize: 300, fileMtime: 3000, cwd: \"/proj\" },\n\t\t\t]);\n\n\t\t\tconst afterMutationsBuf = readFileSync(dbPath);\n\t\t\tassert.deepEqual(\n\t\t\t\tafterMutationsBuf,\n\t\t\t\tinitialStat,\n\t\t\t\t\"File should not have been written yet (debounce window has not elapsed)\",\n\t\t\t);\n\n\t\t\tawait wait(700);\n\n\t\t\tconst afterDebounceBuf = readFileSync(dbPath);\n\t\t\tassert.notDeepEqual(\n\t\t\t\tafterDebounceBuf,\n\t\t\t\tinitialStat,\n\t\t\t\t\"File should have been written after debounce window elapsed\",\n\t\t\t);\n\n\t\t\tconst stats = storage.getStats();\n\t\t\tassert.equal(stats.totalThreads, 3);\n\n\t\t\tstorage.close();\n\t\t} finally {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t}\n\t});\n\n\tit(\"close() flushes pending changes immediately without waiting for debounce\", async () => {\n\t\tconst dir = makeTmpDir();\n\t\tconst dbPath = join(dir, \"test.db\");\n\t\ttry {\n\t\t\tconst storage = await MemoryStorage.create(dbPath);\n\n\t\t\tconst initialBuf = readFileSync(dbPath);\n\n\t\t\tstorage.upsertThreads([\n\t\t\t\t{ threadId: \"t1\", filePath: \"/a.txt\", fileSize: 100, fileMtime: 1000, cwd: \"/proj\" },\n\t\t\t]);\n\n\t\t\tconst beforeCloseBuf = readFileSync(dbPath);\n\t\t\tassert.deepEqual(\n\t\t\t\tbeforeCloseBuf,\n\t\t\t\tinitialBuf,\n\t\t\t\t\"File should not have been written yet (debounce window has not elapsed)\",\n\t\t\t);\n\n\t\t\tstorage.close();\n\n\t\t\tconst afterCloseBuf = readFileSync(dbPath);\n\t\t\tassert.notDeepEqual(\n\t\t\t\tafterCloseBuf,\n\t\t\t\tinitialBuf,\n\t\t\t\t\"File should have been written immediately on close()\",\n\t\t\t);\n\n\t\t\tconst reopened = await MemoryStorage.create(dbPath);\n\t\t\tconst stats = reopened.getStats();\n\t\t\tassert.equal(stats.totalThreads, 1, \"Data should be persisted and readable after close\");\n\t\t\treopened.close();\n\t\t} finally {\n\t\t\trmSync(dir, { recursive: true, force: true });\n\t\t}\n\t});\n});\n"]}

package/packages/pi-coding-agent/src/core/auth-storage.test.ts

@@ -287,7 +287,7 @@ describe("AuthStorage — oauth credential for non-OAuth provider (#2083)", () =
 		assert.equal(key, undefined);
 	});
 
-	it("falls through to env var when openrouter has type:oauth credential", async (t) => {
+	it("falls through to env var when openrouter has type:oauth credential", async () => {
 		const storage = inMemory({
 			openrouter: {
 				type: "oauth",
@@ -299,17 +299,17 @@ describe("AuthStorage — oauth credential for non-OAuth provider (#2083)", () =
 
 		// Simulate OPENROUTER_API_KEY being set via env
 		const origEnv = process.env.OPENROUTER_API_KEY;
-		t.after(() => {
+		try {
+			process.env.OPENROUTER_API_KEY = "sk-or-v1-env-key";
+			const key = await storage.getApiKey("openrouter");
+			assert.equal(key, "sk-or-v1-env-key");
+		} finally {
 			if (origEnv === undefined) {
 				delete process.env.OPENROUTER_API_KEY;
 			} else {
 				process.env.OPENROUTER_API_KEY = origEnv;
 			}
-		});
-
-		process.env.OPENROUTER_API_KEY = "sk-or-v1-env-key";
-		const key = await storage.getApiKey("openrouter");
-		assert.equal(key, "sk-or-v1-env-key");
+		}
 	});
 
 	it("falls through to fallback resolver when openrouter has type:oauth credential", async () => {

package/packages/pi-coding-agent/src/core/extensions/runner.test.ts

@@ -48,37 +48,37 @@ function makeThrowingExtension(eventType: string, error: Error): Extension {
 }
 
 describe("ExtensionRunner.emitToolCall", () => {
-	it("catches throwing extension handler and routes to emitError", async (t) => {
+	it("catches throwing extension handler and routes to emitError", async () => {
 		const dir = mkdtempSync(join(tmpdir(), "runner-test-"));
-		t.after(() => {
-			rmSync(dir, { recursive: true, force: true });
-		});
-
-		const sessionManager = SessionManager.create(dir, dir);
-		const authStorage = AuthStorage.create();
-		const modelRegistry = new ModelRegistry(authStorage, join(dir, "models.json"));
+		try {
+			const sessionManager = SessionManager.create(dir, dir);
+			const authStorage = AuthStorage.create();
+			const modelRegistry = new ModelRegistry(authStorage, join(dir, "models.json"));
 
-		const throwingExt = makeThrowingExtension("tool_call", new Error("handler crashed"));
-		const runtime = makeMinimalRuntime();
-		const runner = new ExtensionRunner([throwingExt], runtime, dir, sessionManager, modelRegistry);
+			const throwingExt = makeThrowingExtension("tool_call", new Error("handler crashed"));
+			const runtime = makeMinimalRuntime();
+			const runner = new ExtensionRunner([throwingExt], runtime, dir, sessionManager, modelRegistry);
 
-		const errors: any[] = [];
-		runner.onError((err) => errors.push(err));
+			const errors: any[] = [];
+			runner.onError((err) => errors.push(err));
 
-		const event: ToolCallEvent = {
-			type: "tool_call",
-			toolCallId: "test-123",
-			toolName: "test_tool",
-			input: {},
-		} as ToolCallEvent;
+			const event: ToolCallEvent = {
+				type: "tool_call",
+				toolCallId: "test-123",
+				toolName: "test_tool",
+				input: {},
+			} as ToolCallEvent;
 
-		const result = await runner.emitToolCall(event);
+			const result = await runner.emitToolCall(event);
 
-		// Should not throw — error is caught and routed to emitError
-		assert.equal(result, undefined);
-		assert.equal(errors.length, 1);
-		assert.equal(errors[0].error, "handler crashed");
-		assert.equal(errors[0].event, "tool_call");
-		assert.equal(errors[0].extensionPath, "/test/throwing-ext");
+			// Should not throw — error is caught and routed to emitError
+			assert.equal(result, undefined);
+			assert.equal(errors.length, 1);
+			assert.equal(errors[0].error, "handler crashed");
+			assert.equal(errors[0].event, "tool_call");
+			assert.equal(errors[0].extensionPath, "/test/throwing-ext");
+		} finally {
+			rmSync(dir, { recursive: true, force: true });
+		}
 	});
 });

package/packages/pi-coding-agent/src/core/fs-utils.test.ts

@@ -1,54 +1,66 @@
 import assert from "node:assert/strict";
-import { describe, it, afterEach } from "node:test";
+import { describe, it } from "node:test";
 import { mkdtempSync, readFileSync, rmSync, existsSync } from "node:fs";
 import { join } from "node:path";
 import { tmpdir } from "node:os";
 import { atomicWriteFileSync } from "./fs-utils.js";
 
 describe("atomicWriteFileSync", () => {
-	let dir: string;
-
-	afterEach(() => {
-		if (dir) {
+	it("writes file content atomically", () => {
+		const dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
+		try {
+			const filePath = join(dir, "test.txt");
+			atomicWriteFileSync(filePath, "hello world");
+			assert.equal(readFileSync(filePath, "utf-8"), "hello world");
+		} finally {
 			rmSync(dir, { recursive: true, force: true });
 		}
 	});
 
-	it("writes file content atomically", () => {
-		dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
-		const filePath = join(dir, "test.txt");
-		atomicWriteFileSync(filePath, "hello world");
-		assert.equal(readFileSync(filePath, "utf-8"), "hello world");
-	});
-
 	it("overwrites existing file atomically", () => {
-		dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
-		const filePath = join(dir, "test.txt");
-		atomicWriteFileSync(filePath, "first");
-		atomicWriteFileSync(filePath, "second");
-		assert.equal(readFileSync(filePath, "utf-8"), "second");
+		const dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
+		try {
+			const filePath = join(dir, "test.txt");
+			atomicWriteFileSync(filePath, "first");
+			atomicWriteFileSync(filePath, "second");
+			assert.equal(readFileSync(filePath, "utf-8"), "second");
+		} finally {
+			rmSync(dir, { recursive: true, force: true });
+		}
 	});
 
 	it("does not leave .tmp file after successful write", () => {
-		dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
-		const filePath = join(dir, "test.txt");
-		atomicWriteFileSync(filePath, "content");
-		assert.equal(existsSync(filePath + ".tmp"), false);
+		const dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
+		try {
+			const filePath = join(dir, "test.txt");
+			atomicWriteFileSync(filePath, "content");
+			assert.equal(existsSync(filePath + ".tmp"), false);
+		} finally {
+			rmSync(dir, { recursive: true, force: true });
+		}
 	});
 
 	it("supports Buffer content", () => {
-		dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
-		const filePath = join(dir, "test.bin");
-		const buf = Buffer.from([0x00, 0x01, 0x02, 0xff]);
-		atomicWriteFileSync(filePath, buf);
-		const result = readFileSync(filePath);
-		assert.deepEqual(result, buf);
+		const dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
+		try {
+			const filePath = join(dir, "test.bin");
+			const buf = Buffer.from([0x00, 0x01, 0x02, 0xff]);
+			atomicWriteFileSync(filePath, buf);
+			const result = readFileSync(filePath);
+			assert.deepEqual(result, buf);
+		} finally {
+			rmSync(dir, { recursive: true, force: true });
+		}
 	});
 
 	it("supports encoding parameter", () => {
-		dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
-		const filePath = join(dir, "test.txt");
-		atomicWriteFileSync(filePath, "utf8 content", "utf-8");
-		assert.equal(readFileSync(filePath, "utf-8"), "utf8 content");
+		const dir = mkdtempSync(join(tmpdir(), "fs-utils-test-"));
+		try {
+			const filePath = join(dir, "test.txt");
+			atomicWriteFileSync(filePath, "utf8 content", "utf-8");
+			assert.equal(readFileSync(filePath, "utf-8"), "utf8 content");
+		} finally {
+			rmSync(dir, { recursive: true, force: true });
+		}
 	});
 });