88 lines
2.3 KiB
TypeScript
88 lines
2.3 KiB
TypeScript
import { beforeEach, afterAll, describe, expect, it, vi } from "vitest";
|
|
|
|
import { sendMessageTelegram } from "./send.js";
|
|
|
|
const originalEnv = process.env.TELEGRAM_BOT_TOKEN;
|
|
const loadWebMediaMock = vi.fn();
|
|
|
|
const apiMock = {
|
|
sendMessage: vi.fn(),
|
|
sendPhoto: vi.fn(),
|
|
sendVideo: vi.fn(),
|
|
sendAudio: vi.fn(),
|
|
sendDocument: vi.fn(),
|
|
};
|
|
|
|
vi.mock("grammy", async (orig) => {
|
|
const actual = await orig();
|
|
return {
|
|
...actual,
|
|
Bot: vi.fn().mockImplementation(() => ({ api: apiMock })),
|
|
InputFile: actual.InputFile,
|
|
};
|
|
});
|
|
|
|
vi.mock("../web/media.js", () => ({
|
|
loadWebMedia: (...args: unknown[]) => loadWebMediaMock(...args),
|
|
}));
|
|
|
|
describe("sendMessageTelegram", () => {
|
|
beforeEach(() => {
|
|
vi.resetAllMocks();
|
|
process.env.TELEGRAM_BOT_TOKEN = "token123";
|
|
});
|
|
|
|
afterAll(() => {
|
|
process.env.TELEGRAM_BOT_TOKEN = originalEnv;
|
|
});
|
|
|
|
it("sends text and returns ids", async () => {
|
|
apiMock.sendMessage.mockResolvedValueOnce({
|
|
message_id: 42,
|
|
chat: { id: 999 },
|
|
});
|
|
|
|
const res = await sendMessageTelegram("12345", "hello", {
|
|
verbose: false,
|
|
api: apiMock as never,
|
|
});
|
|
expect(res).toEqual({ messageId: "42", chatId: "999" });
|
|
expect(apiMock.sendMessage).toHaveBeenCalled();
|
|
});
|
|
|
|
it("throws when token missing", async () => {
|
|
process.env.TELEGRAM_BOT_TOKEN = "";
|
|
await expect(sendMessageTelegram("1", "hi")).rejects.toThrow(
|
|
/TELEGRAM_BOT_TOKEN/,
|
|
);
|
|
});
|
|
|
|
it("throws on api error", async () => {
|
|
apiMock.sendMessage.mockRejectedValueOnce(new Error("bad token"));
|
|
|
|
await expect(sendMessageTelegram("1", "hi", { api: apiMock as never })).rejects.toThrow(
|
|
/bad token/i,
|
|
);
|
|
});
|
|
|
|
it("sends media via appropriate method", async () => {
|
|
loadWebMediaMock.mockResolvedValueOnce({
|
|
buffer: Buffer.from([1, 2, 3]),
|
|
contentType: "image/jpeg",
|
|
kind: "image",
|
|
fileName: "pic.jpg",
|
|
});
|
|
apiMock.sendPhoto.mockResolvedValueOnce({
|
|
message_id: 99,
|
|
chat: { id: 123 },
|
|
});
|
|
const res = await sendMessageTelegram("123", "hello", {
|
|
mediaUrl: "http://example.com/pic.jpg",
|
|
api: apiMock as never,
|
|
});
|
|
expect(res).toEqual({ messageId: "99", chatId: "123" });
|
|
expect(loadWebMediaMock).toHaveBeenCalled();
|
|
expect(apiMock.sendPhoto).toHaveBeenCalled();
|
|
});
|
|
});
|