test(koa): add wip test

Author: chentsulin

package-lock.json

@@ -49,6 +49,7 @@
     "@koa/router": "12.0.0",
     "@opentelemetry/api": "^1.3.0",
     "@opentelemetry/context-async-hooks": "^1.8.0",
+    "@opentelemetry/contrib-test-utils": "^0.35.0",
     "@opentelemetry/sdk-trace-base": "^1.8.0",
     "@opentelemetry/sdk-trace-node": "^1.8.0",
     "@types/mocha": "7.0.2",

plugins/node/opentelemetry-instrumentation-koa/package.json

@@ -55,8 +55,8 @@ export class KoaInstrumentation extends InstrumentationBase<typeof koa> {
     return new InstrumentationNodeModuleDefinition<typeof koa>(
       'koa',
       ['^2.0.0'],
-      module => {
-        const moduleExports =
+      (module: any) => {
+        const moduleExports: typeof koa =
           module[Symbol.toStringTag] === 'Module'
             ? module.default // ESM
             : module; // CommonJS
@@ -74,8 +74,8 @@ export class KoaInstrumentation extends InstrumentationBase<typeof koa> {
         );
         return module;
       },
-      module => {
-        const moduleExports =
+      (module: any) => {
+        const moduleExports: typeof koa =
           module[Symbol.toStringTag] === 'Module'
             ? module.default // ESM
             : module; // CommonJS

plugins/node/opentelemetry-instrumentation-koa/src/instrumentation.ts

@@ -0,0 +1,63 @@
+/*
+ * Copyright The OpenTelemetry Authors
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *      https://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+// Use koa from an ES module:
+//    node --experimental-loader=@opentelemetry/instrumentation/hook.mjs use-koa.mjs
+
+import { createTestNodeSdk } from '@opentelemetry/contrib-test-utils';
+
+import { KoaInstrumentation } from '../../build/src/index.js';
+
+const sdk = createTestNodeSdk({
+  serviceName: 'use-koa',
+  instrumentations: [
+    new KoaInstrumentation()
+  ]
+})
+sdk.start();
+
+import Koa from 'koa';
+import KoaRouter from '@koa/router';
+import * as http from 'http';
+
+const app = new Koa();
+
+app.use(async function simpleResponse(ctx, next) {
+  await next();
+});
+
+const router = new KoaRouter();
+router.get('/post/:id', ctx => {
+  ctx.body = `Post id: ${ctx.params.id}`;
+});
+
+app.use(router.routes());
+
+const server = http.createServer(app.callback());
+await new Promise(resolve => server.listen(0, resolve));
+const port = server.address().port;
+
+await new Promise(resolve => {
+  http.get(`http://localhost:${port}/post/0`, (res) => {
+    res.resume();
+    res.on('end', () => {
+      resolve();
+    });
+  })
+});
+
+await new Promise(resolve => server.close(resolve));
+await sdk.shutdown();

plugins/node/opentelemetry-instrumentation-koa/test/fixtures/use-koa.mjs

@@ -18,6 +18,7 @@ import * as KoaRouter from '@koa/router';
 import { context, trace, Span } from '@opentelemetry/api';
 import { NodeTracerProvider } from '@opentelemetry/sdk-trace-node';
 import { AsyncHooksContextManager } from '@opentelemetry/context-async-hooks';
+import * as testUtils from '@opentelemetry/contrib-test-utils';
 import {
   InMemorySpanExporter,
   SimpleSpanProcessor,
@@ -709,4 +710,26 @@ describe('Koa Instrumentation', () => {
       );
     });
   });
+
+  it('should work with ESM usage', async () => {
+    await testUtils.runTestFixture({
+      cwd: __dirname,
+      argv: ['fixtures/use-koa.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;
+
+        console.log(JSON.stringify(spans, null, 2));
+
+        // TODO
+      },
+    });
+  });
 });