Repository navigation
Expand file tree
/
Copy pathgithub.telemetry.test.ts
More file actions
56 lines (47 loc) · 1.92 KB
/
Copy pathgithub.telemetry.test.ts
File metadata and controls
56 lines (47 loc) · 1.92 KB
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
import { describe, it, expect, beforeEach, afterEach } from 'vitest';
import {
getCircuitTelemetry,
fetchWithRetry,
clearGitHubApiCacheForTests,
getTokenStatsForTests,
} from './github';
describe('GitHub Circuit Breaker Telemetry', () => {
const originalGitHubPat = process.env.GITHUB_PAT;
const originalGitHubToken = process.env.GITHUB_TOKEN;
const originalGitHubTokens = process.env.GITHUB_TOKENS;
beforeEach(() => {
clearGitHubApiCacheForTests();
delete process.env.GITHUB_PAT;
delete process.env.GITHUB_TOKEN;
delete process.env.GITHUB_TOKENS;
});
afterEach(() => {
process.env.GITHUB_PAT = originalGitHubPat;
process.env.GITHUB_TOKEN = originalGitHubToken;
process.env.GITHUB_TOKENS = originalGitHubTokens;
});
it('reports closed circuit by default', () => {
const telemetry = getCircuitTelemetry();
expect(telemetry.isOpen).toBe(false);
expect(telemetry.resetInMs).toBe(0);
});
it('reports open circuit with reset time when all tokens are exhausted', async () => {
const mockToken1 = 'ghp_telemetryToken1AAAAAAAAAAAAAAAAAA';
const mockToken2 = 'ghp_telemetryToken2AAAAAAAAAAAAAAAAAA';
process.env.GITHUB_PAT = `${mockToken1},${mockToken2}`;
delete process.env.GITHUB_TOKEN;
const resetTime1 = Date.now() + 5000;
const resetTime2 = Date.now() + 10000;
const tokenStats = getTokenStatsForTests();
tokenStats.set(mockToken1, { remaining: 0, resetTime: resetTime1 });
tokenStats.set(mockToken2, { remaining: 0, resetTime: resetTime2 });
await expect(fetchWithRetry('https://api.github.com/graphql', { headers: {} })).rejects.toThrow(
'API Rate Limit Exceeded'
);
const telemetry = getCircuitTelemetry();
expect(telemetry.isOpen).toBe(true);
expect(telemetry.resetInMs).toBeGreaterThan(0);
// Should be based on the earliest reset time (resetTime1)
expect(telemetry.resetInMs).toBeLessThanOrEqual(5000);
});
});