From fddf8a6f4ae77fbd0848858c145273376c9ab530 Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Mon, 16 Feb 2026 00:22:36 +0000
Subject: [PATCH] perf(test): fold pi extensions runtime registry tests into
 agents suite

---
 src/agents/context.test.ts                    | 21 ++++++++++++++++++
 .../session-manager-runtime-registry.test.ts  | 22 -------------------
 2 files changed, 21 insertions(+), 22 deletions(-)
 delete mode 100644 src/agents/pi-extensions/session-manager-runtime-registry.test.ts

diff --git a/src/agents/context.test.ts b/src/agents/context.test.ts
index 41111b4bb41..091df223e56 100644
--- a/src/agents/context.test.ts
+++ b/src/agents/context.test.ts
@@ -1,5 +1,6 @@
 import { describe, expect, it } from "vitest";
 import { applyConfiguredContextWindows } from "./context.js";
+import { createSessionManagerRuntimeRegistry } from "./pi-extensions/session-manager-runtime-registry.js";
 
 describe("applyConfiguredContextWindows", () => {
   it("overrides discovered cache values with explicit models.providers contextWindow", () => {
@@ -39,3 +40,23 @@ describe("applyConfiguredContextWindows", () => {
     expect(cache.has("bad/model")).toBe(false);
   });
 });
+
+describe("createSessionManagerRuntimeRegistry", () => {
+  it("stores, reads, and clears values by object identity", () => {
+    const registry = createSessionManagerRuntimeRegistry<{ value: number }>();
+    const key = {};
+    expect(registry.get(key)).toBeNull();
+    registry.set(key, { value: 1 });
+    expect(registry.get(key)).toEqual({ value: 1 });
+    registry.set(key, null);
+    expect(registry.get(key)).toBeNull();
+  });
+
+  it("ignores non-object keys", () => {
+    const registry = createSessionManagerRuntimeRegistry<{ value: number }>();
+    registry.set(null, { value: 1 });
+    registry.set(123, { value: 1 });
+    expect(registry.get(null)).toBeNull();
+    expect(registry.get(123)).toBeNull();
+  });
+});
diff --git a/src/agents/pi-extensions/session-manager-runtime-registry.test.ts b/src/agents/pi-extensions/session-manager-runtime-registry.test.ts
deleted file mode 100644
index 59e004fab7a..00000000000
--- a/src/agents/pi-extensions/session-manager-runtime-registry.test.ts
+++ /dev/null
@@ -1,22 +0,0 @@
-import { describe, expect, it } from "vitest";
-import { createSessionManagerRuntimeRegistry } from "./session-manager-runtime-registry.js";
-
-describe("createSessionManagerRuntimeRegistry", () => {
-  it("stores, reads, and clears values by object identity", () => {
-    const registry = createSessionManagerRuntimeRegistry<{ value: number }>();
-    const key = {};
-    expect(registry.get(key)).toBeNull();
-    registry.set(key, { value: 1 });
-    expect(registry.get(key)).toEqual({ value: 1 });
-    registry.set(key, null);
-    expect(registry.get(key)).toBeNull();
-  });
-
-  it("ignores non-object keys", () => {
-    const registry = createSessionManagerRuntimeRegistry<{ value: number }>();
-    registry.set(null, { value: 1 });
-    registry.set(123, { value: 1 });
-    expect(registry.get(null)).toBeNull();
-    expect(registry.get(123)).toBeNull();
-  });
-});