handle save hook (#45)

Updates #8

___tests___/User.js

@@ -5,11 +5,14 @@ const { Schema } = mongoose;
 const schema = Schema({
   name: String,
   email: { type: String, required: true },
-  created: { type: Date, default: Date.now }
+  created: { type: Date, default: Date.now },
+  saveCount: { type: Number, default: 0}
 });
 
+schema.pre('save', function() {
+  this.saveCount++
+})
+
 const User = mongoose.model('User', schema);
 
 export default User;
-
-

___tests___/index.test.js

@@ -354,6 +354,23 @@ describe('mockingoose', () => {
 					});
 			});
 		});
+
+		it(`save calls its hook correctly`, () => {
+			const mocked = {
+				name: 'save',
+				email: '[email protected]'
+			};
+
+			mockingoose.User.toReturn(null, 'save');
+
+			User.create(mocked).then(user => {
+				expect(user.saveCount).toBe(1);
+				user.name = 'save2';
+				user.save((err, user) => {
+					expect(user.saveCount).toBe(2);
+				})
+			});
+		});
 	});
 
 	describe('check all operations', () => {
@@ -496,4 +513,3 @@ describe('mockingoose', () => {
 		});
 	});
 });
-

src/index.js

@@ -123,8 +123,43 @@ instance.forEach(methodName => {
 
     Object.assign(this, { op, model: { modelName } });
 
-    return mockedReturn.call(this, cb);
-  })
+    const hooks = this.constructor.hooks
+
+    return new Promise((resolve, reject) => {
+      hooks.execPre(op, this, [cb], (err) => {
+        if (err) {
+          reject(err);
+          return;
+        }
+
+        const ret = mockedReturn.call(this, cb);
+
+        if (cb) {
+          hooks.execPost(op, this, [ret], (err) => {
+            if (err) {
+              reject(err);
+              return;
+            }
+
+            resolve(ret);
+          });
+        } else {
+          ret
+            .then((ret) => {
+              hooks.execPost(op, this, [ret], (err) => {
+                if (err) {
+                  reject(err);
+                  return;
+                }
+
+                resolve(ret);
+              });
+            })
+            .catch(reject);
+        }
+      });
+    });
+  });
 });
 
 jest.doMock('mongoose', () => mongoose);