2015-10-31 20:52:36 +00:00

23 lines
644 B
JavaScript

var transform = require("../lib/api/node").transform;
var Plugin = require("../lib/transformation/plugin");
var babel = require("../lib/api/node");
var chai = require("chai");
suite("traversal path", function () {
test("replaceWithSourceString", function () {
var expectCode = "function foo() {}";
var actualCode = transform(expectCode, {
plugins: [new Plugin({
visitor: {
FunctionDeclaration: function (path) {
path.replaceWithSourceString("console.whatever()");
}
}
})]
}).code;
chai.expect(actualCode).to.be.equal("console.whatever();");
});
});