shout-create-fix

tests/test_notification_fix.py

@@ -0,0 +1,78 @@
+"""
+Тест для проверки исправления проблемы с полем 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__])