You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
62 lines
1.9 KiB
62 lines
1.9 KiB
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({ extensionPath, responsive, port }) {
|
|
const args = [`load-extension=${extensionPath}`]
|
|
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
|
|
|