From ff2c24eed28239f28f7b1df2980a67b643168970 Mon Sep 17 00:00:00 2001 From: Jason Laster Date: Wed, 15 Feb 2017 12:05:31 -0800 Subject: [PATCH] Add test for reference paths (#5296) --- 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 7f823ecfd6..ac6e251c26 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 () { @@ -67,5 +80,13 @@ describe("scope", function () { _foo2: { } `).scope.generateUid("foo"), "_foo3"); }); + + it("reference paths", function() { + const path = getIdentifierPath("function square(n) { return n * n}"); + const referencePaths = path.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 }); + }); }); });