Updates tests for depthFactor fix

Author: bkrem

src/Node/tests/index.test.js

@@ -143,21 +143,6 @@ describe('<Node />', () => {
   });
 
 
-  it('mutates the node\'s `y` property according to `depthFactor`, when specified', () => {
-    const depthFactor = 100;
-    const expectedY = nodeData.depth * depthFactor;
-    const renderedComponent = shallow(
-      <Node
-        {...mockProps}
-        orientation="vertical"
-        depthFactor={depthFactor}
-      />
-    );
-
-    expect(renderedComponent.prop('transform')).toBe(`translate(${nodeData.parent.x},${expectedY})`);
-  });
-
-
   it('applies its own x/y coords on `transform` once mounted', () => {
     const fixture = `translate(${nodeData.y},${nodeData.x})`;
     const renderedComponent = mount(

src/Tree/tests/index.test.js

@@ -130,6 +130,24 @@ describe('<Tree />', () => {
   });
 
 
+  it('mutates each node\'s `y` prop according to `depthFactor` when specified', () => {
+    const depthFactor = 100;
+    // const expectedY = nodeData.depth * depthFactor;
+    const renderedComponent = shallow(
+      <Tree
+        data={mockData}
+        orientation="vertical"
+        depthFactor={depthFactor}
+      />
+    );
+
+    const { nodes } = renderedComponent.instance().generateTree(mockData);
+    nodes.forEach((node) => {
+      expect(node.y).toBe(node.depth * depthFactor);
+    });
+  });
+
+
   it('passes `props.orientation` to its <Node /> and <Link /> children', () => {
     const fixture = 'vertical';
     const renderedComponent = shallow(