test-generators.ts

import escape from 'js-string-escape';

interface TestGenerator {
  suiteHeader(suiteName: string): string
  suiteFooter(): string
  test(testName: string, passed: boolean, assertionMessage: string): string
}

class QUnitTestGenerator implements TestGenerator {
  suiteHeader(suiteName) {
    return `QUnit.module('${escape(suiteName)}');\n`;
  }

  suiteFooter() {
    return '';
  }

  test(testName, passed, assertionMessage) {
    return (
      `QUnit.test('${escape(testName)}', function(assert) {\n` +
      `  assert.expect(1);\n` +
      `  assert.ok(${passed}, '${escape(assertionMessage)}');\n` +
      `});\n`
    );
  }
}

class MochaTestGenerator implements TestGenerator {
  suiteHeader(suiteName) {
    return `describe('${escape(suiteName)}', function() {\n`;
  }

  suiteFooter() {
    return '});\n';
  }

  test(testName, passed, assertionMessage) {
    return (
      `  it('${escape(testName)}', function() {\n` +
      this.assertion(passed, assertionMessage) +
      `  });\n`
    );
  }

  assertion(passed, assertionMessage) {
    if (passed) {
      return `    // test passed\n`;
    }

    return (
      `    // test failed\n`+
      `    var error = new chai.AssertionError('${escape(assertionMessage)}');\n` +
      `    error.stack = undefined;\n` +
      `    throw error;\n`
    );
  }
}

let qunit = new QUnitTestGenerator();
let mocha = new MochaTestGenerator();

let generators: { [name: string]: TestGenerator } = { qunit, mocha };

export default generators;