test(instrumentation): add test for koa

Author: dszakallas

scripts/test-instrumentations

@@ -10,4 +10,5 @@ fi
 
 cd ./test/instrumentations
 make test TARGET_VERSIONS=$INSTR_TARGET_VERSIONS \
+    TARGET_LIBRARIES=$INSTR_TARGET_LIBRARIES \
     NODE_INDEX=$NODE_INDEX NODE_TOTAL=$NODE_TOTAL

test/instrumentations/koa/koa.mk

@@ -0,0 +1,52 @@
+target := koa
+
+all = 1.0 1.2 1.4 2.0 2.2
+some = 1.4 2.2
+one = 1.4
+
+versions := $(call configure_targets,$(one),$(some),$(all))
+
+targets := $(targets) $(target)
+
+# set up as target specific variables to avoid collision in other
+# makefiles
+test_suite_$(target) : target := $(target)
+test_suite_$(target) : versions := $(versions)
+test_suite_$(target) : cur_path := $(abspath $(lastword $(MAKEFILE_LIST)))
+test_suite_$(target) : cur_dir := $(dir $(cur_path))
+test_suite_$(target) : cur_base := $(notdir $(cur_path))
+test_suite_$(target) : db_url := pg://localhost/postgres
+test_suite_$(target) : errors := 0
+
+# normally all tasks should be phony
+.PHONY : before_$(target) \
+	$(versions:%=before_version_$(target)_%) \
+	$(versions:%=test_$(target)_%) \
+	$(versions:%=after_version_$(target)_%) \
+	after_$(target) \
+	test_suite_$(target)
+
+# 'this should run before the test
+before_$(target) : before
+	@echo '*---------------*'
+	@echo '| koa           |'
+	@echo '*---------------*'
+
+# this should run after `before` for each of the version targets
+# before the test
+$(versions:%=before_version_$(target)_%) : before_$(target)
+	@npm i koa@$(subst before_version_$(target)_,,$@)
+
+# run the test for each of the versions
+$(versions:%=test_$(target)_%) : test_% : before_version_%
+	$(MOCHA) $(addprefix $(cur_dir), *.spec.js) || exit 0;
+
+# this should run before `after` for each of the version targets
+# after the test suite
+$(versions:%=after_version_$(target)_%) : $(versions:%=test_$(target)_%)
+
+# this should run after the test
+after_$(target) : $(versions:%=after_version_$(target)_%)
+
+## this is the whole test suite
+test_suite_$(target) : after_$(target)

test/instrumentations/koa/koa.spec.js

@@ -0,0 +1,42 @@
+'use strict'
+
+require('../test-setup.spec')
+
+var sinon = require('sinon')
+var http = require('http')
+var expect = require('chai').expect
+var Shimmer = require('../../../lib/utils/shimmer')
+var Module = require('module')
+
+describe('The koa wrapper module', function () {
+  var sandbox = sinon.sandbox.create()
+  var fakeAgent = {
+    tracer: {
+      collector: {
+        userSentError: sandbox.spy()
+      }
+    }
+  }
+  var koa
+  beforeEach(function () {
+    sandbox.reset()
+    require.cache = {}
+    Shimmer.unwrap(Module, '_load')
+    require('../../../lib/instrumentations').create({
+      instrumentations: { redis: './trace-instrumentation-koa' },
+      agent: fakeAgent
+    })
+    koa = require('koa')
+  })
+  it('reports unhandled errors', function () {
+    var app = koa()
+    app.use(function * (next) {
+      throw new Error('But you shouldn\'t worry about it.')
+      // yield next
+    })
+    app.listen(3000)
+    http.get('http://localhost:3000', function () {
+      expect(fakeAgent.tracer.collector.userSentError).to.have.been.calledOnce
+    })
+  })
+})