2017-01-14 09:48:52 -05:00

65 lines
2.2 KiB
JavaScript

const traverse = require("../lib").default;
const assert = require("assert");
const parse = require("babylon").parse;
function getPath(code) {
const ast = parse(code);
let path;
traverse(ast, {
Program: function (_path) {
path = _path;
_path.stop();
}
});
return path;
}
describe("scope", function () {
describe("binding paths", function () {
it("function declaration id", function () {
assert.ok(getPath("function foo() {}").scope.getBinding("foo").path.type === "FunctionDeclaration");
});
it("function expression id", function () {
assert.ok(getPath("(function foo() {})").get("body")[0].get("expression").scope.getBinding("foo").path.type === "FunctionExpression");
});
it("function param", function () {
assert.ok(getPath("(function (foo) {})").get("body")[0].get("expression").scope.getBinding("foo").path.type === "Identifier");
});
it("variable declaration", function () {
assert.ok(getPath("var foo = null;").scope.getBinding("foo").path.type === "VariableDeclarator");
assert.ok(getPath("var { foo } = null;").scope.getBinding("foo").path.type === "VariableDeclarator");
assert.ok(getPath("var [ foo ] = null;").scope.getBinding("foo").path.type === "VariableDeclarator");
assert.ok(getPath("var { bar: [ foo ] } = null;").scope.getBinding("foo").path.type === "VariableDeclarator");
});
it("purity", function () {
assert.ok(getPath("({ x: 1 })").get("body")[0].get("expression").isPure());
});
test("label", function () {
assert.strictEqual(getPath("foo: { }").scope.getBinding("foo"), undefined);
assert.strictEqual(getPath("foo: { }").scope.getLabel("foo").type, "LabeledStatement");
assert.strictEqual(getPath("foo: { }").scope.getLabel("toString"), undefined);
assert.strictEqual(getPath(`
foo: { }
`).scope.generateUid("foo"), "_foo");
});
test("generateUid collision check with labels", function () {
assert.strictEqual(getPath(`
_foo: { }
`).scope.generateUid("foo"), "_foo2");
assert.strictEqual(getPath(`
_foo: { }
_foo1: { }
_foo2: { }
`).scope.generateUid("foo"), "_foo3");
});
});
});