Рефакторинг + убраны Mock'и для сообщений + тесты