From 3a2c4fd099493c00d6238625fa8b661f52d11914 Mon Sep 17 00:00:00 2001 From: YeonGyu-Kim Date: Fri, 3 Apr 2026 17:37:17 +0900 Subject: [PATCH] fix(tests): type config-handler test spy restore Avoid for the MCP env allowlist spy restore path so the config-handler test keeps the same behavior with specific typing. Ultraworked with [Sisyphus](https://github.com/code-yeongyu/oh-my-openagent) Co-authored-by: Sisyphus --- src/plugin-handlers/config-handler.test.ts | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/src/plugin-handlers/config-handler.test.ts b/src/plugin-handlers/config-handler.test.ts index 050a5ab69..3c0af3a84 100644 --- a/src/plugin-handlers/config-handler.test.ts +++ b/src/plugin-handlers/config-handler.test.ts @@ -31,6 +31,8 @@ function createPluginConfig(overrides: Partial = {}): OhMyOp } } +let setAdditionalAllowedMcpEnvVarsSpy: ReturnType | undefined + beforeEach(() => { spyOn(agents, "createBuiltinAgents" as any).mockResolvedValue({ sisyphus: { name: "sisyphus", prompt: "test", mode: "primary" }, @@ -57,7 +59,7 @@ beforeEach(() => { spyOn(agentLoader, "loadProjectAgents" as any).mockReturnValue({}) spyOn(mcpLoader, "loadMcpConfigs" as any).mockResolvedValue({ servers: {} }) - spyOn(mcpLoader, "setAdditionalAllowedMcpEnvVars").mockImplementation(() => {}) + setAdditionalAllowedMcpEnvVarsSpy = spyOn(mcpLoader, "setAdditionalAllowedMcpEnvVars").mockImplementation(() => {}) spyOn(pluginLoader, "loadAllPluginComponents" as any).mockResolvedValue({ commands: {}, @@ -104,7 +106,7 @@ afterEach(() => { ;(agentLoader.loadUserAgents as any)?.mockRestore?.() ;(agentLoader.loadProjectAgents as any)?.mockRestore?.() ;(mcpLoader.loadMcpConfigs as any)?.mockRestore?.() - ;(mcpLoader.setAdditionalAllowedMcpEnvVars as any)?.mockRestore?.() + setAdditionalAllowedMcpEnvVarsSpy?.mockRestore() ;(pluginLoader.loadAllPluginComponents as any)?.mockRestore?.() ;(mcpModule.createBuiltinMcps as any)?.mockRestore?.() ;(shared.log as any)?.mockRestore?.()