Improve unit test stability

wojtekmaj · Mar 2, 2023 · cd7ef12 · cd7ef12
1 parent a8aab0d
commit cd7ef12
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 17 deletions.
diff --git a/src/Calendar/Navigation.spec.jsx b/src/Calendar/Navigation.spec.jsx
@@ -20,29 +20,29 @@ describe('Navigation', () => {
   it('renders prev2, prev, drill up, next and next2 buttons', () => {
     const { container } = render(<Navigation {...defaultProps} view="month" />);
 
-    const children = [...container.firstChild.children];
+    const children = [...container.firstElementChild.children];
 
     const [prev2, prev, drillUp, next, next2] = children;
 
     expect(children).toHaveLength(5);
-    expect(prev2.type).toBe('button');
-    expect(prev.type).toBe('button');
-    expect(drillUp.type).toBe('button');
-    expect(next.type).toBe('button');
-    expect(next2.type).toBe('button');
+    expect(prev2.getAttribute('type')).toBe('button');
+    expect(prev.getAttribute('type')).toBe('button');
+    expect(drillUp.getAttribute('type')).toBe('button');
+    expect(next.getAttribute('type')).toBe('button');
+    expect(next2.getAttribute('type')).toBe('button');
   });
 
   it('renders prev, drill up, next and buttons only for century view', () => {
     const { container } = render(<Navigation {...defaultProps} view="century" />);
 
-    const children = [...container.firstChild.children];
+    const children = [...container.firstElementChild.children];
 
     const [prev, drillUp, next] = children;
 
     expect(children).toHaveLength(3);
-    expect(prev.type).toBe('button');
-    expect(drillUp.type).toBe('button');
-    expect(next.type).toBe('button');
+    expect(prev.getAttribute('type')).toBe('button');
+    expect(drillUp.getAttribute('type')).toBe('button');
+    expect(next.getAttribute('type')).toBe('button');
   });
 
   it('displays proper title for month view', () => {
@@ -97,7 +97,7 @@ describe('Navigation', () => {
       />,
     );
 
-    const [prev2, prev, , next, next2] = [...container.firstChild.children];
+    const [prev2, prev, , next, next2] = [...container.firstElementChild.children];
 
     expect(prev2).toHaveTextContent('prev2Label');
     expect(prev).toHaveTextContent('prevLabel');
@@ -110,7 +110,7 @@ describe('Navigation', () => {
       <Navigation {...defaultProps} navigationAriaLive="polite" view="month" />,
     );
 
-    const [, , navigation] = [...container.firstChild.children];
+    const [, , navigation] = [...container.firstElementChild.children];
 
     expect(navigation).toHaveAttribute('aria-live', 'polite');
   });
@@ -128,7 +128,7 @@ describe('Navigation', () => {
       />,
     );
 
-    const [prev2, prev, navigation, next, next2] = [...container.firstChild.children];
+    const [prev2, prev, navigation, next, next2] = [...container.firstElementChild.children];
 
     expect(prev2).toHaveAccessibleName('prev2AriaLabel');
     expect(prev).toHaveAccessibleName('prevAriaLabel');

diff --git a/src/Flex.spec.jsx b/src/Flex.spec.jsx
@@ -13,7 +13,7 @@ describe('Flex', () => {
       </Flex>,
     );
 
-    const wrapper = container.firstChild;
+    const wrapper = container.firstElementChild;
 
     expect(wrapper).toHaveStyle('display: flex');
     expect(wrapper).toHaveStyle('flex-wrap: nowrap');
@@ -28,7 +28,7 @@ describe('Flex', () => {
       </Flex>,
     );
 
-    const wrapper = container.firstChild;
+    const wrapper = container.firstElementChild;
 
     expect(wrapper).toHaveStyle('display: flex');
     expect(wrapper).toHaveStyle('flex-wrap: wrap');
@@ -43,7 +43,7 @@ describe('Flex', () => {
       </Flex>,
     );
 
-    const children = [...container.firstChild.children];
+    const children = [...container.firstElementChild.children];
 
     expect(children).toHaveLength(3);
     expect(children[0]).toHaveTextContent('Hey');
@@ -59,7 +59,7 @@ describe('Flex', () => {
       </Flex>,
     );
 
-    const children = [...container.firstChild.children];
+    const children = [...container.firstElementChild.children];
 
     children.forEach((child) => expect(child).toHaveStyle('flex-basis: 33.333333333333336%'));
     expect(children[0]).toHaveStyle('margin-left: 33.333333333333336%');