testplane
Version:
Tests framework based on mocha and wdio
94 lines • 4.47 kB
JavaScript
;
var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
var desc = Object.getOwnPropertyDescriptor(m, k);
if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
desc = { enumerable: true, get: function() { return m[k]; } };
}
Object.defineProperty(o, k2, desc);
}) : (function(o, m, k, k2) {
if (k2 === undefined) k2 = k;
o[k2] = m[k];
}));
var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
Object.defineProperty(o, "default", { enumerable: true, value: v });
}) : function(o, v) {
o["default"] = v;
});
var __importStar = (this && this.__importStar) || function (mod) {
if (mod && mod.__esModule) return mod;
var result = {};
if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
__setModuleDefault(result, mod);
return result;
};
Object.defineProperty(exports, "__esModule", { value: true });
exports.WebdriverPool = void 0;
const browser_1 = require("../utils/browser");
class WebdriverPool {
constructor() {
this.driverProcess = new Map();
this.portToDriverProcess = new Map();
}
async getWebdriver(browserName, browserVersion, { debug = false } = {}) {
const browserNameNormalized = (0, browser_1.getNormalizedBrowserName)(browserName);
if (!browserNameNormalized) {
throw new Error([
`Couldn't run browser driver for "${browserName}", as this browser is not supported`,
`Supported browsers: "chrome", "firefox", "safari", "MicrosoftEdge"`,
].join("\n"));
}
const { resolveBrowserVersion } = await Promise.resolve().then(() => __importStar(require("../browser-installer")));
const browserVersionNormalized = browserVersion || (await resolveBrowserVersion(browserNameNormalized));
const wdProcesses = this.driverProcess.get(browserNameNormalized)?.get(browserVersionNormalized) ?? {};
for (const port in wdProcesses) {
if (!wdProcesses[port].isBusy) {
wdProcesses[port].isBusy = true;
return {
gridUrl: wdProcesses[port].gridUrl,
free: () => this.freeWebdriver(port),
kill: () => this.killWebdriver(browserNameNormalized, browserVersionNormalized, port),
getPid: () => wdProcesses[port].process.pid,
};
}
}
return this.createWebdriverProcess(browserNameNormalized, browserVersionNormalized, { debug });
}
freeWebdriver(port) {
const wdProcess = this.portToDriverProcess.get(port);
if (wdProcess) {
wdProcess.isBusy = false;
}
}
killWebdriver(browserName, browserVersion, port) {
const wdProcess = this.portToDriverProcess.get(port);
const nodes = this.driverProcess.get(browserName)?.get(browserVersion);
if (wdProcess && nodes) {
wdProcess.process.kill();
this.portToDriverProcess.delete(port);
delete nodes[port];
}
}
async createWebdriverProcess(browserName, browserVersion, { debug = false } = {}) {
const { runBrowserDriver } = await Promise.resolve().then(() => __importStar(require("../browser-installer")));
const driver = await runBrowserDriver(browserName, browserVersion, { debug });
if (!this.driverProcess.has(browserName)) {
this.driverProcess.set(browserName, new Map());
}
if (!this.driverProcess.get(browserName)?.has(browserVersion)) {
this.driverProcess.get(browserName)?.set(browserVersion, {});
}
const nodes = this.driverProcess.get(browserName)?.get(browserVersion);
const node = { process: driver.process, gridUrl: driver.gridUrl, isBusy: true };
nodes[driver.port] = node;
this.portToDriverProcess.set(String(driver.port), node);
return {
gridUrl: driver.gridUrl,
free: () => this.freeWebdriver(String(driver.port)),
kill: () => this.killWebdriver(browserName, browserVersion, String(driver.port)),
getPid: () => driver.process.pid,
};
}
}
exports.WebdriverPool = WebdriverPool;
//# sourceMappingURL=webdriver-pool.js.map