diff --git a/src/flows/search-setup.test.ts b/src/flows/search-setup.test.ts index 9779f0fef1c2..503e338d2bde 100644 --- a/src/flows/search-setup.test.ts +++ b/src/flows/search-setup.test.ts @@ -173,8 +173,6 @@ describe("runSearchSetupFlow", () => { }); it("localizes setup copy for web search provider selection", async () => { - const previousLocale = process.env.OPENCLAW_LOCALE; - process.env.OPENCLAW_LOCALE = "zh-CN"; const note = vi.fn(async () => {}); const select = vi.fn().mockResolvedValueOnce("__skip__"); const prompter = createWizardPrompter({ @@ -182,19 +180,13 @@ describe("runSearchSetupFlow", () => { select: select as never, }); - try { + await withEnvAsync({ OPENCLAW_LOCALE: "zh-CN" }, async () => { await runSearchSetupFlow( { plugins: { allow: ["xai"] } }, createNonExitingRuntime(), prompter, ); - } finally { - if (previousLocale === undefined) { - delete process.env.OPENCLAW_LOCALE; - } else { - process.env.OPENCLAW_LOCALE = previousLocale; - } - } + }); expect(note).toHaveBeenCalledWith(expect.stringContaining("在线查询资料"), "网页搜索"); expect(select).toHaveBeenCalledWith(