From c237a36db1f7730109764db1b5d6f7516a58b891 Mon Sep 17 00:00:00 2001
From: Benjamin Coe <ben@npmjs.com>
Date: Sun, 6 Dec 2015 14:07:27 -0800
Subject: [PATCH] the combination of #73 and #66 broke tests on master

---
 index.js                 | 11 +++++++++--
 test/source-map-cache.js | 26 ++++++++++++++------------
 2 files changed, 23 insertions(+), 14 deletions(-)

diff --git a/index.js b/index.js
index f57df0af9..0e7195265 100755
--- a/index.js
+++ b/index.js
@@ -45,9 +45,16 @@ function NYC (opts) {
 }
 
 NYC.prototype._loadAdditionalModules = function () {
-  require.main.paths.push(path.resolve(this.cwd, '/node_modules'))
+  var _this = this
   this.require.forEach(function (r) {
-    require(r)
+    try {
+      // first attempt to require the module relative to
+      // the directory being instrumented.
+      require(path.resolve(_this.cwd, './node_modules', r))
+    } catch (e) {
+      // now try other locations, .e.g, the nyc node_modules folder.
+      require(r)
+    }
   })
 }
 
diff --git a/test/source-map-cache.js b/test/source-map-cache.js
index b061e40ac..7e63feb35 100644
--- a/test/source-map-cache.js
+++ b/test/source-map-cache.js
@@ -7,6 +7,8 @@ var fixture = require('source-map-fixtures').inline('branching')
 // Coverage for the fixture is stored relative to the root directory. Here
 // compute the path to the fixture file relative to the root directory.
 var relpath = './' + path.relative(path.join(__dirname, '..'), fixture.file)
+// the sourcemap itself remaps the path.
+var mappedPath = './' + path.join(path.dirname(relpath), '../src/branching.js')
 // Compute the number of lines in the original source, excluding any line break
 // at the end of the file.
 var maxLine = fixture.sourceContentSync().trimRight().split(/\r?\n/).length
@@ -24,15 +26,15 @@ describe('source-map-cache', function () {
   describe('statements', function () {
     it('drops statements that have no mapping back to the original source code', function () {
       var mappedCoverage = sourceMapCache.applySourceMaps(coverage)
-      Object.keys(mappedCoverage[relpath].s)
+      Object.keys(mappedCoverage[mappedPath].s)
         .should.be.lt(coverage[relpath].s)
-      Object.keys(mappedCoverage[relpath].statementMap).length
-        .should.equal(Object.keys(mappedCoverage[relpath].s).length)
+      Object.keys(mappedCoverage[mappedPath].statementMap).length
+        .should.equal(Object.keys(mappedCoverage[mappedPath].s).length)
     })
 
     it('maps all statements back to their original loc', function () {
       var mappedCoverage = sourceMapCache.applySourceMaps(coverage)
-      var statements = _.values(mappedCoverage[relpath].statementMap)
+      var statements = _.values(mappedCoverage[mappedPath].statementMap)
       var maxStatement = _.max(statements, function (s) {
         return Math.max(s.start.line, s.end.line)
       })
@@ -43,15 +45,15 @@ describe('source-map-cache', function () {
   describe('functions', function () {
     it('drops functions that have no mapping back to the original source code', function () {
       var mappedCoverage = sourceMapCache.applySourceMaps(coverage)
-      Object.keys(mappedCoverage[relpath].f)
+      Object.keys(mappedCoverage[mappedPath].f)
         .should.be.lt(coverage[relpath].f)
-      Object.keys(mappedCoverage[relpath].fnMap).length
-        .should.equal(Object.keys(mappedCoverage[relpath].f).length)
+      Object.keys(mappedCoverage[mappedPath].fnMap).length
+        .should.equal(Object.keys(mappedCoverage[mappedPath].f).length)
     })
 
     it('maps all functions back to their original loc', function () {
       var mappedCoverage = sourceMapCache.applySourceMaps(coverage)
-      var functions = _.values(mappedCoverage[relpath].fnMap)
+      var functions = _.values(mappedCoverage[mappedPath].fnMap)
       var maxFunction = _.max(functions, function (f) {
         return f.line
       })
@@ -62,15 +64,15 @@ describe('source-map-cache', function () {
   describe('branches', function () {
     it('drops branches that have no mapping back to the original source code', function () {
       var mappedCoverage = sourceMapCache.applySourceMaps(coverage)
-      Object.keys(mappedCoverage[relpath].b)
+      Object.keys(mappedCoverage[mappedPath].b)
         .should.be.lt(coverage[relpath].b)
-      Object.keys(mappedCoverage[relpath].branchMap).length
-        .should.equal(Object.keys(mappedCoverage[relpath].b).length)
+      Object.keys(mappedCoverage[mappedPath].branchMap).length
+        .should.equal(Object.keys(mappedCoverage[mappedPath].b).length)
     })
 
     it('maps all branches back to their original loc', function () {
       var mappedCoverage = sourceMapCache.applySourceMaps(coverage)
-      var branches = _.values(mappedCoverage[relpath].branchMap)
+      var branches = _.values(mappedCoverage[mappedPath].branchMap)
       var maxBranch = _.max(branches, function (b) {
         return b.line
       })