From 3c3e39684ef752f1cfc26573fb459cda250747f5 Mon Sep 17 00:00:00 2001
From: Ayaan Zaidi <hi@obviy.us>
Date: Sat, 30 May 2026 09:23:49 +0530
Subject: [PATCH] test(agents): cover flushed cli context engine session

---
 src/agents/cli-runner.context-engine.test.ts | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/agents/cli-runner.context-engine.test.ts b/src/agents/cli-runner.context-engine.test.ts
index 28af255128f4..3014cc805409 100644
--- a/src/agents/cli-runner.context-engine.test.ts
+++ b/src/agents/cli-runner.context-engine.test.ts
@@ -1,5 +1,5 @@
 import type { AgentMessage } from "openclaw/plugin-sdk/agent-core";
-import { beforeEach, describe, expect, it, vi } from "vitest";
+import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
 import type { ContextEngine } from "../context-engine/types.js";
 import type { PreparedCliRunContext } from "./cli-runner/types.js";
 
@@ -122,7 +122,7 @@ function expectMessageText(message: AgentMessage | undefined, expected: string):
 }
 
 describe("runPreparedCliAgent context engine lifecycle", () => {
-  beforeEach(() => {
+  beforeEach(async () => {
     executePreparedCliRunMock.mockReset();
     executePreparedCliRunMock.mockResolvedValue({
       text: " final answer ",
@@ -140,6 +140,16 @@ describe("runPreparedCliAgent context engine lifecycle", () => {
     loadCliSessionHistoryMessagesMock.mockResolvedValue([]);
     getGlobalHookRunnerMock.mockReset();
     getGlobalHookRunnerMock.mockReturnValue(null);
+    const { restoreCliRunnerTestDeps, setCliRunnerTestDeps } = await import("./cli-runner.js");
+    restoreCliRunnerTestDeps();
+    setCliRunnerTestDeps({
+      claudeCliSessionTranscriptHasContent: vi.fn(async () => true),
+    });
+  });
+
+  afterEach(async () => {
+    const { restoreCliRunnerTestDeps } = await import("./cli-runner.js");
+    restoreCliRunnerTestDeps();
   });
 
   it("finalizes successful CLI turns with the active context engine", async () => {