diff --git a/lib/index.js b/lib/index.js
index 7cb7bae..918515e 100644
--- a/lib/index.js
+++ b/lib/index.js
@@ -28,7 +28,7 @@ _mongoose2.default.createConnection = jest.fn().mockReturnValue({
   model: _mongoose2.default.model.bind(_mongoose2.default)
 });
 
-var ops = ['find', 'findOne', 'count', 'distinct', 'findOneAndUpdate', 'findOneAndRemove', 'remove', 'deleteOne', 'deleteMany'];
+var ops = ['find', 'findOne', 'count', 'distinct', 'findOneAndUpdate', 'findOneAndRemove', 'remove', 'update', 'deleteOne', 'deleteMany'];
 
 var mockedReturn = function mockedReturn(cb) {
   var op = this.op,
@@ -68,6 +68,7 @@ var mockedReturn = function mockedReturn(cb) {
 ops.forEach(function (op) {
   _mongoose2.default.Query.prototype[op] = jest.fn().mockImplementation(function (criteria, doc, options, callback) {
     switch (arguments.length) {
+      case 4:
       case 3:
         if (typeof options === 'function') {
           callback = options;
@@ -104,6 +105,23 @@ _mongoose2.default.Query.prototype.exec = jest.fn().mockImplementation(function
   return mockedReturn.call(this, cb);
 });
 
+_mongoose2.default.Aggregate.prototype.exec = jest.fn().mockImplementation(function cb(cb) {
+  var modelName = this._model.modelName;
+
+
+  var mock = mockingoose.__mocks[modelName] && mockingoose.__mocks[modelName].aggregate;
+
+  var err = null;
+
+  if (mock instanceof Error) err = mock;
+
+  if (cb) return cb(err, mock);
+
+  if (err) return Promise.reject(err);
+
+  return Promise.resolve(mock);
+});
+
 var instance = ['remove', 'save'];
 
 instance.forEach(function (methodName) {
diff --git a/lib/index.js.map b/lib/index.js.map
index 1c1efdc..73eb22b 100644
--- a/lib/index.js.map
+++ b/lib/index.js.map
@@ -1 +1 @@
-{"version":3,"sources":["../src/index.js"],"names":["test","version","Promise","connect","jest","fn","mockImplementation","resolve","createConnection","mockReturnValue","on","once","then","catch","model","bind","ops","mockedReturn","cb","op","modelName","_mongooseOptions","Model","mock","mockingoose","__mocks","err","Error","includes","Array","isArray","map","item","lean","toObject","reject","forEach","Query","prototype","criteria","doc","options","callback","arguments","length","undefined","exec","call","instance","methodName","constructor","Object","assign","doMock","target","resetAll","toJSON","traps","get","prop","hasOwnProperty","Reflect","toReturn","o","reset","Proxy"],"mappings":";;;;;;AAAA;;;;;;;;AAEA,IAAG,CAAC,KAAKA,IAAL,CAAU,mBAASC,OAAnB,CAAJ,EAAiC,mBAASC,OAAT,GAAmBA,OAAnB;;AAEjC,mBAASC,OAAT,GAAmBC,KAAKC,EAAL,GAAUC,kBAAV,CAA6B;AAAA,SAAMJ,QAAQK,OAAR,EAAN;AAAA,CAA7B,CAAnB;AACA,mBAASC,gBAAT,GAA4BJ,KACzBC,EADyB,GAEzBI,eAFyB,CAET;AACfC,MAAIN,KAAKC,EAAL,EADW;AAEfM,QAAMP,KAAKC,EAAL,EAFS;AAGfO,MAHe,gBAGVL,OAHU,EAGD;AAAE,WAAOL,QAAQK,OAAR,CAAgBA,QAAQ,IAAR,CAAhB,CAAP;AAAwC,GAHzC;AAIfM,OAJe,oBAIP,CAAE,CAJK;;AAKfC,SAAO,mBAASA,KAAT,CAAeC,IAAf;AALQ,CAFS,CAA5B;;AAUA,IAAMC,MAAM,CACV,MADU,EAEV,SAFU,EAGV,OAHU,EAIV,UAJU,EAKV,kBALU,EAMV,kBANU,EAOV,QAPU,EAQV,WARU,EASV,YATU,CAAZ;;AAYA,IAAMC,eAAe,SAAfA,YAAe,CAAUC,EAAV,EAAc;AAAA,MACzBC,EADyB,GAC2B,IAD3B,CACzBA,EADyB;AAAA,MACZC,SADY,GAC2B,IAD3B,CACrBN,KADqB,CACZM,SADY;AAAA,0BAC2B,IAD3B,CACCC,gBADD;AAAA,MACCA,gBADD,qCACoB,EADpB;;AAEjC,MAAMC,QAAQ,mBAASR,KAAT,CAAeM,SAAf,CAAd;;AAEA,MAAIG,OAAOC,YAAYC,OAAZ,CAAoBL,SAApB,KAAkCI,YAAYC,OAAZ,CAAoBL,SAApB,EAA+BD,EAA/B,CAA7C;;AAEA,MAAIO,MAAM,IAAV;;AAEA,MAAIH,gBAAgBI,KAApB,EAA2BD,MAAMH,IAAN;;AAE3B,MAAI,CAACA,IAAD,IAASJ,OAAO,MAApB,EAA4B;AAAEI,WAAO,IAAP;AAAa;;AAE3C,MAAIA,QAAQA,gBAAgBD,KAAhB,KAA0B,KAAlC,IAA4C,CAAC,CAAC,QAAD,EAAW,OAAX,EAAoBM,QAApB,CAA6BT,EAA7B,CAAjD,EAAoF;AAClFI,WAAOM,MAAMC,OAAN,CAAcP,IAAd,IAAsBA,KAAKQ,GAAL,CAAS;AAAA,aAAQ,IAAIT,KAAJ,CAAUU,IAAV,CAAR;AAAA,KAAT,CAAtB,GAA0D,IAAIV,KAAJ,CAAUC,IAAV,CAAjE;;AAEA,QAAIF,iBAAiBY,IAArB,EAA2BV,OAAOM,MAAMC,OAAN,CAAcP,IAAd,IAAsBA,KAAKQ,GAAL,CAAS;AAAA,aAAQC,KAAKE,QAAL,EAAR;AAAA,KAAT,CAAtB,GAA0DX,KAAKW,QAAL,EAAjE;AAC5B;;AAED,MAAIhB,EAAJ,EAAQ,OAAOA,GAAGQ,GAAH,EAAQH,IAAR,CAAP;;AAER,MAAIG,GAAJ,EAAS,OAAOxB,QAAQiC,MAAR,CAAeT,GAAf,CAAP;;AAET,SAAOxB,QAAQK,OAAR,CAAgBgB,IAAhB,CAAP;AACD,CAvBD;;AAyBAP,IAAIoB,OAAJ,CAAY,cAAM;AAChB,qBAASC,KAAT,CAAeC,SAAf,CAAyBnB,EAAzB,IAA+Bf,KAAKC,EAAL,GAAUC,kBAAV,CAA6B,UAAUiC,QAAV,EAAoBC,GAApB,EAAyBC,OAAzB,EAAkCC,QAAlC,EAA4C;AACtG,YAAQC,UAAUC,MAAlB;AACE,WAAK,CAAL;AACE,YAAI,OAAOH,OAAP,KAAmB,UAAvB,EAAmC;AACjCC,qBAAWD,OAAX;AACAA,oBAAU,EAAV;AACD;AACD;AACF,WAAK,CAAL;AACE,YAAI,OAAOD,GAAP,KAAe,UAAnB,EAA+B;AAC7BE,qBAAWF,GAAX;AACAA,gBAAMD,QAAN;AACAA,qBAAWM,SAAX;AACD;AACDJ,kBAAUI,SAAV;AACA;AACF,WAAK,CAAL;AACE,YAAI,OAAON,QAAP,KAAoB,UAAxB,EAAoC;AAClCG,qBAAWH,QAAX;AACAA,qBAAWE,UAAUD,MAAMK,SAA3B;AACD,SAHD,MAGO;AACLL,gBAAMD,QAAN;AACAA,qBAAWE,UAAUI,SAArB;AACD;AAtBL;;AAyBA,SAAK1B,EAAL,GAAUA,EAAV;;AAEA,QAAI,CAACuB,QAAL,EAAe,OAAO,IAAP;;AAEf,WAAO,KAAKI,IAAL,CAAUC,IAAV,CAAe,IAAf,EAAqBL,QAArB,CAAP;AACD,GA/B8B,CAA/B;AAgCD,CAjCD;;AAmCA,mBAASL,KAAT,CAAeC,SAAf,CAAyBQ,IAAzB,GAAgC1C,KAAKC,EAAL,GAAUC,kBAAV,CAA6B,SAASY,EAAT,CAAYA,EAAZ,EAAgB;AAC3E,SAAOD,aAAa8B,IAAb,CAAkB,IAAlB,EAAwB7B,EAAxB,CAAP;AACD,CAF+B,CAAhC;;AAIA,IAAM8B,WAAW,CACf,QADe,EAEf,MAFe,CAAjB;;AAKAA,SAASZ,OAAT,CAAiB,sBAAc;AAC7B,qBAASd,KAAT,CAAegB,SAAf,CAAyBW,UAAzB,IAAuC7C,KAAKC,EAAL,GAAUC,kBAAV,CAA6B,UAAUmC,OAAV,EAAmBvB,EAAnB,EAAuB;AACzF,QAAMC,KAAK8B,UAAX;AADyF,QAEjF7B,SAFiF,GAEnE,KAAK8B,WAF8D,CAEjF9B,SAFiF;;;AAIzF,QAAI,OAAOqB,OAAP,KAAmB,UAAvB,EAAmCvB,KAAKuB,OAAL;;AAEnCU,WAAOC,MAAP,CAAc,IAAd,EAAoB,EAAEjC,MAAF,EAAML,OAAO,EAAEM,oBAAF,EAAb,EAApB;;AAEA,WAAOH,aAAa8B,IAAb,CAAkB,IAAlB,EAAwB7B,EAAxB,CAAP;AACD,GATsC,CAAvC;AAUD,CAXD;;AAaAd,KAAKiD,MAAL,CAAY,UAAZ,EAAwB;AAAA;AAAA,CAAxB;;AAEA,IAAMC,SAAS;AACb7B,WAAS,EADI;AAEb8B,UAFa,sBAEF;AAAE,SAAK9B,OAAL,GAAe,EAAf;AAAoB,GAFpB;AAGb+B,QAHa,oBAGJ;AAAE,WAAO,KAAK/B,OAAZ;AAAsB;AAHpB,CAAf;;AAMA,IAAMgC,QAAQ;AACZC,KADY,eACRJ,MADQ,EACAK,IADA,EACM;AAChB,QAAIL,OAAOM,cAAP,CAAsBD,IAAtB,CAAJ,EAAiC,OAAOE,QAAQH,GAAR,CAAYJ,MAAZ,EAAoBK,IAApB,CAAP;;AAEjC,WAAO;AACLG,cADK,oBACIC,CADJ,EACoB;AAAA,YAAb5C,EAAa,uEAAR,MAAQ;;AACvBmC,eAAO7B,OAAP,CAAemC,cAAf,CAA8BD,IAA9B,IACIL,OAAO7B,OAAP,CAAekC,IAAf,EAAqBxC,EAArB,IAA2B4C,CAD/B,GAEIT,OAAO7B,OAAP,CAAekC,IAAf,wBAA0BxC,EAA1B,EAA+B4C,CAA/B,CAFJ;;AAIA,eAAO,IAAP;AACD,OAPI;AASLC,WATK,iBASC7C,EATD,EASK;AACRA,cAAM,OAAOmC,OAAO7B,OAAP,CAAekC,IAAf,EAAqBxC,EAArB,CAAb,IAAyC,OAAOmC,OAAO7B,OAAP,CAAekC,IAAf,CAAhD;;AAEA,eAAO,IAAP;AACD,OAbI;AAeLH,YAfK,oBAeI;AACP,eAAOF,OAAO7B,OAAP,CAAekC,IAAf,KAAwB,EAA/B;AACD;AAjBI,KAAP;AAmBD;AAvBW,CAAd;;AA0BA,IAAMnC,cAAc,IAAIyC,KAAJ,CAAUX,MAAV,EAAkBG,KAAlB,CAApB;;kBAEejC,W","file":"index.js","sourcesContent":["import mongoose from 'mongoose';\r\n\r\nif(!/^5/.test(mongoose.version)) mongoose.Promise = Promise;\r\n\r\nmongoose.connect = jest.fn().mockImplementation(() => Promise.resolve());\r\nmongoose.createConnection = jest\r\n  .fn()\r\n  .mockReturnValue({\r\n    on: jest.fn(),\r\n    once: jest.fn(),\r\n    then(resolve) { return Promise.resolve(resolve(this)); },\r\n    catch() {},\r\n    model: mongoose.model.bind(mongoose),\r\n  });\r\n\r\nconst ops = [\r\n  'find',\r\n  'findOne',\r\n  'count',\r\n  'distinct',\r\n  'findOneAndUpdate',\r\n  'findOneAndRemove',\r\n  'remove',\r\n  'deleteOne',\r\n  'deleteMany',\r\n];\r\n\r\nconst mockedReturn = function (cb) {\r\n  const { op, model: { modelName }, _mongooseOptions = {} } = this;\r\n  const Model = mongoose.model(modelName);\r\n\r\n  let mock = mockingoose.__mocks[modelName] && mockingoose.__mocks[modelName][op];\r\n\r\n  let err = null;\r\n\r\n  if (mock instanceof Error) err = mock;\r\n\r\n  if (!mock && op === 'save') { mock = this;}\r\n\r\n  if (mock && mock instanceof Model === false && (!['update', 'count'].includes(op))) {\r\n    mock = Array.isArray(mock) ? mock.map(item => new Model(item)) : new Model(mock);\r\n\r\n    if (_mongooseOptions.lean) mock = Array.isArray(mock) ? mock.map(item => item.toObject()) : mock.toObject();\r\n  }\r\n\r\n  if (cb) return cb(err, mock);\r\n\r\n  if (err) return Promise.reject(err);\r\n\r\n  return Promise.resolve(mock);\r\n};\r\n\r\nops.forEach(op => {\r\n  mongoose.Query.prototype[op] = jest.fn().mockImplementation(function (criteria, doc, options, callback) {\r\n    switch (arguments.length) {\r\n      case 3:\r\n        if (typeof options === 'function') {\r\n          callback = options;\r\n          options = {};\r\n        }\r\n        break;\r\n      case 2:\r\n        if (typeof doc === 'function') {\r\n          callback = doc;\r\n          doc = criteria;\r\n          criteria = undefined;\r\n        }\r\n        options = undefined;\r\n        break;\r\n      case 1:\r\n        if (typeof criteria === 'function') {\r\n          callback = criteria;\r\n          criteria = options = doc = undefined;\r\n        } else {\r\n          doc = criteria;\r\n          criteria = options = undefined;\r\n        }\r\n    }\r\n\r\n    this.op = op;\r\n\r\n    if (!callback) return this;\r\n\r\n    return this.exec.call(this, callback);\r\n  });\r\n});\r\n\r\nmongoose.Query.prototype.exec = jest.fn().mockImplementation(function cb(cb) {\r\n  return mockedReturn.call(this, cb);\r\n});\r\n\r\nconst instance = [\r\n  'remove',\r\n  'save'\r\n];\r\n\r\ninstance.forEach(methodName => {\r\n  mongoose.Model.prototype[methodName] = jest.fn().mockImplementation(function (options, cb) {\r\n    const op = methodName;\r\n    const { modelName } = this.constructor;\r\n\r\n    if (typeof options === 'function') cb = options;\r\n\r\n    Object.assign(this, { op, model: { modelName } });\r\n\r\n    return mockedReturn.call(this, cb);\r\n  })\r\n});\r\n\r\njest.doMock('mongoose', () => mongoose);\r\n\r\nconst target = {\r\n  __mocks: {},\r\n  resetAll() { this.__mocks = {}; },\r\n  toJSON() { return this.__mocks; },\r\n};\r\n\r\nconst traps = {\r\n  get(target, prop) {\r\n    if (target.hasOwnProperty(prop)) return Reflect.get(target, prop);\r\n\r\n    return {\r\n      toReturn(o, op = 'find') {\r\n        target.__mocks.hasOwnProperty(prop)\r\n          ? target.__mocks[prop][op] = o\r\n          : target.__mocks[prop] = { [op]: o };\r\n\r\n        return this;\r\n      },\r\n\r\n      reset(op) {\r\n        op && delete target.__mocks[prop][op] || delete target.__mocks[prop];\r\n\r\n        return this;\r\n      },\r\n\r\n      toJSON() {\r\n        return target.__mocks[prop] || {};\r\n      },\r\n    };\r\n  },\r\n};\r\n\r\nconst mockingoose = new Proxy(target, traps);\r\n\r\nexport default mockingoose;\r\n"]}
\ No newline at end of file
+{"version":3,"sources":["../src/index.js"],"names":["test","version","Promise","connect","jest","fn","mockImplementation","resolve","createConnection","mockReturnValue","on","once","then","catch","model","bind","ops","mockedReturn","cb","op","modelName","_mongooseOptions","Model","mock","mockingoose","__mocks","err","Error","includes","Array","isArray","map","item","lean","toObject","reject","forEach","Query","prototype","criteria","doc","options","callback","arguments","length","undefined","exec","call","Aggregate","_model","aggregate","instance","methodName","constructor","Object","assign","doMock","target","resetAll","toJSON","traps","get","prop","hasOwnProperty","Reflect","toReturn","o","reset","Proxy"],"mappings":";;;;;;AAAA;;;;;;;;AAEA,IAAG,CAAC,KAAKA,IAAL,CAAU,mBAASC,OAAnB,CAAJ,EAAiC,mBAASC,OAAT,GAAmBA,OAAnB;;AAEjC,mBAASC,OAAT,GAAmBC,KAAKC,EAAL,GAAUC,kBAAV,CAA6B;AAAA,SAAMJ,QAAQK,OAAR,EAAN;AAAA,CAA7B,CAAnB;AACA,mBAASC,gBAAT,GAA4BJ,KACzBC,EADyB,GAEzBI,eAFyB,CAET;AACfC,MAAIN,KAAKC,EAAL,EADW;AAEfM,QAAMP,KAAKC,EAAL,EAFS;AAGfO,MAHe,gBAGVL,OAHU,EAGD;AAAE,WAAOL,QAAQK,OAAR,CAAgBA,QAAQ,IAAR,CAAhB,CAAP;AAAwC,GAHzC;AAIfM,OAJe,oBAIP,CAAE,CAJK;;AAKfC,SAAO,mBAASA,KAAT,CAAeC,IAAf;AALQ,CAFS,CAA5B;;AAUA,IAAMC,MAAM,CACV,MADU,EAEV,SAFU,EAGV,OAHU,EAIV,UAJU,EAKV,kBALU,EAMV,kBANU,EAOV,QAPU,EAQV,QARU,EASV,WATU,EAUV,YAVU,CAAZ;;AAaA,IAAMC,eAAe,SAAfA,YAAe,CAAUC,EAAV,EAAc;AAAA,MACzBC,EADyB,GAC2B,IAD3B,CACzBA,EADyB;AAAA,MACZC,SADY,GAC2B,IAD3B,CACrBN,KADqB,CACZM,SADY;AAAA,0BAC2B,IAD3B,CACCC,gBADD;AAAA,MACCA,gBADD,qCACoB,EADpB;;AAEjC,MAAMC,QAAQ,mBAASR,KAAT,CAAeM,SAAf,CAAd;;AAEA,MAAIG,OAAOC,YAAYC,OAAZ,CAAoBL,SAApB,KAAkCI,YAAYC,OAAZ,CAAoBL,SAApB,EAA+BD,EAA/B,CAA7C;;AAEA,MAAIO,MAAM,IAAV;;AAEA,MAAIH,gBAAgBI,KAApB,EAA2BD,MAAMH,IAAN;;AAE3B,MAAI,CAACA,IAAD,IAASJ,OAAO,MAApB,EAA4B;AAAEI,WAAO,IAAP;AAAa;;AAE3C,MAAIA,QAAQA,gBAAgBD,KAAhB,KAA0B,KAAlC,IAA4C,CAAC,CAAC,QAAD,EAAW,OAAX,EAAoBM,QAApB,CAA6BT,EAA7B,CAAjD,EAAoF;AAClFI,WAAOM,MAAMC,OAAN,CAAcP,IAAd,IAAsBA,KAAKQ,GAAL,CAAS;AAAA,aAAQ,IAAIT,KAAJ,CAAUU,IAAV,CAAR;AAAA,KAAT,CAAtB,GAA0D,IAAIV,KAAJ,CAAUC,IAAV,CAAjE;;AAEA,QAAIF,iBAAiBY,IAArB,EAA2BV,OAAOM,MAAMC,OAAN,CAAcP,IAAd,IAAsBA,KAAKQ,GAAL,CAAS;AAAA,aAAQC,KAAKE,QAAL,EAAR;AAAA,KAAT,CAAtB,GAA0DX,KAAKW,QAAL,EAAjE;AAC5B;;AAED,MAAIhB,EAAJ,EAAQ,OAAOA,GAAGQ,GAAH,EAAQH,IAAR,CAAP;;AAER,MAAIG,GAAJ,EAAS,OAAOxB,QAAQiC,MAAR,CAAeT,GAAf,CAAP;;AAET,SAAOxB,QAAQK,OAAR,CAAgBgB,IAAhB,CAAP;AACD,CAvBD;;AAyBAP,IAAIoB,OAAJ,CAAY,cAAM;AAChB,qBAASC,KAAT,CAAeC,SAAf,CAAyBnB,EAAzB,IAA+Bf,KAAKC,EAAL,GAAUC,kBAAV,CAA6B,UAAUiC,QAAV,EAAoBC,GAApB,EAAyBC,OAAzB,EAAkCC,QAAlC,EAA4C;AACtG,YAAQC,UAAUC,MAAlB;AACE,WAAK,CAAL;AACA,WAAK,CAAL;AACE,YAAI,OAAOH,OAAP,KAAmB,UAAvB,EAAmC;AACjCC,qBAAWD,OAAX;AACAA,oBAAU,EAAV;AACD;AACD;AACF,WAAK,CAAL;AACE,YAAI,OAAOD,GAAP,KAAe,UAAnB,EAA+B;AAC7BE,qBAAWF,GAAX;AACAA,gBAAMD,QAAN;AACAA,qBAAWM,SAAX;AACD;AACDJ,kBAAUI,SAAV;AACA;AACF,WAAK,CAAL;AACE,YAAI,OAAON,QAAP,KAAoB,UAAxB,EAAoC;AAClCG,qBAAWH,QAAX;AACAA,qBAAWE,UAAUD,MAAMK,SAA3B;AACD,SAHD,MAGO;AACLL,gBAAMD,QAAN;AACAA,qBAAWE,UAAUI,SAArB;AACD;AAvBL;;AA0BA,SAAK1B,EAAL,GAAUA,EAAV;;AAEA,QAAI,CAACuB,QAAL,EAAe,OAAO,IAAP;;AAEf,WAAO,KAAKI,IAAL,CAAUC,IAAV,CAAe,IAAf,EAAqBL,QAArB,CAAP;AACD,GAhC8B,CAA/B;AAiCD,CAlCD;;AAoCA,mBAASL,KAAT,CAAeC,SAAf,CAAyBQ,IAAzB,GAAgC1C,KAAKC,EAAL,GAAUC,kBAAV,CAA6B,SAASY,EAAT,CAAYA,EAAZ,EAAgB;AAC3E,SAAOD,aAAa8B,IAAb,CAAkB,IAAlB,EAAwB7B,EAAxB,CAAP;AACD,CAF+B,CAAhC;;AAIA,mBAAS8B,SAAT,CAAmBV,SAAnB,CAA6BQ,IAA7B,GAAoC1C,KAAKC,EAAL,GAAUC,kBAAV,CAA6B,SAASY,EAAT,CAAYA,EAAZ,EAAgB;AAAA,MAC9DE,SAD8D,GAC9C,IAD8C,CACxE6B,MADwE,CAC9D7B,SAD8D;;;AAGhF,MAAIG,OAAOC,YAAYC,OAAZ,CAAoBL,SAApB,KAAkCI,YAAYC,OAAZ,CAAoBL,SAApB,EAA+B8B,SAA5E;;AAEA,MAAIxB,MAAM,IAAV;;AAEA,MAAIH,gBAAgBI,KAApB,EAA2BD,MAAMH,IAAN;;AAE3B,MAAIL,EAAJ,EAAQ,OAAOA,GAAGQ,GAAH,EAAQH,IAAR,CAAP;;AAER,MAAIG,GAAJ,EAAS,OAAOxB,QAAQiC,MAAR,CAAeT,GAAf,CAAP;;AAET,SAAOxB,QAAQK,OAAR,CAAgBgB,IAAhB,CAAP;AACA,CAdmC,CAApC;;AAgBA,IAAM4B,WAAW,CACf,QADe,EAEf,MAFe,CAAjB;;AAKAA,SAASf,OAAT,CAAiB,sBAAc;AAC7B,qBAASd,KAAT,CAAegB,SAAf,CAAyBc,UAAzB,IAAuChD,KAAKC,EAAL,GAAUC,kBAAV,CAA6B,UAAUmC,OAAV,EAAmBvB,EAAnB,EAAuB;AACzF,QAAMC,KAAKiC,UAAX;AADyF,QAEjFhC,SAFiF,GAEnE,KAAKiC,WAF8D,CAEjFjC,SAFiF;;;AAIzF,QAAI,OAAOqB,OAAP,KAAmB,UAAvB,EAAmCvB,KAAKuB,OAAL;;AAEnCa,WAAOC,MAAP,CAAc,IAAd,EAAoB,EAAEpC,MAAF,EAAML,OAAO,EAAEM,oBAAF,EAAb,EAApB;;AAEA,WAAOH,aAAa8B,IAAb,CAAkB,IAAlB,EAAwB7B,EAAxB,CAAP;AACD,GATsC,CAAvC;AAUD,CAXD;;AAaAd,KAAKoD,MAAL,CAAY,UAAZ,EAAwB;AAAA;AAAA,CAAxB;;AAEA,IAAMC,SAAS;AACbhC,WAAS,EADI;AAEbiC,UAFa,sBAEF;AAAE,SAAKjC,OAAL,GAAe,EAAf;AAAoB,GAFpB;AAGbkC,QAHa,oBAGJ;AAAE,WAAO,KAAKlC,OAAZ;AAAsB;AAHpB,CAAf;;AAMA,IAAMmC,QAAQ;AACZC,KADY,eACRJ,MADQ,EACAK,IADA,EACM;AAChB,QAAIL,OAAOM,cAAP,CAAsBD,IAAtB,CAAJ,EAAiC,OAAOE,QAAQH,GAAR,CAAYJ,MAAZ,EAAoBK,IAApB,CAAP;;AAEjC,WAAO;AACLG,cADK,oBACIC,CADJ,EACoB;AAAA,YAAb/C,EAAa,uEAAR,MAAQ;;AACvBsC,eAAOhC,OAAP,CAAesC,cAAf,CAA8BD,IAA9B,IACIL,OAAOhC,OAAP,CAAeqC,IAAf,EAAqB3C,EAArB,IAA2B+C,CAD/B,GAEIT,OAAOhC,OAAP,CAAeqC,IAAf,wBAA0B3C,EAA1B,EAA+B+C,CAA/B,CAFJ;;AAIA,eAAO,IAAP;AACD,OAPI;AASLC,WATK,iBASChD,EATD,EASK;AACRA,cAAM,OAAOsC,OAAOhC,OAAP,CAAeqC,IAAf,EAAqB3C,EAArB,CAAb,IAAyC,OAAOsC,OAAOhC,OAAP,CAAeqC,IAAf,CAAhD;;AAEA,eAAO,IAAP;AACD,OAbI;AAeLH,YAfK,oBAeI;AACP,eAAOF,OAAOhC,OAAP,CAAeqC,IAAf,KAAwB,EAA/B;AACD;AAjBI,KAAP;AAmBD;AAvBW,CAAd;;AA0BA,IAAMtC,cAAc,IAAI4C,KAAJ,CAAUX,MAAV,EAAkBG,KAAlB,CAApB;;kBAEepC,W","file":"index.js","sourcesContent":["import mongoose from 'mongoose';\r\n\r\nif(!/^5/.test(mongoose.version)) mongoose.Promise = Promise;\r\n\r\nmongoose.connect = jest.fn().mockImplementation(() => Promise.resolve());\r\nmongoose.createConnection = jest\r\n  .fn()\r\n  .mockReturnValue({\r\n    on: jest.fn(),\r\n    once: jest.fn(),\r\n    then(resolve) { return Promise.resolve(resolve(this)); },\r\n    catch() {},\r\n    model: mongoose.model.bind(mongoose),\r\n  });\r\n\r\nconst ops = [\r\n  'find',\r\n  'findOne',\r\n  'count',\r\n  'distinct',\r\n  'findOneAndUpdate',\r\n  'findOneAndRemove',\r\n  'remove',\r\n  'update',\r\n  'deleteOne',\r\n  'deleteMany',\r\n];\r\n\r\nconst mockedReturn = function (cb) {\r\n  const { op, model: { modelName }, _mongooseOptions = {} } = this;\r\n  const Model = mongoose.model(modelName);\r\n\r\n  let mock = mockingoose.__mocks[modelName] && mockingoose.__mocks[modelName][op];\r\n\r\n  let err = null;\r\n\r\n  if (mock instanceof Error) err = mock;\r\n\r\n  if (!mock && op === 'save') { mock = this;}\r\n\r\n  if (mock && mock instanceof Model === false && (!['update', 'count'].includes(op))) {\r\n    mock = Array.isArray(mock) ? mock.map(item => new Model(item)) : new Model(mock);\r\n\r\n    if (_mongooseOptions.lean) mock = Array.isArray(mock) ? mock.map(item => item.toObject()) : mock.toObject();\r\n  }\r\n\r\n  if (cb) return cb(err, mock);\r\n\r\n  if (err) return Promise.reject(err);\r\n\r\n  return Promise.resolve(mock);\r\n};\r\n\r\nops.forEach(op => {\r\n  mongoose.Query.prototype[op] = jest.fn().mockImplementation(function (criteria, doc, options, callback) {\r\n    switch (arguments.length) {\r\n      case 4:\r\n      case 3:\r\n        if (typeof options === 'function') {\r\n          callback = options;\r\n          options = {};\r\n        }\r\n        break;\r\n      case 2:\r\n        if (typeof doc === 'function') {\r\n          callback = doc;\r\n          doc = criteria;\r\n          criteria = undefined;\r\n        }\r\n        options = undefined;\r\n        break;\r\n      case 1:\r\n        if (typeof criteria === 'function') {\r\n          callback = criteria;\r\n          criteria = options = doc = undefined;\r\n        } else {\r\n          doc = criteria;\r\n          criteria = options = undefined;\r\n        }\r\n    }\r\n\r\n    this.op = op;\r\n\r\n    if (!callback) return this;\r\n\r\n    return this.exec.call(this, callback);\r\n  });\r\n});\r\n\r\nmongoose.Query.prototype.exec = jest.fn().mockImplementation(function cb(cb) {\r\n  return mockedReturn.call(this, cb);\r\n});\r\n\r\nmongoose.Aggregate.prototype.exec = jest.fn().mockImplementation(function cb(cb) {\r\n\tconst { _model: { modelName } } = this;\r\n\r\n\tlet mock = mockingoose.__mocks[modelName] && mockingoose.__mocks[modelName].aggregate;\r\n\r\n\tlet err = null;\r\n\r\n\tif (mock instanceof Error) err = mock;\r\n\r\n\tif (cb) return cb(err, mock);\r\n\r\n\tif (err) return Promise.reject(err);\r\n\r\n\treturn Promise.resolve(mock);\r\n});\r\n\r\nconst instance = [\r\n  'remove',\r\n  'save'\r\n];\r\n\r\ninstance.forEach(methodName => {\r\n  mongoose.Model.prototype[methodName] = jest.fn().mockImplementation(function (options, cb) {\r\n    const op = methodName;\r\n    const { modelName } = this.constructor;\r\n\r\n    if (typeof options === 'function') cb = options;\r\n\r\n    Object.assign(this, { op, model: { modelName } });\r\n\r\n    return mockedReturn.call(this, cb);\r\n  })\r\n});\r\n\r\njest.doMock('mongoose', () => mongoose);\r\n\r\nconst target = {\r\n  __mocks: {},\r\n  resetAll() { this.__mocks = {}; },\r\n  toJSON() { return this.__mocks; },\r\n};\r\n\r\nconst traps = {\r\n  get(target, prop) {\r\n    if (target.hasOwnProperty(prop)) return Reflect.get(target, prop);\r\n\r\n    return {\r\n      toReturn(o, op = 'find') {\r\n        target.__mocks.hasOwnProperty(prop)\r\n          ? target.__mocks[prop][op] = o\r\n          : target.__mocks[prop] = { [op]: o };\r\n\r\n        return this;\r\n      },\r\n\r\n      reset(op) {\r\n        op && delete target.__mocks[prop][op] || delete target.__mocks[prop];\r\n\r\n        return this;\r\n      },\r\n\r\n      toJSON() {\r\n        return target.__mocks[prop] || {};\r\n      },\r\n    };\r\n  },\r\n};\r\n\r\nconst mockingoose = new Proxy(target, traps);\r\n\r\nexport default mockingoose;\r\n"]}
\ No newline at end of file