vultr/openclaw
1
0
Fork 0
mirror of https://github.com/openclaw/openclaw.git synced 2026-05-06 06:20:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

test(browser): mirror route URL guard in existing-session helper

Browse Source
This commit is contained in:
Peter Steinberger
2026-05-04 22:29:08 +01:00
parent 0909df1a4f
commit 4556707cb7
1 changed files with 32 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
extensions/browser/src/browser/routes/existing-session.test-support.ts
View File

@@ -1,4 +1,9 @@
import { vi } from "vitest";
import {
assertBrowserNavigationResultAllowed,
withBrowserNavigationPolicy,
} from "../navigation-guard.js";
import type { BrowserRouteContext } from "../server-context.js";
import type { BrowserRequest } from "./types.js";
export const existingSessionRouteState = {
@@ -37,14 +42,33 @@ export function createExistingSessionAgentSharedModule() {
typeof body.targetId === "string" ? body.targetId : undefined,
),
withPlaywrightRouteContext: vi.fn(),
withRouteTabContext: vi.fn(async ({ run }: { run: (args: unknown) => Promise<void> }) => {
await run({
profileCtx: existingSessionRouteState.profileCtx,
cdpUrl: "http://127.0.0.1:18800",
tab: existingSessionRouteState.tab,
resolveTabUrl: vi.fn(async (fallbackUrl?: string) => fallbackUrl ?? routeStateUrl()),
});
}),
withRouteTabContext: vi.fn(
async ({
ctx,
enforceCurrentUrlAllowed,
run,
}: {
ctx: BrowserRouteContext;
enforceCurrentUrlAllowed?: boolean;
run: (args: unknown) => Promise<void>;
}) => {
if (enforceCurrentUrlAllowed) {
const ssrfPolicyOpts = withBrowserNavigationPolicy(ctx.state().resolved.ssrfPolicy);
if (ssrfPolicyOpts.ssrfPolicy) {
await assertBrowserNavigationResultAllowed({
url: existingSessionRouteState.tab.url,
...ssrfPolicyOpts,
});
}
}
await run({
profileCtx: existingSessionRouteState.profileCtx,
cdpUrl: "http://127.0.0.1:18800",
tab: existingSessionRouteState.tab,
resolveTabUrl: vi.fn(async (fallbackUrl?: string) => fallbackUrl ?? routeStateUrl()),
});
},
),
};
}