3746273eda
* Path#ensureBlock keeps path context
This ensures that if you're inside an ArrowFunction with an expression body (say, you're on the BooleanLiteral in `() => true`), you don't suddenly lose your path context after inserting a variable.
This is because of 82d8aded8e (diff-9e0668ad44535be897b934e7077ecea5R14). Basically, an innocent `Scope#push` caused my visitor to suddenly stop working. Now, we mutate the Path so it's still in the tree.
* Tests
70 lines
2.1 KiB
JavaScript
70 lines
2.1 KiB
JavaScript
import traverse from "../lib";
|
|
import assert from "assert";
|
|
import { parse } from "babylon";
|
|
import generate from "babel-generator";
|
|
import * as t from "babel-types";
|
|
|
|
function getPath(code) {
|
|
const ast = parse(code);
|
|
let path;
|
|
traverse(ast, {
|
|
Program: function(_path) {
|
|
path = _path.get("body.0");
|
|
_path.stop();
|
|
},
|
|
});
|
|
|
|
return path;
|
|
}
|
|
|
|
function generateCode(path) {
|
|
return generate(path.parentPath.node).code;
|
|
}
|
|
|
|
describe("conversion", function() {
|
|
describe("ensureBlock", function() {
|
|
it("throws converting node without body to block", function() {
|
|
const rootPath = getPath("true;");
|
|
|
|
assert.throws(() => {
|
|
rootPath.ensureBlock();
|
|
}, /Can't convert node without a body/);
|
|
});
|
|
|
|
it("throws converting already block array", function() {
|
|
const rootPath = getPath("function test() { true; }").get("body");
|
|
assert.throws(() => {
|
|
rootPath.ensureBlock();
|
|
}, /Can't convert array path to a block statement/);
|
|
});
|
|
|
|
it("converts arrow function with expression body to block", function() {
|
|
const rootPath = getPath("() => true").get("expression");
|
|
rootPath.ensureBlock();
|
|
assert.equal(generateCode(rootPath), "() => {\n return true;\n};");
|
|
});
|
|
|
|
it("preserves arrow function body's context", function() {
|
|
const rootPath = getPath("() => true").get("expression");
|
|
const body = rootPath.get("body");
|
|
rootPath.ensureBlock();
|
|
body.replaceWith(t.booleanLiteral(false));
|
|
assert.equal(generateCode(rootPath), "() => {\n return false;\n};");
|
|
});
|
|
|
|
it("converts for loop with statement body to block", function() {
|
|
const rootPath = getPath("for (;;) true;");
|
|
rootPath.ensureBlock();
|
|
assert.equal(generateCode(rootPath), "for (;;) {\n true;\n}");
|
|
});
|
|
|
|
it("preserves for loop body's context", function() {
|
|
const rootPath = getPath("for (;;) true;");
|
|
const body = rootPath.get("body");
|
|
rootPath.ensureBlock();
|
|
body.replaceWith(t.booleanLiteral(false));
|
|
assert.equal(generateCode(rootPath), "for (;;) {\n false;\n}");
|
|
});
|
|
});
|
|
});
|