add support for Q

src/plugins/index.js

@@ -16,6 +16,7 @@ module.exports = {
   'mysql': require('./mysql'),
   'mysql2': require('./mysql2'),
   'pg': require('./pg'),
+  'q': require('./q'),
   'redis': require('./redis'),
   'restify': require('./restify')
 }

src/plugins/q.js

@@ -0,0 +1,16 @@
+'use strict'
+
+const tx = require('./util/promise')
+
+module.exports = [
+  {
+    name: 'q',
+    versions: ['1'],
+    patch (Q, tracer, config) {
+      this.wrap(Q.makePromise.prototype, 'then', tx.createWrapThen(tracer, config))
+    },
+    unpatch (Q) {
+      this.unwrap(Q.makePromise.prototype, 'then')
+    }
+  }
+]

test/plugins/q.spec.js

@@ -0,0 +1,85 @@
+'use strict'
+
+const agent = require('./agent')
+const plugin = require('../../src/plugins/q')
+
+wrapIt()
+
+describe('Plugin', () => {
+  let Q
+  let tracer
+
+  describe('q', () => {
+    withVersions(plugin, 'q', version => {
+      beforeEach(() => {
+        tracer = require('../..')
+      })
+
+      afterEach(() => {
+        return agent.close()
+      })
+
+      describe('without configuration', () => {
+        beforeEach(() => {
+          return agent.load(plugin, 'q')
+            .then(() => {
+              Q = require(`../../versions/q@${version}`).get()
+            })
+        })
+
+        it('should run the then() callback in context where then() was called', () => {
+          if (process.env.DD_CONTEXT_PROPAGATION === 'false') return
+
+          const span = {}
+          const deferred = Q.defer()
+          const promise = deferred.promise
+
+          setImmediate(() => {
+            tracer.scopeManager().activate({})
+            deferred.resolve()
+          })
+
+          tracer.scopeManager().activate(span)
+
+          return promise
+            .then(() => {
+              tracer.scopeManager().activate({})
+            })
+            .then(() => {
+              const scope = tracer.scopeManager().active()
+
+              expect(scope).to.not.be.null
+              expect(scope.span()).to.equal(span)
+            })
+        })
+
+        it('should run the catch() callback in context where catch() was called', () => {
+          if (process.env.DD_CONTEXT_PROPAGATION === 'false') return
+
+          const span = {}
+          const deferred = Q.defer()
+          const promise = deferred.promise
+
+          setImmediate(() => {
+            tracer.scopeManager().activate({})
+            deferred.reject(new Error())
+          })
+
+          tracer.scopeManager().activate(span)
+
+          return promise
+            .catch(err => {
+              tracer.scopeManager().activate({})
+              throw err
+            })
+            .catch(() => {
+              const scope = tracer.scopeManager().active()
+
+              expect(scope).to.not.be.null
+              expect(scope.span()).to.equal(span)
+            })
+        })
+      })
+    })
+  })
+})