use shallow in tests

airbnb · May 29, 2020 · 619a003 · 619a003
1 parent 0e40447
commit 619a003
Showing 1 changed file with 10 additions and 6 deletions.
diff --git a/packages/vx-tooltip/test/TooltipWithBounds.test.tsx b/packages/vx-tooltip/test/TooltipWithBounds.test.tsx
@@ -1,23 +1,27 @@
 import React from 'react';
-import { mount } from 'enzyme';
-import { TooltipWithBounds, defaultStyles, Tooltip } from '../src';
+import { shallow } from 'enzyme';
+import { TooltipWithBounds, defaultStyles } from '../src';
 
 describe('<TooltipWithBounds />', () => {
   test('it should be defined', () => {
     expect(TooltipWithBounds).toBeDefined();
   });
 
   it('should render the Tooltip with default styles by default', () => {
-    const wrapper = mount(<TooltipWithBounds>Hello</TooltipWithBounds>);
-    const styles = wrapper.find(Tooltip).props().style as any;
+    const wrapper = shallow(<TooltipWithBounds>Hello</TooltipWithBounds>, {
+      disableLifecycleMethods: true,
+    }).dive();
+    const styles = wrapper.find('Tooltip').props().style as any;
     Object.entries(defaultStyles).forEach(([key, value]) => {
       expect(styles[key]).toBe(value);
     });
   });
 
   it('should render the tooltip without default styles if unstyled is set to true', () => {
-    const wrapper = mount(<TooltipWithBounds unstyled>Hello</TooltipWithBounds>);
-    const styles = wrapper.find(Tooltip).props().style as any;
+    const wrapper = shallow(<TooltipWithBounds unstyled>Hello</TooltipWithBounds>, {
+      disableLifecycleMethods: true,
+    }).dive();
+    const styles = wrapper.find('Tooltip').props().style as any;
     Object.keys(defaultStyles).forEach(key => {
       expect(styles[key]).toBeUndefined();
     });