added "migration" event

tj · Apr 22, 2011 · 54b0b79 · 54b0b79
1 parent 491c432
commit 54b0b79
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/lib/set.js b/lib/set.js
@@ -146,6 +146,7 @@ Set.prototype._migrate = function(direction, fn){
       return;
     }
 
+    self.emit('migration', migration, direction);
     migration[direction](function(err){
       next(err, migrations.shift());
     });

diff --git a/test/test.migrate.js b/test/test.migrate.js
@@ -120,9 +120,26 @@ function testMigrationEvents() {
     next();
   });
 
+  var migrations = []
+    , expectedMigrations = [
+      'add guy ferrets'
+    , 'add girl ferrets'
+    , 'add emails'
+    , 'add dogs'
+    , 'adjust emails'];
+
+  set.on('migration', function(migration, direction){
+    migrations.push(migration.title);
+    direction.should.be.a('string');
+  });
+
   set.up(function(){
     db.pets[0].email.should.equal('tobi@lb.com');
+    migrations.should.eql(expectedMigrations);
+
+    migrations = [];
     set.down(function(){
+      migrations.should.eql(expectedMigrations.reverse());
       assertNoPets();
     });
   });