Change API

Author: Philipp Alferov

index.js

@@ -33,11 +33,15 @@ function orderByParents(list) {
   return parents;
 }
 
-module.exports = function(obj) {
-  if (!util.isArray(obj)) {
+module.exports = function(options) {
+  var config = util._extend({ parentProperty: 'parent_id', data: [] }, options);
+  var data = config.data;
+
+  if (!util.isArray(data)) {
     throw new Error('Expected an object but got an invalid argument');
   }
-  var cloned = obj.slice();
+
+  var cloned = data.slice();
   var ordered = orderByParents(cloned);
   return createTree(ordered, ordered[0]);
 };

test/test.js

@@ -10,7 +10,7 @@ describe('parent pointer array to tree', function() {
   describe('expected behavior', function() {
 
     before(function() {
-      current = toTree(initial);
+      current = toTree({ data: initial });
     });
 
     it('should not modify passed object', function() {
@@ -44,12 +44,12 @@ describe('parent pointer array to tree', function() {
 
   describe('with incorrect arguments', function() {
     it('should return an empty array if the empty array passed', function() {
-      expect(toTree([])).to.be.deep.equal([]);
+      expect(toTree({ data: [] })).to.be.deep.equal([]);
     });
 
     it('should throw an error if wrong arguments passed', function() {
-      expect(toTree.bind(null, 'string')).to.throw(/invalid argument/);
-      expect(toTree.bind(null, {})).to.throw(/invalid argument/);
+      expect(toTree.bind(null, { data: 'string' })).to.throw(/invalid argument/);
+      expect(toTree.bind(null, { data: {} })).to.throw(/invalid argument/);
     });
 
   })