pinpoint-apm#92 mock datasender

feelform · Sep 12, 2023 · 9ccf0d6 · 9ccf0d6
1 parent 01ced0c
commit 9ccf0d6
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/grpc/grpc-idl b/grpc/grpc-idl
diff --git a/test/instrumentation/module/express.test.js b/test/instrumentation/module/express.test.js
@@ -14,7 +14,6 @@ const DefaultAnnotationKey = require('../../../lib/constant/annotation-key').Def
 const apiMetaService = require('../../../lib/context/api-meta-service')
 const MethodDescriptorBuilder = require('../../../lib/context/method-descriptor-builder')
 const dataSenderMock = require('../../support/data-sender-mock')
-apiMetaService.init(dataSenderMock())
 
 const TEST_ENV = {
   host: 'localhost',

diff --git a/test/support/agent-singleton-mock.js b/test/support/agent-singleton-mock.js
@@ -17,6 +17,7 @@ const httpShared = require('../../lib/instrumentation/http-shared')
 const traceContext = require('../../lib/context/trace-context')
 const contextManager = require('../../lib/context/context-manager')
 const activeTrace = require('../../lib/metric/active-trace')
+const apiMetaService = require('../../lib/context/api-meta-service')
 
 class MockAgent extends Agent {
     startSchedule(agentId, agentStartTime) {
@@ -31,6 +32,7 @@ class MockAgent extends Agent {
 
     bindHttp(json) {
         this.cleanHttp()
+        apiMetaService.init(dataSenderMock())
 
         if (!json) {
             json = require('../pinpoint-config-test')
@@ -43,7 +45,7 @@ class MockAgent extends Agent {
         const http = require('http')
         log.debug('shimming http.Server.prototype.emit function')
         shimmer.wrap(http && http.Server && http.Server.prototype, 'emit', httpShared.instrumentRequest(agent, 'http'))
-      
+
         log.debug('shimming http.request function')
         shimmer.wrap(http, 'request', httpShared.traceOutgoingRequest(agent, 'http'))
 
@@ -69,7 +71,7 @@ class MockAgent extends Agent {
         this.traceContext = traceContext.init(this.agentInfo, this.dataSender, this.config)
     }
 
-    cleanHttp() {        
+    cleanHttp() {
         const http = require('http')
         shimmer.unwrap(http && http.Server && http.Server.prototype, 'emit')
         shimmer.unwrap(http, 'request')