-
Notifications
You must be signed in to change notification settings - Fork 2
/
Utils.test.tsx
81 lines (66 loc) · 2.34 KB
/
Utils.test.tsx
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
import { describe, test, expect, vi, afterEach } from "vitest";
import { DataPoller } from "./Poller/DataPoller";
describe("DataPoller", () => {
const fetcher = vi.fn();
const onPoll = vi.fn();
const createPoller = (interval: number, pollAttempts: number) => {
return new DataPoller({
fetcher,
onPoll,
interval,
pollAttempts,
});
};
afterEach(() => {
vi.clearAllMocks();
});
test("should resolve with correctDataStateFound true when correct data state is found", async () => {
fetcher.mockResolvedValueOnce("invalid value");
onPoll.mockReturnValueOnce(false);
fetcher.mockResolvedValueOnce("valid value");
onPoll.mockReturnValueOnce(true);
const poller = createPoller(50, 5);
const result = await poller.startPollingData();
expect(result).toEqual({
correctDataStateFound: true,
maxAttemptsReached: false,
});
expect(fetcher).toHaveBeenCalledTimes(2);
expect(onPoll).toHaveBeenCalledTimes(2);
});
test("should resolve with maxAttemptsReached true when max attempts reached", async () => {
fetcher.mockResolvedValue("invalid value");
onPoll.mockReturnValue(false);
const poller = createPoller(50, 5);
const result = await poller.startPollingData();
expect(result).toEqual({
correctDataStateFound: false,
maxAttemptsReached: true,
});
expect(fetcher).toHaveBeenCalledTimes(5);
expect(onPoll).toHaveBeenCalledTimes(5);
});
test("should stop polling when correct data state is found before reaching max attempts", async () => {
fetcher.mockResolvedValueOnce("invalid value");
onPoll.mockReturnValueOnce(false);
fetcher.mockResolvedValueOnce("valid value");
onPoll.mockReturnValueOnce(true);
const poller = createPoller(50, 3);
const result = await poller.startPollingData();
expect(result).toEqual({
correctDataStateFound: true,
maxAttemptsReached: false,
});
expect(fetcher).toHaveBeenCalledTimes(2);
expect(onPoll).toHaveBeenCalledTimes(2);
});
test("should stop polling when interval is reached", async () => {
setInterval(() => onPoll.mockReturnValue(true), 300);
const poller = createPoller(50, 3);
const result = await poller.startPollingData();
expect(result).toEqual({
correctDataStateFound: false,
maxAttemptsReached: true,
});
});
});