Switched to vi.useFakeTimers() from spy

objectcomputing · Nov 5, 2024 · 4399622 · 4399622
1 parent bedd97f
commit 4399622
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 8 deletions.
diff --git a/web-ui/src/pages/AnniversaryReportPage.test.jsx b/web-ui/src/pages/AnniversaryReportPage.test.jsx
@@ -14,26 +14,28 @@ const userStateWithPermission = {
 
 it('renders correctly', () => {
   const mockDate = new Date(2022, 1, 1);
-  const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
+  vi.useFakeTimers();
+  vi.setSystemTime(mockDate);
 
   snapshot(
     <AppContextProvider value={userStateWithPermission}>
       <AnniversaryReportPage />
     </AppContextProvider>
   );
 
-  spy.mockRestore();
+  vi.useRealTimers();
 });
 
 it('renders an error if user does not have appropriate permission', () => {
   const mockDate = new Date(2022, 1, 1);
-  const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
+  vi.useFakeTimers();
+  vi.setSystemTime(mockDate);
 
   snapshot(
     <AppContextProvider>
       <AnniversaryReportPage />
     </AppContextProvider>
   );
 
-  spy.mockRestore();
+  vi.useRealTimers();
 });
diff --git a/web-ui/src/pages/BirthdayReportPage.test.jsx b/web-ui/src/pages/BirthdayReportPage.test.jsx
@@ -1,4 +1,5 @@
 import React from 'react';
+import { vi } from 'vitest';
 import BirthdayReportPage from './BirthdayReportPage';
 import { AppContextProvider } from '../context/AppContext';
 
@@ -14,26 +15,28 @@ const userStateWithPermission = {
 
 it('renders correctly', () => {
   const mockDate = new Date(2022, 1, 1);
-  const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
+  vi.useFakeTimers();
+  vi.setSystemTime(mockDate);
 
   snapshot(
     <AppContextProvider value={userStateWithPermission}>
       <BirthdayReportPage />
     </AppContextProvider>
   );
 
-  spy.mockRestore();
+  vi.useRealTimers();
 });
 
 it('renders an error if user does not have appropriate permission', () => {
   const mockDate = new Date(2022, 1, 1);
-  const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
+  vi.useFakeTimers();
+  vi.setSystemTime(mockDate);
 
   snapshot(
     <AppContextProvider>
       <BirthdayReportPage />
     </AppContextProvider>
   );
 
-  spy.mockRestore();
+  vi.useRealTimers();
 });