test(postgres): fix event emitter tests

RisingStack · Jan 19, 2017 · 4e7b120 · 4e7b120
1 parent c71bfe7
commit 4e7b120
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/instrumentations/pg/pg.spec.js b/test/instrumentations/pg/pg.spec.js
@@ -121,8 +121,10 @@ describe('pg module wrapper', function () {
         throw err
       }
 
-      var shimmerWrapStub = sandbox.stub(Shimmer, 'wrap', function (nodule, name) {
+      var shimmerWrapStub = sandbox.stub(Shimmer, 'wrap', function (nodule, name, cb) {
+        expect(cb).to.be.a('function')
         var qryArguments = [qryString]
+        cb(client.query).apply(client, qryArguments)
 
         expect(fakeWrapQuery).to.have.been.calledWith(
           client.query,
@@ -161,8 +163,10 @@ describe('pg module wrapper', function () {
         throw err
       }
 
-      var shimmerWrapStub = sandbox.stub(Shimmer, 'wrap', function (nodule, name) {
+      var shimmerWrapStub = sandbox.stub(Shimmer, 'wrap', function (nodule, name, cb) {
+        expect(cb).to.be.a('function')
         var qryArguments = [qryString]
+        cb(client.query).apply(client, qryArguments)
 
         expect(fakeWrapQuery).to.have.been.calledWith(
           client.query,