babel/test/generation.js
2014-11-04 08:03:52 +11:00

37 lines
1.2 KiB
JavaScript

var generate = require("../lib/6to5/generator");
var traverse = require("../lib/6to5/traverse");
var assert = require("assert");
var helper = require("./_helper");
var util = require("../lib/6to5/util");
var chai = require("chai");
var _ = require("lodash");
suite("generation", function () {
test("completeness", function () {
_.each(traverse.VISITOR_KEYS, function (keys, type) {
assert.ok(!!generate.CodeGenerator.prototype[type], type + " should exist");
});
_.each(generate.CodeGenerator.prototype, function (fn, type) {
if (!/[A-Z]/.test(type[0])) return;
assert.ok(traverse.VISITOR_KEYS[type], type + " should not exist");
});
});
});
_.each(helper.get("generation"), function (testSuite) {
suite("generation/" + testSuite.title, function () {
_.each(testSuite.tests, function (task) {
test(task.title, function () {
var expect = task.expect;
var actual = task.actual;
var actualAst = util.parseNoProperties(actual.loc, actual.code);
var actualCode = generate(actual, actualAst).code;
chai.expect(actualCode).to.equal(expect.code, actual.loc + " !== " + expect.loc);
});
});
});
});