کارشناس تست واحد TypeScript فرانت‌اند

import { ChatMessage, ChatMessageMap } from '@/types/chatMessage';
import { LLMRoleType } from '@/types/llm';
import { MetaData } from '@/types/meta';
import { nanoid } from '@/utils/uuid';

interface AddMessage {
id?: string;
message: string;
meta?: MetaData;
parentId?: string;
quotaId?: string;
role: LLMRoleType;
type: 'addMessage';
}

interface DeleteMessage {
id: string;
type: 'deleteMessage';
}

interface ResetMessages {
topicId?: string;
type: 'resetMessages';
}

interface UpdateMessage {
id: string;
key: keyof ChatMessage;
type: 'updateMessage';
value: ChatMessage[keyof ChatMessage];
}
interface UpdateMessageExtra {
id: string;
key: string;
type: 'updateMessageExtra';
value: any;
}

export type MessageDispatch =
| AddMessage
| DeleteMessage
| ResetMessages
| UpdateMessage
| UpdateMessageExtra;

export const messagesReducer = (
state: ChatMessageMap,
payload: MessageDispatch,
): ChatMessageMap => {
switch (payload.type) {
case 'addMessage': {
return produce(state, (draftState) => {
const mid = payload.id || nanoid();

draftState[mid] = {
  content: payload.message,
  createAt: Date.now(),
  id: mid,
  meta: payload.meta || {},
  parentId: payload.parentId,
  quotaId: payload.quotaId,
  role: payload.role,
  updateAt: Date.now(),
};

// @ts-ignore
message[key] = value;
message.updateAt = Date.now();

if (!message.extra) {
  message.extra = { [key]: value } as any;
} else {
  message.extra[key] = value;
}

message.updateAt = Date.now();

const messages = Object.values(draftState).filter((message) => {
  // اگر topicId وجود نداشته باشد، به معنی پاک کردن پیام‌های گفتگو پیش‌فرض است
  if (!topicId) return !message.topicId;

  return message.topicId === topicId;
});

// حذف پیام‌های یافت شده
for (const message of messages) {
  delete draftState[message.id];
}

}
};