Как правильно писать WebSocket API автотесты на Python

Значимый прорыв формирует отрасль ИИ: sound_right 10 минут назад Как правильно писать WebSocket API автотесты на Python Средний 38 мин 346 Тестирование IT-систем * Тестирование веб-сервисов * Python * GitHub * Туториал ВступлениеWebSocket API часто тестируют как что-то отдельное: подключились к сокету, отправили сообщение, получили ответ. HTTP API при этом живёт в другой части тестов: создали пользователя, проверили эндпоинт, сравнили JSON. В небольших примерах такой подход выглядит нормально, но в реальных сервисах HTTP и WebSocket обычно работают вместе.

Например, пользователь регистрируется через REST API, затем подключается к WebSocket-каналу, получает историю сообщений, отправляет новое сообщение, а остальные участники получают событие в реальном времени. Формально это два разных транспорта, но с точки зрения продукта — один сценарий. У него есть общие пользователи, общие идентификаторы, общий контракт данных и ожидаемый порядок событий.

Технические детали

В этой статье разберём, как писать автотесты для такого API на Python: не в формате «hello world с одним сокетом», а как полноценный тестовый слой вокруг небольшого FastAPI-приложения с REST и WebSocket. В качестве примера будет использоваться простой чат: HTTP-эндпоинты отвечают за пользователей, сброс состояния и вспомогательные операции, а WebSocket — за события чата. Основной фокус будет не на продакшен-архитектуре сервиса, а на тестируемости.

Поэтому демо-приложение намеренно простое: без базы данных, с in-memory-хранилищем и предсказуемыми эндпоинтами. Это позволяет сосредоточиться на том, как устроить сами автотесты: контракты, HTTP- и WebSocket-клиенты, фикстуры, строгие проверки событий, Allure-шаги и запуск в CI. В итоге получится не набор разрозненных проверок, а единый подход: создать пользователя через HTTP, подключить его к WebSocket по тому же user_id, проверить порядок входящих событий и убедиться, что тесты читаются как сценарий, а не как набор случайных json.

loads, send и receive. Используемые технологииПеред тем как переходить к структуре проекта и тестам, зафиксируем стек. В примере используется небольшой FastAPI-сервис и отдельный слой автотестов на Python.

Отраслевые последствия

Все зависимости подобраны так, чтобы показать именно тестирование REST + WebSocket API, а не усложнять демо инфраструктурой, базой данных или внешними брокерами. БиблиотекаЗачем используетсяfastapiОсновной фреймворк для демо-приложения. На нём реализованы HTTP-эндпоинты и WebSocket-ручка чата.

uvicornASGI-сервер для локального запуска FastAPI-приложения и запуска сервиса в CI перед тестами. pydanticОписание моделей данных: пользователей, сообщений, входящих и исходящих WebSocket-событий. В тестах используется для валидации контрактов.

pydantic-settingsУправление настройками тестов через переменные окружения: HTTP URL, WebSocket URL и другие параметры запуска. pytestОсновной тестовый фреймворк. На нём строятся фикстуры, маркеры и сами сценарии.

Этот прогресс даёт важные сигналы о будущем отрасли, и технологический мир внимательно наблюдает.