118 lines
4.1 KiB
TypeScript
118 lines
4.1 KiB
TypeScript
import { describe, expect, test } from 'bun:test';
|
|
import { desktopHostProbe, desktopHostsGet, desktopHostsSet, redactSensitiveUrl, resolveDesktopHostUrl } from './desktopHosts';
|
|
|
|
const withDesktopBridge = async <T>(handler: (cmd: string, args: Record<string, unknown>) => unknown | Promise<unknown>, run: () => Promise<T>): Promise<T> => {
|
|
const previousWindow = Object.getOwnPropertyDescriptor(globalThis, 'window');
|
|
Object.defineProperty(globalThis, 'window', {
|
|
configurable: true,
|
|
value: {
|
|
__OPENCHAMBER_DESKTOP__: {
|
|
invoke: handler,
|
|
},
|
|
},
|
|
});
|
|
try {
|
|
return await run();
|
|
} finally {
|
|
if (previousWindow) {
|
|
Object.defineProperty(globalThis, 'window', previousWindow);
|
|
} else {
|
|
Reflect.deleteProperty(globalThis, 'window');
|
|
}
|
|
}
|
|
};
|
|
|
|
describe('resolveDesktopHostUrl', () => {
|
|
test('keeps regular host URLs unchanged', () => {
|
|
expect(resolveDesktopHostUrl('https://example.com/app?x=1')).toEqual({
|
|
persistedUrl: 'https://example.com/app?x=1',
|
|
redeemUrl: null,
|
|
kind: 'normal-host',
|
|
});
|
|
});
|
|
|
|
test('detects tunnel connect links and stores only origin', () => {
|
|
expect(resolveDesktopHostUrl('https://example.trycloudflare.com/connect?t=secret-token')).toEqual({
|
|
persistedUrl: 'https://example.trycloudflare.com',
|
|
redeemUrl: 'https://example.trycloudflare.com/connect?t=secret-token',
|
|
kind: 'tunnel-connect-link',
|
|
});
|
|
});
|
|
|
|
test('detects tunnel connect links with trailing slash', () => {
|
|
expect(resolveDesktopHostUrl('https://example.trycloudflare.com/connect/?t=secret-token#section')).toEqual({
|
|
persistedUrl: 'https://example.trycloudflare.com',
|
|
redeemUrl: 'https://example.trycloudflare.com/connect/?t=secret-token',
|
|
kind: 'tunnel-connect-link',
|
|
});
|
|
});
|
|
|
|
test('redacts tunnel tokens from labels', () => {
|
|
expect(redactSensitiveUrl('https://example.trycloudflare.com/connect?t=secret-token')).toBe(
|
|
'https://example.trycloudflare.com/connect?t=%5BREDACTED%5D',
|
|
);
|
|
});
|
|
});
|
|
|
|
describe('desktop host runtime headers', () => {
|
|
test('parses persisted request headers from desktop config', async () => {
|
|
await withDesktopBridge(async (cmd) => {
|
|
expect(cmd).toBe('desktop_hosts_get');
|
|
return {
|
|
hosts: [{
|
|
id: 'remote-1',
|
|
label: 'Remote',
|
|
url: 'https://remote.example',
|
|
requestHeaders: {
|
|
' CF-Access-Client-Id ': ' client-id ',
|
|
Authorization: 'Bearer should-not-be-read',
|
|
'Bad:Name': 'bad',
|
|
},
|
|
}],
|
|
defaultHostId: 'remote-1',
|
|
initialHostChoiceCompleted: true,
|
|
};
|
|
}, async () => {
|
|
const config = await desktopHostsGet();
|
|
expect(config.hosts[0]?.requestHeaders).toEqual({
|
|
'CF-Access-Client-Id': 'client-id',
|
|
});
|
|
});
|
|
});
|
|
|
|
test('passes request headers through host save and probe IPC calls', async () => {
|
|
const calls: Array<{ cmd: string; args: Record<string, unknown> }> = [];
|
|
await withDesktopBridge(async (cmd, args) => {
|
|
calls.push({ cmd, args });
|
|
if (cmd === 'desktop_host_probe') return { status: 'ok', latencyMs: 7 };
|
|
return null;
|
|
}, async () => {
|
|
const requestHeaders = { 'CF-Access-Client-Id': 'client-id' };
|
|
await desktopHostsSet({
|
|
hosts: [{ id: 'remote-1', label: 'Remote', url: 'https://remote.example', requestHeaders }],
|
|
defaultHostId: 'remote-1',
|
|
});
|
|
const probe = await desktopHostProbe('https://remote.example', { requestHeaders });
|
|
expect(probe).toEqual({ status: 'ok', latencyMs: 7 });
|
|
});
|
|
|
|
expect(calls[0]).toEqual({
|
|
cmd: 'desktop_hosts_set',
|
|
args: {
|
|
input: {
|
|
hosts: [{ id: 'remote-1', label: 'Remote', url: 'https://remote.example', requestHeaders: { 'CF-Access-Client-Id': 'client-id' } }],
|
|
defaultHostId: 'remote-1',
|
|
initialHostChoiceCompleted: undefined,
|
|
},
|
|
},
|
|
});
|
|
expect(calls[1]).toEqual({
|
|
cmd: 'desktop_host_probe',
|
|
args: {
|
|
url: 'https://remote.example',
|
|
requestHeaders: { 'CF-Access-Client-Id': 'client-id' },
|
|
},
|
|
});
|
|
});
|
|
});
|