forked from babel/babel
/
family.js
116 lines (100 loc) · 3.72 KB
/
family.js
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import traverse from "../lib";
import { parse } from "@babel/parser";
import * as t from "@babel/types";
describe("path/family", function () {
describe("getBindingIdentifiers", function () {
const ast = parse("var a = 1, {b} = c, [d] = e; function f() {}");
let nodes = {},
paths = {},
outerNodes = {},
outerPaths = {};
traverse(ast, {
VariableDeclaration(path) {
nodes = path.getBindingIdentifiers();
paths = path.getBindingIdentifierPaths();
},
FunctionDeclaration(path) {
outerNodes = path.getOuterBindingIdentifiers();
outerPaths = path.getOuterBindingIdentifierPaths();
},
});
it("should contain keys of nodes in paths", function () {
Object.keys(nodes).forEach(id => {
expect(hop(paths, id)).toBe(true);
});
});
it("should contain outer bindings", function () {
Object.keys(outerNodes).forEach(id => {
expect(hop(outerPaths, id)).toBe(true);
});
});
it("should return paths", function () {
Object.keys(paths).forEach(id => {
expect(paths[id].node).toBeTruthy();
expect(paths[id].type).toBe(paths[id].node.type);
});
Object.keys(outerPaths).forEach(id => {
expect(outerPaths[id].node).toBeTruthy();
expect(outerPaths[id].type).toBe(outerPaths[id].node.type);
});
});
it("should match paths and nodes returned for the same ast", function () {
Object.keys(nodes).forEach(id => {
expect(nodes[id]).toBe(paths[id].node);
});
});
it("should match paths and nodes returned for outer Bindings", function () {
Object.keys(outerNodes).forEach(id => {
expect(outerNodes[id]).toBe(outerPaths[id].node);
});
});
});
describe("getSibling", function () {
const ast = parse(
"var a = 1, {b} = c, [d] = e; function f() {} function g() {}",
);
let sibling = {},
lastSibling = {};
traverse(ast, {
VariableDeclaration(path) {
sibling = path.getSibling(path.key);
lastSibling = sibling.getNextSibling().getNextSibling();
},
});
it("should return traverse sibling nodes", function () {
expect(sibling.getNextSibling().node).toBeTruthy();
expect(lastSibling.getPrevSibling().node).toBeTruthy();
expect(sibling.getPrevSibling().node).toBeFalsy();
expect(lastSibling.getNextSibling().node).toBeFalsy();
});
it("should return all preceding and succeeding sibling nodes", function () {
expect(sibling.getAllNextSiblings().length).toBeTruthy();
expect(lastSibling.getAllPrevSiblings().length).toBeTruthy();
expect(sibling.getAllNextSiblings()).toHaveLength(2);
expect(lastSibling.getAllPrevSiblings()).toHaveLength(2);
});
it("should initialize path.scope when needed", function () {
const ast = parse("if (0) {}");
let testHasScope = false;
let consequentHasScope = false;
traverse(ast, {
IfStatement(path) {
// @babel/traverse pre-traverses the whole tree to populate the initial
// scope. Thus, it pre-caches paths for all the original nodes.
// We need to introduce two new nodes to avoid using the cached paths
// that already have the path.scope property.
path.set("test", t.identifier("a"));
path.set("consequent", t.expressionStatement(t.identifier("b")));
const testPath = path.get("test");
testHasScope = !!testPath.scope;
consequentHasScope = !!testPath.getSibling("consequent").scope;
},
});
expect(testHasScope).toBe(true);
expect(consequentHasScope).toBe(true);
});
});
});
function hop(o, key) {
return Object.hasOwnProperty.call(o, key);
}