codeceptjs/lib/assert/equal.js

Modern Era Acceptance Testing Framework for NodeJS

const Assertion = require('../assert');
const AssertionFailedError = require('./error');
const template = require('../utils').template;
const output = require('../output');

class EqualityAssertion extends Assertion {
  constructor(params) {
    const comparator = function (a, b) {
      if (b.length === 0) {
        b = '';
      }
      return a === b;
    };
    super(comparator, params);
    this.params.type = 'to equal';
  }

  getException() {
    const params = this.params;
    params.jar = template(params.jar, params);
    const err = new AssertionFailedError(params, '{{customMessage}}expected {{jar}} "{{expected}}" {{type}} "{{actual}}"');
    err.showDiff = false;
    if (typeof err.cliMessage === 'function') {
      err.message = err.cliMessage();
    }
    err.cliMessage = () => {
      const msg = err.template
        .replace('{{jar}}', output.colors.bold('{{jar}}'));
      return template(msg, this.params);
    };
    return err;
  }

  addAssertParams() {
    this.params.expected = arguments[0];
    this.params.actual = arguments[1];
    this.params.customMessage = arguments[2] ? `${arguments[2]}\n\n` : '';
  }
}

module.exports = {
  Assertion: EqualityAssertion,
  equals: jar => new EqualityAssertion({ jar }),
  urlEquals: (baseUrl) => {
    const assert = new EqualityAssertion({ jar: 'url of current page' });
    assert.comparator = function (expected, actual) {
      if (expected.indexOf('http') !== 0) {
        actual = actual.slice(actual.indexOf(baseUrl) + baseUrl.length);
      }
      return actual === expected;
    };
    return assert;
  },
  fileEquals: file => new EqualityAssertion({ file, jar: 'contents of {{file}}' }),
};