aboutsummaryrefslogblamecommitdiffstats
path: root/test/e2e/func.js
blob: 13dfb82f942ee463adaaca0297531e776f621146 (plain) (tree)
1
2
3
4
5
6
7
8
9
10
                       
                      
                              
                        
                            

                                                 
                                               

                                                                 
 

                  
                          
                           
                         




                        

 


                                                          
 











                                                                                                           
                                       



                                                    
                                                                




                                                     
                                                             






                                                   
                                         
                                                                                


                                




                                                   

        

            
 
                                   

                                        





















                                                                                                                                                                                                                                                            
 









                                                                                          
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.By

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.schedule(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)
}