test(node): Add tests for genericPoolIntegration

Signed-off-by: Kaung Zin Hein <[email protected]>

Author: Zen-cronic

dev-packages/node-integration-tests/package.json

@@ -44,6 +44,7 @@
     "cors": "^2.8.5",
     "cron": "^3.1.6",
     "express": "^4.17.3",
+    "generic-pool": "3.9.0",
     "graphql": "^16.3.0",
     "http-terminator": "^3.2.0",
     "ioredis": "^5.4.1",

dev-packages/node-integration-tests/suites/tracing/genericPool/scenario.js

@@ -0,0 +1,71 @@
+const { loggingTransport } = require('@sentry-internal/node-integration-tests');
+const Sentry = require('@sentry/node');
+
+Sentry.init({
+  dsn: 'https://[email protected]/1337',
+  release: '1.0',
+  tracesSampleRate: 1.0,
+  transport: loggingTransport,
+});
+
+// Stop the process from exiting before the transaction is sent
+setInterval(() => {}, 1000);
+
+const mysql = require('mysql');
+const genericPool = require('generic-pool');
+
+const factory = {
+  create: function () {
+    return mysql.createConnection({
+      user: 'root',
+      password: 'docker',
+    });
+  },
+  destroy: function (client) {
+    client.end(err => {
+      if (err) {
+        // eslint-disable-next-line no-console
+        console.error('Error while disconnecting MySQL:', err);
+      }
+    });
+  },
+};
+
+const opts = {
+  max: 10,
+  min: 2,
+};
+
+const myPool = genericPool.createPool(factory, opts);
+
+async function run() {
+  await Sentry.startSpan(
+    {
+      op: 'transaction',
+      name: 'Test Transaction',
+    },
+    async () => {
+      try {
+        const client1 = await myPool.acquire();
+        const client2 = await myPool.acquire();
+
+        client1.query('SELECT NOW()', function () {
+          myPool.release(client1);
+        });
+
+        client2.query('SELECT 1 + 1 AS solution', function () {
+          myPool.release(client2);
+        });
+      } catch (err) {
+        // eslint-disable-next-line no-console
+        console.error('Error while pooling MySQL:', err);
+      } finally {
+        await myPool.drain();
+        await myPool.clear();
+      }
+    },
+  );
+}
+
+// eslint-disable-next-line @typescript-eslint/no-floating-promises
+run();

dev-packages/node-integration-tests/suites/tracing/genericPool/test.ts

@@ -0,0 +1,34 @@
+import { cleanupChildProcesses, createRunner } from '../../../utils/runner';
+
+describe('genericPool auto instrumentation', () => {
+  afterAll(() => {
+    cleanupChildProcesses();
+  });
+
+  test('should auto-instrument `genericPool` package when calling pool.require()', done => {
+    const EXPECTED_TRANSACTION = {
+      transaction: 'Test Transaction',
+      spans: expect.arrayContaining([
+        expect.objectContaining({
+          description: 'generic-pool.aquire',
+          origin: 'auto.db.otel.generic-pool',
+          data: {
+            'sentry.origin': 'auto.db.otel.generic-pool',
+          },
+          status: 'ok',
+        }),
+
+        expect.objectContaining({
+          description: 'generic-pool.aquire',
+          origin: 'auto.db.otel.generic-pool',
+          data: {
+            'sentry.origin': 'auto.db.otel.generic-pool',
+          },
+          status: 'ok',
+        }),
+      ]),
+    };
+
+    createRunner(__dirname, 'scenario.js').expect({ transaction: EXPECTED_TRANSACTION }).start(done);
+  });
+});

packages/node/package.json

@@ -73,6 +73,7 @@
     "@opentelemetry/instrumentation-express": "0.41.1",
     "@opentelemetry/instrumentation-fastify": "0.38.0",
     "@opentelemetry/instrumentation-fs": "0.14.0",
+    "@opentelemetry/instrumentation-generic-pool": "0.38.0",
     "@opentelemetry/instrumentation-graphql": "0.42.0",
     "@opentelemetry/instrumentation-hapi": "0.40.0",
     "@opentelemetry/instrumentation-http": "0.52.1",
@@ -85,7 +86,6 @@
     "@opentelemetry/instrumentation-nestjs-core": "0.39.0",
     "@opentelemetry/instrumentation-pg": "0.43.0",
     "@opentelemetry/instrumentation-redis-4": "0.41.0",
-    "@@opentelemetry/instrumentation-generic-pool": "0.38.0",
     "@opentelemetry/resources": "^1.25.1",
     "@opentelemetry/sdk-trace-base": "^1.25.1",
     "@opentelemetry/semantic-conventions": "^1.25.1",