fix(react-components): update viewportAdapter tests for month and min…

…utes
awslabs · Nov 1, 2023 · a269626 · a269626
1 parent fb2caed
commit a269626
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/packages/react-components/src/components/time-sync/viewportAdapter.spec.ts b/packages/react-components/src/components/time-sync/viewportAdapter.spec.ts
@@ -152,7 +152,9 @@ describe('getViewportStartOnBackwardRelative', () => {
       type: 'relative',
     });
 
-    expect(currentDate.getTime() - newDate.getTime()).toEqual(300000);
+    const result = currentDate.getTime() - newDate.getTime();
+    expect(result).toBeGreaterThanOrEqual(299999);
+    expect(result).toBeLessThanOrEqual(300000);
 
     currentDate = new Date();
     newDate = getViewportStartOnBackwardRelative({
@@ -193,6 +195,13 @@ describe('getViewportStartOnBackwardRelative', () => {
       unit: 'month',
       type: 'relative',
     });
-    expect(currentDate.getTime() - newDate.getTime()).toEqual(2678400000);
+
+    if ([0, 1, 3, 5, 6, 7, 8, 10].includes(currentDate.getMonth())) {
+      expect(currentDate.getTime() - newDate.getTime()).toEqual(2678400000); //previous month is 31 days
+    } else if (currentDate.getMonth() === 2) {
+      expect(currentDate.getTime() - newDate.getTime()).toEqual(2419200000 || 2505600000); //previous month is february
+    } else {
+      expect(currentDate.getTime() - newDate.getTime()).toEqual(2592000000); //previous month is 30 days
+    }
   });
 });