Move tests

open-telemetry · Dec 6, 2024 · 44e37ea · 44e37ea
1 parent 21840f9
commit 44e37ea
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 74 deletions.
diff --git a/plugins/node/opentelemetry-instrumentation-pg/test/fixtures/use-pg.mjs b/plugins/node/opentelemetry-instrumentation-pg/test/fixtures/use-pg.mjs
@@ -40,7 +40,6 @@ const sdk = createTestNodeSdk({
 sdk.start();
 
 import pg from 'pg';
-
 const client = new pg.Client(CONFIG);
 
 await client.connect();

diff --git a/plugins/node/opentelemetry-instrumentation-pg/test/pg.esm.test.ts b/plugins/node/opentelemetry-instrumentation-pg/test/pg.esm.test.ts
diff --git a/plugins/node/opentelemetry-instrumentation-pg/test/pg.test.ts b/plugins/node/opentelemetry-instrumentation-pg/test/pg.test.ts
@@ -1087,4 +1087,31 @@ describe('pg', () => {
       });
     });
   });
+
+  it('should work with ESM usage', async () => {
+    await testUtils.runTestFixture({
+      cwd: __dirname,
+      argv: ['fixtures/use-pg.mjs'],
+      env: {
+        NODE_OPTIONS:
+          '--experimental-loader=@opentelemetry/instrumentation/hook.mjs',
+        NODE_NO_WARNINGS: '1',
+      },
+      checkResult: (err, stdout, stderr) => {
+        assert.ifError(err);
+      },
+      checkCollector: (collector: testUtils.TestCollector) => {
+        const spans = collector.sortedSpans;
+
+        assert.strictEqual(spans.length, 3);
+
+        assert.strictEqual(spans[0].name, 'pg.connect');
+        assert.strictEqual(spans[0].kind, 3);
+        assert.strictEqual(spans[1].name, 'test-span');
+        assert.strictEqual(spans[1].kind, 1);
+        assert.strictEqual(spans[2].name, 'pg.query:SELECT postgres');
+        assert.strictEqual(spans[2].kind, 3);
+      },
+    });
+  });
 });