67 lines
2.2 KiB
TypeScript
67 lines
2.2 KiB
TypeScript
import fs from 'node:fs/promises';
|
|
import os from 'node:os';
|
|
import path from 'node:path';
|
|
|
|
import { describe, expect, it } from 'vitest';
|
|
|
|
import { NotificationStateStore } from '../src/platforms/xiaohongshu/notification-state.js';
|
|
import type { CommentNotification } from '../src/platforms/xiaohongshu/types.js';
|
|
|
|
function makeNotification(overrides?: Partial<CommentNotification>): CommentNotification {
|
|
return {
|
|
userId: 'u1',
|
|
nickname: 'tester',
|
|
avatar: 'https://example.com/a.png',
|
|
content: '你好',
|
|
type: '评论了你的笔记',
|
|
time: '1分钟前',
|
|
feedId: 'feed123',
|
|
xsecToken: 'token123',
|
|
noteImage: 'https://example.com/note.png',
|
|
...overrides,
|
|
};
|
|
}
|
|
|
|
describe('notification-state store', () => {
|
|
it('upserts notifications and tracks status transitions', async () => {
|
|
const tempDir = await fs.mkdtemp(path.join(os.tmpdir(), 'social-mcp-notif-'));
|
|
try {
|
|
const store = new NotificationStateStore(tempDir, 'test.db');
|
|
const n1 = makeNotification();
|
|
|
|
const first = store.upsertNotifications([n1]);
|
|
expect(first).toEqual({ fetched: 1, inserted: 1, updated: 0 });
|
|
|
|
const second = store.upsertNotifications([n1]);
|
|
expect(second).toEqual({ fetched: 1, inserted: 0, updated: 1 });
|
|
|
|
const openTasks = store.listByStatuses(['new'], 10);
|
|
expect(openTasks).toHaveLength(1);
|
|
expect(openTasks[0]?.notification.userId).toBe('u1');
|
|
|
|
const fp = store.findOpenFingerprint('u1', '你好');
|
|
expect(fp).toBeTypeOf('string');
|
|
|
|
if (!fp) {
|
|
throw new Error('fingerprint should not be null');
|
|
}
|
|
|
|
store.markPending(fp);
|
|
expect(store.listByStatuses(['pending'], 10)).toHaveLength(1);
|
|
|
|
store.markFailed(fp, 'network error');
|
|
const failed = store.listByStatuses(['failed'], 10);
|
|
expect(failed).toHaveLength(1);
|
|
expect(failed[0]?.retryCount).toBe(1);
|
|
expect(failed[0]?.errorMessage).toBe('network error');
|
|
|
|
store.markReplied(fp, '收到，感谢反馈');
|
|
const replied = store.listByStatuses(['replied'], 10);
|
|
expect(replied).toHaveLength(1);
|
|
expect(replied[0]?.replyContent).toBe('收到，感谢反馈');
|
|
} finally {
|
|
await fs.rm(tempDir, { recursive: true, force: true });
|
|
}
|
|
});
|
|
});
|