chore: migrated koa versioned tests to use agent_helper, moved symbol…

…s to lib/symbols, consolidated koa instrumentation into one folder

lib/instrumentation/koa/lib/instrumentation.js → lib/instrumentation/koa/instrumentation.js

@@ -4,7 +4,7 @@
  */
 
 'use strict'
-const symbols = require('./symbols')
+const symbols = require('../../symbols')
 
 module.exports = function initialize(shim, Koa) {
   // Koa's exports are different depending on using CJS or MJS - https://github.com/koajs/koa/issues/1513
@@ -70,34 +70,49 @@ function wrapMiddleware(shim, middleware) {
   )
 }
 
-function wrapCreateContext(shim, fn, fnName, context) {
-  // Many of the properties on the `context` object are just aliases for the same
-  // property on the `request` or `response` objects. We take advantage of this
-  // by just intercepting the `request` or `response` property and don't touch
-  // the `context` property.
-  //
-  // See: https://github.com/koajs/koa/blob/master/lib/context.js#L186-L241
+/**
+ * Many of the properties on the `context` object are just aliases for the same
+ * property on the `request` or `response` objects. We take advantage of this
+ * by just intercepting the `request` or `response` property and don't touch
+ * the `context` property.
+ * See: https://github.com/koajs/koa/blob/master/lib/context.js#L186-L241
+ *
+ * @param {Shim} shim instance of shim
+ * @param {function} _fn createContext function
+ * @param {string} _fnName name of function
+ * @param {object} context koa ctx object
+ */
+function wrapCreateContext(shim, _fn, _fnName, context) {
+  wrapResponseBody(shim, context)
+  wrapMatchedRoute(shim, context)
+  wrapResponseStatus(shim, context)
+}
 
+function wrapResponseBody(shim, context) {
   // The `context.body` and `context.response.body` properties are how users set
   // the response contents. It is roughly equivalent to `res.send()` in Express.
   // Under the hood, these set the `_body` property on the `context.response`.
-  context[symbols.body] = context.response.body
-  context[symbols.bodySet] = false
+  context[symbols.koaBody] = context.response.body
+  context[symbols.koaBodySet] = false
+
   Object.defineProperty(context.response, '_body', {
-    get: () => context[symbols.body],
+    get: () => context[symbols.koaBody],
     set: function setBody(val) {
       if (!context[symbols.koaRouter]) {
         shim.savePossibleTransactionName(context.req)
       }
-      context[symbols.body] = val
-      context[symbols.bodySet] = true
+      context[symbols.koaBody] = val
+      context[symbols.koaBodySet] = true
     }
   })
+}
 
-  context[symbols.matchedRoute] = null
+function wrapMatchedRoute(shim, context) {
+  context[symbols.koaMatchedRoute] = null
   context[symbols.koaRouter] = false
+
   Object.defineProperty(context, '_matchedRoute', {
-    get: () => context[symbols.matchedRoute],
+    get: () => context[symbols.koaMatchedRoute],
     set: (val) => {
       const match = getLayerForTransactionName(context)
 
@@ -111,7 +126,7 @@ function wrapCreateContext(shim, fn, fnName, context) {
         if (currentSegment) {
           const transaction = currentSegment.transaction
 
-          if (context[symbols.matchedRoute]) {
+          if (context[symbols.koaMatchedRoute]) {
             transaction.nameState.popPath()
           }
 
@@ -120,13 +135,15 @@ function wrapCreateContext(shim, fn, fnName, context) {
         }
       }
 
-      context[symbols.matchedRoute] = val
+      context[symbols.koaMatchedRoute] = val
       // still true if somehow match is undefined because we are
       // using koa-router naming and don't want to allow default naming
       context[symbols.koaRouter] = true
     }
   })
+}
 
+function wrapResponseStatus(shim, context) {
   // Sometimes people just set `context.status` or `context.response.status`
   // without setting a body. When this happens we'll want to use that as the
   // response point to name the transaction. `context.status` is just an alias
@@ -143,7 +160,7 @@ function wrapCreateContext(shim, fn, fnName, context) {
   Object.defineProperty(context.response, 'status', {
     get: () => statusDescriptor.get.call(context.response),
     set: function setStatus(val) {
-      if (!context[symbols.bodySet] && !context[symbols.koaRouter]) {
+      if (!context[symbols.koaBodySet] && !context[symbols.koaRouter]) {
         shim.savePossibleTransactionName(context.req)
       }
       return statusDescriptor.set.call(this, val)

lib/instrumentation/koa/nr-hooks.js

@@ -12,24 +12,24 @@ module.exports = [
     type: InstrumentationDescriptor.TYPE_WEB_FRAMEWORK,
     moduleName: 'koa',
     shimName: 'koa',
-    onRequire: require('./lib/instrumentation')
+    onRequire: require('./instrumentation')
   },
   {
     type: InstrumentationDescriptor.TYPE_WEB_FRAMEWORK,
     moduleName: 'koa-router',
     shimName: 'koa',
-    onRequire: require('./lib/router-instrumentation')
+    onRequire: require('./router-instrumentation')
   },
   {
     type: InstrumentationDescriptor.TYPE_WEB_FRAMEWORK,
     moduleName: '@koa/router',
     shimName: 'koa',
-    onRequire: require('./lib/router-instrumentation')
+    onRequire: require('./router-instrumentation')
   },
   {
     type: InstrumentationDescriptor.TYPE_WEB_FRAMEWORK,
     moduleName: 'koa-route',
     shimName: 'koa',
-    onRequire: require('./lib/route-instrumentation')
+    onRequire: require('./route-instrumentation')
   }
 ]

lib/instrumentation/koa/lib/route-instrumentation.js → lib/instrumentation/koa/route-instrumentation.js

@@ -5,7 +5,7 @@
 
 'use strict'
 
-const { METHODS } = require('./http-methods')
+const { METHODS } = require('../http-methods')
 
 module.exports = function instrumentRoute(shim, route) {
   shim.setFramework(shim.KOA)

lib/instrumentation/koa/lib/router-instrumentation.js → lib/instrumentation/koa/router-instrumentation.js

@@ -4,7 +4,7 @@
  */
 
 'use strict'
-const symbols = require('./symbols')
+const symbols = require('../../symbols')
 
 module.exports = function instrumentRouter(shim, Router) {
   shim.setFramework(shim.KOA)

lib/instrumentation/superagent.js

@@ -5,8 +5,7 @@
 
 'use strict'
 
-const http = require('http')
-const METHODS = http.METHODS.map((method) => method.toLowerCase())
+const { METHODS } = require('./http-methods')
 
 module.exports = function instrument(agent, superagent, moduleName, shim) {
   shim.wrapExport(superagent, function wrapRequest(shim, request) {

lib/symbols.js

@@ -11,6 +11,10 @@ module.exports = {
   databaseName: Symbol('databaseName'),
   disableDT: Symbol('Disable distributed tracing'), // description for backwards compatibility
   executorContext: Symbol('executorContext'),
+  koaBody: Symbol('body'),
+  koaBodySet: Symbol('bodySet'),
+  koaRouter: Symbol('koaRouter'),
+  koaMatchedRoute: Symbol('matchedRoute'),
   name: Symbol('name'),
   onceExecuted: Symbol('onceExecuted'),
   offTheRecord: Symbol('offTheRecord'),

test/lib/agent_helper.js

@@ -653,3 +653,12 @@ helper.isSupportedVersion = function isSupportedVersion(version) {
 helper.destroyProxyAgent = function destroyProxyAgent() {
   require('../../lib/collector/http-agents').proxyAgent().destroy()
 }
+
+/**
+ * Gets a shim instance for a package.
+ * @param {object} pkg exported obj that is instrumented
+ * @returns The existing or newly created shim.
+ */
+helper.getShim = function getShim(pkg) {
+  return pkg?.[symbols.shim]
+}

test/unit/instrumentation/koa/instrumentation.test.js

@@ -7,7 +7,7 @@
 
 const tap = require('tap')
 const sinon = require('sinon')
-const initialize = require('../../../../lib/instrumentation/koa/lib/instrumentation')
+const initialize = require('../../../../lib/instrumentation/koa/instrumentation')
 
 tap.test('Koa instrumentation', (t) => {
   t.beforeEach((t) => {

test/unit/instrumentation/koa/koa.test.js

@@ -14,7 +14,7 @@ tap.beforeEach((t) => {
   t.context.agent = helper.instrumentMockedAgent({
     moduleName: 'koa',
     type: InstrumentationDescriptor.TYPE_WEB_FRAMEWORK,
-    onRequire: require('../../../../lib/instrumentation/koa/lib/instrumentation'),
+    onRequire: require('../../../../lib/instrumentation/koa/instrumentation'),
     shimName: 'koa'
   })
 

test/unit/instrumentation/koa/route.test.js

@@ -6,7 +6,7 @@
 'use strict'
 
 const tap = require('tap')
-const { METHODS } = require('../../../../lib/instrumentation/koa/lib/http-methods')
+const { METHODS } = require('../../../../lib/instrumentation/http-methods')
 const helper = require('../../../lib/agent_helper')
 const InstrumentationDescriptor = require('../../../../lib/instrumentation-descriptor')
 const symbols = require('../../../../lib/symbols')
@@ -15,7 +15,7 @@ tap.beforeEach((t) => {
   t.context.agent = helper.instrumentMockedAgent({
     moduleName: 'koa-route',
     type: InstrumentationDescriptor.TYPE_WEB_FRAMEWORK,
-    onRequire: require('../../../../lib/instrumentation/koa/lib/route-instrumentation'),
+    onRequire: require('../../../../lib/instrumentation/koa/route-instrumentation'),
     shimName: 'koa'
   })
 

test/unit/instrumentation/koa/router.test.js

@@ -7,8 +7,8 @@
 
 const tap = require('tap')
 
-const instrumentation = require('../../../../lib/instrumentation/koa/lib/router-instrumentation')
-const { METHODS } = require('../../../../lib/instrumentation/koa/lib/http-methods')
+const instrumentation = require('../../../../lib/instrumentation/koa/router-instrumentation')
+const { METHODS } = require('../../../../lib/instrumentation/http-methods')
 const helper = require('../../../lib/agent_helper')
 const InstrumentationDescriptor = require('../../../../lib/instrumentation-descriptor')
 const symbols = require('../../../../lib/symbols')