feat: subs and views
This commit is contained in:
Artem Tsyrulnikov
@@ -19,6 +19,8 @@ class MockDatabase:
|
||||
self.external_channels: dict[uuid.UUID, domain.ExternalChannel] = {}
|
||||
self.creatives: dict[uuid.UUID, domain.Creative] = {}
|
||||
self.purchases: dict[uuid.UUID, domain.Purchase] = {}
|
||||
self.subscriptions: dict[uuid.UUID, domain.Subscription] = {}
|
||||
self.views_snapshots: dict[uuid.UUID, domain.ViewsSnapshot] = {}
|
||||
self.m2m_target_external: dict[uuid.UUID, list[uuid.UUID]] = {}
|
||||
|
||||
def transaction(self) -> AbstractAsyncContextManager[None]:
|
||||
@@ -205,6 +207,8 @@ class MockDatabase:
|
||||
purchase.subscriptions_count = 0
|
||||
if not hasattr(purchase, 'status') or purchase.status is None:
|
||||
purchase.status = domain.PurchaseStatus.ACTIVE
|
||||
if not hasattr(purchase, 'views_availability') or purchase.views_availability is None:
|
||||
purchase.views_availability = domain.PostViewsAvailability.UNKNOWN
|
||||
# Загружаем связанные объекты
|
||||
target_ch = self.target_channels.get(purchase.target_channel_id)
|
||||
if target_ch:
|
||||
@@ -257,6 +261,61 @@ class MockDatabase:
|
||||
return True
|
||||
return False
|
||||
|
||||
async def get_purchase_by_invite_link(self, invite_link: str) -> domain.Purchase | None:
|
||||
for purchase in self.purchases.values():
|
||||
if purchase.invite_link == invite_link:
|
||||
return purchase
|
||||
return None
|
||||
|
||||
async def create_subscription(self, subscription: domain.Subscription) -> domain.Subscription:
|
||||
if not subscription.id:
|
||||
subscription.id = uuid.uuid4()
|
||||
if not hasattr(subscription, 'created_at') or subscription.created_at is None:
|
||||
subscription.created_at = datetime.datetime.now(datetime.UTC)
|
||||
if not hasattr(subscription, 'updated_at') or subscription.updated_at is None:
|
||||
subscription.updated_at = datetime.datetime.now(datetime.UTC)
|
||||
# Загружаем связанный объект
|
||||
purchase = self.purchases.get(subscription.purchase_id)
|
||||
if purchase:
|
||||
subscription.purchase = purchase
|
||||
self.subscriptions[subscription.id] = subscription
|
||||
return subscription
|
||||
|
||||
async def get_subscription_by_user_and_purchase(
|
||||
self, user_telegram_id: int, purchase_id: uuid.UUID
|
||||
) -> domain.Subscription | None:
|
||||
for subscription in self.subscriptions.values():
|
||||
if subscription.user_telegram_id == user_telegram_id and subscription.purchase_id == purchase_id:
|
||||
return subscription
|
||||
return None
|
||||
|
||||
async def create_views_snapshot(self, snapshot: domain.ViewsSnapshot) -> domain.ViewsSnapshot:
|
||||
if not snapshot.id:
|
||||
snapshot.id = uuid.uuid4()
|
||||
if not hasattr(snapshot, 'created_at') or snapshot.created_at is None:
|
||||
snapshot.created_at = datetime.datetime.now(datetime.UTC)
|
||||
if not hasattr(snapshot, 'updated_at') or snapshot.updated_at is None:
|
||||
snapshot.updated_at = datetime.datetime.now(datetime.UTC)
|
||||
purchase = self.purchases.get(snapshot.purchase_id)
|
||||
if purchase:
|
||||
snapshot.purchase = purchase
|
||||
self.views_snapshots[snapshot.id] = snapshot
|
||||
return snapshot
|
||||
|
||||
async def get_views_history(
|
||||
self,
|
||||
purchase_id: uuid.UUID,
|
||||
*,
|
||||
from_date: datetime.datetime | None = None,
|
||||
to_date: datetime.datetime | None = None,
|
||||
) -> list[domain.ViewsSnapshot]:
|
||||
result = [s for s in self.views_snapshots.values() if s.purchase_id == purchase_id]
|
||||
if from_date:
|
||||
result = [s for s in result if s.fetched_at >= from_date]
|
||||
if to_date:
|
||||
result = [s for s in result if s.fetched_at <= to_date]
|
||||
return sorted(result, key=lambda s: s.fetched_at)
|
||||
|
||||
|
||||
class MockTelegramWriter:
|
||||
def __init__(self) -> None:
|
||||
@@ -279,6 +338,10 @@ class MockTelegramWriter:
|
||||
link_type = 'approval' if requires_approval else 'public'
|
||||
return f'https://t.me/+{link_type}_link_{self.invite_link_counter}_{chat_id}'
|
||||
|
||||
async def get_post_views(self, post_url: str) -> int | None:
|
||||
"""Mock implementation returns None (TDLib not integrated)."""
|
||||
return None
|
||||
|
||||
|
||||
class MockJWTEncoder:
|
||||
def encode_access_token(self, user_id: uuid.UUID, telegram_id: int, username: str | None = None) -> str:
|
||||
|
||||
342
tests/test_subscriptions.py
Normal file
342
tests/test_subscriptions.py
Normal file
@@ -0,0 +1,342 @@
|
||||
"""Тесты обработки подписок пользователей через invite_link."""
|
||||
|
||||
import datetime
|
||||
|
||||
import pytest
|
||||
|
||||
from src import domain, usecase
|
||||
from tests.conftest import MockDatabase
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_subscription_success(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Первая подписка пользователя по invite_link успешно создаётся."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=50000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
invite_link = 'https://t.me/+unique_link_123'
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link=invite_link,
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
subscriptions_count=0,
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
# Симулируем подписку пользователя
|
||||
subscriber_id = 999888777
|
||||
subscriber_username = 'new_subscriber'
|
||||
await usecases.handle_subscription(
|
||||
user_telegram_id=subscriber_id,
|
||||
username=subscriber_username,
|
||||
invite_link=invite_link,
|
||||
)
|
||||
|
||||
# Проверяем, что подписка создана
|
||||
subscription = await mock_database.get_subscription_by_user_and_purchase(subscriber_id, purchase.id)
|
||||
assert subscription is not None
|
||||
assert subscription.user_telegram_id == subscriber_id
|
||||
assert subscription.username == subscriber_username
|
||||
assert subscription.invite_link == invite_link
|
||||
|
||||
# Проверяем, что счётчик подписок у закупа увеличился
|
||||
updated_purchase = await mock_database.get_purchase(user.id, purchase.id)
|
||||
assert updated_purchase is not None
|
||||
assert updated_purchase.subscriptions_count == 1
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_subscription_duplicate_ignored(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Повторная подписка того же пользователя игнорируется - не создаёт дубликат."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=30000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
invite_link = 'https://t.me/+unique_link_456'
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link=invite_link,
|
||||
invite_link_type=domain.InviteLinkType.APPROVAL,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
subscriptions_count=0,
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
subscriber_id = 888777666
|
||||
subscriber_username = 'duplicate_user'
|
||||
|
||||
# Первая подписка
|
||||
await usecases.handle_subscription(
|
||||
user_telegram_id=subscriber_id,
|
||||
username=subscriber_username,
|
||||
invite_link=invite_link,
|
||||
)
|
||||
|
||||
# Проверяем счётчик
|
||||
updated_purchase = await mock_database.get_purchase(user.id, purchase.id)
|
||||
assert updated_purchase is not None
|
||||
assert updated_purchase.subscriptions_count == 1
|
||||
|
||||
# Повторная попытка подписки того же пользователя
|
||||
await usecases.handle_subscription(
|
||||
user_telegram_id=subscriber_id,
|
||||
username=subscriber_username,
|
||||
invite_link=invite_link,
|
||||
)
|
||||
|
||||
# Проверяем, что счётчик не увеличился
|
||||
updated_purchase_again = await mock_database.get_purchase(user.id, purchase.id)
|
||||
assert updated_purchase_again is not None
|
||||
assert updated_purchase_again.subscriptions_count == 1 # Остался 1, не увеличился
|
||||
|
||||
# Проверяем, что в БД только одна подписка этого пользователя
|
||||
all_subscriptions = [s for s in mock_database.subscriptions.values() if s.user_telegram_id == subscriber_id]
|
||||
assert len(all_subscriptions) == 1
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_handle_subscription_invalid_invite_link(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Подписка по несуществующему invite_link не создаётся."""
|
||||
fake_invite_link = 'https://t.me/+fake_link_999'
|
||||
|
||||
# Пытаемся обработать подписку по несуществующей ссылке
|
||||
await usecases.handle_subscription(
|
||||
user_telegram_id=777666555,
|
||||
username='unknown_user',
|
||||
invite_link=fake_invite_link,
|
||||
)
|
||||
|
||||
# Проверяем, что подписки не создано
|
||||
assert len(mock_database.subscriptions) == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_multiple_users_subscription_same_purchase(
|
||||
usecases: usecase.Usecase, mock_database: MockDatabase
|
||||
) -> None:
|
||||
"""Несколько пользователей подписываются через один закуп - каждый учитывается."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=40000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Viral Creative',
|
||||
text='Viral content',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
invite_link = 'https://t.me/+popular_link_789'
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link=invite_link,
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
subscriptions_count=0,
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
# Подписываются три разных пользователя
|
||||
users_data = [
|
||||
(111222333, 'user_one'),
|
||||
(444555666, 'user_two'),
|
||||
(777888999, 'user_three'),
|
||||
]
|
||||
|
||||
for telegram_id, username in users_data:
|
||||
await usecases.handle_subscription(
|
||||
user_telegram_id=telegram_id,
|
||||
username=username,
|
||||
invite_link=invite_link,
|
||||
)
|
||||
|
||||
# Проверяем, что создано 3 подписки
|
||||
purchase_subscriptions = [s for s in mock_database.subscriptions.values() if s.purchase_id == purchase.id]
|
||||
assert len(purchase_subscriptions) == 3
|
||||
|
||||
# Проверяем, что счётчик правильный
|
||||
updated_purchase = await mock_database.get_purchase(user.id, purchase.id)
|
||||
assert updated_purchase is not None
|
||||
assert updated_purchase.subscriptions_count == 3
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_subscription_tracking_across_different_purchases(
|
||||
usecases: usecase.Usecase, mock_database: MockDatabase
|
||||
) -> None:
|
||||
"""Один пользователь может подписаться через разные закупы - каждая подписка учитывается отдельно."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=25000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
# Создаём два разных закупа с разными invite_link
|
||||
purchase1 = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link_a',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
subscriptions_count=0,
|
||||
)
|
||||
purchase2 = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link_b',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
subscriptions_count=0,
|
||||
)
|
||||
await mock_database.create_purchase(purchase1)
|
||||
await mock_database.create_purchase(purchase2)
|
||||
|
||||
subscriber_id = 555444333
|
||||
|
||||
# Пользователь подписывается через первый закуп
|
||||
await usecases.handle_subscription(
|
||||
user_telegram_id=subscriber_id,
|
||||
username='multi_subscriber',
|
||||
invite_link=purchase1.invite_link,
|
||||
)
|
||||
|
||||
# Тот же пользователь "подписывается" через второй закуп
|
||||
# (технически невозможно, но тестируем логику)
|
||||
await usecases.handle_subscription(
|
||||
user_telegram_id=subscriber_id,
|
||||
username='multi_subscriber',
|
||||
invite_link=purchase2.invite_link,
|
||||
)
|
||||
|
||||
# Проверяем, что у каждого закупа свой счётчик
|
||||
updated_purchase1 = await mock_database.get_purchase(user.id, purchase1.id)
|
||||
updated_purchase2 = await mock_database.get_purchase(user.id, purchase2.id)
|
||||
|
||||
assert updated_purchase1 is not None
|
||||
assert updated_purchase1.subscriptions_count == 1
|
||||
|
||||
assert updated_purchase2 is not None
|
||||
assert updated_purchase2.subscriptions_count == 1
|
||||
|
||||
# Проверяем, что создано 2 подписки
|
||||
user_subscriptions = [s for s in mock_database.subscriptions.values() if s.user_telegram_id == subscriber_id]
|
||||
assert len(user_subscriptions) == 2
|
||||
424
tests/test_views.py
Normal file
424
tests/test_views.py
Normal file
@@ -0,0 +1,424 @@
|
||||
"""Тесты работы с просмотрами рекламных постов."""
|
||||
|
||||
import datetime
|
||||
|
||||
import pytest
|
||||
|
||||
from src import domain, dto, usecase
|
||||
from tests.conftest import MockDatabase
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_fetch_views_no_ad_post_url(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Попытка получить просмотры для закупа без ad_post_url возвращает unavailable."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=50000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link1',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
ad_post_url=None, # Нет ссылки на пост
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
input_data = dto.FetchViewsInput(
|
||||
purchase_id=purchase.id,
|
||||
user_id=user.id,
|
||||
)
|
||||
|
||||
result = await usecases.fetch_views(input=input_data)
|
||||
|
||||
assert result.views_count is None
|
||||
assert result.views_availability == domain.PostViewsAvailability.UNAVAILABLE
|
||||
assert result.error_message == 'No ad_post_url specified'
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_fetch_views_api_unavailable(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Если Telegram API недоступен (возвращает None), статус = unavailable."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=30000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link2',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
ad_post_url='https://t.me/externalchannel/123',
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
input_data = dto.FetchViewsInput(
|
||||
purchase_id=purchase.id,
|
||||
user_id=user.id,
|
||||
)
|
||||
|
||||
# Mock telegram writer returns None (TDLib not integrated)
|
||||
result = await usecases.fetch_views(input=input_data)
|
||||
|
||||
assert result.views_count is None
|
||||
assert result.views_availability == domain.PostViewsAvailability.UNAVAILABLE
|
||||
assert 'Failed to fetch views' in result.error_message
|
||||
|
||||
# Проверяем, что purchase обновлён
|
||||
updated_purchase = await mock_database.get_purchase(user.id, purchase.id)
|
||||
assert updated_purchase is not None
|
||||
assert updated_purchase.views_availability == domain.PostViewsAvailability.UNAVAILABLE
|
||||
assert updated_purchase.last_views_fetch_at is not None
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_update_views_manually(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Ручное обновление просмотров создаёт снимок и обновляет закуп."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=40000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link3',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
ad_post_url='https://t.me/externalchannel/456',
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
input_data = dto.UpdateViewsManuallyInput(
|
||||
purchase_id=purchase.id,
|
||||
user_id=user.id,
|
||||
views_count=15000,
|
||||
)
|
||||
|
||||
result = await usecases.update_views_manually(input=input_data)
|
||||
|
||||
# Проверяем возвращённый результат
|
||||
assert result.views_count == 15000
|
||||
assert result.views_availability == domain.PostViewsAvailability.MANUAL
|
||||
assert result.last_views_fetch_at is not None
|
||||
|
||||
# Проверяем, что создан снимок
|
||||
snapshots = [s for s in mock_database.views_snapshots.values() if s.purchase_id == purchase.id]
|
||||
assert len(snapshots) == 1
|
||||
assert snapshots[0].views_count == 15000
|
||||
|
||||
# Проверяем, что закуп обновлён
|
||||
updated_purchase = await mock_database.get_purchase(user.id, purchase.id)
|
||||
assert updated_purchase is not None
|
||||
assert updated_purchase.views_count == 15000
|
||||
assert updated_purchase.views_availability == domain.PostViewsAvailability.MANUAL
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_views_history_empty(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Получение истории просмотров для закупа без снимков возвращает пустой список."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=25000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link4',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
input_data = dto.GetViewsHistoryInput(
|
||||
purchase_id=purchase.id,
|
||||
user_id=user.id,
|
||||
)
|
||||
|
||||
result = await usecases.get_views_history(input=input_data)
|
||||
|
||||
assert len(result.snapshots) == 0
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_views_history_with_snapshots(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Получение истории просмотров возвращает снимки в хронологическом порядке."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=35000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link5',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
# Создаём несколько снимков
|
||||
now = datetime.datetime.now(datetime.UTC)
|
||||
snapshot1 = domain.ViewsSnapshot(
|
||||
purchase_id=purchase.id,
|
||||
views_count=1000,
|
||||
fetched_at=now - datetime.timedelta(hours=2),
|
||||
)
|
||||
snapshot2 = domain.ViewsSnapshot(
|
||||
purchase_id=purchase.id,
|
||||
views_count=1500,
|
||||
fetched_at=now - datetime.timedelta(hours=1),
|
||||
)
|
||||
snapshot3 = domain.ViewsSnapshot(
|
||||
purchase_id=purchase.id,
|
||||
views_count=2000,
|
||||
fetched_at=now,
|
||||
)
|
||||
await mock_database.create_views_snapshot(snapshot1)
|
||||
await mock_database.create_views_snapshot(snapshot2)
|
||||
await mock_database.create_views_snapshot(snapshot3)
|
||||
|
||||
input_data = dto.GetViewsHistoryInput(
|
||||
purchase_id=purchase.id,
|
||||
user_id=user.id,
|
||||
)
|
||||
|
||||
result = await usecases.get_views_history(input=input_data)
|
||||
|
||||
assert len(result.snapshots) == 3
|
||||
# Проверяем сортировку по времени (старые первые)
|
||||
assert result.snapshots[0].views_count == 1000
|
||||
assert result.snapshots[1].views_count == 1500
|
||||
assert result.snapshots[2].views_count == 2000
|
||||
|
||||
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_views_history_with_date_filter(usecases: usecase.Usecase, mock_database: MockDatabase) -> None:
|
||||
"""Фильтрация истории просмотров по датам работает корректно."""
|
||||
user = domain.User(telegram_id=123456, username='testuser')
|
||||
await mock_database.create_user(user)
|
||||
|
||||
target_channel = domain.TargetChannel(
|
||||
telegram_id=-1001234567890,
|
||||
title='Target Channel',
|
||||
username='targetchannel',
|
||||
user_id=user.id,
|
||||
is_active=True,
|
||||
)
|
||||
await mock_database.upsert_target_channel(target_channel)
|
||||
|
||||
external_channel = domain.ExternalChannel(
|
||||
telegram_id=-1009876543210,
|
||||
title='External Channel',
|
||||
username='externalchannel',
|
||||
description=None,
|
||||
subscribers_count=28000,
|
||||
user_id=user.id,
|
||||
)
|
||||
await mock_database.create_external_channel(external_channel)
|
||||
|
||||
creative = domain.Creative(
|
||||
name='Test Creative',
|
||||
text='Test text',
|
||||
target_channel_id=target_channel.id,
|
||||
user_id=user.id,
|
||||
status=domain.CreativeStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_creative(creative)
|
||||
|
||||
purchase = domain.Purchase(
|
||||
target_channel_id=target_channel.id,
|
||||
external_channel_id=external_channel.id,
|
||||
creative_id=creative.id,
|
||||
user_id=user.id,
|
||||
placement_date=datetime.datetime.now(datetime.UTC),
|
||||
invite_link='https://t.me/+link6',
|
||||
invite_link_type=domain.InviteLinkType.PUBLIC,
|
||||
status=domain.PurchaseStatus.ACTIVE,
|
||||
)
|
||||
await mock_database.create_purchase(purchase)
|
||||
|
||||
# Создаём снимки с разными датами
|
||||
now = datetime.datetime.now(datetime.UTC)
|
||||
snapshot1 = domain.ViewsSnapshot(
|
||||
purchase_id=purchase.id,
|
||||
views_count=1000,
|
||||
fetched_at=now - datetime.timedelta(days=3),
|
||||
)
|
||||
snapshot2 = domain.ViewsSnapshot(
|
||||
purchase_id=purchase.id,
|
||||
views_count=1500,
|
||||
fetched_at=now - datetime.timedelta(days=1),
|
||||
)
|
||||
snapshot3 = domain.ViewsSnapshot(
|
||||
purchase_id=purchase.id,
|
||||
views_count=2000,
|
||||
fetched_at=now,
|
||||
)
|
||||
await mock_database.create_views_snapshot(snapshot1)
|
||||
await mock_database.create_views_snapshot(snapshot2)
|
||||
await mock_database.create_views_snapshot(snapshot3)
|
||||
|
||||
# Запрашиваем историю за последние 2 дня
|
||||
input_data = dto.GetViewsHistoryInput(
|
||||
purchase_id=purchase.id,
|
||||
user_id=user.id,
|
||||
from_date=now - datetime.timedelta(days=2),
|
||||
)
|
||||
|
||||
result = await usecases.get_views_history(input=input_data)
|
||||
|
||||
# Должны вернуться только snapshot2 и snapshot3
|
||||
assert len(result.snapshots) == 2
|
||||
assert result.snapshots[0].views_count == 1500
|
||||
assert result.snapshots[1].views_count == 2000
|
||||
Reference in New Issue
Block a user