增加TG代理选项,增加测试用例

This commit is contained in:
Shinku
2026-03-20 01:03:21 +08:00
parent 54203d64a3
commit c5d55f0ab0
3 changed files with 157 additions and 0 deletions
@@ -55,6 +55,7 @@ describe('settings and auth events', () => {
(config as any).telegramApiBaseUrl = 'https://api.telegram.org';
(config as any).telegramBotToken = '';
(config as any).telegramChatId = '';
(config as any).telegramUseSystemProxy = false;
});
afterAll(async () => {
@@ -219,6 +220,32 @@ describe('settings and auth events', () => {
expect(body.message).toContain('Telegram API Base URL');
});
it('persists and returns telegram use system proxy from runtime settings', async () => {
const updateResponse = await app.inject({
method: 'PUT',
url: '/api/settings/runtime',
payload: {
telegramUseSystemProxy: true,
},
});
expect(updateResponse.statusCode).toBe(200);
const updated = updateResponse.json() as { telegramUseSystemProxy?: boolean };
expect(updated.telegramUseSystemProxy).toBe(true);
expect((config as any).telegramUseSystemProxy).toBe(true);
const saved = await db.select().from(schema.settings).where(eq(schema.settings.key, 'telegram_use_system_proxy')).get();
expect(saved?.value).toBe(JSON.stringify(true));
const getResponse = await app.inject({
method: 'GET',
url: '/api/settings/runtime',
});
expect(getResponse.statusCode).toBe(200);
const runtime = getResponse.json() as { telegramUseSystemProxy?: boolean };
expect(runtime.telegramUseSystemProxy).toBe(true);
});
it('persists and returns routing fallback unit cost from runtime settings', async () => {
const updateResponse = await app.inject({
method: 'PUT',
+60
View File
@@ -17,12 +17,24 @@ vi.mock('undici', () => ({
fetch: (...args: unknown[]) => fetchMock(...args),
}));
const withExplicitProxyRequestInitMock = vi.fn(
(_proxyUrl: unknown, options?: Record<string, unknown>) => {
if (_proxyUrl) return { ...(options || {}), dispatcher: 'mock-proxy-dispatcher' };
return options ?? {};
},
);
vi.mock('./siteProxy.js', () => ({
withExplicitProxyRequestInit: (...args: unknown[]) => withExplicitProxyRequestInitMock(...args),
}));
describe('notifyService', () => {
beforeEach(async () => {
vi.resetModules();
sendMailMock.mockReset();
createTransportMock.mockClear();
fetchMock.mockReset();
withExplicitProxyRequestInitMock.mockClear();
const { config } = await import('../config.js');
config.notifyCooldownSec = 300;
@@ -35,6 +47,8 @@ describe('notifyService', () => {
(config as any).telegramEnabled = false;
(config as any).telegramBotToken = '';
(config as any).telegramChatId = '';
(config as any).telegramUseSystemProxy = false;
config.systemProxyUrl = '';
config.smtpEnabled = true;
config.smtpHost = 'smtp.example.com';
config.smtpPort = 465;
@@ -222,4 +236,50 @@ describe('notifyService', () => {
}),
);
});
it('applies system proxy dispatcher when telegramUseSystemProxy is enabled', async () => {
const { config } = await import('../config.js');
(config as any).telegramEnabled = true;
(config as any).telegramBotToken = '123456:telegram-token';
(config as any).telegramChatId = '-1001234567890';
(config as any).telegramUseSystemProxy = true;
config.systemProxyUrl = 'http://127.0.0.1:7890';
config.smtpEnabled = false;
fetchMock.mockResolvedValue({ ok: true, status: 200 });
const { sendNotification } = await import('./notifyService.js');
await sendNotification('测试通知', 'proxy-test', 'info', { bypassThrottle: true, throwOnFailure: true });
expect(withExplicitProxyRequestInitMock).toHaveBeenCalledWith(
'http://127.0.0.1:7890',
expect.objectContaining({ method: 'POST' }),
);
expect(fetchMock).toHaveBeenCalledWith(
expect.stringContaining('/sendMessage'),
expect.objectContaining({ dispatcher: 'mock-proxy-dispatcher' }),
);
});
it('does not apply proxy dispatcher when telegramUseSystemProxy is disabled', async () => {
const { config } = await import('../config.js');
(config as any).telegramEnabled = true;
(config as any).telegramBotToken = '123456:telegram-token';
(config as any).telegramChatId = '-1001234567890';
(config as any).telegramUseSystemProxy = false;
config.systemProxyUrl = 'http://127.0.0.1:7890';
config.smtpEnabled = false;
fetchMock.mockResolvedValue({ ok: true, status: 200 });
const { sendNotification } = await import('./notifyService.js');
await sendNotification('测试通知', 'no-proxy-test', 'info', { bypassThrottle: true, throwOnFailure: true });
expect(withExplicitProxyRequestInitMock).toHaveBeenCalledWith(
null,
expect.objectContaining({ method: 'POST' }),
);
const fetchOptions = fetchMock.mock.calls[0]?.[1] as Record<string, unknown>;
expect(fetchOptions.dispatcher).toBeUndefined();
});
});