follower-notification

tests/test_follow_sse_notifications.py

@@ -0,0 +1,145 @@
+"""
+Тест SSE уведомлений о новых подписчиках
+"""
+from __future__ import annotations
+
+from unittest.mock import patch
+
+import orjson
+import pytest
+
+from services.notify import notify_follower
+from storage.redis import redis
+
+
+@pytest.mark.asyncio
+async def test_follow_sse_notification_format():
+    """
+    Тест формата SSE уведомления о новой подписке
+    """
+    # Мокаем Redis publish, чтобы перехватить отправляемые данные
+    published_data = []
+
+    async def mock_publish(channel: str, data: bytes) -> None:
+        published_data.append((channel, orjson.loads(data)))
+
+    with patch.object(redis, 'publish', side_effect=mock_publish):
+        # Данные подписавшегося пользователя
+        follower_data = {
+            "id": 123,
+            "name": "Test Follower",
+            "slug": "test-follower",
+            "pic": "https://example.com/avatar.jpg"
+        }
+
+        target_author_id = 456
+
+        # Отправляем уведомление
+        await notify_follower(
+            follower=follower_data,
+            author_id=target_author_id,
+            action="follow"
+        )
+
+        # Проверяем, что данные отправлены правильно
+        assert len(published_data) == 1
+        channel, data = published_data[0]
+
+        # Проверяем канал
+        assert channel == f"follower:{target_author_id}"
+
+        # Проверяем формат данных согласно обновленной спецификации фронтенда
+        assert data["action"] == "create"
+        assert data["entity"] == "follower"
+        assert data["payload"]["follower_id"] == 123
+        assert data["payload"]["following_id"] == target_author_id
+        assert "id" in data["payload"]
+        assert "created_at" in data["payload"]
+
+        print(f"✅ SSE уведомление отправлено правильно: {data}")
+
+
+@pytest.mark.asyncio
+async def test_unfollow_sse_notification_format():
+    """
+    Тест формата SSE уведомления об отписке
+    """
+    published_data = []
+
+    async def mock_publish(channel: str, data: bytes) -> None:
+        published_data.append((channel, orjson.loads(data)))
+
+    with patch.object(redis, 'publish', side_effect=mock_publish):
+        # Данные отписавшегося пользователя
+        follower_data = {
+            "id": 789,
+            "name": "Test Unfollower",
+            "slug": "test-unfollower",
+            "pic": "https://example.com/avatar2.jpg"
+        }
+
+        target_author_id = 101
+
+        # Отправляем уведомление об отписке
+        await notify_follower(
+            follower=follower_data,
+            author_id=target_author_id,
+            action="unfollow"
+        )
+
+        # Проверяем формат для отписки
+        assert len(published_data) == 1
+        channel, data = published_data[0]
+
+        assert channel == f"follower:{target_author_id}"
+
+        # Для отписки action должен быть "delete"
+        assert data["action"] == "delete"
+        assert data["entity"] == "follower"
+        assert data["payload"]["follower_id"] == 789
+        assert data["payload"]["following_id"] == target_author_id
+        assert "id" in data["payload"]
+        assert "created_at" in data["payload"]
+
+        print(f"✅ SSE уведомление об отписке отправлено правильно: {data}")
+
+
+@pytest.mark.asyncio
+async def test_custom_subscription_id():
+    """
+    Тест передачи пользовательского ID подписки
+    """
+    published_data = []
+
+    async def mock_publish(channel: str, data: bytes) -> None:
+        published_data.append((channel, orjson.loads(data)))
+
+    with patch.object(redis, 'publish', side_effect=mock_publish):
+        # Данные подписчика
+        follower_data = {
+            "id": 777,
+            "name": "Test User",
+            "slug": "test-user",
+            "pic": "https://example.com/avatar.jpg"
+        }
+
+        target_author_id = 333
+        custom_subscription_id = 12345
+
+        # Отправляем уведомление с пользовательским ID
+        await notify_follower(
+            follower=follower_data,
+            author_id=target_author_id,
+            action="follow",
+            subscription_id=custom_subscription_id
+        )
+
+        # Проверяем, что передается правильный subscription_id
+        assert len(published_data) == 1
+        channel, data = published_data[0]
+
+        assert data["payload"]["id"] == custom_subscription_id
+        assert data["payload"]["follower_id"] == 777
+        assert data["payload"]["following_id"] == target_author_id
+
+        print(f"✅ Передан пользовательский subscription_id: {custom_subscription_id}")