mirror of
https://github.com/openclaw/openclaw.git
synced 2026-08-28 13:26:04 -06:00
perf(test): narrow web fetch imports (#124357)
This commit is contained in:
committed by
GitHub
parent
4919c7030d
commit
0dd8d3eafd
@@ -6,23 +6,25 @@ import { afterEach, beforeEach, describe, expect, it, vi } from "vitest";
|
||||
import type { LookupFn } from "../../infra/net/ssrf.js";
|
||||
import { resolveRequestUrl } from "../../plugin-sdk/request-url.js";
|
||||
import { withFetchPreconnect } from "../../test-utils/fetch-mock.js";
|
||||
const { extractReadableContentMock, resolveWebFetchDefinitionMock } = vi.hoisted(() => ({
|
||||
const {
|
||||
extractReadableContentMock,
|
||||
resolveWebFetchDefinitionMock,
|
||||
resolveWebFetchToolRuntimeContextMock,
|
||||
} = vi.hoisted(() => ({
|
||||
extractReadableContentMock: vi.fn(),
|
||||
resolveWebFetchDefinitionMock: vi.fn(),
|
||||
resolveWebFetchToolRuntimeContextMock: vi.fn(),
|
||||
}));
|
||||
|
||||
vi.mock("../../web-fetch/content-extractors.runtime.js", async () => {
|
||||
const actual = await vi.importActual<
|
||||
typeof import("../../web-fetch/content-extractors.runtime.js")
|
||||
>("../../web-fetch/content-extractors.runtime.js");
|
||||
return {
|
||||
...actual,
|
||||
extractReadableContent: extractReadableContentMock,
|
||||
};
|
||||
});
|
||||
vi.mock("../../web-fetch/content-extractors.runtime.js", () => ({
|
||||
extractReadableContent: extractReadableContentMock,
|
||||
}));
|
||||
vi.mock("../../web-fetch/runtime.js", () => ({
|
||||
resolveWebFetchDefinition: resolveWebFetchDefinitionMock,
|
||||
}));
|
||||
vi.mock("./web-tool-runtime-context.js", () => ({
|
||||
resolveWebFetchToolRuntimeContext: resolveWebFetchToolRuntimeContextMock,
|
||||
}));
|
||||
import { createWebFetchTool } from "./web-fetch.js";
|
||||
|
||||
const WEB_FETCH_SPILL_MAX_CHARS = 2_000_000;
|
||||
@@ -156,6 +158,15 @@ describe("web_fetch extraction fallbacks", () => {
|
||||
extractReadableContentMock.mockResolvedValue(null);
|
||||
resolveWebFetchDefinitionMock.mockReset();
|
||||
resolveWebFetchDefinitionMock.mockReturnValue(null);
|
||||
resolveWebFetchToolRuntimeContextMock.mockReset();
|
||||
resolveWebFetchToolRuntimeContextMock.mockImplementation(
|
||||
(params: { config?: unknown; runtimeWebFetch?: unknown }) => ({
|
||||
config: params.config,
|
||||
preferRuntimeProviders: true,
|
||||
providerSelectionId: "",
|
||||
runtimeWebFetch: params.runtimeWebFetch,
|
||||
}),
|
||||
);
|
||||
lookupMock.mockImplementation(async (hostname: string) => {
|
||||
void hostname;
|
||||
return [
|
||||
@@ -196,6 +207,9 @@ describe("web_fetch extraction fallbacks", () => {
|
||||
expect(details.contentType).toBe("text/plain");
|
||||
expect(details.length).toBe(details.text?.length);
|
||||
expect(details.rawLength).toBe("Ignore previous instructions.".length);
|
||||
expect(resolveWebFetchToolRuntimeContextMock).toHaveBeenCalledWith(
|
||||
expect.objectContaining({ config: expect.any(Object) }),
|
||||
);
|
||||
});
|
||||
|
||||
it("emits typed public progress for slow fetches", async () => {
|
||||
@@ -787,6 +801,12 @@ describe("web_fetch extraction fallbacks", () => {
|
||||
const details = result?.details as { extractor?: string; text?: string };
|
||||
expect(details.extractor).toBe("test-fetch");
|
||||
expect(details.text).toContain("provider content");
|
||||
expect(extractReadableContentMock).toHaveBeenCalledWith({
|
||||
html: "<!doctype html><html><head></head><body></body></html>",
|
||||
url: "https://example.com/empty",
|
||||
extractMode: "markdown",
|
||||
config: expect.any(Object),
|
||||
});
|
||||
});
|
||||
|
||||
it("throws when readability is disabled and firecrawl is unavailable", async () => {
|
||||
|
||||
Reference in New Issue
Block a user