Объяснить, как тестировать структуру с websocket через pytest

В рамках студенческого проекта надо написать тесты для структуры с websocket. Само приложение пишется на FastApi, а тесты пишутся с помощью pytest.

Вообще реализуется сервис типа 'chat', работает он на websocket, есть модели сообщений. Пытался написать тесты как в документации по fastapi, но там websocket используются без дополнительных надстроек. Для работы проект пришлось включить websocket в структуру формата
Dict[str, Dict[int, Dict[int, WebSocket]]]

Возникают вопросы с тем, как это тестировать.

Просьба отправлять телеграм для связи.