37 lines
761 B
JavaScript
37 lines
761 B
JavaScript
import traverse from "../lib";
|
|
import { parse } from "@babel/parser";
|
|
import generate from "@babel/generator";
|
|
|
|
function getPath(code) {
|
|
const ast = parse(code);
|
|
let path;
|
|
traverse(ast, {
|
|
Program: function(_path) {
|
|
path = _path;
|
|
_path.stop();
|
|
},
|
|
});
|
|
|
|
return path;
|
|
}
|
|
|
|
function generateCode(path) {
|
|
return generate(path.node).code;
|
|
}
|
|
|
|
describe("removal", function() {
|
|
describe("ArrowFunction", function() {
|
|
it("remove body", function() {
|
|
const rootPath = getPath("x = () => b;");
|
|
const path = rootPath
|
|
.get("body")[0]
|
|
.get("expression")
|
|
.get("right");
|
|
const body = path.get("body");
|
|
body.remove();
|
|
|
|
expect(generateCode(rootPath)).toBe("x = () => {};");
|
|
});
|
|
});
|
|
});
|