From 148067da21a52a81a71072505016837846691e50 Mon Sep 17 00:00:00 2001 From: Jason Laster Date: Fri, 10 Feb 2017 18:23:07 -0500 Subject: [PATCH] Add test for reference paths --- packages/babel-traverse/test/scope.js | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/packages/babel-traverse/test/scope.js b/packages/babel-traverse/test/scope.js index db51118fcfe8..9c844ec6b888 100644 --- a/packages/babel-traverse/test/scope.js +++ b/packages/babel-traverse/test/scope.js @@ -14,6 +14,19 @@ function getPath(code) { return path; } +function getIdentifierPath(code) { + const ast = parse(code); + let nodePath; + traverse(ast, { + Identifier: function(path) { + nodePath = path; + path.stop(); + } + }); + + return nodePath; +} + describe("scope", function () { describe("binding paths", function () { it("function declaration id", function () { @@ -60,5 +73,13 @@ describe("scope", function () { _foo2: { } `).scope.generateUid("foo"), "_foo3"); }); + + it("reference paths", function() { + const node = getIdentifierPath("function square(n) { return n * n}"); + const referencePaths = node.context.scope.bindings.n.referencePaths; + assert.equal(referencePaths.length, 2); + assert.deepEqual(referencePaths[0].node.loc.start, { line: 1, column:28 }); + assert.deepEqual(referencePaths[1].node.loc.start, { line: 1, column:32 }); + }); }); });