Duck type Promise type - Fix #24

Author: SBoudrias

lib/run-context.js

@@ -126,6 +126,25 @@ RunContext.prototype.toPromise = function () {
   }.bind(this));
 };
 
+/**
+ * Promise `.then()` duck typing
+ * @return {Promise}
+ */
+RunContext.prototype.then = function () {
+  var promise = this.toPromise();
+  return promise.then.apply(promise, arguments);
+};
+
+
+/**
+ * Promise `.catch()` duck typing
+ * @return {Promise}
+ */
+RunContext.prototype.catch = function () {
+  var promise = this.toPromise();
+  return promise.catch.apply(promise, arguments);
+};
+
 /**
  * Clean the provided directory, then change directory into it
  * @param  {String} dirPath - Directory path (relative to CWD). Prefer passing an absolute

test/run-context.js

@@ -227,6 +227,40 @@ describe('RunContext', function () {
     });
   });
 
+  describe('#then()', function () {
+    it('handle success', function () {
+      return this.ctx.then(function (dir) {
+        assert.equal(this.ctx.targetDirectory, dir);
+      }.bind(this));
+    });
+
+    it('handles errors', function () {
+      var error = new Error();
+      var Dummy = helpers.createDummyGenerator();
+      var execSpy = sinon.stub().throws(error);
+      Dummy.prototype.exec = execSpy;
+      var ctx = new RunContext(Dummy);
+
+      return ctx.then(function () {}, function (err) {
+        assert.equal(err, error);
+      });
+    });
+  });
+
+  describe('#catch()', function () {
+    it('handles errors', function () {
+      var error = new Error();
+      var Dummy = helpers.createDummyGenerator();
+      var execSpy = sinon.stub().throws(error);
+      Dummy.prototype.exec = execSpy;
+      var ctx = new RunContext(Dummy);
+
+      return ctx.catch(function (err) {
+        assert.equal(err, error);
+      });
+    });
+  });
+
   describe('#inDir()', function () {
     beforeEach(function () {
       process.chdir(__dirname);