59 lines
2.0 KiB
TypeScript
59 lines
2.0 KiB
TypeScript
import { afterEach, beforeEach, describe, expect, it, vi } from "bun:test";
|
|
import { hookFetch } from "@oh-my-pi/pi-utils";
|
|
import { _resetSettingsForTest, Settings } from "../../../src/config/settings";
|
|
import * as toolsManager from "../../../src/utils/tools-manager";
|
|
import { handleYouTube } from "../../../src/web/scrapers/youtube";
|
|
|
|
describe("handleYouTube with Parallel extract", () => {
|
|
beforeEach(async () => {
|
|
_resetSettingsForTest();
|
|
process.env.PARALLEL_API_KEY = "test-parallel-key";
|
|
await Settings.init({ inMemory: true, overrides: { "providers.parallelFetch": true } });
|
|
});
|
|
|
|
afterEach(() => {
|
|
_resetSettingsForTest();
|
|
vi.restoreAllMocks();
|
|
delete process.env.PARALLEL_API_KEY;
|
|
});
|
|
|
|
it("returns Parallel extract content before yt-dlp fallback", async () => {
|
|
const ensureToolSpy = vi.spyOn(toolsManager, "ensureTool");
|
|
using _hook = hookFetch(input => {
|
|
const url = String(input);
|
|
if (url === "https://api.parallel.ai/v1beta/extract") {
|
|
return new Response(
|
|
JSON.stringify({
|
|
extract_id: "extract-youtube-1",
|
|
results: [
|
|
{
|
|
url: "https://www.youtube.com/watch?v=dQw4w9WgXcQ",
|
|
title: "Video page",
|
|
excerpts: [
|
|
"Parallel summary for the video page that is comfortably longer than one hundred characters. ".repeat(
|
|
2,
|
|
),
|
|
],
|
|
full_content: null,
|
|
},
|
|
],
|
|
errors: [],
|
|
warnings: null,
|
|
usage: null,
|
|
}),
|
|
{ status: 200, headers: { "Content-Type": "application/json" } },
|
|
);
|
|
}
|
|
return new Response("unexpected", { status: 500 });
|
|
});
|
|
|
|
const result = await handleYouTube("https://youtu.be/dQw4w9WgXcQ", 10);
|
|
expect(result?.method).toBe("parallel");
|
|
expect(result?.finalUrl).toBe("https://www.youtube.com/watch?v=dQw4w9WgXcQ");
|
|
expect(result?.contentType).toBe("text/markdown");
|
|
expect(result?.content).toContain("Parallel summary for the video page");
|
|
expect(result?.notes).toContain("Used Parallel extract for YouTube");
|
|
expect(ensureToolSpy).not.toHaveBeenCalled();
|
|
});
|
|
});
|