added test for new setData + updated examples accordingly

jsful · Jul 5, 2016 · 9841a82 · 9841a82
1 parent f3c6db7
commit 9841a82
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 12 deletions.
diff --git a/example/react-counter/components/counter-control.js b/example/react-counter/components/counter-control.js
@@ -3,13 +3,13 @@ import Treeful from '../../../src/treeful';
 
 export default class CounterControl extends Component {
 	increment() {
-		Treeful.setData('count', Treeful.getData('count') + 1);
+		Treeful.setData('count', (e) => e + 1);
 		// Or you may use a helper function
 		// Treeful.incrementData('count');
 	}
 
 	decrement() {
-		Treeful.setData('count', Treeful.getData('count') - 1);
+		Treeful.setData('count', (e) => e - 1);
 		// Or you may use a helper function
 		// Treeful.decrementData('count');
 	}

diff --git a/example/react-todo/components/todo-controller.js b/example/react-todo/components/todo-controller.js
@@ -3,16 +3,15 @@ import Treeful from '../../../src/treeful';
 
 export default class TodoController extends Component {
 	addTodo() {
-		const data = Treeful.getData('todos');
-
-		data.push({ 
-			name: this.refs.name.value,
-			color: this.refs.color.value,
-			id: data.length + 1
+		Treeful.setData('todos', (e) => {
+			e.push({
+				name: this.refs.name.value,
+				color: this.refs.color.value,
+				id: e.length + 1
+			});
+			return e;
 		});
 
-		Treeful.setData('todos', data);
-
 		// Or you may use a helper function
 
 		// Treeful.pushData('todos', {

diff --git a/example/vanilla-counter/index.js b/example/vanilla-counter/index.js
@@ -5,13 +5,13 @@ Treeful.addNode('count', 0);
 let unsubscribe;
 
 const onClickInc = () => {
-	Treeful.setData('count', Treeful.getData('count') + 1);
+	Treeful.setData('count', (e) => e + 1);
 	// Or you may use a helper function
 	// Treeful.incrementData('count');
 };
 
 const onClickDec = () => {
-	Treeful.setData('count', Treeful.getData('count') - 1);
+	Treeful.setData('count', (e) => e - 1);
 	// Or you may use a helper function
 	// Treeful.decrementData('count');
 };

diff --git a/src/treeful.js b/src/treeful.js
@@ -199,6 +199,12 @@ class Treeful {
 			}
 		};
 
+		const checkIfDataIsFunction = (data) => {
+			if(isType(data, 'function')) {
+				throw new TypeError('Data cannot be a function.');
+			}
+		};
+
 		const checkDataType = (data, type) => {
 			if(!isType(data, type)) {
 				throw new TypeError('Data type must be a(n) ' + type + '.');

diff --git a/test/treeful.test.js b/test/treeful.test.js
@@ -86,6 +86,15 @@ describe('treeful', () => {
 		Treeful.destroy();
 	});
 
+	it('accepts function and sets data of a node', () => {
+		Treeful.addNode('1', 1);
+		Treeful.setData('1', (e) => {
+			return e + 1;
+		});
+		expect(Treeful.getData('1')).toEqual(2);
+		Treeful.destroy();
+	});
+
 	/** subscribe **/
 
 	it('calls callback functions when a node\'s data is changed, and passes data', () => {