feat: Add --set-node-options option

This completely bypasses spawn-wrap and is enabled by default.

bin/nyc.js

@@ -4,9 +4,7 @@ const configUtil = require('../lib/config-util')
 const foreground = require('foreground-child')
 const NYC = require('../index.js')
 const processArgs = require('../lib/process-args')
-
-const sw = require('spawn-wrap')
-const wrapper = require.resolve('./wrap.js')
+const resolveFrom = require('resolve-from')
 
 // parse configuration and command-line arguments;
 // we keep these values in a few different forms,
@@ -34,12 +32,8 @@ if ([
   } else {
     nyc.createTempDirectory()
   }
-  if (argv.all) nyc.addAllFiles()
 
-  var env = {
-    // Support running nyc as a user without HOME (e.g. linux 'nobody'),
-    // https://github.com/istanbuljs/nyc/issues/951
-    SPAWN_WRAP_SHIM_ROOT: process.env.SPAWN_WRAP_SHIM_ROOT || process.env.XDG_CACHE_HOME || require('os').homedir(),
+  const env = {
     NYC_CONFIG: JSON.stringify(argv),
     NYC_CWD: process.cwd()
   }
@@ -50,7 +44,29 @@ if ([
     env.BABEL_DISABLE_CACHE = process.env.BABEL_DISABLE_CACHE = '1'
   }
 
-  sw([wrapper], env)
+  if (argv.setNodeOptions) {
+    const { preloadAppend, propagateEnv } = require('node-preload')
+
+    nyc.require.forEach(requireModule => {
+      const mod = resolveFrom.silent(nyc.cwd, requireModule) || requireModule
+      preloadAppend(mod)
+      require(mod)
+    })
+    preloadAppend(require.resolve('../lib/wrap.js'))
+    Object.assign(propagateEnv, env)
+  }
+
+  if (argv.all) nyc.addAllFiles()
+
+  if (!argv.setNodeOptions) {
+    const wrapper = require.resolve('./wrap.js')
+    // Support running nyc as a user without HOME (e.g. linux 'nobody'),
+    // https://github.com/istanbuljs/nyc/issues/951
+    env.SPAWN_WRAP_SHIM_ROOT = process.env.SPAWN_WRAP_SHIM_ROOT || process.env.XDG_CACHE_HOME || require('os').homedir()
+    const sw = require('spawn-wrap')
+
+    sw([wrapper], env)
+  }
 
   // Both running the test script invocation and the check-coverage run may
   // set process.exitCode. Keep track so that both children are run, but

bin/wrap.js

@@ -1,29 +1,2 @@
-var sw = require('spawn-wrap')
-var NYC = require('../index.js')
-
-var config = {}
-if (process.env.NYC_CONFIG) config = JSON.parse(process.env.NYC_CONFIG)
-config.isChildProcess = true
-
-config._processInfo = {
-  pid: process.pid,
-  ppid: process.ppid,
-  parent: process.env.NYC_PROCESS_ID || null
-}
-if (process.env.NYC_PROCESSINFO_EXTERNAL_ID) {
-  config._processInfo.externalId = process.env.NYC_PROCESSINFO_EXTERNAL_ID
-  delete process.env.NYC_PROCESSINFO_EXTERNAL_ID
-}
-
-if (process.env.NYC_CONFIG_OVERRIDE) {
-  const override = JSON.parse(process.env.NYC_CONFIG_OVERRIDE)
-  config = {
-    ...config,
-    ...override
-  }
-  process.env.NYC_CONFIG = JSON.stringify(config)
-}
-
-;(new NYC(config)).wrap()
-
-sw.runMain()
+require('../lib/wrap')
+require('spawn-wrap').runMain()

index.js

@@ -120,6 +120,10 @@ class NYC {
   }
 
   _loadAdditionalModules () {
+    if (this.config.setNodeOptions) {
+      return
+    }
+
     this.require.forEach(requireModule => {
       // Attempt to require the module relative to the directory being instrumented.
       // Then try other locations, e.g. the nyc node_modules folder.
@@ -334,6 +338,12 @@ class NYC {
 
   wrap (bin) {
     process.env.NYC_PROCESS_ID = this.processInfo.uuid
+    // The else branch is a bug with the spawn-wrap method where
+    // we cannot force propagation of NYC_PROCESS_ID.
+    if (this.config.setNodeOptions) {
+      const { propagateEnv } = require('node-preload')
+      propagateEnv.NYC_PROCESS_ID = this.processInfo.uuid
+    }
     this._addRequireHooks()
     this._wrapExit()
     this._loadAdditionalModules()

lib/config-util.js

@@ -236,6 +236,12 @@ Config.buildYargs = function (cwd) {
       type: 'boolean',
       global: false
     })
+    .option('set-node-options', {
+      describe: 'set process.env.NODE_OPTIONS instead of using spawn-wrap',
+      default: true,
+      type: 'boolean',
+      global: false
+    })
     .option('clean', {
       describe: 'should the .nyc_output folder be cleaned before executing tests',
       default: true,

lib/wrap.js

@@ -0,0 +1,29 @@
+const NYC = require('../index.js')
+
+let config = {}
+if (process.env.NYC_CONFIG) {
+  config = JSON.parse(process.env.NYC_CONFIG)
+}
+config.isChildProcess = true
+
+config._processInfo = {
+  pid: process.pid,
+  ppid: process.ppid,
+  parent: process.env.NYC_PROCESS_ID || null
+}
+
+if (process.env.NYC_PROCESSINFO_EXTERNAL_ID) {
+  config._processInfo.externalId = process.env.NYC_PROCESSINFO_EXTERNAL_ID
+  delete process.env.NYC_PROCESSINFO_EXTERNAL_ID
+}
+
+if (process.env.NYC_CONFIG_OVERRIDE) {
+  const override = JSON.parse(process.env.NYC_CONFIG_OVERRIDE)
+  config = {
+    ...config,
+    ...override
+  }
+  process.env.NYC_CONFIG = JSON.stringify(config)
+}
+
+;(new NYC(config)).wrap()

package.json

@@ -85,6 +85,7 @@
     "js-yaml": "^3.13.1",
     "make-dir": "^3.0.0",
     "merge-source-map": "^1.1.0",
+    "node-preload": "^0.1.2",
     "resolve-from": "^5.0.0",
     "rimraf": "^3.0.0",
     "signal-exit": "^3.0.2",

tap-snapshots/test-nyc-integration.js-TAP.test.js

@@ -98,7 +98,7 @@ All files                        |       0 |        0 |       0 |       0 |
   external-instrumenter.js       |       0 |        0 |       0 |       0 | 1                       
   gc.js                          |       0 |      100 |     100 |       0 | 2,3                     
   half-covered-failing.js        |       0 |        0 |     100 |       0 | 1,3,5,6,7,8             
-  selfspawn-fibonacci.js         |       0 |        0 |       0 |       0 | ...19,21,24,25,26,27,28 
+  selfspawn-fibonacci.js         |       0 |        0 |       0 |       0 | ...21,23,26,27,28,29,30 
   skip-full.js                   |       0 |      100 |     100 |       0 | 1,2                     
   test.js                        |       0 |        0 |       0 |       0 |                         
  cli/fakebin                     |       0 |      100 |     100 |       0 |                         
@@ -129,8 +129,8 @@ exports[`test/nyc-integration.js TAP --ignore-class-method skips methods that ma
 ---------------------------------|---------|----------|---------|---------|-------------------------
 File                             | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s       
 ---------------------------------|---------|----------|---------|---------|-------------------------
-All files                        |    1.45 |        0 |       5 |    1.89 |                         
- cli                             |    2.08 |        0 |    5.56 |    3.13 |                         
+All files                        |    1.44 |        0 |       5 |    1.87 |                         
+ cli                             |    2.06 |        0 |    5.56 |    3.08 |                         
   args.js                        |       0 |      100 |     100 |       0 | 1                       
   by-arg2.js                     |       0 |        0 |     100 |       0 | 1,2,3,4,5,7             
   classes.js                     |   66.67 |      100 |      50 |   66.67 | 6                       
@@ -143,7 +143,7 @@ All files                        |    1.45 |        0 |       5 |    1.89 |
   gc.js                          |       0 |      100 |     100 |       0 | 2,3                     
   half-covered-failing.js        |       0 |        0 |     100 |       0 | 1,3,5,6,7,8             
   half-covered.js                |       0 |        0 |     100 |       0 | 1,3,5,6,7,8             
-  selfspawn-fibonacci.js         |       0 |        0 |       0 |       0 | ...19,21,24,25,26,27,28 
+  selfspawn-fibonacci.js         |       0 |        0 |       0 |       0 | ...21,23,26,27,28,29,30 
   skip-full.js                   |       0 |      100 |     100 |       0 | 1,2                     
  cli/fakebin                     |       0 |      100 |     100 |       0 |                         
   npm-template.js                |       0 |      100 |     100 |       0 | 2,3,4,7,9               
@@ -182,13 +182,35 @@ All files        |      50 |       50 |     100 |      50 |
 
 `
 
+exports[`test/nyc-integration.js TAP --set-node-options=false is functional > stdout 1`] = `
+3
+------------------------|---------|----------|---------|---------|-------------------
+File                    | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
+------------------------|---------|----------|---------|---------|-------------------
+All files               |    91.3 |       70 |     100 |     100 |                   
+ selfspawn-fibonacci.js |    91.3 |       70 |     100 |     100 | 6,27,29           
+------------------------|---------|----------|---------|---------|-------------------
+
+`
+
+exports[`test/nyc-integration.js TAP --set-node-options=true is functional > stdout 1`] = `
+3
+------------------------|---------|----------|---------|---------|-------------------
+File                    | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
+------------------------|---------|----------|---------|---------|-------------------
+All files               |    91.3 |       70 |     100 |     100 |                   
+ selfspawn-fibonacci.js |    91.3 |       70 |     100 |     100 | 6,27,29           
+------------------------|---------|----------|---------|---------|-------------------
+
+`
+
 exports[`test/nyc-integration.js TAP --show-process-tree displays a tree of spawned processes > stdout 1`] = `
 3
 ------------------------|---------|----------|---------|---------|-------------------
 File                    | % Stmts | % Branch | % Funcs | % Lines | Uncovered Line #s 
 ------------------------|---------|----------|---------|---------|-------------------
-All files               |   90.91 |       70 |     100 |     100 |                   
- selfspawn-fibonacci.js |   90.91 |       70 |     100 |     100 | 4,25,27           
+All files               |    91.3 |       70 |     100 |     100 |                   
+ selfspawn-fibonacci.js |    91.3 |       70 |     100 |     100 | 6,27,29           
 ------------------------|---------|----------|---------|---------|-------------------
 nyc
 │   100 % Lines
@@ -199,17 +221,17 @@ nyc
   │ ├─┬ node ./selfspawn-fibonacci.js 3
   │ │ │   100 % Lines
   │ │ ├── node ./selfspawn-fibonacci.js 2
-  │ │ │     31.58 % Lines
+  │ │ │     35 % Lines
   │ │ └── node ./selfspawn-fibonacci.js 1
-  │ │       26.32 % Lines
+  │ │       30 % Lines
   │ └── node ./selfspawn-fibonacci.js 2
-  │       31.58 % Lines
+  │       35 % Lines
   └─┬ node ./selfspawn-fibonacci.js 3
     │   100 % Lines
     ├── node ./selfspawn-fibonacci.js 2
-    │     31.58 % Lines
+    │     35 % Lines
     └── node ./selfspawn-fibonacci.js 1
-          26.32 % Lines
+          30 % Lines
 
 
 `

test/add-all-files.js

@@ -65,7 +65,10 @@ t.test('transpiles .js files added via addAllFiles', async t => {
     'utf-8'
   )
 
-  const nyc = new NYC(configUtil.buildYargs(fixtures).parse(['--require', transpileHook]))
+  const nyc = new NYC(configUtil.buildYargs(fixtures).parse([
+    '--set-node-options=false',
+    '--require', transpileHook
+  ]))
   nyc.reset()
   nyc.addAllFiles()
 
@@ -88,6 +91,7 @@ t.test('does not attempt to transpile files when they are excluded', async t =>
   )
 
   const nyc = new NYC(configUtil.buildYargs(fixtures).parse([
+    '--set-node-options=false',
     `--require=${transpileHook}`,
     '--extension=.do-not-transpile',
     '--include=needs-transpile.do-not-transpile'
@@ -108,6 +112,7 @@ t.test('transpiles non-.js files added via addAllFiles', async t => {
   )
 
   const nyc = (new NYC(configUtil.buildYargs(fixtures).parse([
+    '--set-node-options=false',
     `--require=${transpileHook}`,
     '--extension=.whatever'
   ])))

test/fixtures/cli/selfspawn-fibonacci.js

@@ -1,6 +1,8 @@
 'use strict';
 var cp = require('child_process');
 
+process.env = {};
+
 var index = +process.argv[2] || 0
 if (index <= 1) {
   console.log(0)

test/nyc-integration.js

@@ -186,6 +186,14 @@ t.test('--show-process-tree displays a tree of spawned processes', t => testSucc
   args: ['--show-process-tree', process.execPath, 'selfspawn-fibonacci.js', '5']
 }))
 
+t.test('--set-node-options=true is functional', t => testSuccess(t, {
+  args: ['--set-node-options=true', process.execPath, 'selfspawn-fibonacci.js', '5']
+}))
+
+t.test('--set-node-options=false is functional', t => testSuccess(t, {
+  args: ['--set-node-options=false', process.execPath, 'selfspawn-fibonacci.js', '5']
+}))
+
 t.test('can run "npm test" which directly invokes a test file', t => testSuccess(t, {
   args: ['npm', 'test'],
   cwd: path.resolve(fixturesCLI, 'run-npm-test')