codeceptjs/lib/step/func.js

Supercharged End 2 End Testing Framework for NodeJS

const BaseStep = require('./base')
const store = require('../store')

/**
 * Function executed as a step
 */
class FuncStep extends BaseStep {
  // this is actual function that should be executed within step
  setCallable(fn) {
    this.fn = fn
  }

  // helper is optional, if we need to allow step to access helper methods
  setHelper(helper) {
    this.helper = helper
  }

  run() {
    if (!this.fn) throw new Error('Function is not set')

    // we wrap that function to track time and status
    // and disable it in dry run mode
    this.args = Array.prototype.slice.call(arguments)
    this.startTime = +Date.now()

    if (store.dryRun) {
      this.setStatus('success')
      // should we add Proxy and dry run resolver here?
      return Promise.resolve(true)
    }

    let result
    try {
      result = this.fn.apply(this.helper, this.args)
      this.setStatus('success')
      this.endTime = +Date.now()
    } catch (err) {
      this.endTime = +Date.now()
      this.setStatus('failed')
      throw err
    }
    return result
  }
}

module.exports = FuncStep