feat: updated fake agent to register ritm hooks

newrelic · Aug 9, 2023 · e4f4074 · e4f4074
1 parent 0342e4d
commit e4f4074
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/lib/agent.js b/lib/agent.js
@@ -134,12 +134,12 @@ TestAgent.makeFullyInstrumented = function makeInstrumented(conf, setState) {
  */
 TestAgent.prototype.instrument = function instrument(instrumentFull = false) {
   shimmer.debug = true
-  shimmer.patchModule(this.agent)
   if (instrumentFull) {
     shimmer.bootstrapInstrumentation(this.agent)
   } else {
     shimmer.registerCoreInstrumentation(this.agent)
   }
+  shimmer.registerHooks(this.agent)
 
   return this
 }
@@ -153,7 +153,7 @@ TestAgent.prototype.instrument = function instrument(instrumentFull = false) {
  * that tests aren't stepping on eachothers' toes.
  */
 TestAgent.prototype.unload = function unload() {
-  shimmer.unpatchModule()
+  shimmer.removeHooks()
   shimmer.unwrapAll()
   shimmer.debug = false
   shimmer.registeredInstrumentations = Object.create(null)