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);
  });
});