diff --git a/src/agents/tools/telegram-actions.e2e.test.ts b/src/agents/tools/telegram-actions.e2e.test.ts index c4e26f403..42d2b9d2f 100644 --- a/src/agents/tools/telegram-actions.e2e.test.ts +++ b/src/agents/tools/telegram-actions.e2e.test.ts @@ -1,5 +1,6 @@ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest"; import type { OpenClawConfig } from "../../config/config.js"; +import { captureEnv } from "../../test-utils/env.js"; import { handleTelegramAction, readTelegramButtons } from "./telegram-actions.js"; const reactMessageTelegram = vi.fn(async () => ({ ok: true })); @@ -12,7 +13,7 @@ const sendStickerTelegram = vi.fn(async () => ({ chatId: "123", })); const deleteMessageTelegram = vi.fn(async () => ({ ok: true })); -const originalToken = process.env.TELEGRAM_BOT_TOKEN; +let envSnapshot: ReturnType; vi.mock("../../telegram/send.js", () => ({ reactMessageTelegram: (...args: Parameters) => @@ -50,6 +51,7 @@ describe("handleTelegramAction", () => { } beforeEach(() => { + envSnapshot = captureEnv(["TELEGRAM_BOT_TOKEN"]); reactMessageTelegram.mockClear(); sendMessageTelegram.mockClear(); sendStickerTelegram.mockClear(); @@ -58,11 +60,7 @@ describe("handleTelegramAction", () => { }); afterEach(() => { - if (originalToken === undefined) { - delete process.env.TELEGRAM_BOT_TOKEN; - } else { - process.env.TELEGRAM_BOT_TOKEN = originalToken; - } + envSnapshot.restore(); }); it("adds reactions when reactionLevel is minimal", async () => {