improve state and method chaining

HBM · Sep 2, 2016 · 206f429 · 206f429
1 parent 06247d0
commit 206f429
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/lib/jet/peer/method.js b/lib/jet/peer/method.js
@@ -24,6 +24,7 @@ Method.prototype.on = function (event, cb) {
     } else {
       this._dispatcher = this.createAsyncDispatcher(cb)
     }
+    return this
   } else {
     throw new Error('event not available')
   }

diff --git a/lib/jet/peer/state.js b/lib/jet/peer/state.js
@@ -138,6 +138,7 @@ State.prototype.on = function (event, cb) {
     } else {
       this._dispatcher = this.createAsyncDispatcher(cb)
     }
+    return this
   } else {
     throw new Error('event not available')
   }

diff --git a/test/peer-test.js b/test/peer-test.js
@@ -154,6 +154,16 @@ describe('Jet module', function () {
       expect(state.path).to.be.a('function')
     })
 
+    it('new jet.State().on returns self', function () {
+      var state = new jet.State(randomPath(), 123)
+      expect(state.on('set', function () {})).to.equal(state)
+    })
+
+    it('new jet.Method().on returns self', function () {
+      var method = new jet.Method(randomPath())
+      expect(method.on('call', function () {})).to.equal(method)
+    })
+
     it('can add, fetch and set a state', function (done) {
       var random = randomPath()
       var state = new jet.State(random, 123)