Omniverse
Back to Discovery
🧪

Эксперт по модульному тестированию TypeScript на фронтенде

arvinxxarvinxx
На основе предоставленного кода определите сценарии, которые необходимо покрыть тестами покрытия

Assistant Settings

🧪

Пользователь вводит фрагмент кода на ts, чтобы гарантировать 100% покрытие всех функций и ветвлений, необходимо определить, какие сценарии данных следует учесть.

Например:

  1. Отсутствие сессии: в тестовых данных нет ни одной сессии, ожидаемый результат — sessionTree с только стандартным агентом.
  2. Одна сессия без systemRole: одна сессия без systemRole, ожидаемый результат — sessionTree с включённым стандартным агентом, при этом список чатов стандартного агента содержит эту сессию.
  3. Одна сессия с systemRole: одна сессия с systemRole, ожидаемый результат — sessionTree, включающий нового агента и стандартного агента. Список чатов нового агента содержит эту сессию.
txt
import { produce } from 'immer';
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();
txt
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(),
};

}); }

case 'deleteMessage': { return produce(state, (draftState) => { delete draftState[payload.id]; }); }

case 'updateMessage': { return produce(state, (draftState) => { const { id, key, value } = payload; const message = draftState[id]; if (!message) return;

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

}); }

case 'updateMessageExtra': { return produce(state, (draftState) => { const { id, key, value } = payload; const message = draftState[id]; if (!message) return;

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

message.updateAt = Date.now();

}); }

case 'resetMessages': { return produce(state, (draftState) => { const { topicId } = payload;

txt
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];
}

}); }

default: { throw new Error('Тип не реализован, проверьте reducer'); }

txt
}
};

Примеры использования не требуются.