fix test

Author: DocX

src/index.js

@@ -81,18 +81,22 @@ class ChartComponent extends React.Component {
       return true;
     }
 
-    const nextData = this.transformDataProp()
+    const nextData = this.transformDataProp(nextProps)
     return !isEqual(this.shadowDataProp, nextData);
   }
 
   componentWillUnmount() {
     this.chart_instance.destroy();
   }
 
-  transformDataProp() {
-    const { data } = this.props;
-    const node = ReactDOM.findDOMNode(this);
-    return (typeof(data) == "function") ? data(node) : data;
+  transformDataProp(props) {
+    const { data } = props;
+    if (typeof(data) == "function") {
+      const node = ReactDOM.findDOMNode(this);
+      return data(node)
+    } else {
+      return data;
+    }
   }
 
   // Chart.js directly mutates the data.dataset objects by adding _meta proprerty
@@ -104,7 +108,7 @@ class ChartComponent extends React.Component {
       return;
     }
 
-    const data = this.transformDataProp();
+    const data = this.transformDataProp(this.props);
 
     this.shadowDataProp = {
       ...data,
@@ -117,7 +121,7 @@ class ChartComponent extends React.Component {
   updateChart() {
     const {options} = this.props;
 
-    const data = this.memoizeDataProps();
+    const data = this.memoizeDataProps(this.props);
 
     if (!this.chart_instance) return;
 

test/__tests__/Chart_spec.js

@@ -209,4 +209,17 @@ describe('<Chart />', () => {
 
     expect(onElementsClick.called).to.equal(true);
   });
+
+  describe('props.data function', () => {
+    it('calls data func with canvas node', () => {
+      const resultData = { test: 1 }
+      const dataFn = sinon.spy((canvas) => resultData);
+      const wrapper = mountComponent({ data: dataFn });
+
+      const canvas = wrapper.find('canvas').at(0).node;
+
+      expect(dataFn.callCount).to.equal(1);
+      expect(dataFn.calledWith(canvas)).to.equal(true);
+    });
+  })
 });