116 lines
3.4 KiB
TypeScript
116 lines
3.4 KiB
TypeScript
import { QqbotRateLimitService } from '@/modules/qqbot/core/send/qqbot-rate-limit.service';
|
|
|
|
function createService(config: Record<string, number | string | undefined>) {
|
|
return new QqbotRateLimitService({
|
|
get: jest.fn((key: string) => config[key]),
|
|
} as any);
|
|
}
|
|
|
|
describe('QqbotRateLimitService', () => {
|
|
beforeEach(() => {
|
|
jest.useFakeTimers();
|
|
jest.setSystemTime(new Date('2026-06-12T00:00:00.000Z'));
|
|
});
|
|
|
|
afterEach(() => {
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
it('queues a repeated target send instead of rejecting it immediately', async () => {
|
|
const service = createService({
|
|
QQBOT_SEND_GLOBAL_INTERVAL_MS: 1000,
|
|
QQBOT_SEND_JITTER_MS: 0,
|
|
QQBOT_SEND_MAX_QUEUE_WAIT_MS: 10000,
|
|
QQBOT_SEND_TARGET_INTERVAL_MS: 3000,
|
|
});
|
|
|
|
await expect(
|
|
(service as any).waitForSendSlot('bot-1', 'group-1'),
|
|
).resolves.toEqual({ waitMs: 0 });
|
|
|
|
const secondSend = (service as any).waitForSendSlot('bot-1', 'group-1');
|
|
|
|
jest.advanceTimersByTime(2999);
|
|
await Promise.resolve();
|
|
await expect(Promise.race([secondSend, Promise.resolve('pending')]))
|
|
.resolves.toBe('pending');
|
|
|
|
jest.advanceTimersByTime(1);
|
|
await Promise.resolve();
|
|
await expect(secondSend).resolves.toEqual({ waitMs: 3000 });
|
|
});
|
|
|
|
it('rejects when the queued send would wait beyond the configured budget', async () => {
|
|
const service = createService({
|
|
QQBOT_SEND_GLOBAL_INTERVAL_MS: 1000,
|
|
QQBOT_SEND_JITTER_MS: 0,
|
|
QQBOT_SEND_MAX_QUEUE_WAIT_MS: 1000,
|
|
QQBOT_SEND_TARGET_INTERVAL_MS: 3000,
|
|
});
|
|
|
|
await (service as any).waitForSendSlot('bot-1', 'group-1');
|
|
|
|
await expect(
|
|
(service as any).waitForSendSlot('bot-1', 'group-1'),
|
|
).rejects.toMatchObject({
|
|
response: expect.objectContaining({
|
|
msg: 'QQBot 发送队列繁忙，请稍后再试',
|
|
}),
|
|
});
|
|
});
|
|
|
|
it('does not reserve a slot for a rejected queued send', async () => {
|
|
const service = createService({
|
|
QQBOT_SEND_GLOBAL_INTERVAL_MS: 1000,
|
|
QQBOT_SEND_JITTER_MS: 0,
|
|
QQBOT_SEND_MAX_QUEUE_WAIT_MS: 1000,
|
|
QQBOT_SEND_TARGET_INTERVAL_MS: 3000,
|
|
});
|
|
|
|
await (service as any).waitForSendSlot('bot-1', 'group-1');
|
|
|
|
await expect(
|
|
(service as any).waitForSendSlot('bot-1', 'group-1'),
|
|
).rejects.toBeDefined();
|
|
|
|
jest.advanceTimersByTime(3000);
|
|
|
|
await expect(
|
|
(service as any).waitForSendSlot('bot-1', 'group-1'),
|
|
).resolves.toEqual({ waitMs: 0 });
|
|
});
|
|
|
|
it('does not let one target queue block another target global slot', async () => {
|
|
const service = createService({
|
|
QQBOT_SEND_GLOBAL_INTERVAL_MS: 1000,
|
|
QQBOT_SEND_JITTER_MS: 0,
|
|
QQBOT_SEND_MAX_QUEUE_WAIT_MS: 10000,
|
|
QQBOT_SEND_TARGET_INTERVAL_MS: 3000,
|
|
});
|
|
|
|
await (service as any).waitForSendSlot('bot-1', 'group-1');
|
|
|
|
const sameTargetSend = (service as any).waitForSendSlot(
|
|
'bot-1',
|
|
'group-1',
|
|
);
|
|
const otherTargetSend = (service as any).waitForSendSlot(
|
|
'bot-1',
|
|
'group-2',
|
|
);
|
|
|
|
jest.advanceTimersByTime(999);
|
|
await Promise.resolve();
|
|
await expect(Promise.race([otherTargetSend, Promise.resolve('pending')]))
|
|
.resolves.toBe('pending');
|
|
|
|
jest.advanceTimersByTime(1);
|
|
await Promise.resolve();
|
|
await expect(otherTargetSend).resolves.toEqual({ waitMs: 1000 });
|
|
|
|
jest.advanceTimersByTime(2000);
|
|
await Promise.resolve();
|
|
await expect(sameTargetSend).resolves.toEqual({ waitMs: 3000 });
|
|
});
|
|
});
|