Skip to content

Commit 9e96e6a

Browse files
committed
Switched to vi.useFakeTimers() from spy
1 parent e8091fd commit 9e96e6a

File tree

2 files changed

+13
-8
lines changed

2 files changed

+13
-8
lines changed

web-ui/src/pages/AnniversaryReportPage.test.jsx

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -14,26 +14,28 @@ const userStateWithPermission = {
1414

1515
it('renders correctly', () => {
1616
const mockDate = new Date(2022, 1, 1);
17-
const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
17+
vi.useFakeTimers();
18+
vi.setSystemTime(mockDate);
1819

1920
snapshot(
2021
<AppContextProvider value={userStateWithPermission}>
2122
<AnniversaryReportPage />
2223
</AppContextProvider>
2324
);
2425

25-
spy.mockRestore();
26+
vi.useRealTimers();
2627
});
2728

2829
it('renders an error if user does not have appropriate permission', () => {
2930
const mockDate = new Date(2022, 1, 1);
30-
const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
31+
vi.useFakeTimers();
32+
vi.setSystemTime(mockDate);
3133

3234
snapshot(
3335
<AppContextProvider>
3436
<AnniversaryReportPage />
3537
</AppContextProvider>
3638
);
3739

38-
spy.mockRestore();
40+
vi.useRealTimers();
3941
});

web-ui/src/pages/BirthdayReportPage.test.jsx

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
import React from 'react';
2+
import { vi } from 'vitest';
23
import BirthdayReportPage from './BirthdayReportPage';
34
import { AppContextProvider } from '../context/AppContext';
45

@@ -14,26 +15,28 @@ const userStateWithPermission = {
1415

1516
it('renders correctly', () => {
1617
const mockDate = new Date(2022, 1, 1);
17-
const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
18+
vi.useFakeTimers();
19+
vi.setSystemTime(mockDate);
1820

1921
snapshot(
2022
<AppContextProvider value={userStateWithPermission}>
2123
<BirthdayReportPage />
2224
</AppContextProvider>
2325
);
2426

25-
spy.mockRestore();
27+
vi.useRealTimers();
2628
});
2729

2830
it('renders an error if user does not have appropriate permission', () => {
2931
const mockDate = new Date(2022, 1, 1);
30-
const spy = vi.spyOn(global, 'Date').mockImplementation(() => mockDate);
32+
vi.useFakeTimers();
33+
vi.setSystemTime(mockDate);
3134

3235
snapshot(
3336
<AppContextProvider>
3437
<BirthdayReportPage />
3538
</AppContextProvider>
3639
);
3740

38-
spy.mockRestore();
41+
vi.useRealTimers();
3942
});

0 commit comments

Comments
 (0)