Move playwright driver impl to class

Part of #136064
This commit is contained in:
Daniel Imms 2021-11-09 08:45:08 -08:00
parent 28da3051b3
commit 628f486b1d

View file

@ -11,6 +11,7 @@ import { promisify } from 'util';
import { IDriver, IDisposable } from './driver';
import { URI } from 'vscode-uri';
import * as kill from 'tree-kill';
import { IElement, ILocaleInfo, ILocalizedStrings } from '.';
const width = 1200;
const height = 800;
@ -35,69 +36,96 @@ const vscodeToPlaywrightKey: { [key: string]: string } = {
let traceCounter = 1;
function buildDriver(browser: playwright.Browser, context: playwright.BrowserContext, page: playwright.Page): IDriver {
const driver: IDriver = {
_serviceBrand: undefined,
getWindowIds: () => {
return Promise.resolve([1]);
},
capturePage: () => Promise.resolve(''),
reloadWindow: (windowId) => Promise.resolve(),
exitApplication: async () => {
try {
await context.tracing.stop({ path: join(logsPath, `playwright-trace-${traceCounter++}.zip`) });
} catch (error) {
console.warn(`Failed to stop playwright tracing.`); // do not fail the build when this fails
}
await browser.close();
await teardown();
return new PlaywrightDriver(browser, context, page);
}
return false;
},
dispatchKeybinding: async (windowId, keybinding) => {
const chords = keybinding.split(' ');
for (let i = 0; i < chords.length; i++) {
const chord = chords[i];
if (i > 0) {
await timeout(100);
}
const keys = chord.split('+');
const keysDown: string[] = [];
for (let i = 0; i < keys.length; i++) {
if (keys[i] in vscodeToPlaywrightKey) {
keys[i] = vscodeToPlaywrightKey[keys[i]];
}
await page.keyboard.down(keys[i]);
keysDown.push(keys[i]);
}
while (keysDown.length > 0) {
await page.keyboard.up(keysDown.pop()!);
}
}
class PlaywrightDriver implements IDriver {
_serviceBrand: undefined;
await timeout(100);
},
click: async (windowId, selector, xoffset, yoffset) => {
const { x, y } = await driver.getElementXY(windowId, selector, xoffset, yoffset);
await page.mouse.click(x + (xoffset ? xoffset : 0), y + (yoffset ? yoffset : 0));
},
doubleClick: async (windowId, selector) => {
await driver.click(windowId, selector, 0, 0);
await timeout(60);
await driver.click(windowId, selector, 0, 0);
await timeout(100);
},
setValue: async (windowId, selector, text) => page.evaluate(`window.driver.setValue('${selector}', '${text}')`).then(undefined),
getTitle: (windowId) => page.evaluate(`window.driver.getTitle()`),
isActiveElement: (windowId, selector) => page.evaluate(`window.driver.isActiveElement('${selector}')`),
getElements: (windowId, selector, recursive) => page.evaluate(`window.driver.getElements('${selector}', ${recursive})`),
getElementXY: (windowId, selector, xoffset?, yoffset?) => page.evaluate(`window.driver.getElementXY('${selector}', ${xoffset}, ${yoffset})`),
typeInEditor: (windowId, selector, text) => page.evaluate(`window.driver.typeInEditor('${selector}', '${text}')`),
getTerminalBuffer: (windowId, selector) => page.evaluate(`window.driver.getTerminalBuffer('${selector}')`),
writeInTerminal: (windowId, selector, text) => page.evaluate(`window.driver.writeInTerminal('${selector}', '${text}')`),
getLocaleInfo: (windowId) => page.evaluate(`window.driver.getLocaleInfo()`),
getLocalizedStrings: (windowId) => page.evaluate(`window.driver.getLocalizedStrings()`)
};
return driver;
constructor(
private readonly _browser: playwright.Browser,
private readonly _context: playwright.BrowserContext,
private readonly _page: playwright.Page
) {
}
async getWindowIds() { return [1]; }
async capturePage() { return ''; }
async reloadWindow(windowId: number) { }
async exitApplication() {
try {
await this._context.tracing.stop({ path: join(logsPath, `playwright-trace-${traceCounter++}.zip`) });
} catch (error) {
console.warn(`Failed to stop playwright tracing.`); // do not fail the build when this fails
}
await this._browser.close();
await teardown();
return false;
}
async dispatchKeybinding(windowId: number, keybinding: string) {
const chords = keybinding.split(' ');
for (let i = 0; i < chords.length; i++) {
const chord = chords[i];
if (i > 0) {
await timeout(100);
}
const keys = chord.split('+');
const keysDown: string[] = [];
for (let i = 0; i < keys.length; i++) {
if (keys[i] in vscodeToPlaywrightKey) {
keys[i] = vscodeToPlaywrightKey[keys[i]];
}
await this._page.keyboard.down(keys[i]);
keysDown.push(keys[i]);
}
while (keysDown.length > 0) {
await this._page.keyboard.up(keysDown.pop()!);
}
}
await timeout(100);
}
async click(windowId: number, selector: string, xoffset?: number | undefined, yoffset?: number | undefined) {
const { x, y } = await this.getElementXY(windowId, selector, xoffset, yoffset);
await this._page.mouse.click(x + (xoffset ? xoffset : 0), y + (yoffset ? yoffset : 0));
}
async doubleClick(windowId: number, selector: string) {
await this.click(windowId, selector, 0, 0);
await timeout(60);
await this.click(windowId, selector, 0, 0);
await timeout(100);
}
async setValue(windowId: number, selector: string, text: string) {
await this._page.evaluate(`window.driver.setValue('${selector}', '${text}')`);
}
async getTitle(windowId: number) {
return this._page.evaluate<string>(`window.driver.getTitle()`);
}
isActiveElement(windowId: number, selector: string) {
return this._page.evaluate<boolean>(`window.driver.isActiveElement('${selector}')`);
}
getElements(windowId: number, selector: string, recursive?: boolean) {
return this._page.evaluate<IElement[]>(`window.driver.getElements('${selector}', ${recursive})`);
}
getElementXY(windowId: number, selector: string, xoffset?: number, yoffset?: number) {
return this._page.evaluate<{ x: number, y: number }>(`window.driver.getElementXY('${selector}', ${xoffset}, ${yoffset})`);
}
async typeInEditor(windowId: number, selector: string, text: string) {
await this._page.evaluate(`window.driver.typeInEditor('${selector}', '${text}')`);
}
getTerminalBuffer(windowId: number, selector: string) {
return this._page.evaluate<string[]>(`window.driver.getTerminalBuffer('${selector}')`);
}
async writeInTerminal(windowId: number, selector: string, text: string) {
await this._page.evaluate(`window.driver.writeInTerminal('${selector}', '${text}')`);
}
getLocaleInfo(windowId: number) {
return this._page.evaluate<ILocaleInfo>(`window.driver.getLocaleInfo()`);
}
getLocalizedStrings(windowId: number) {
return this._page.evaluate<ILocalizedStrings>(`window.driver.getLocalizedStrings()`);
}
}
function timeout(ms: number): Promise<void> {