From 5aa4062dcabc2fdac1c93df09e71d702ec7ec81e Mon Sep 17 00:00:00 2001 From: cyanos3 Date: Tue, 5 Nov 2013 17:44:10 -0700 Subject: Tree: let the width parameter also be a function that returns an int --- test/tree.unit.js | 18 ++++++++++++++++-- 1 file changed, 16 insertions(+), 2 deletions(-) (limited to 'test') 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 -- cgit v1.2.3