Update: Add test for a custom registry's set method returning undefined

Author: phated

test/registry.js

@@ -20,6 +20,18 @@ CustomRegistry.prototype.get = noop;
 CustomRegistry.prototype.set = noop;
 CustomRegistry.prototype.tasks = noop;
 
+function SetNoReturnRegistry() {
+  this._tasks = {};
+}
+SetNoReturnRegistry.prototype.init = noop;
+SetNoReturnRegistry.prototype.get = function(name) {
+  return this.tasks[name];
+};
+SetNoReturnRegistry.prototype.set = function(name, fn) {
+  this.tasks[name] = fn;
+};
+SetNoReturnRegistry.prototype.tasks = noop;
+
 function InvalidRegistry() {}
 
 describe('registry', function() {
@@ -175,4 +187,15 @@ describe('registry', function() {
     });
   });
 
+  it('does not require the `set` method to return a task', function(done) {
+    var taker = new Undertaker();
+    taker.registry(new SetNoReturnRegistry());
+    taker.task('test', noop);
+    taker.on('start', function(data) {
+      expect(data.name).to.equal('test');
+      done();
+    });
+    taker.series('test')();
+  });
+
 });