import generator from "../../babel-generator";
import template from "../lib";
import chai from "chai";

const comments = "// Sum two numbers\nconst add = (a, b) => a + b;";

describe("templating", function() {
  it("import statement will cause parser to throw by default", function() {
    chai
      .expect(function() {
        template("import foo from 'foo'")({});
      })
      .to.throw();
  });

  it("import statements are allowed with sourceType: module", function() {
    chai
      .expect(function() {
        template("import foo from 'foo'", { sourceType: "module" })({});
      })
      .not.to.throw();
  });

  it("should strip comments by default", function() {
    const code = "const add = (a, b) => a + b;";
    const output = template(comments)();
    chai.expect(generator(output).code).to.be.equal(code);
  });

  it("should preserve comments with a flag", function() {
    const output = template(comments, { preserveComments: true })();
    chai.expect(generator(output).code).to.be.equal(comments);
  });
});