Отсутствие тестирования

[x2d7]

---

• Отсутствует тестирование edge-кейсов для функционалаTools
• Отсутствует тестирование генерации - прерываний и других событий, которые должны предсказуемо себя вести в случаях ошибок
• Отсутствует тестирование конвертации диалога, а также над другим функционалом, который связан с хранением сообщений/истории событий

Стоит учесть, что некоторые аспекты сейчас нельзя так просто протестировать в силу текущего устройства кода, поэтому также следует провести рефакторинг, который позволит это сделать (например, Stream токенов из OpenAI API)

---

Необходимо прописать тесты и добавить их в CI

[x2d7]

Первый пакет, который требует тестирования — chat/tools.
Нет смысла начинать реализацию до merge #8 в dev

---

Tools

Модуль представляет из себя 2 коротких файла, один из которых - стандартный файл types.go, который представляет из себя небольшой набор структур и методов, которые определяют работу с данными для реализации инструментов.
Второй же файл tools.go содержит методы для поддержания работы со структурами первого файла во внешнем коде, а также их практическим использованием.

types.go

Метод	Тест	Результат
Add	Дубликат Name	Ошибка дублирования имени
Add	Инструмент без названия	Ошибка идентификации инструмента
Add	Стандартное добавление	Успех
Add	Добавление инструмента без описания	Успех
Remove	Несуществующий элемент	Ошибка
Remove	Стандартное удаление	Успех
Snapshot	Override в Tool.Name на уровне Tools	Возврат []Tool с локальными Name
Snapshot	Серия Add и Remove	Ожидаемое наполнение на каждом этапе
Snapshot	Проверка безопасности владения []Tool	Успешное копирование элементов, ссылки не пересекаются

tools.go

Файл содержит 2 метода — NewTool и Execute, которые отвечают за создание нового объекта структуры Tool с удобной инициализацией, а также непосредственным исполнением внутренней логики без непосредственного обращения соответственно.

Функция	Тест	Результат
NewTool	Создание и вызов Tool с struct T (далее - structT)	Успех
NewTool	Создание и вызов Tool с primitive T (далее - primT)	Успех
NewTool	structT с вложенной структурой	Успех
NewTool	structT с embedded структурой	Успех
NewTool	primT c map (primitive value)	Успех
NewTool	primT c map (struct value)	Успех
NewTool	primT c slice (primitive value)	Успех
NewTool	primT c slice (struct case value)	Успех
NewTool	primT c map	Успех
NewTool	structT с рекурсией	Успех

Перед добавлением тестов для Execute стоит прояснить: NewTool генерирует для примитивных типов такую же схему, как для структур. Нет смысла разделять их на отдельные вызовы. Если работает генерация схемы, значит будет работать и Execute.

Функция	Тест	Результат
Execute	Обычный вызов	Успех, ok bool = true
Execute	Вызов несуществующего инструмента	string с ошибкой (error:...), ok bool = false
Execute	Вызов инструмента, который вызывает ошибку	string с ошибкой (error:...), ok bool = false

[x2d7]

Edge cases

NewTool

Функция	Тест	Результат
NewTool	Поле с типом func (StructWithFunc)	Ошибка: panic during schema generation
NewTool	Поле с типом chan (StructWithChan)	Ошибка: panic during schema generation
NewTool	пустое имя (EmptyName)	Успех (id = "")
NewTool	Указатель (PointerType)	Успех
NewTool	interface{} (InterfaceType)	Успех
NewTool	Анонимная структура	Успех
NewTool	int примитив	Успех
NewTool	bool примитив	Успех
NewTool	float64 примитив	Успех

Execute

Функция	Тест	Результат
Execute	Malformed JSON	ok=false, ошибка unmarshal
Execute	Пустые аргументы (EmptyArguments)	Зависит от типа T
Execute	TypeMismatch (несоответствие типов)	ok=false, ошибка unmarshal
Execute	MissingRequiredFields	ok=true (zero values)
Execute	PartialFields	ok=true

[x2d7]

GetSchema

Функция	Тест	Результат
GetSchema	Вызов дважды на том же инструменте (кеш)	Успех, возвращает кэшированное значение
GetSchema	Panic во время генерации схемы (func/chan)	Ошибка: "panic during schema generation"

[x2d7]

Add с AddOptions

Функция	Тест	Результат
Add	WithAutoIncrement - несколько тулов с одинаковым ID	Успех, ID: test, test_1, test_2
Add	WithAutoIncrement + WithStartIncrement(0)	Успех, ID: test_0, test_1, test_2
nextID	ID не существует в мапе	Возвращает оригинальный ID
nextID	ID существует в мапе (startIncrement=1)	Генерирует test_3
nextID	ID существует в мапе (startIncrement=0)	Генерирует test_2

[x2d7]

Chat

Модуль содержит основную логику чата: Events (события), Messages (коллекция событий), ApproveWaiter (механизм одобрения tool calls), Senders (типы отправителей) и Helpers (вспомогательные функции для работы с чатом).

types.go (Messages, ApproveWaiter)

Messages

Mетод	Тест	Результат
NewMessages	Создание нового объекта	Пустой Events
AddEvent	Добавление одного ивента	Успех, длина = 1
AddEvent	Добавление нескольких ивентов	Успех, длина = N
AddEvent	Конкурентное добавление (goroutines)	Нет race condition
Snapshot Пустой снэпшот		Пустой срез
Snapshot	Получение копии	Изменения в оригинале не влияют на копию
Snapshot	Модификация копии	Оригинал не меняется

ApproveWaiter

Функция	Тест	Результат
NewApproveWaiter	Создание	Не nil, канал verdicts создан
Attach	Привязка к EventNewToolCall	approval установлен
Wait	Ожидание 0 вердиктов	Канал сразу закрыт
Wait	Получение вердиктов	Канал получает N вердиктов
Wait	Отмена контекста	Канал закрывается
Wait	Отмена контекста внутри цикла	Канал закрывается после получения N вердиктов
Wait	Inner ctx.Done() после получения вердикта	Канал закрывается, вердикт не отправляется
Resolve	Отправка вердикта accept=true	Wait получает Accepted=true
Resolve	Отправка вердикта accept=false	Wait получает Accepted=false
EventNewToolCall.Resolve	Вызов без approval	Нет паники
EventNewToolCall.Resolve	Повторный вызов	Второй вызов игнорируется

[x2d7]

chat.go (Complete, Session)

Complete

Метод	Тест	Результат
Complete	Успешная генерация (токены)	All tokens received
Complete	Успешная генерация (tool call)	EventNewToolCall received
Complete	nil stream returned	EventNewError with ErrNilStreaming
Complete	Ошибка при итерации	Last event is EventNewError
SyncInput	Возвращает новый клиент	New client is different instance
SyncInput	С добавленными сообщениями	New client contains new messages
SyncInput	С добавленными тулами	New client contains new tools

Session

Метод	Тест	Результат
Session	Накопленные токены добавляются в историю	EventNewToken in Messages
Session	Накопленные tool calls добавляются	EventNewToolCall in Messages
Session	Отправка EventCompletionEnded	EventCompletionEnded sent
Session	Tool call с Refusal	EventNewRefusal added to history
Session	Resolve(true) → выполнение тула	EventNewToolMessage with Success=true
Session	Resolve(true) → вызов несуществующего тула	EventNewToolMessage with Success=false
Session	Resolve(false) → отказ	EventNewToolMessage with "User declined..."

helpers.go (Helpers)

Функция	Тест	Результат
AddMessage	SenderUser	EventNewUserMessage added
AddMessage	SenderAssistant	EventNewAssistantMessage added
AddMessage	SenderSystem	EventNewSystemMessage added
AppendEvent	Добавление события	Event in Messages
SendUserStream	Успешная отправка	Session started
SendAssistantStream	Успешная отправка	Session started
SendSystemStream	Успешная отправка	Session started

[x2d7]

connect/openai

Пакет реализует адаптер к OpenAI API. Содержит два файла: openai.go — конвертация данных чата в формат OpenAI и управление клиентом; stream.go — обёртка над SSE-стримом OpenAI.

Рефакторинг: поле SSEStream в OpenAIStream заменено с конкретного типа *ssestream.Stream[openai.ChatCompletionChunk] на приватный интерфейс sseStreamer. Это позволяет делать мок для SSE-потока в тестах без реального API.

openai.go

isEmpty

Функция	Тест	Результат
isEmpty	Нулевая структура	true
isEmpty	OfUser заполнен	false
isEmpty	OfAssistant заполнен	false
isEmpty	OfSystem заполнен	false
isEmpty	OfTool заполнен	false

openAIMessages.findLastAssistantMessage

Метод	Тест	Результат
findLastAssistantMessage	Пустой список	nil
findLastAssistantMessage	Только User и System сообщения	nil
findLastAssistantMessage	Один Assistant	Указатель на него
findLastAssistantMessage	Несколько Assistant	Указатель на последний

openAIMessages.Add

Метод	Тест	Результат
Add	EventNewAssistantMessage	OfAssistant != nil
Add	EventNewSystemMessage	OfSystem != nil
Add	EventNewUserMessage	OfUser != nil
Add	EventNewRefusal	OfAssistant с refusal-контентом
Add	EventNewToolMessage	OfTool != nil
Add	EventNewToolCall — нет предшествующего Assistant	Создаётся новый Assistant с tool call
Add	EventNewToolCall — есть предшествующий Assistant	Tool call добавляется к существующему Assistant
Add	Два EventNewToolCall подряд	Оба tool call в одном Assistant
Add	Неизвестный тип (EventNewToken)	Ничего не добавляется
Add	Последовательность System → User → Assistant → User	4 сообщения корректных типов

ConvertTools

Функция	Тест	Результат
ConvertTools	Пустые Tools	Пустой срез
ConvertTools	Один инструмент	Name и Description корректны
ConvertTools	Несколько инструментов	Все инструменты присутствуют
ConvertTools	Инструмент со схемой	Parameters != nil

OpenAIClient.SyncInput

Метод	Тест	Результат
SyncInput	Возвращает новый экземпляр	result != original
SyncInput	Оригинальный клиент не изменяется	Params.Messages оригинала не растёт
SyncInput	Сообщения синхронизированы	Params.Messages содержит все сообщения чата
SyncInput	Тулы синхронизированы	Params.Tools содержит все инструменты чата
SyncInput	Пустой чат	Params.Messages = 0, Params.Tools = 0
SyncInput	Сохранение Model, APIKey, Endpoint	Поля скопированы без изменений

stream.go

OpenAIStream._handleRawChunk

Метод	Тест	Результат
_handleRawChunk	Только content	EventNewToken с корректным Content
_handleRawChunk	Только refusal	EventNewRefusal с корректным Content
_handleRawChunk	Один tool call	EventNewToolCall с корректными ID, Name, Content
_handleRawChunk	Несколько tool calls	Несколько EventNewToolCall
_handleRawChunk	content + refusal + tool call одновременно	3 события корректных типов
_handleRawChunk	Пустая delta	Пустой список, без ошибки

OpenAIStream.handleRawChunk (decorator)

Метод	Тест	Результат
handleRawChunk	Пустые события от _handleRawChunk	Ошибка "empty events"
handleRawChunk	Непустые события	Пробрасывает без изменений

OpenAIStream.Next

Метод	Тест	Результат
Next	Один чанк с контентом	true, событие доступно в Current()
Next	Один чанк с двумя событиями (content+refusal)	Первый Next() — из SSE, второй — из очереди
Next	SSE исчерпан (пустой стрим)	false
Next	SSE завершился без ошибки	Err() == nil
Next	SSE вернул ошибку	false, Err() == ошибка SSE
Next	Чанк с пустой delta	false, Err() == "empty events"
Next	err уже установлен	false, SSE.Next() не вызывается
Next	Несколько чанков	Все события получены в правильном порядке

OpenAIStream.Current, Err, Close

Метод	Тест	Результат
Current	До вызова Next()	nil
Current	После вызова Next()	Ненулевое событие
Err	Изначально	nil
Err	После ошибки SSE	Ошибка от SSE
Close	Делегирование в SSEStream	SSEStream.Close() вызван