Tree: let the width parameter also be a function that returns an int

1 files changed, 16 insertions, 2 deletions

diff --git a/test/tree.unit.js b/test/tree.unit.js
index dc6c9530..3af144eb 100644
--- a/test/tree.unit.js
+++ b/test/tree.unit.js
@@ -91,6 +91,20 @@ describe("tree.js", function () {
             Faker.random.first_name.restore();
         });
 
-    });
+        it("can accept a function for the width", function () {
+            var widthFuncCalled = 0;
+            var widthFunc = function () {
+                widthFuncCalled = widthFuncCalled + 1;
+                return 2;
+            };
+
+            var tree = Faker.Tree.createTree(2, widthFunc, proto);
+            assert.equal(widthFuncCalled, 3);
+
+            Faker.random.first_name.restore();
 
-});
+        });
+
+
+    });
+});
\ No newline at end of file