1
0
mirror of https://github.com/kremalicious/metamask-extension.git synced 2024-12-23 09:52:26 +01:00
metamask-extension/test/e2e/webdriver/chrome.js
Mark Stacey d8ec5f19f6
Use .zip for Firefox e2e tests (#10056)
The Firefox e2e tests now use the `.zip` file for testing the
extension. We've found this to produce more similar results to
production, compared to the old method of loading the unzipped
directory.

Passing in a `.zip` file to the Chrome driver didn't seem to work. I
didn't investigate this further to see if it was possible, but I'm not
sure it makes a difference on Chrome anyway.
2020-12-11 12:54:17 -03:30

63 lines
1.9 KiB
JavaScript

const { Builder } = require('selenium-webdriver')
const chrome = require('selenium-webdriver/chrome')
/**
* 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')
}
const options = new chrome.Options().addArguments(args)
const builder = new Builder().forBrowser('chrome').setChromeOptions(options)
if (port) {
const service = new chrome.ServiceBuilder().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}`,
}
}
/**
* @constructor
* @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 === "${extensionName}") {
return extensions[i].getAttribute("id")
}
}
return undefined
`)
}
}
module.exports = ChromeDriver