Merge pull request #19 from Santinell/master

Added support of pre\post hooks

Author: Alex Ford

.travis.yml

@@ -3,5 +3,7 @@ services:
   - mongodb
 node_js:
   - "0.10"
+  - "0.12"
+  - "4.2"
 notifications:
   email: false

index.js

@@ -130,6 +130,17 @@ module.exports = exports = {
                             schema.plugin(record.plugin, record.options);
                         }
 
+                        //Add pre hooks
+                        for (var key in modelData.pre) {
+                            var record = modelData.pre[key];
+                            schema.pre(key, record);
+                        }
+
+                        //Add post hooks
+                        for (var key in modelData.post) {
+                            var record = modelData.post[key];
+                            schema.post(key, record);
+                        }
 
                         // If autoIncrementIds:true then utilize mongoose-auto-increment plugin for this model.
                         if (settings.autoIncrementNumberIds) {

test/dbmodels/Author.js

@@ -8,9 +8,25 @@ exports.schema = {
     },
     birthday: Date,
     gender: String,
-    bio: String
+    bio: String,
+    updateCounter: {type: Number, default: 0}
 };
 
+
+exports.pre = {
+    save: function (next) {
+        this.updateCounter++;
+        next();
+    }
+};
+
+exports.post = {
+    save: function (record) {
+        console.log("Post-hook is works!");
+    }
+};
+
+
 exports.methods = {
     formatBirthday: function (formatString) {
         return moment(this.birthday).format(formatString);
@@ -46,4 +62,4 @@ exports.virtuals = {
                 this.name.first = fullName;
         }
     }
-}
+}

test/test.js

@@ -48,7 +48,7 @@ describe("simpledb", function () {
         // We aren't specifying a models directory and the default models directory doesn't exist so this
         // should fail with a "readdir" error.
         should.exist(err);
-        ['ENOENT, readdir', 'ENOENT, scandir'].should.include(err.message.substr(0, 15));
+        ['ENOENT, readdir', 'ENOENT, scandir', 'ENOENT: no such'].should.include(err.message.substr(0, 15));
         done();
       });
     });
@@ -58,7 +58,7 @@ describe("simpledb", function () {
         // We aren't specifying a models directory and the default models directory doesn't exist so this
         // should fail with a "readdir" error.
         should.exist(err);
-        ['ENOENT, readdir', 'ENOENT, scandir'].should.include(err.message.substr(0, 15));
+        ['ENOENT, readdir', 'ENOENT, scandir', 'ENOENT: no such'].should.include(err.message.substr(0, 15));
         done();
       });
     });
@@ -194,6 +194,30 @@ describe("simpledb", function () {
 
     });
 
+    it("should correct work with pre and post hooks", function (done) {
+
+      simpledb.init(options, function (err, db) {
+
+        var author = new db.Author({
+          name: {
+            first: 'Alex',
+            last: 'Ford'
+          },
+          birthday: new Date('3/2/1987')
+        });
+        author.updateCounter.should.equal(0);
+        author.save();
+
+        db.Author.findOne(function (err, record) {
+          record.updateCounter.should.equal(1);
+          done();
+        });
+
+      });
+
+
+    });
+
     it("should correct set settings of mongoose-auto-increment plugin", function (done) {
 
       var localOptions = extend({}, options, { modelsDir: path.join(__dirname, 'pluginsmodels'), autoIncrementSettings: { startAt: 5, field: 'id' } });