added env var for enabling diagnostic code

newrelic · Oct 7, 2019 · adba864 · adba864
1 parent 8a194e5
commit adba864
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/lib/config/default.js b/lib/config/default.js
@@ -179,8 +179,8 @@ exports.config = () => ({
     filter_cache_limit: 1000
   },
 
-  code_injector: {
-    diagnostics: {
+  diagnostics: {
+    code_injector: {
       enabled: false,
       internal_file_pattern: /nodejs_agent\/(?:(?!test))|\/node_modules\/(?:@)?newrelic/
     }

diff --git a/lib/config/env.js b/lib/config/env.js
@@ -51,6 +51,11 @@ const ENV_MAPPING = {
       include: 'NEW_RELIC_TRANSACTION_EVENTS_ATTRIBUTES_INCLUDE'
     }
   },
+  diagnostics: {
+    code_injector: {
+      enabled: 'NEW_RELIC_DIAGNOSTICS_CODE_INJECTOR_ENABLED'
+    }
+  },
   transaction_tracer: {
     attributes: {
       enabled: 'NEW_RELIC_TRANSACTION_TRACER_ATTRIBUTES_ENABLED',
@@ -186,6 +191,7 @@ const BOOLEAN_VARS = new Set([
   'NEW_RELIC_TRACER_ENABLED',
   'NEW_RELIC_TRANSACTION_TRACER_ATTRIBUTES_ENABLED',
   'NEW_RELIC_TRANSACTION_EVENTS_ATTRIBUTES_ENABLED',
+  'NEW_RELIC_DIAGNOSTICS_CODE_INJECTOR_ENABLED',
   'NEW_RELIC_TRANSACTION_SEGMENTS_ATTRIBUTES_ENABLED',
   'NEW_RELIC_DEBUG_METRICS',
   'NEW_RELIC_DEBUG_TRACER',

diff --git a/test/unit/config/config.test.js b/test/unit/config/config.test.js
@@ -176,6 +176,13 @@ describe('the agent configuration', function() {
       })
     })
 
+    it('should pick up on diagnostics code injection', function() {
+      idempotentEnv({'NEW_RELIC_DIAGNOSTICS_CODE_INJECTOR_ENABLED': true}, function(tc) {
+        should.exist(tc.diagnostics.code_injector.enabled)
+        expect(tc.diagnostics.code_injector.enabled).to.equal(true)
+      })
+    })
+
     it('should pick up the billing hostname', function() {
       idempotentEnv({'NEW_RELIC_UTILIZATION_LOGICAL_PROCESSORS': 123}, function(tc) {
         should.exist(tc.utilization.logical_processors)