Adds onClick callback + test (#19)

Author: bkrem

src/Tree/index.js

@@ -189,7 +189,7 @@ export default class Tree extends React.Component {
       targetNode._collapsed
         ? this.expandNode(targetNode)
         : this.collapseNode(targetNode);
-      this.setState({ data });
+      this.setState({ data }, this.props.onClick);
     }
   }
 
@@ -291,6 +291,7 @@ export default class Tree extends React.Component {
 }
 
 Tree.defaultProps = {
+  onClick: undefined,
   orientation: 'horizontal',
   translate: { x: 0, y: 0 },
   pathFunc: 'diagonal',
@@ -322,6 +323,7 @@ Tree.defaultProps = {
 
 Tree.propTypes = {
   data: PropTypes.array.isRequired,
+  onClick: PropTypes.func,
   orientation: PropTypes.oneOf([
     'horizontal',
     'vertical',

src/Tree/tests/index.test.js

@@ -5,58 +5,9 @@ import { shallow, mount } from 'enzyme';
 import Node from '../../Node';
 import Link from '../../Link';
 import Tree from '../index';
+import { mockData, mockData2 } from './mockData';
 
 describe('<Tree />', () => {
-  const mockData = [
-    {
-      name: 'Top Level',
-      parent: 'null',
-      attributes: {
-        keyA: 'val A',
-        keyB: 'val B',
-        keyC: 'val C',
-      },
-      children: [
-        {
-          name: 'Level 2: A',
-          parent: 'Top Level',
-          attributes: {
-            keyA: 'val A',
-            keyB: 'val B',
-            keyC: 'val C',
-          },
-        },
-        {
-          name: 'Level 2: B',
-          parent: 'Top Level',
-        },
-      ],
-    },
-  ];
-
-  const mockData2 = [
-    {
-      name: 'Top Level',
-      parent: 'null',
-      attributes: {
-        keyA: 'val A',
-        keyB: 'val B',
-        keyC: 'val C',
-      },
-      children: [
-        {
-          name: 'Level 2: A',
-          parent: 'Top Level',
-          attributes: {
-            keyA: 'val A',
-            keyB: 'val B',
-            keyC: 'val C',
-          },
-        },
-      ],
-    },
-  ];
-
   jest.spyOn(Tree.prototype, 'generateTree');
   jest.spyOn(Tree.prototype, 'assignInternalProperties');
   jest.spyOn(Tree.prototype, 'handleNodeToggle');
@@ -243,4 +194,19 @@ describe('<Tree />', () => {
     expect(zoomableComponent.find('.rd3t-tree-container').hasClass('rd3t-grabbable')).toBe(true);
     expect(nonZoomableComponent.find('.rd3t-tree-container').hasClass('rd3t-grabbable')).toBe(false);
   });
+
+
+  it('calls the onClick callback when a node is toggled', () => {
+    const onClickSpy = jest.fn();
+    const renderedComponent = mount(
+      <Tree
+        data={mockData}
+        onClick={onClickSpy}
+      />
+    );
+
+    renderedComponent.find(Node).first().simulate('click');
+
+    expect(onClickSpy).toHaveBeenCalledTimes(1);
+  });
 });

src/Tree/tests/mockData.js

@@ -0,0 +1,54 @@
+const mockData = [
+  {
+    name: 'Top Level',
+    parent: 'null',
+    attributes: {
+      keyA: 'val A',
+      keyB: 'val B',
+      keyC: 'val C',
+    },
+    children: [
+      {
+        name: 'Level 2: A',
+        parent: 'Top Level',
+        attributes: {
+          keyA: 'val A',
+          keyB: 'val B',
+          keyC: 'val C',
+        },
+      },
+      {
+        name: 'Level 2: B',
+        parent: 'Top Level',
+      },
+    ],
+  },
+];
+
+const mockData2 = [
+  {
+    name: 'Top Level',
+    parent: 'null',
+    attributes: {
+      keyA: 'val A',
+      keyB: 'val B',
+      keyC: 'val C',
+    },
+    children: [
+      {
+        name: 'Level 2: A',
+        parent: 'Top Level',
+        attributes: {
+          keyA: 'val A',
+          keyB: 'val B',
+          keyC: 'val C',
+        },
+      },
+    ],
+  },
+];
+
+export {
+  mockData,
+  mockData2,
+};