From cd7ef12d5d1ce4a19d82152ddd73bf385506246e Mon Sep 17 00:00:00 2001
From: Wojciech Maj <kontakt@wojtekmaj.pl>
Date: Wed, 1 Mar 2023 14:40:14 +0000
Subject: [PATCH] Improve unit test stability

---
 src/Calendar/Navigation.spec.jsx | 26 +++++++++++++-------------
 src/Flex.spec.jsx                |  8 ++++----
 2 files changed, 17 insertions(+), 17 deletions(-)

diff --git a/src/Calendar/Navigation.spec.jsx b/src/Calendar/Navigation.spec.jsx
index 22efa6c4..25d27688 100644
--- 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
index c93e9ab9..4841391f 100644
--- 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%');