Make setGraph chainable.

webpro-patch · Sep 24, 2014 · 051eecc · 051eecc
1 parent 3bf0a4d
commit 051eecc
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 0 deletions.
diff --git a/lib/base-graph.js b/lib/base-graph.js
@@ -90,6 +90,7 @@ BaseGraph.prototype.setGraph = function(label) {
   } else {
     delete this._graphLabel;
   }
+  return this;
 };
 
 BaseGraph.prototype.updateGraph = function(fn) {

diff --git a/test/base-graph-test.js b/test/base-graph-test.js
@@ -34,6 +34,11 @@ exports.tests = function(GraphConstructor) {
         g.setGraph();
         expect(g.getGraph()).to.be.undefined;
       });
+
+      it("is chainable", function() {
+        var g2 = g.setGraph("foo");
+        expect(g).to.equal(g2);
+      });
     });
 
     describe("updateGraph", function() {
@@ -48,6 +53,11 @@ exports.tests = function(GraphConstructor) {
         g.updateGraph(function() {});
         expect(g.getGraph()).to.be.undefined;
       });
+
+      it("is chainable", function() {
+        var g2 = g.updateGraph(function(old) { return old + "-new"; });
+        expect(g).to.equal(g2);
+      });
     });
 
     describe("hasNode", function() {