From 368469688b1ff4e1a347da7e202daa606cd17dbe Mon Sep 17 00:00:00 2001
From: Peter Steinberger <steipete@gmail.com>
Date: Wed, 27 May 2026 07:55:01 +0100
Subject: [PATCH] test: fix main ci expectations

---
 src/acp/control-plane/manager.test.ts | 6 +++---
 src/commands/agent/session.test.ts    | 4 ++--
 2 files changed, 5 insertions(+), 5 deletions(-)

diff --git a/src/acp/control-plane/manager.test.ts b/src/acp/control-plane/manager.test.ts
index 422ea013567..f2f6e1c55c2 100644
--- a/src/acp/control-plane/manager.test.ts
+++ b/src/acp/control-plane/manager.test.ts
@@ -274,11 +274,11 @@ function extractStateUpsertPersistenceOptions(): Array<{
     };
     const current = readySessionMeta();
     const next = payload.mutate(current, { acp: current });
-    if (next?.state && payload.skipMaintenance === true && payload.takeCacheOwnership === true) {
+    if (next?.state && payload.skipMaintenance && payload.takeCacheOwnership) {
       options.push({
         state: next.state,
-        ...(payload.skipMaintenance === true ? { skipMaintenance: true } : {}),
-        ...(payload.takeCacheOwnership === true ? { takeCacheOwnership: true } : {}),
+        ...(payload.skipMaintenance ? { skipMaintenance: true } : {}),
+        ...(payload.takeCacheOwnership ? { takeCacheOwnership: true } : {}),
       });
     }
   }
diff --git a/src/commands/agent/session.test.ts b/src/commands/agent/session.test.ts
index be619cfdd25..90962fe0376 100644
--- a/src/commands/agent/session.test.ts
+++ b/src/commands/agent/session.test.ts
@@ -147,7 +147,7 @@ describe("resolveSessionKeyForRequest", () => {
     expect(result.storePath).toBe(SHARED_STORE_PATH);
     expect(result.sessionStore["agent:mybot:main"]?.sessionId).toBe("legacy-session-id");
     expect(mocks.loadSessionStore).toHaveBeenCalledTimes(1);
-    expect(mocks.loadSessionStore).toHaveBeenCalledWith(SHARED_STORE_PATH);
+    expect(mocks.loadSessionStore).toHaveBeenCalledWith(SHARED_STORE_PATH, undefined);
   });
 
   it("prefers the configured default-agent session over legacy main-store rows", () => {
@@ -266,7 +266,7 @@ describe("resolveSessionKeyForRequest", () => {
     expect(result.sessionKey).toBe("agent:mybot:explicit:target-session-id");
     expect(result.storePath).toBe(MYBOT_STORE_PATH);
     expect(mocks.loadSessionStore).toHaveBeenCalledTimes(1);
-    expect(mocks.loadSessionStore).toHaveBeenCalledWith(MYBOT_STORE_PATH);
+    expect(mocks.loadSessionStore).toHaveBeenCalledWith(MYBOT_STORE_PATH, undefined);
   });
 
   it("returns correct sessionStore when session found in non-primary agent store", () => {