Ben Briggs c94abcc170 Add support for preserving comments in babel-template. (#3689)
* Add support for preserving comments in babel-template.

* Add an API section to babel-template.
2016-09-01 10:50:25 -04:00

31 lines
1023 B
JavaScript

var generator = require('../../babel-generator').default;
var template = require("../lib");
var chai = require("chai");
var comments = "// Sum two numbers\nconst add = (a, b) => a + b;";
suite("templating", function () {
test("import statement will cause parser to throw by default", function () {
chai.expect(function () {
template("import foo from 'foo'")({});
}).to.throw();
});
test("import statements are allowed with sourceType: module", function () {
chai.expect(function () {
template("import foo from 'foo'", {sourceType: 'module'})({});
}).not.to.throw();
});
test("should strip comments by default", function () {
var code = "const add = (a, b) => a + b;"
var output = template(comments)();
chai.expect(generator(output).code).to.be.equal(code);
});
test("should preserve comments with a flag", function () {
var output = template(comments, {preserveComments: true})();
chai.expect(generator(output).code).to.be.equal(comments);
});
});