85 lines
2.5 KiB
TypeScript
85 lines
2.5 KiB
TypeScript
/**
|
|
* Copyright (c) Microsoft Corporation.
|
|
*
|
|
* Licensed under the Apache License, Version 2.0 (the "License");
|
|
* you may not use this file except in compliance with the License.
|
|
* You may obtain a copy of the License at
|
|
*
|
|
* http://www.apache.org/licenses/LICENSE-2.0
|
|
*
|
|
* Unless required by applicable law or agreed to in writing, software
|
|
* distributed under the License is distributed on an "AS IS" BASIS,
|
|
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
* See the License for the specific language governing permissions and
|
|
* limitations under the License.
|
|
*/
|
|
|
|
import { test, expect } from './fixtures';
|
|
import type { Client } from '@modelcontextprotocol/sdk/client/index.js';
|
|
|
|
for (const mode of ['isolated', 'persistent']) {
|
|
test(`should work with recordVideo (${mode})`, async ({ startClient, server }, testInfo) => {
|
|
const videosDir = testInfo.outputPath('videos');
|
|
|
|
const { client } = await startClient({
|
|
config: {
|
|
browser: {
|
|
contextOptions: {
|
|
recordVideo: {
|
|
dir: videosDir,
|
|
size: { width: 800, height: 600 },
|
|
},
|
|
}
|
|
}
|
|
},
|
|
args: [
|
|
...(mode === 'isolated' ? ['--isolated'] : []),
|
|
],
|
|
});
|
|
|
|
await navigateToTestPage(client, server);
|
|
await produceFrames(client);
|
|
await closeBrowser(client);
|
|
});
|
|
}
|
|
|
|
async function navigateToTestPage(client: Client, server: any) {
|
|
expect(await client.callTool({
|
|
name: 'browser_navigate',
|
|
arguments: { url: server.HELLO_WORLD },
|
|
})).toHaveResponse({
|
|
code: expect.stringContaining(`page.goto('http://localhost`),
|
|
});
|
|
}
|
|
|
|
async function closeBrowser(client: Client) {
|
|
expect(await client.callTool({
|
|
name: 'browser_close',
|
|
})).toHaveResponse({
|
|
code: expect.stringContaining(`page.close()`),
|
|
});
|
|
}
|
|
|
|
async function produceFrames(client: Client) {
|
|
expect(await client.callTool({
|
|
name: 'browser_evaluate',
|
|
arguments: {
|
|
function: `async () => {
|
|
async function rafraf(count) {
|
|
for (let i = 0; i < count; i++)
|
|
await new Promise(f => requestAnimationFrame(() => requestAnimationFrame(f)));
|
|
}
|
|
document.body.style.backgroundColor = "red";
|
|
await rafraf(30);
|
|
document.body.style.backgroundColor = "green";
|
|
await rafraf(30);
|
|
document.body.style.backgroundColor = "blue";
|
|
await rafraf(30);
|
|
return 'ok';
|
|
}`,
|
|
},
|
|
})).toHaveResponse({
|
|
result: '"ok"',
|
|
});
|
|
}
|