From 5aab2d69fefa0d3ed7f6dcdda10da024969e8a1f Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Sat, 9 May 2026 14:07:09 +0100
Subject: [PATCH] test: tighten embedded extension runtime assertion

---
 src/agents/pi-embedded-runner/extensions.test.ts | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/src/agents/pi-embedded-runner/extensions.test.ts b/src/agents/pi-embedded-runner/extensions.test.ts
index 2c223d240b1..239274e2700 100644
--- a/src/agents/pi-embedded-runner/extensions.test.ts
+++ b/src/agents/pi-embedded-runner/extensions.test.ts
@@ -41,7 +41,10 @@ function expectSafeguardRuntime(
   const { factories, sessionManager } = buildSafeguardFactories(cfg);
 
   expect(factories).toContain(compactionSafeguardExtension);
-  expect(getCompactionSafeguardRuntime(sessionManager)).toMatchObject(expectedRuntime);
+  const runtime = getCompactionSafeguardRuntime(sessionManager);
+  expect(runtime?.contextWindowTokens).toBe(200_000);
+  expect(runtime?.qualityGuardEnabled).toBe(expectedRuntime.qualityGuardEnabled);
+  expect(runtime?.qualityGuardMaxRetries).toBe(expectedRuntime.qualityGuardMaxRetries);
 }
 
 describe("buildEmbeddedExtensionFactories", () => {