test: mocking

Author: aryamohanan

package-lock.json

@@ -226,7 +226,7 @@
     "proxyquire": "^2.1.3",
     "read-pkg-up": "^7.0.1",
     "recursive-copy": "^2.0.14",
-    "redis": "^5.0.0",
+    "redis": "^5.0.1",
     "redis-v3": "npm:redis@^3.1.2",
     "redis-v4": "npm:redis@^4.7.0",
     "restify": "^11.1.0",

package.json

@@ -5,6 +5,7 @@
 'use strict';
 
 const mock = require('@instana/core/test/test_util/mockRequire');
+const hook = require('../../../../../core/src/util/hook');
 
 const REDIS_PKG = process.env.REDIS_PKG;
 const REDIS_VERSION = process.env.REDIS_VERSION;
@@ -23,3 +24,24 @@ if (REDIS_PKG === 'redis') {
 if (mockedModuleName !== REDIS_PKG) {
   mock(REDIS_PKG, mockedModuleName);
 }
+
+const originalOnFileLoad = hook.onFileLoad;
+
+hook.onFileLoad = function onFileLoad() {
+  const source = arguments[0]?.source;
+
+  if (
+    mockedModuleName.startsWith('@redis/client') &&
+    (source === '\\/@redis\\/client\\/dist\\/lib\\/commands\\/index.js' ||
+      source === '\\/@redis\\/client\\/dist\\/lib\\/cluster\\/commands.js')
+  ) {
+    const updatedSource = source.replace('@redis\\/client', mockedModuleName);
+
+    const regex = new RegExp(updatedSource);
+    arguments[0] = regex;
+
+    return originalOnFileLoad.apply(this, arguments);
+  }
+
+  return originalOnFileLoad.apply(this, arguments);
+};