feat(mongo): add more operations

Author: gergelyke

lib/instrumentations/mongodb.js

@@ -2,8 +2,29 @@ var Shimmer = require('../utils/shimmer')
 var consts = require('../consts')
 
 var COLLECTION_OPERATIONS = [
-  'find',
-  'findOne'
+  'aggregate',
+  'count',
+  'deleteMany',
+  'deleteOne',
+  'distinct',
+  'findAndModify',
+  'findAndRemove',
+  'findOne',
+  'findOneAndDelete',
+  'findOneAndReplace',
+  'findOneAndUpdate',
+  'geoHaystackSearch',
+  'geoNear',
+  'group',
+  'insert',
+  'insertMany',
+  'insertOne',
+  'mapReduce',
+  'remove',
+  'save',
+  'update',
+  'updateMany',
+  'updateOne'
 ]
 
 module.exports = function (mongodb, agent) {
@@ -63,3 +84,5 @@ module.exports = function (mongodb, agent) {
 
   return mongodb
 }
+
+module.exports._COLLECTION_OPERATIONS = COLLECTION_OPERATIONS

lib/instrumentations/mongodb.spec.js

@@ -4,11 +4,6 @@ var expect = require('chai').expect
 var wrapper = require('./mongodb')
 var Shimmer = require('../utils/shimmer')
 
-var COLLECTION_OPERATIONS = [
-  'find',
-  'findOne'
-]
-
 describe('The mongodb wrapper module', function () {
   it('should wrap collection\'s operations', function () {
     var shimmerWrapStub = this.sandbox.stub(Shimmer, 'wrap')
@@ -23,7 +18,7 @@ describe('The mongodb wrapper module', function () {
     expect(shimmerWrapStub).to.have.been.calledWith(
       fakeMongo.Collection.prototype,
       'mongodb.Collection.prototype',
-      COLLECTION_OPERATIONS
+      wrapper._COLLECTION_OPERATIONS
     )
   })