Disable and enable instrumentation on create.

open-telemetry · Dec 6, 2024 · 4dfc927 · 4dfc927
1 parent 76cefbb
commit 4dfc927
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 3 deletions.
diff --git a/plugins/node/opentelemetry-instrumentation-pg/test/pg-pool.test.ts b/plugins/node/opentelemetry-instrumentation-pg/test/pg-pool.test.ts
@@ -114,6 +114,10 @@ describe('pg-pool', () => {
   function create(config: PgInstrumentationConfig = {}) {
     instrumentation.setConfig(config);
     instrumentation.enable();
+
+    // Disable and enable the instrumentation to visit unwrap calls
+    instrumentation.disable();
+    instrumentation.enable();
   }
 
   let pool: pgPool<pg.Client>;
@@ -540,7 +544,7 @@ describe('pg-pool', () => {
           );
           assert.strictEqual(
             metrics[1].dataPoints[0].attributes[
-              ATTR_DB_CLIENT_CONNECTION_STATE
+            ATTR_DB_CLIENT_CONNECTION_STATE
             ],
             'used'
           );
@@ -551,7 +555,7 @@ describe('pg-pool', () => {
           );
           assert.strictEqual(
             metrics[1].dataPoints[1].attributes[
-              ATTR_DB_CLIENT_CONNECTION_STATE
+            ATTR_DB_CLIENT_CONNECTION_STATE
             ],
             'idle'
           );
@@ -764,7 +768,7 @@ describe('pg-pool', () => {
           );
           assert.strictEqual(
             metrics[1].dataPoints[0].attributes[
-              ATTR_DB_CLIENT_CONNECTION_STATE
+            ATTR_DB_CLIENT_CONNECTION_STATE
             ],
             'used'
           );

diff --git a/plugins/node/opentelemetry-instrumentation-pg/test/pg.test.ts b/plugins/node/opentelemetry-instrumentation-pg/test/pg.test.ts
@@ -108,6 +108,10 @@ describe('pg', () => {
   function create(config: PgInstrumentationConfig = {}) {
     instrumentation.setConfig(config);
     instrumentation.enable();
+
+    // Disable and enable the instrumentation to visit unwrap calls
+    instrumentation.disable();
+    instrumentation.enable();
   }
 
   let postgres: typeof pg;