test: tighten tokenjuice middleware assertion

steipete · steipete · commit 2bdb77472624 · 2026-05-09T04:22:45.000+01:00
diff --git a/extensions/tokenjuice/index.test.ts b/extensions/tokenjuice/index.test.ts
@@ -48,8 +48,8 @@ describe("tokenjuice bundled plugin", () => {
 
     expect(createTokenjuiceOpenClawEmbeddedExtension).toHaveBeenCalledTimes(1);
     expect(tokenjuiceFactory).toHaveBeenCalledTimes(1);
-    expect(registerAgentToolResultMiddleware).toHaveBeenCalledWith(expect.any(Function), {
-      runtimes: ["pi", "codex"],
-    });
+    const registration = registerAgentToolResultMiddleware.mock.calls[0];
+    expect(typeof registration?.[0]).toBe("function");
+    expect(registration?.[1]).toEqual({ runtimes: ["pi", "codex"] });
   });
 });
