kafkajs

Author: mydea

dev-packages/node-integration-tests/suites/tracing/kafkajs/instrument.mjs

@@ -0,0 +1,9 @@
+import * as Sentry from '@sentry/node';
+import { loggingTransport } from '@sentry-internal/node-integration-tests';
+
+Sentry.init({
+  dsn: 'https://[email protected]/1337',
+  release: '1.0',
+  tracesSampleRate: 1.0,
+  transport: loggingTransport,
+});

dev-packages/node-integration-tests/suites/tracing/kafkajs/scenario.js renamed to dev-packages/node-integration-tests/suites/tracing/kafkajs/scenario.mjs

@@ -1,17 +1,7 @@
-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 { Kafka } = require('kafkajs');
+import { Kafka } from 'kafkajs';
 
 async function run() {
   const kafka = new Kafka({
@@ -54,10 +44,6 @@ async function run() {
       },
     ],
   });
-
-  // Wait for the message to be received
-  await new Promise(resolve => setTimeout(resolve, 5000));
 }
 
-// eslint-disable-next-line @typescript-eslint/no-floating-promises
 run();

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

@@ -1,54 +1,56 @@
-import { afterAll, describe, expect, test } from 'vitest';
-import { cleanupChildProcesses, createRunner } from '../../../utils/runner';
+import { afterAll, describe, expect } from 'vitest';
+import { cleanupChildProcesses, createEsmAndCjsTests } from '../../../utils/runner';
 
 describe('kafkajs', () => {
   afterAll(() => {
     cleanupChildProcesses();
   });
 
-  test('traces producers and consumers', { timeout: 60_000 }, async () => {
-    await createRunner(__dirname, 'scenario.js')
-      .withDockerCompose({
-        workingDirectory: [__dirname],
-        readyMatches: ['9092'],
-      })
-      .expect({
-        transaction: {
-          transaction: 'test-topic',
-          contexts: {
-            trace: expect.objectContaining({
-              op: 'message',
-              status: 'ok',
-              data: expect.objectContaining({
-                'messaging.system': 'kafka',
-                'messaging.destination': 'test-topic',
-                'otel.kind': 'PRODUCER',
-                'sentry.op': 'message',
-                'sentry.origin': 'auto.kafkajs.otel.producer',
+  createEsmAndCjsTests(__dirname, 'scenario.mjs', 'instrument.mjs', (createRunner, test) => {
+    test('traces producers and consumers', { timeout: 60_000 }, async () => {
+      await createRunner()
+        .withDockerCompose({
+          workingDirectory: [__dirname],
+          readyMatches: ['9092'],
+        })
+        .expect({
+          transaction: {
+            transaction: 'test-topic',
+            contexts: {
+              trace: expect.objectContaining({
+                op: 'message',
+                status: 'ok',
+                data: expect.objectContaining({
+                  'messaging.system': 'kafka',
+                  'messaging.destination': 'test-topic',
+                  'otel.kind': 'PRODUCER',
+                  'sentry.op': 'message',
+                  'sentry.origin': 'auto.kafkajs.otel.producer',
+                }),
               }),
-            }),
+            },
           },
-        },
-      })
-      .expect({
-        transaction: {
-          transaction: 'test-topic',
-          contexts: {
-            trace: expect.objectContaining({
-              op: 'message',
-              status: 'ok',
-              data: expect.objectContaining({
-                'messaging.system': 'kafka',
-                'messaging.destination': 'test-topic',
-                'otel.kind': 'CONSUMER',
-                'sentry.op': 'message',
-                'sentry.origin': 'auto.kafkajs.otel.consumer',
+        })
+        .expect({
+          transaction: {
+            transaction: 'test-topic',
+            contexts: {
+              trace: expect.objectContaining({
+                op: 'message',
+                status: 'ok',
+                data: expect.objectContaining({
+                  'messaging.system': 'kafka',
+                  'messaging.destination': 'test-topic',
+                  'otel.kind': 'CONSUMER',
+                  'sentry.op': 'message',
+                  'sentry.origin': 'auto.kafkajs.otel.consumer',
+                }),
               }),
-            }),
+            },
           },
-        },
-      })
-      .start()
-      .completed();
+        })
+        .start()
+        .completed();
+    });
   });
 });