136 lines
4.1 KiB
TypeScript
136 lines
4.1 KiB
TypeScript
import type { QqbotNormalizedMessage } from '@/modules/qqbot/core/contract/qqbot.types';
|
|
import { createPlugin } from '@/modules/qqbot/plugins/repeater/src';
|
|
|
|
/**
|
|
* 创建 复读插件对象或配置。
|
|
* @param text - 待匹配文本；生成 测试对象。
|
|
* @returns 创建后的 复读插件对象或配置。
|
|
*/
|
|
function createMessage(text: string): QqbotNormalizedMessage {
|
|
return {
|
|
eventTime: new Date(),
|
|
groupId: 'group-1',
|
|
messageId: `message-${text}-${Date.now()}`,
|
|
messageText: text,
|
|
messageType: 'group',
|
|
rawEvent: {},
|
|
rawMessage: text,
|
|
selfId: 'bot-1',
|
|
targetId: 'group-1',
|
|
userId: 'user-1',
|
|
};
|
|
}
|
|
|
|
/**
|
|
* 创建 复读插件对象或配置。
|
|
* @param config - config 输入；构造 Jest mock 返回值。
|
|
*/
|
|
function createService(config: Record<string, number | string | undefined>) {
|
|
const sendService = {
|
|
sendText: jest.fn().mockResolvedValue({ status: 'ok' }),
|
|
};
|
|
const service = createPlugin({
|
|
host: {
|
|
bindEventPlugin: jest.fn(),
|
|
getBoundEventPluginKeys: jest.fn().mockResolvedValue(['repeater']),
|
|
getConfig: <T = string>(key: string) => config[key] as T,
|
|
sendText: sendService.sendText,
|
|
unbindEventPlugin: jest.fn(),
|
|
warn: jest.fn(),
|
|
},
|
|
manifest: {
|
|
events: [],
|
|
name: '复读机',
|
|
pluginKey: 'repeater',
|
|
version: '1.0.0',
|
|
},
|
|
now: () => Date.now(),
|
|
});
|
|
return { sendService, service };
|
|
}
|
|
|
|
describe('RepeaterPluginService risk-control defaults', () => {
|
|
beforeEach(() => {
|
|
jest.useFakeTimers();
|
|
jest.setSystemTime(new Date('2026-06-12T00:00:00.000Z'));
|
|
});
|
|
|
|
afterEach(() => {
|
|
jest.useRealTimers();
|
|
});
|
|
|
|
it('uses a conservative default threshold before repeating', async () => {
|
|
const { sendService, service } = createService({});
|
|
|
|
await service.handleMessage(createMessage('哈'));
|
|
await service.handleMessage(createMessage('哈'));
|
|
await service.handleMessage(createMessage('哈'));
|
|
|
|
expect(sendService.sendText).not.toHaveBeenCalled();
|
|
|
|
await service.handleMessage(createMessage('哈'));
|
|
|
|
expect(sendService.sendText).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it('applies a conversation interval after one repeated send', async () => {
|
|
const { sendService, service } = createService({
|
|
QQBOT_REPEATER_CONFIG_CACHE_TTL_MS: 600000,
|
|
QQBOT_REPEATER_MIN_INTERVAL_MS: 600000,
|
|
QQBOT_REPEATER_THRESHOLD: 4,
|
|
});
|
|
|
|
for (let index = 0; index < 4; index += 1) {
|
|
await service.handleMessage(createMessage('哈'));
|
|
}
|
|
for (let index = 0; index < 4; index += 1) {
|
|
await service.handleMessage(createMessage('呀'));
|
|
}
|
|
|
|
expect(sendService.sendText).toHaveBeenCalledTimes(1);
|
|
|
|
jest.setSystemTime(new Date(Date.now() + 600000));
|
|
await service.handleMessage(createMessage('呀'));
|
|
|
|
expect(sendService.sendText).toHaveBeenCalledTimes(2);
|
|
});
|
|
|
|
it('applies the conversation interval even when repeated send fails', async () => {
|
|
const { sendService, service } = createService({
|
|
QQBOT_REPEATER_CONFIG_CACHE_TTL_MS: 600000,
|
|
QQBOT_REPEATER_MIN_INTERVAL_MS: 600000,
|
|
QQBOT_REPEATER_THRESHOLD: 4,
|
|
});
|
|
sendService.sendText.mockRejectedValueOnce(new Error('send failed'));
|
|
|
|
for (let index = 0; index < 4; index += 1) {
|
|
await service.handleMessage(createMessage('哈'));
|
|
}
|
|
for (let index = 0; index < 4; index += 1) {
|
|
await service.handleMessage(createMessage('呀'));
|
|
}
|
|
|
|
expect(sendService.sendText).toHaveBeenCalledTimes(1);
|
|
});
|
|
|
|
it('keeps the conversation interval after a non-repeatable message', async () => {
|
|
const { sendService, service } = createService({
|
|
QQBOT_REPEATER_CONFIG_CACHE_TTL_MS: 600000,
|
|
QQBOT_REPEATER_MIN_INTERVAL_MS: 600000,
|
|
QQBOT_REPEATER_THRESHOLD: 4,
|
|
});
|
|
|
|
for (let index = 0; index < 4; index += 1) {
|
|
await service.handleMessage(createMessage('哈'));
|
|
}
|
|
|
|
await service.handleMessage(createMessage('!help'));
|
|
|
|
for (let index = 0; index < 4; index += 1) {
|
|
await service.handleMessage(createMessage('呀'));
|
|
}
|
|
|
|
expect(sendService.sendText).toHaveBeenCalledTimes(1);
|
|
});
|
|
});
|