Automation test runner in poi.js app with puppeteer

test-runner.js

const path = require('path')
const connect = require('connect')
const superstatic = require('superstatic')
const puppeteer = require('puppeteer')
const poi = require('poi')
const poiConfig = require('../../poi.config')

const rootDir = path.join(__dirname, '../..')

const poiApp = poi({
  mode: 'test',
  entry: poiConfig.entry,
  html: poiConfig.html,
  autoprefixer: poiConfig.autoprefixer,
})

const runner = (port = 4444) => {
  const app = connect().use(
    superstatic({
      public: path.join(rootDir, 'dist'),
    })
  )

  const server = app.listen(port)
  server.addListener('listening', () => {
    puppeteer.launch().then(browser => {
      browser
        .newPage()
        .then(page => {
          return page.goto(`http://localhost:${port}`).then(() => {
            return page
              .waitForSelector('#login-form', { visible: true })
              .then(() => {
                return page.screenshot({ path: 'page.png' })
              })
          })
        })
        .then(() => browser.close())
        .then(() => server.close())
    })
  })
}

poiApp
  .prepare()
  .then(() => poiApp.build())
  .then(build => {
    runner()
  })