Merge pull request #61 from vdemedes/co-generators

Use `co` for running generator functions

lib/babel.js

@@ -3,7 +3,8 @@
 var resolveFrom = require('resolve-from');
 
 var options = {
-	only: /(test|test\-.+|test\/.+)\.js$/
+	only: /(test|test\-.+|test\/.+)\.js$/,
+	blacklist: ['regenerator']
 };
 
 try {

lib/test.js

@@ -4,6 +4,7 @@ var Promise = require('bluebird');
 var setImmediate = require('set-immediate-shim');
 var fnName = require('fn-name');
 var assert = require('./assert');
+var co = require('co');
 
 function Test(title, fn) {
 	if (!(this instanceof Test)) {
@@ -16,7 +17,7 @@ function Test(title, fn) {
 	}
 
 	this.title = title || fnName(fn) || '[anonymous]';
-	this.fn = isGenerator(fn) ? Promise.coroutine(fn) : fn;
+	this.fn = isGenerator(fn) ? co.wrap(fn) : fn;
 	this.assertCount = 0;
 	this.planCount = null;
 	this.duration = null;

package.json

@@ -63,6 +63,7 @@
     "babel-core": "^5.8.23",
     "bluebird": "^2.9.34",
     "chalk": "^1.0.0",
+    "co": "^4.6.0",
     "figures": "^1.3.5",
     "fn-name": "^2.0.0",
     "globby": "^3.0.1",