1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-04 23:14:56 +01:00
metamask-extension/test/e2e/webdriver/chrome.js

94 lines
2.8 KiB
JavaScript
Raw Normal View History

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({ responsive, port }) {
const args = [`load-extension=dist/chrome`];
if (responsive) {
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
*/
2020-11-03 00:41:28 +01:00
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
*/
2020-11-03 00:41:28 +01:00
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;