111 lines
3.7 KiB
JavaScript
111 lines
3.7 KiB
JavaScript
require('chromedriver')
|
|
require('geckodriver')
|
|
const fs = require('fs-extra')
|
|
const os = require('os')
|
|
const path = require('path')
|
|
const pify = require('pify')
|
|
const prependFile = pify(require('prepend-file'))
|
|
const webdriver = require('selenium-webdriver')
|
|
const Command = require('selenium-webdriver/lib/command').Command
|
|
|
|
const { By } = webdriver
|
|
|
|
module.exports = {
|
|
delay,
|
|
createModifiedTestBuild,
|
|
setupBrowserAndExtension,
|
|
verboseReportOnFailure,
|
|
buildChromeWebDriver,
|
|
buildFirefoxWebdriver,
|
|
installWebExt,
|
|
getExtensionIdChrome,
|
|
getExtensionIdFirefox,
|
|
}
|
|
|
|
function delay (time) {
|
|
return new Promise(resolve => setTimeout(resolve, time))
|
|
}
|
|
|
|
async function createModifiedTestBuild ({ browser, srcPath }) {
|
|
// copy build to test-builds directory
|
|
const extPath = path.resolve(`test-builds/${browser}`)
|
|
await fs.ensureDir(extPath)
|
|
await fs.copy(srcPath, extPath)
|
|
// inject METAMASK_TEST_CONFIG setting default test network
|
|
const config = { NetworkController: { provider: { type: 'localhost' } } }
|
|
await prependFile(`${extPath}/background.js`, `window.METAMASK_TEST_CONFIG=${JSON.stringify(config)};\n`)
|
|
return { extPath }
|
|
}
|
|
|
|
async function setupBrowserAndExtension ({ browser, extPath }) {
|
|
let driver, extensionId, extensionUri
|
|
|
|
if (browser === 'chrome') {
|
|
driver = buildChromeWebDriver(extPath)
|
|
extensionId = await getExtensionIdChrome(driver)
|
|
extensionUri = `chrome-extension://${extensionId}/home.html`
|
|
} else if (browser === 'firefox') {
|
|
driver = buildFirefoxWebdriver()
|
|
await installWebExt(driver, extPath)
|
|
await delay(700)
|
|
extensionId = await getExtensionIdFirefox(driver)
|
|
extensionUri = `moz-extension://${extensionId}/home.html`
|
|
} else {
|
|
throw new Error(`Unknown Browser "${browser}"`)
|
|
}
|
|
|
|
return { driver, extensionId, extensionUri }
|
|
}
|
|
|
|
function buildChromeWebDriver (extPath) {
|
|
const tmpProfile = fs.mkdtempSync(path.join(os.tmpdir(), 'mm-chrome-profile'))
|
|
return new webdriver.Builder()
|
|
.withCapabilities({
|
|
chromeOptions: {
|
|
args: [
|
|
`load-extension=${extPath}`,
|
|
`user-data-dir=${tmpProfile}`,
|
|
],
|
|
binary: process.env.SELENIUM_CHROME_BINARY,
|
|
},
|
|
})
|
|
.build()
|
|
}
|
|
|
|
function buildFirefoxWebdriver () {
|
|
return new webdriver.Builder().build()
|
|
}
|
|
|
|
async function getExtensionIdChrome (driver) {
|
|
await driver.get('chrome://extensions')
|
|
const extensionId = await driver.executeScript('return document.querySelector("extensions-manager").shadowRoot.querySelector("extensions-view-manager extensions-item-list").shadowRoot.querySelector("extensions-item:nth-child(2)").getAttribute("id")')
|
|
return extensionId
|
|
}
|
|
|
|
async function getExtensionIdFirefox (driver) {
|
|
await driver.get('about:debugging#addons')
|
|
const extensionId = await driver.findElement(By.css('dd.addon-target-info-content:nth-child(6) > span:nth-child(1)')).getText()
|
|
return extensionId
|
|
}
|
|
|
|
async function installWebExt (driver, extension) {
|
|
const cmd = await new Command('moz-install-web-ext')
|
|
.setParameter('path', path.resolve(extension))
|
|
.setParameter('temporary', true)
|
|
|
|
await driver.getExecutor()
|
|
.defineCommand(cmd.getName(), 'POST', '/session/:sessionId/moz/addon/install')
|
|
|
|
return await driver.execute(cmd, 'installWebExt(' + extension + ')')
|
|
}
|
|
|
|
async function verboseReportOnFailure ({ browser, driver, title }) {
|
|
const artifactDir = `./test-artifacts/${browser}/${title}`
|
|
const filepathBase = `${artifactDir}/test-failure`
|
|
await fs.ensureDir(artifactDir)
|
|
const screenshot = await driver.takeScreenshot()
|
|
await fs.writeFile(`${filepathBase}-screenshot.png`, screenshot, { encoding: 'base64' })
|
|
const htmlSource = await driver.getPageSource()
|
|
await fs.writeFile(`${filepathBase}-dom.html`, htmlSource)
|
|
}
|