1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-23 01:39:44 +01:00
metamask-extension/test/e2e/webdriver/chrome.js
Pedro Figueiredo bfbe10ba28
e2e test for action metrics (#18347)
* e2e test for action metrics

* add tests to implementing sw restart delay tracking

* wip

* wip

* wip

* wip

* remove unneeded code

* remove scuttling changes contemplated in #18989
2023-05-04 10:38:09 +04:00

94 lines
2.9 KiB
JavaScript

const { Builder } = require('selenium-webdriver');
const chrome = require('selenium-webdriver/chrome');
const proxy = require('selenium-webdriver/proxy');
/**
* Proxy host to use for HTTPS requests
*
* @type {string}
*/
const HTTPS_PROXY_HOST = '127.0.0.1:8000';
/**
* A wrapper around a {@code WebDriver} instance exposing Chrome-specific functionality
*/
class ChromeDriver {
static async build({ openDevToolsForTabs, port }) {
const args = [`load-extension=dist/chrome`];
if (openDevToolsForTabs) {
args.push('--auto-open-devtools-for-tabs');
}
if (process.env.ENABLE_MV3) {
args.push('--log-level=0');
args.push('--enable-logging');
args.push(`--user-data-dir=${process.cwd()}/test-artifacts/chrome`);
} else {
args.push('--log-level=3');
}
const options = new chrome.Options().addArguments(args);
options.setProxy(proxy.manual({ https: HTTPS_PROXY_HOST }));
options.setAcceptInsecureCerts(true);
options.setUserPreferences({
'download.default_directory': `${process.cwd()}/test-artifacts/downloads`,
});
const builder = new Builder()
.forBrowser('chrome')
.setChromeOptions(options);
const service = new chrome.ServiceBuilder();
// Enables Chrome logging. Default: enabled
// Especially useful for discovering why Chrome has crashed, but can also
// be useful for revealing console errors (from the page or background).
if (process.env.ENABLE_CHROME_LOGGING !== 'false') {
service.setStdio('inherit').enableChromeLogging();
}
if (port) {
service.setPort(port);
}
builder.setChromeService(service);
const driver = builder.build();
const chromeDriver = new ChromeDriver(driver);
const extensionId = await chromeDriver.getExtensionIdByName('MetaMask');
return {
driver,
extensionUrl: `chrome-extension://${extensionId}`,
};
}
/**
* @param {!ThenableWebDriver} driver - a {@code WebDriver} instance
*/
constructor(driver) {
this._driver = driver;
}
/**
* Returns the extension ID for the given extension name
*
* @param {string} extensionName - the extension name
* @returns {Promise<string|undefined>} the extension ID
*/
async getExtensionIdByName(extensionName) {
await this._driver.get('chrome://extensions');
return await this._driver.executeScript(`
const extensions = document.querySelector("extensions-manager").shadowRoot
.querySelector("extensions-item-list").shadowRoot
.querySelectorAll("extensions-item")
for (let i = 0; i < extensions.length; i++) {
const extension = extensions[i].shadowRoot
const name = extension.querySelector('#name').textContent
if (name.startsWith("${extensionName}")) {
return extensions[i].getAttribute("id")
}
}
return undefined
`);
}
}
module.exports = ChromeDriver;