From aebf24aeb8c7297876422582b82eeb5f6d1bf4f8 Mon Sep 17 00:00:00 2001
From: Jason Laster <jason.laster.11@gmail.com>
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..e873691f7a55 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() {
+      var node = getIdentifierPath("function square(n) { return n * n}");
+      var 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 });
+    })
   });
 });