watson/umami
1
0
Fork 0
mirror of https://github.com/umami-software/umami.git synced 2026-02-04 04:37:11 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Fixed tests.

Browse source
This commit is contained in:
Mike Cao 2025-02-01 20:04:43 -08:00
parent 6ab7746b7d
commit 7e2d257293
2 changed files with 5 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

8
src/lib/__tests__/detect.test.ts
View file

@ -6,17 +6,17 @@ const IP = '127.0.0.1';
test('getIpAddress: Custom header', () => { test('getIpAddress: Custom header', () => {
process.env.CLIENT_IP_HEADER = 'x-custom-ip-header'; process.env.CLIENT_IP_HEADER = 'x-custom-ip-header';
expect(detect.getIpAddress({ headers: { 'x-custom-ip-header': IP } } as any)).toEqual(IP); expect(detect.getIpAddress(new Headers({ 'x-custom-ip-header': IP }))).toEqual(IP);
}); });
test('getIpAddress: CloudFlare header', () => { test('getIpAddress: CloudFlare header', () => {
expect(detect.getIpAddress({ headers: { 'cf-connecting-ip': IP } } as any)).toEqual(IP); expect(detect.getIpAddress(new Headers({ 'cf-connecting-ip': IP }))).toEqual(IP);
}); });
test('getIpAddress: Standard header', () => { test('getIpAddress: Standard header', () => {
expect(detect.getIpAddress({ headers: { 'x-forwarded-for': IP } } as any)).toEqual(IP); expect(detect.getIpAddress(new Headers({ 'x-forwarded-for': IP }))).toEqual(IP);
}); });
test('getIpAddress: No header', () => { test('getIpAddress: No header', () => {
expect(detect.getIpAddress({ headers: {} } as any)).toEqual(null); expect(detect.getIpAddress(new Headers())).toEqual(null);
}); });

2
src/lib/detect.ts
View file

@ -28,7 +28,7 @@ export function getIpAddress(headers: Headers) {
const ip = headers.get(header); const ip = headers.get(header);
if (header === 'x-forwarded-for') { if (header === 'x-forwarded-for') {
return ip?.split[',']?.[0]?.trim(); return ip?.split(',')?.[0]?.trim();
} }
if (header === 'forwarded') { if (header === 'forwarded') {