97 lines
3.3 KiB
TypeScript
97 lines
3.3 KiB
TypeScript
import { afterEach, describe, expect, it, vi } from "vitest";
|
|
import { downloadTelegramFile, getTelegramFile, type TelegramFileInfo } from "./download.js";
|
|
import { resetTelegramFetchStateForTests, resolveTelegramFetch } from "./fetch.js";
|
|
|
|
const setDefaultAutoSelectFamily = vi.hoisted(() => vi.fn());
|
|
|
|
vi.mock("node:net", async () => {
|
|
const actual = await vi.importActual<typeof import("node:net")>("node:net");
|
|
return {
|
|
...actual,
|
|
setDefaultAutoSelectFamily,
|
|
};
|
|
});
|
|
|
|
const originalFetch = globalThis.fetch;
|
|
|
|
afterEach(() => {
|
|
resetTelegramFetchStateForTests();
|
|
setDefaultAutoSelectFamily.mockReset();
|
|
vi.unstubAllEnvs();
|
|
vi.clearAllMocks();
|
|
if (originalFetch) {
|
|
globalThis.fetch = originalFetch;
|
|
} else {
|
|
delete (globalThis as { fetch?: typeof fetch }).fetch;
|
|
}
|
|
});
|
|
|
|
describe("resolveTelegramFetch", () => {
|
|
it("returns wrapped global fetch when available", async () => {
|
|
const fetchMock = vi.fn(async () => ({}));
|
|
globalThis.fetch = fetchMock as unknown as typeof fetch;
|
|
const resolved = resolveTelegramFetch();
|
|
expect(resolved).toBeTypeOf("function");
|
|
});
|
|
|
|
it("prefers proxy fetch when provided", async () => {
|
|
const fetchMock = vi.fn(async () => ({}));
|
|
const resolved = resolveTelegramFetch(fetchMock as unknown as typeof fetch);
|
|
expect(resolved).toBeTypeOf("function");
|
|
});
|
|
|
|
it("honors env enable override", async () => {
|
|
vi.stubEnv("OPENCLAW_TELEGRAM_ENABLE_AUTO_SELECT_FAMILY", "1");
|
|
globalThis.fetch = vi.fn(async () => ({})) as unknown as typeof fetch;
|
|
resolveTelegramFetch();
|
|
expect(setDefaultAutoSelectFamily).toHaveBeenCalledWith(true);
|
|
});
|
|
|
|
it("uses config override when provided", async () => {
|
|
globalThis.fetch = vi.fn(async () => ({})) as unknown as typeof fetch;
|
|
resolveTelegramFetch(undefined, { network: { autoSelectFamily: true } });
|
|
expect(setDefaultAutoSelectFamily).toHaveBeenCalledWith(true);
|
|
});
|
|
|
|
it("env disable override wins over config", async () => {
|
|
vi.stubEnv("OPENCLAW_TELEGRAM_ENABLE_AUTO_SELECT_FAMILY", "0");
|
|
vi.stubEnv("OPENCLAW_TELEGRAM_DISABLE_AUTO_SELECT_FAMILY", "1");
|
|
globalThis.fetch = vi.fn(async () => ({})) as unknown as typeof fetch;
|
|
resolveTelegramFetch(undefined, { network: { autoSelectFamily: true } });
|
|
expect(setDefaultAutoSelectFamily).toHaveBeenCalledWith(false);
|
|
});
|
|
});
|
|
|
|
describe("telegram download", () => {
|
|
it("fetches file info", async () => {
|
|
const json = vi.fn().mockResolvedValue({ ok: true, result: { file_path: "photos/1.jpg" } });
|
|
vi.spyOn(globalThis, "fetch" as never).mockResolvedValueOnce({
|
|
ok: true,
|
|
status: 200,
|
|
statusText: "OK",
|
|
json,
|
|
} as Response);
|
|
const info = await getTelegramFile("tok", "fid");
|
|
expect(info.file_path).toBe("photos/1.jpg");
|
|
});
|
|
|
|
it("downloads and saves", async () => {
|
|
const info: TelegramFileInfo = {
|
|
file_id: "fid",
|
|
file_path: "photos/1.jpg",
|
|
};
|
|
const arrayBuffer = async () => new Uint8Array([1, 2, 3, 4]).buffer;
|
|
vi.spyOn(globalThis, "fetch" as never).mockResolvedValueOnce({
|
|
ok: true,
|
|
status: 200,
|
|
statusText: "OK",
|
|
body: true,
|
|
arrayBuffer,
|
|
headers: { get: () => "image/jpeg" },
|
|
} as Response);
|
|
const saved = await downloadTelegramFile("tok", info, 1024 * 1024);
|
|
expect(saved.path).toBeTruthy();
|
|
expect(saved.contentType).toBe("image/jpeg");
|
|
});
|
|
});
|