Add done callback

Ryan Schumacher · Ryan Schumacher · commit 9a8af8de7fcc · 2015-01-27T09:57:51.000-05:00
diff --git a/lib/helpers.js b/lib/helpers.js
@@ -27,13 +27,17 @@ _beforeEach.withApp = function(app) {
 }
 
 _beforeEach.cleanDatasource = function(dsName) {
-  beforeEach(function() {
+  beforeEach(function(done) {
     if(!dsName) dsName = 'db';
 
-    if (this.app) {
+    if (typeof this.app === 'function'
+        && typeof this.app.datasources === 'object'
+        && typeof this.app.datasources[dsName] === 'object') {
       this.app.datasources[dsName].automigrate();
       this.app.datasources[dsName].connector.ids = {};
     }
+    
+    done();
   });
 }
 

Original file line number	Diff line number	Diff line change
`@@ -27,13 +27,17 @@ _beforeEach.withApp = function(app) {`
`27`	`27`	`}`
`28`	`28`
`29`	`29`	`_beforeEach.cleanDatasource = function(dsName) {`
`30`		`- beforeEach(function() {`
	`30`	`+ beforeEach(function(done) {`
`31`	`31`	`if(!dsName) dsName = 'db';`
`32`	`32`
`33`		`- if (this.app) {`
	`33`	`+ if (typeof this.app === 'function'`
	`34`	`+ && typeof this.app.datasources === 'object'`
	`35`	`+ && typeof this.app.datasources[dsName] === 'object') {`
`34`	`36`	`this.app.datasources[dsName].automigrate();`
`35`	`37`	`this.app.datasources[dsName].connector.ids = {};`
`36`	`38`	`}`
	`39`	`+`
	`40`	`+ done();`
`37`	`41`	`});`
`38`	`42`	`}`
`39`	`43`