mirror of
https://github.com/web-infra-dev/midscene.git
synced 2025-07-07 00:51:46 +00:00
edf0871fea
* feat(tab-control): enhance the configuration to limit AI from opening new tabs during operations, preventing failures. * chore: optimize evaluate error * chore: resolve navigation error * fix(browser): add forceSameTabNavigation config toe limit open new tab * chore: upgrade vitest version * fix: typo (#390) --------- Co-authored-by: yuyutaotao <167746126+yuyutaotao@users.noreply.github.com>
75 lines
2.0 KiB
TypeScript
75 lines
2.0 KiB
TypeScript
import {
|
||
AgentOverChromeBridge,
|
||
getBridgePageInCliSide,
|
||
} from '@/bridge-mode/agent-cli-side';
|
||
import { describe, expect, it, vi } from 'vitest';
|
||
|
||
vi.setConfig({
|
||
testTimeout: 60 * 1000,
|
||
});
|
||
const sleep = (ms: number) => new Promise((resolve) => setTimeout(resolve, ms));
|
||
const describeIf = process.env.BRIDGE_MODE ? describe : describe.skip;
|
||
|
||
describeIf(
|
||
'fully functional agent in server(cli) side',
|
||
{
|
||
timeout: 3 * 60 * 10,
|
||
},
|
||
() => {
|
||
it('basic', async () => {
|
||
const page = getBridgePageInCliSide();
|
||
expect(page).toBeDefined();
|
||
|
||
// server should be destroyed as well
|
||
await page.destroy();
|
||
});
|
||
|
||
it('page in cli side', async () => {
|
||
const page = getBridgePageInCliSide();
|
||
|
||
// make sure the extension bridge is launched before timeout
|
||
await page.connectNewTabWithUrl('https://www.baidu.com');
|
||
|
||
// sleep 3s
|
||
await sleep(3000);
|
||
|
||
await page.destroy();
|
||
});
|
||
|
||
it('agent in cli side, new tab', async () => {
|
||
const agent = new AgentOverChromeBridge();
|
||
|
||
await agent.connectNewTabWithUrl('https://www.bing.com');
|
||
await sleep(3000);
|
||
|
||
await agent.ai('type "AI 101" and hit Enter and scroll down');
|
||
await sleep(3000);
|
||
|
||
await agent.aiAssert('there are some search results');
|
||
await agent.destroy();
|
||
});
|
||
|
||
it('agent in cli side, current tab', async () => {
|
||
const agent = new AgentOverChromeBridge();
|
||
await agent.connectCurrentTab();
|
||
await sleep(3000);
|
||
const answer = await agent.aiQuery(
|
||
'name of the current page? return {name: string}',
|
||
);
|
||
|
||
console.log(answer);
|
||
expect(answer.name).toBeTruthy();
|
||
await agent.destroy();
|
||
});
|
||
|
||
it('agent in cli side, current tab, limit popup to current page', async () => {
|
||
const agent = new AgentOverChromeBridge();
|
||
await agent.connectCurrentTab({ forceSameTabNavigation: true });
|
||
|
||
await agent.ai('click "文库"，sleep 1500ms，type "AI 101" and hit Enter');
|
||
await sleep(3000);
|
||
await agent.destroy();
|
||
});
|
||
},
|
||
);
|