@contrast/assess 1.41.0 → 1.42.0

package/lib/dataflow/propagation/install/path/join-and-resolve.js

@@ -16,7 +16,6 @@
 'use strict';
 
 const { isString, primordials: { ArrayPrototypeJoin } } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { createMergedTags } = require('../../../tag-utils');
 const { patchType } = require('../../common');
 const { createArgTagsInResult, excludeExtensionDotFromTags } = require('./common');
@@ -26,7 +25,7 @@ module.exports = function(core) {
     depHooks,
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker },
     },
@@ -34,7 +33,7 @@ module.exports = function(core) {
 
   core.assess.dataflow.propagation.pathInstrumentation.joinAndResolve = {
     install() {
-      depHooks.resolve({ name: 'path' }, (path) => {
+      depHooks.resolve({ name: 'path', version: '*' }, (path) => {
         for (const method of ['join', 'resolve']) {
           for (const os of ['posix', 'win32']) {
             const name = `path.${os}.${method}`;
@@ -47,7 +46,7 @@ module.exports = function(core) {
               post(data) {
                 const { args: origArgs, result, hooked, orig } = data;
 
-                if (!result || !getSourceContext(PROPAGATOR)) return;
+                if (!result || !getPropagatorContext()) return;
 
                 const pathSegments = [...origArgs].reverse();
                 const args = [];

package/lib/dataflow/propagation/install/path/normalize.js

@@ -15,16 +15,15 @@
 
 'use strict';
 const { isString } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { patchType } = require('../../common');
-const { createArgTagsInResult, excludeExtensionDotFromTags, } = require('./common');
+const { createArgTagsInResult, excludeExtensionDotFromTags } = require('./common');
 
 module.exports = function(core) {
   const {
     depHooks,
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker },
     },
@@ -32,7 +31,7 @@ module.exports = function(core) {
 
   core.assess.dataflow.propagation.pathInstrumentation.normalize = {
     install() {
-      depHooks.resolve({ name: 'path' }, (path) => {
+      depHooks.resolve({ name: 'path', version: '*' }, (path) => {
         for (const os of ['posix', 'win32']) {
           const isWin32 = os === 'win32';
 
@@ -42,7 +41,7 @@ module.exports = function(core) {
             usePerf: 'sync',
             post(data) {
               const { args, result, name, hooked, orig } = data;
-              if (!result || !getSourceContext(PROPAGATOR)) return;
+              if (!result || !getPropagatorContext()) return;
 
               const pathStr = args[0];
 

package/lib/dataflow/propagation/install/path/parse.js

@@ -15,7 +15,6 @@
 
 'use strict';
 const { isString } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { createSubsetTags } = require('../../../tag-utils');
 const { patchType } = require('../../common');
 const { excludeExtensionDotFromTags } = require('./common');
@@ -25,7 +24,7 @@ module.exports = function(core) {
     depHooks,
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       inspect, // TODO NODE-3455: remove
       eventFactory: { createPropagationEvent },
       dataflow: { tracker },
@@ -34,7 +33,7 @@ module.exports = function(core) {
 
   core.assess.dataflow.propagation.pathInstrumentation.parse = {
     install() {
-      depHooks.resolve({ name: 'path' }, (path) => {
+      depHooks.resolve({ name: 'path', version: '*' }, (path) => {
         for (const os of ['posix', 'win32']) {
           const isWin32 = os === 'win32';
 
@@ -44,7 +43,7 @@ module.exports = function(core) {
             usePerf: 'sync',
             post(data) {
               const { args, result, name: patchName, hooked, orig } = data;
-              if (!result || !getSourceContext(PROPAGATOR)) return;
+              if (!result || !getPropagatorContext()) return;
 
               const [path] = args;
               if (!path || !isString(path)) return;

package/lib/dataflow/propagation/install/path/relative.js

@@ -16,16 +16,15 @@
 'use strict';
 
 const { isString } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { patchType } = require('../../common');
-const { createArgTagsInResult, excludeExtensionDotFromTags, } = require('./common');
+const { createArgTagsInResult, excludeExtensionDotFromTags } = require('./common');
 
 module.exports = function(core) {
   const {
     depHooks,
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker },
     },
@@ -33,7 +32,7 @@ module.exports = function(core) {
 
   core.assess.dataflow.propagation.pathInstrumentation.relative = {
     install() {
-      depHooks.resolve({ name: 'path' }, (path) => {
+      depHooks.resolve({ name: 'path', version: '*' }, (path) => {
         for (const os of ['posix', 'win32']) {
           const isWin32 = os === 'win32';
 
@@ -43,7 +42,7 @@ module.exports = function(core) {
             usePerf: 'sync',
             post(data) {
               const { args, result, name, hooked, orig } = data;
-              if (!result || !getSourceContext(PROPAGATOR)) return;
+              if (!result || !getPropagatorContext()) return;
 
               const [fromStr, toStr] = args;
 

package/lib/dataflow/propagation/install/path/toNamespacedPath.js

@@ -16,7 +16,6 @@
 'use strict';
 
 const { isString } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { patchType } = require('../../common');
 const { createArgTagsInResult, excludeExtensionDotFromTags } = require('./common');
 
@@ -25,7 +24,7 @@ module.exports = function(core) {
     depHooks,
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker },
     },
@@ -33,7 +32,7 @@ module.exports = function(core) {
 
   core.assess.dataflow.propagation.pathInstrumentation.toNamespacedPath = {
     install() {
-      depHooks.resolve({ name: 'path' }, (path) => {
+      depHooks.resolve({ name: 'path', version: '*' }, (path) => {
         for (const os of ['posix', 'win32']) {
 
           patcher.patch(path[os], 'toNamespacedPath', {
@@ -42,7 +41,7 @@ module.exports = function(core) {
             usePerf: 'sync',
             post(data) {
               const { args, result, name, hooked, orig } = data;
-              if (!result || !getSourceContext(PROPAGATOR)) return;
+              if (!result || !getPropagatorContext()) return;
 
               const pathStr = args[0];
 

package/lib/dataflow/propagation/install/pug/index.js

@@ -14,7 +14,6 @@
  */
 'use strict';
 
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { patchType } = require('../../common');
 
 /** @type {import('@contrast/rewriter').RewriteOpts} */
@@ -28,19 +27,19 @@ module.exports = function (core) {
     rewriter,
     depHooks,
     scopes: { instrumentation },
-    assess: { getSourceContext },
+    assess: { getPropagatorContext },
   } = core;
 
   const pugInstrumentation = {
     install() {
       depHooks.resolve(
-        { name: 'pug' },
+        { name: 'pug', version: '<4' },
         (pug) => patcher.patch(pug, 'compile', {
           name: 'pug.compile',
           patchType,
           usePerf: 'tbd',
           pre(data) {
-            if (!getSourceContext(PROPAGATOR)) return;
+            if (!getPropagatorContext()) return;
 
             const opts = data.args[1] || {};
             const plugins = opts.plugins || [];

package/lib/dataflow/propagation/install/pug-runtime-escape.js

@@ -16,7 +16,6 @@
 'use strict';
 
 const { DataflowTag: { WEAK_URL_ENCODED } } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../constants');
 const { createFullLengthCopyTags } = require('../../tag-utils');
 const { patchType, createModuleLabel } = require('../common');
 
@@ -25,7 +24,7 @@ module.exports = function(core) {
     patcher,
     depHooks,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
     }
@@ -33,7 +32,7 @@ module.exports = function(core) {
 
   return core.assess.dataflow.propagation.pugRuntimeEscape = {
     install() {
-      depHooks.resolve({ name: 'pug-runtime' }, (pugRuntime, version) => {
+      depHooks.resolve({ name: 'pug-runtime', version: '<4' }, (pugRuntime, version) => {
         const name = 'pug-runtime.escape';
 
         patcher.patch(pugRuntime, 'escape', {
@@ -42,7 +41,7 @@ module.exports = function(core) {
           usePerf: 'sync',
           post(data) {
             const { args, result, hooked, orig } = data;
-            if (!result || !args[0] || !getSourceContext(PROPAGATOR)) return;
+            if (!result || !args[0] || !getPropagatorContext()) return;
 
             const argInfo = tracker.getData(args[0]);
 

package/lib/dataflow/propagation/install/querystring/escape.js

@@ -15,14 +15,13 @@
 'use strict';
 
 const { DataflowTag: { URL_ENCODED } } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { createFullLengthCopyTags } = require('../../../tag-utils');
 const { patchType } = require('../../common');
 
 module.exports = function(core) {
   const {
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       inspect, // TODO NODE-3455: remove
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
@@ -33,7 +32,7 @@ module.exports = function(core) {
 
   return core.assess.dataflow.propagation.querystringInstrumentation.escape = {
     install() {
-      depHooks.resolve({ name: 'querystring' }, (_export) => {
+      depHooks.resolve({ name: 'querystring', version: '<1' }, (_export) => {
         patcher.patch(_export, 'escape', {
           name: 'querystring.escape',
           patchType,
@@ -45,7 +44,7 @@ module.exports = function(core) {
             const strInfo = tracker.getData(value);
             if (!strInfo) return;
 
-            if (!getSourceContext(PROPAGATOR)) return;
+            if (!getPropagatorContext()) return;
 
             let tags;
             if (value !== data.result) {

package/lib/dataflow/propagation/install/querystring/escape.test.js

@@ -20,7 +20,7 @@ describe('assess dataflow propagation querystring escape', function () {
     tracker = core.assess.dataflow.tracker;
 
     querystring = require('querystring');
-    core.depHooks.resolve.withArgs({ name: 'querystring' }).yields(querystring);
+    core.depHooks.resolve.yields(querystring);
 
     core.assess.dataflow.propagation.querystringInstrumentation.escape.install();
   });

package/lib/dataflow/propagation/install/querystring/parse.js

@@ -20,7 +20,6 @@ const {
   DataflowTag: { URL_ENCODED },
   primordials: { ArrayPrototypeJoin },
 } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { createSubsetTags, createAppendTags, getAdjustedUntrackedValue } = require('../../../tag-utils');
 const { patchType } = require('../../common');
 
@@ -29,7 +28,7 @@ module.exports = function(core) {
     patcher,
     depHooks,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
     }
@@ -100,14 +99,14 @@ module.exports = function(core) {
 
   return core.assess.dataflow.propagation.querystringInstrumentation.parse = {
     install() {
-      depHooks.resolve({ name: 'querystring' }, (module) => {
+      depHooks.resolve({ name: 'querystring', version: '<1' }, (module) => {
         ['parse', 'decode'].forEach((method) => {
           patcher.patch(module, method, {
             name: `querystring.${method}`,
             patchType,
             usePerf: 'sync',
             pre(data) {
-              if (!data.args[0] || !getSourceContext(PROPAGATOR)) return;
+              if (!data.args[0] || !getPropagatorContext()) return;
 
               const trackingData = tracker.getData(data.args[0]);
               if (!trackingData) return;

package/lib/dataflow/propagation/install/querystring/parse.test.js

@@ -21,7 +21,7 @@ describe('assess dataflow propagation querystring parse', function () {
     querystring = require('querystring');
     core.assess.dataflow.propagation.querystringInstrumentation.parse.install();
     core.assess.dataflow.propagation.stringInstrumentation.concat.install();
-    core.depHooks.resolve.withArgs({ name: 'querystring' }).yield(querystring);
+    core.depHooks.resolve.yield(querystring);
   });
 
   afterEach(function () {

package/lib/dataflow/propagation/install/querystring/stringify.js

@@ -16,7 +16,6 @@
 
 const querystring = require('querystring');
 const { isString } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const utils = require('../../../tag-utils');
 const { patchType } = require('../../common');
 
@@ -25,7 +24,7 @@ const moduleName = 'querystring';
 module.exports = function(core) {
   const {
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       inspect, // TODO NODE-3455: remove
       dataflow: { tracker },
       eventFactory: { createPropagationEvent },
@@ -38,7 +37,7 @@ module.exports = function(core) {
    * Adds custom encoding function to capture key/value tags and history during stringification
    */
   function pre(data) {
-    if (!getSourceContext(PROPAGATOR)) return;
+    if (!getPropagatorContext()) return;
 
     const [input] = data.args;
     const escape = typeof data.args[3]?.encodeURIComponent === 'function'
@@ -142,7 +141,7 @@ module.exports = function(core) {
 
   return core.assess.dataflow.propagation.querystringInstrumentation.stringify = {
     install() {
-      depHooks.resolve({ name: moduleName }, (_export) => {
+      depHooks.resolve({ name: moduleName, version: '<1' }, (_export) => {
         for (const methodName of ['encode', 'stringify']) {
           patcher.patch(_export, methodName, {
             name: `querystring.${methodName}`,

package/lib/dataflow/propagation/install/querystring/stringify.test.js

@@ -14,7 +14,7 @@ describe('assess dataflow propagation querystring stringify', function () {
     tracker = core.assess.dataflow.tracker;
 
     querystring = require('querystring');
-    core.depHooks.resolve.withArgs({ name: 'querystring' }).yields(querystring);
+    core.depHooks.resolve.yields(querystring);
 
     core.assess.dataflow.propagation.stringInstrumentation.concat.install();
     core.assess.dataflow.propagation.querystringInstrumentation.escape.install();

package/lib/dataflow/propagation/install/reg-exp-prototype-exec.js

@@ -15,7 +15,6 @@
 
 'use strict';
 
-const { InstrumentationType: { PROPAGATOR } } = require('../../../constants');
 const { createSubsetTags, getAdjustedUntrackedValue } = require('../../tag-utils');
 const { patchType } = require('../common');
 
@@ -23,7 +22,7 @@ module.exports = function(core) {
   const {
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker },
     },
@@ -86,7 +85,7 @@ module.exports = function(core) {
             !obj ||
             !args[0] ||
             !result?.length ||
-            !getSourceContext(PROPAGATOR)
+            !getPropagatorContext()
           )
             return;
 

package/lib/dataflow/propagation/install/send.js

@@ -21,7 +21,7 @@ module.exports = function (core) {
   const {
     depHooks,
     patcher,
-    assess: { getSourceContext }
+    assess: { getPropagatorContext }
   } = core;
 
   const send = {};
@@ -38,7 +38,7 @@ module.exports = function (core) {
           patchType,
           pre(data) {
             const { args } = data;
-            if (!getSourceContext()) return;
+            if (!getPropagatorContext()) return;
 
             const untrackedPath = StringPrototypeSlice.call(` ${args[0]}`, 1);
             args[0] = untrackedPath;
@@ -49,7 +49,7 @@ module.exports = function (core) {
   }
 
   send.install = function () {
-    depHooks.resolve({ name: 'send' }, patchSendModule);
+    depHooks.resolve({ name: 'send', version: '<2' }, patchSendModule);
   };
 
   return send;

package/lib/dataflow/propagation/install/sequelize/query-generator.js

@@ -31,7 +31,7 @@ const DIALECTS = [
 module.exports = function(core) {
   const {
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
     },
@@ -43,7 +43,7 @@ module.exports = function(core) {
     install() {
       for (const dialect of DIALECTS) {
         depHooks.resolve(
-          { name: 'sequelize', file: `lib/dialects/${dialect}/query-generator.js` },
+          { name: 'sequelize', version: '<7', file: `lib/dialects/${dialect}/query-generator.js` },
           (_export) => {
             const constructor = _export.name;
 
@@ -53,7 +53,7 @@ module.exports = function(core) {
               usePerf: 'sync',
               post(data) {
                 const strInfo = tracker.getData(data.result);
-                if (!strInfo || !getSourceContext()) return;
+                if (!strInfo || !getPropagatorContext()) return;
 
                 const { value } = strInfo;
                 const event = createPropagationEvent({

package/lib/dataflow/propagation/install/sequelize/query-generator.test.js

@@ -1,6 +1,7 @@
 'use strict';
 
 const { expect } = require('chai');
+const sinon = require('sinon');
 const {
   DataflowTag: { UNTRUSTED, SQL_ENCODED }
 } = require('@contrast/common');
@@ -45,7 +46,7 @@ describe('assess dataflow propagation sequelize QueryGenerators', function() {
     },
   ].forEach(({ Ctor, file }) => {
     it(`${Ctor.name}.prototype.quoteIdentifier adds sql-encoded tag`, function() {
-      core.depHooks.resolve.withArgs({ name: 'sequelize', file }).yields(Ctor);
+      core.depHooks.resolve.withArgs(sinon.match({ name: 'sequelize', file })).yields(Ctor);
       component.install();
 
       simulateRequestScope(() => {

package/lib/dataflow/propagation/install/sequelize/sql-string.js

@@ -27,7 +27,7 @@ module.exports = function(core) {
     patcher,
     depHooks,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker },
     },
@@ -54,7 +54,7 @@ module.exports = function(core) {
   return core.assess.dataflow.propagation.sequelizeInstrumentation.sqlString = {
     install() {
       depHooks.resolve(
-        { name: 'sequelize', file: 'lib/sql-string.js' },
+        { name: 'sequelize', version: '<7', file: 'lib/sql-string.js' },
         (sqlString, version) => {
           const origEscape = sqlString.escape;
           const name = 'sequelize.escape';
@@ -70,7 +70,7 @@ module.exports = function(core) {
                 !result ||
                 !args[0] ||
                 !isString(args[0]) ||
-                !getSourceContext()
+                !getPropagatorContext()
               ) return;
 
               const argInfo = tracker.getData(args[0]);
@@ -124,7 +124,7 @@ module.exports = function(core) {
                 !result ||
                 !args[0] ||
                 !isString(args[0]) ||
-                !getSourceContext()
+                !getPropagatorContext()
               ) return;
 
               const resultInfo = tracker.getData(result);
@@ -216,7 +216,7 @@ module.exports = function(core) {
                 !result ||
                 !args[0] ||
                 !isString(args[0]) ||
-                !getSourceContext()
+                !getPropagatorContext()
               ) return;
 
               const resultInfo = tracker.getData(result);

package/lib/dataflow/propagation/install/sql-template-strings.js

@@ -23,7 +23,7 @@ module.exports = function(core) {
     patcher,
     depHooks,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
     }
@@ -31,7 +31,7 @@ module.exports = function(core) {
 
   return core.assess.dataflow.propagation.sqlTemplateStrings = {
     install() {
-      depHooks.resolve({ name: 'sql-template-strings' }, (sqlTemplateStrings, version) => {
+      depHooks.resolve({ name: 'sql-template-strings', version: '<3' }, (sqlTemplateStrings, version) => {
         const name = 'sql-template-strings.SQL';
 
         patcher.patch(sqlTemplateStrings, 'SQL', {
@@ -40,7 +40,7 @@ module.exports = function(core) {
           usePerf: 'sync',
           post(data) {
             const { args, result, hooked, orig } = data;
-            if (!result || !args[0] || !getSourceContext()) return;
+            if (!result || !args[0] || !getPropagatorContext()) return;
 
             const argInfo = tracker.getData(args[0]);
 

package/lib/dataflow/propagation/install/string/concat.js

@@ -16,7 +16,6 @@
 'use strict';
 
 const { primordials: { ArrayPrototypeJoin } } = require('@contrast/common');
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { createAppendTags, getAdjustedUntrackedValue } = require('../../../tag-utils');
 const { patchType } = require('../../common');
 
@@ -24,7 +23,7 @@ module.exports = function(core) {
   const {
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
     }
@@ -40,7 +39,7 @@ module.exports = function(core) {
         usePerf: 'sync',
         post(data) {
           const { obj, result, hooked, orig } = data;
-          if (!result || !getSourceContext(PROPAGATOR)) return;
+          if (!result || !getPropagatorContext()) return;
 
           const rInfo = tracker.getData(result);
           if (rInfo) {

package/lib/dataflow/propagation/install/string/format-methods.js

@@ -15,14 +15,13 @@
 
 'use strict';
 
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { patchType } = require('../../common');
 
 module.exports = function(core) {
   const {
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
     }
@@ -39,7 +38,7 @@ module.exports = function(core) {
           usePerf: 'sync',
           post(data) {
             const { obj, result, hooked, orig } = data;
-            if (!result || !getSourceContext(PROPAGATOR)) return;
+            if (!result || !getPropagatorContext()) return;
 
             const objInfo = tracker.getData(obj);
 

package/lib/dataflow/propagation/install/string/html-methods.js

@@ -14,7 +14,6 @@
  */
 
 'use strict';
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { createAppendTags } = require('../../../tag-utils');
 const { patchType } = require('../../common');
 const htmlTagsLengths = {
@@ -33,7 +32,7 @@ module.exports = function(core) {
     patcher,
     assess: {
       inspect, // TODO NODE-3455: remove
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: { tracker }
     }
@@ -64,7 +63,7 @@ module.exports = function(core) {
         usePost: true,
         post(data) {
           const { args, obj, result, hooked, orig } = data;
-          if (!result || !getSourceContext(PROPAGATOR)) return;
+          if (!result || !getPropagatorContext()) return;
 
           const objInfo = tracker.getData(obj);
           const history = objInfo ? new Set([objInfo]) : new Set();
@@ -121,7 +120,7 @@ module.exports = function(core) {
           patchType,
           post(data) {
             const { obj, result, hooked, orig } = data;
-            if (!result || !getSourceContext(PROPAGATOR)) return;
+            if (!result || !getPropagatorContext()) return;
 
             const objInfo = tracker.getData(obj);
 

package/lib/dataflow/propagation/install/string/match-all.js

@@ -14,7 +14,6 @@
  */
 
 'use strict';
-const { InstrumentationType: { PROPAGATOR } } = require('../../../../constants');
 const { createSubsetTags } = require('../../../tag-utils');
 const { patchType } = require('../../common');
 
@@ -22,7 +21,7 @@ module.exports = function(core) {
   const {
     patcher,
     assess: {
-      getSourceContext,
+      getPropagatorContext,
       eventFactory: { createPropagationEvent },
       dataflow: {
         tracker,
@@ -83,7 +82,7 @@ module.exports = function(core) {
             !obj ||
             !args[0] ||
             typeof obj !== 'string' ||
-            !getSourceContext(PROPAGATOR)
+            !getPropagatorContext()
           )
             return origFn();