Make removeEdge chainable

kkowar · Sep 24, 2014 · 5405c82 · 5405c82
1 parent 3e1d0ce
commit 5405c82
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/lib/graph.js b/lib/graph.js
@@ -380,6 +380,7 @@ Graph.prototype.removeEdge = function(v, w, name) {
     delete this._out[v][e];
     this._edgeCount--;
   }
+  return this;
 };
 
 function decrementOrRemoveEntry(map, k) {

diff --git a/test/graph-test.js b/test/graph-test.js
@@ -642,6 +642,11 @@ describe("Graph", function() {
       expect(g.predecessors("b")).to.eql(["a"]);
       expect(g.neighbors("b")).to.eql(["a"]);
     });
+
+    it("is chainable", function() {
+      g.setEdge("a", "b");
+      expect(g.removeEdge("a", "b")).to.equal(g);
+    });
   });
 
   describe("inEdges", function() {