79 lines
3.5 KiB
Python
79 lines
3.5 KiB
Python
"""
|
||
Тест для проверки исправления проблемы с полем kind в уведомлениях.
|
||
"""
|
||
|
||
import pytest
|
||
from unittest.mock import patch, MagicMock
|
||
|
||
from services.notify import save_notification
|
||
from orm.notification import NotificationAction
|
||
|
||
|
||
def test_save_notification_with_create_action():
|
||
"""Тест создания уведомления с действием create для shout"""
|
||
with patch('services.notify.local_session') as mock_session:
|
||
mock_session_instance = MagicMock()
|
||
mock_session.return_value.__enter__.return_value = mock_session_instance
|
||
|
||
# Тестируем с действием create для shout
|
||
save_notification("create", "shout", {"id": 1})
|
||
|
||
# Проверяем, что уведомление создано с правильным kind
|
||
mock_session_instance.add.assert_called_once()
|
||
notification = mock_session_instance.add.call_args[0][0]
|
||
assert notification.kind == NotificationAction.CREATE
|
||
assert notification.action == "create"
|
||
assert notification.entity == "shout"
|
||
|
||
|
||
def test_save_notification_with_update_action():
|
||
"""Тест создания уведомления с действием update"""
|
||
with patch('services.notify.local_session') as mock_session:
|
||
mock_session_instance = MagicMock()
|
||
mock_session.return_value.__enter__.return_value = mock_session_instance
|
||
|
||
# Тестируем с действием update
|
||
save_notification("update", "shout", {"id": 1})
|
||
|
||
# Проверяем, что уведомление создано с правильным kind
|
||
mock_session_instance.add.assert_called_once()
|
||
notification = mock_session_instance.add.call_args[0][0]
|
||
assert notification.kind == NotificationAction.UPDATE
|
||
assert notification.action == "update"
|
||
assert notification.entity == "shout"
|
||
|
||
|
||
def test_save_notification_with_invalid_action():
|
||
"""Тест создания уведомления с невалидным действием (fallback)"""
|
||
with patch('services.notify.local_session') as mock_session:
|
||
mock_session_instance = MagicMock()
|
||
mock_session.return_value.__enter__.return_value = mock_session_instance
|
||
|
||
# Тестируем с невалидным действием
|
||
save_notification("invalid_action", "shout", {"id": 1})
|
||
|
||
# Проверяем, что уведомление создано с fallback значением
|
||
mock_session_instance.add.assert_called_once()
|
||
notification = mock_session_instance.add.call_args[0][0]
|
||
assert notification.kind == "invalid_action" # fallback
|
||
assert notification.action == "invalid_action"
|
||
assert notification.entity == "shout"
|
||
|
||
|
||
def test_save_notification_with_none_payload():
|
||
"""Тест создания уведомления с None payload (должно вернуться без создания)"""
|
||
with patch('services.notify.local_session') as mock_session:
|
||
mock_session_instance = MagicMock()
|
||
mock_session.return_value.__enter__.return_value = mock_session_instance
|
||
|
||
# Тестируем с None payload
|
||
save_notification("create", "shout", None)
|
||
|
||
# Проверяем, что уведомление не создавалось
|
||
mock_session_instance.add.assert_not_called()
|
||
mock_session_instance.commit.assert_not_called()
|
||
|
||
|
||
if __name__ == "__main__":
|
||
pytest.main([__file__])
|