Merge branch 'dev' of ssh://git.commumedia.org:2222/alex/headquarter into dev
This commit is contained in:
Alex Blank
@@ -0,0 +1,334 @@
|
||||
"""Unit tests for NotificationService."""
|
||||
|
||||
import uuid
|
||||
from datetime import datetime, timedelta, timezone
|
||||
|
||||
import pytest
|
||||
from sqlalchemy import select
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from src.models.notification import Notification
|
||||
from src.models.user import User
|
||||
from src.services.notification_service import NotificationService
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
def notification_service() -> NotificationService:
|
||||
return NotificationService()
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def user_a(db_session: AsyncSession) -> User:
|
||||
user = User(
|
||||
id=uuid.uuid4(),
|
||||
email="user-a@headquarter.local",
|
||||
name="User A",
|
||||
authentik_id=f"authentik-{uuid.uuid4()}",
|
||||
avatar_url=None,
|
||||
)
|
||||
db_session.add(user)
|
||||
await db_session.commit()
|
||||
return user
|
||||
|
||||
|
||||
@pytest.fixture
|
||||
async def user_b(db_session: AsyncSession) -> User:
|
||||
user = User(
|
||||
id=uuid.uuid4(),
|
||||
email="user-b@headquarter.local",
|
||||
name="User B",
|
||||
authentik_id=f"authentik-{uuid.uuid4()}",
|
||||
avatar_url=None,
|
||||
)
|
||||
db_session.add(user)
|
||||
await db_session.commit()
|
||||
return user
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_create_notification(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
notification = await notification_service.create_notification(
|
||||
db_session,
|
||||
user_a.id,
|
||||
category="instance",
|
||||
severity="info",
|
||||
title="Container started",
|
||||
message="Instance is running",
|
||||
source_type="tool_instances",
|
||||
source_id=uuid.uuid4(),
|
||||
metadata={"key": "value"},
|
||||
)
|
||||
|
||||
assert notification.user_id == user_a.id
|
||||
assert notification.category == "instance"
|
||||
assert notification.severity == "info"
|
||||
assert notification.title == "Container started"
|
||||
assert notification.message == "Instance is running"
|
||||
assert notification.source_type == "tool_instances"
|
||||
assert notification.notification_metadata == {"key": "value"}
|
||||
assert notification.read_at is None
|
||||
assert notification.dismissed_at is None
|
||||
assert notification.created_at is not None
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_notifications_orders_by_created_at_desc(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
n1 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="First"
|
||||
)
|
||||
n1.created_at = datetime.now(timezone.utc) - timedelta(seconds=2)
|
||||
await db_session.commit()
|
||||
await db_session.refresh(n1)
|
||||
|
||||
n2 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Second"
|
||||
)
|
||||
n2.created_at = datetime.now(timezone.utc) - timedelta(seconds=1)
|
||||
await db_session.commit()
|
||||
await db_session.refresh(n2)
|
||||
|
||||
n3 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Third"
|
||||
)
|
||||
|
||||
items, total = await notification_service.list_notifications(db_session, user_a.id)
|
||||
|
||||
assert total == 3
|
||||
assert [item.id for item in items] == [n3.id, n2.id, n1.id]
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_notifications_excludes_dismissed(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
n1 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Visible"
|
||||
)
|
||||
n2 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Dismissed"
|
||||
)
|
||||
await notification_service.dismiss(db_session, n2.id, user_a.id)
|
||||
|
||||
items, total = await notification_service.list_notifications(db_session, user_a.id)
|
||||
|
||||
assert total == 1
|
||||
assert items[0].id == n1.id
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_notifications_unread_only(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
n1 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Unread"
|
||||
)
|
||||
n2 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Read"
|
||||
)
|
||||
await notification_service.mark_read(db_session, n2.id, user_a.id)
|
||||
|
||||
items, total = await notification_service.list_notifications(
|
||||
db_session, user_a.id, unread_only=True
|
||||
)
|
||||
|
||||
assert total == 1
|
||||
assert items[0].id == n1.id
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_unread_count(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
for i in range(5):
|
||||
n = await notification_service.create_notification(
|
||||
db_session,
|
||||
user_a.id,
|
||||
category="instance",
|
||||
severity="info",
|
||||
title=f"Notification {i}",
|
||||
)
|
||||
if i >= 3:
|
||||
await notification_service.mark_read(db_session, n.id, user_a.id)
|
||||
|
||||
count = await notification_service.get_unread_count(db_session, user_a.id)
|
||||
|
||||
assert count == 3
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_mark_read_sets_read_at(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
n = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Unread"
|
||||
)
|
||||
|
||||
updated = await notification_service.mark_read(db_session, n.id, user_a.id)
|
||||
|
||||
assert updated.read_at is not None
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_mark_all_read_affects_all_unread(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
for i in range(4):
|
||||
await notification_service.create_notification(
|
||||
db_session,
|
||||
user_a.id,
|
||||
category="instance",
|
||||
severity="info",
|
||||
title=f"Notification {i}",
|
||||
)
|
||||
|
||||
marked = await notification_service.mark_all_read(db_session, user_a.id)
|
||||
|
||||
assert marked == 4
|
||||
count = await notification_service.get_unread_count(db_session, user_a.id)
|
||||
assert count == 0
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_dismiss_sets_dismissed_at(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
n = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="To dismiss"
|
||||
)
|
||||
|
||||
await notification_service.dismiss(db_session, n.id, user_a.id)
|
||||
|
||||
result = await db_session.execute(
|
||||
select(Notification).where(Notification.id == n.id)
|
||||
)
|
||||
row = result.scalar_one()
|
||||
assert row.dismissed_at is not None
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_mark_read_wrong_owner_raises(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
user_b: User,
|
||||
) -> None:
|
||||
n = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Owned by A"
|
||||
)
|
||||
|
||||
with pytest.raises(ValueError, match="Notification not found"):
|
||||
await notification_service.mark_read(db_session, n.id, user_b.id)
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_dismiss_wrong_owner_raises(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
user_b: User,
|
||||
) -> None:
|
||||
n = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Owned by A"
|
||||
)
|
||||
|
||||
with pytest.raises(ValueError, match="Notification not found"):
|
||||
await notification_service.dismiss(db_session, n.id, user_b.id)
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_list_notifications_mute_categories(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title="Instance"
|
||||
)
|
||||
n2 = await notification_service.create_notification(
|
||||
db_session, user_a.id, category="system", severity="info", title="System"
|
||||
)
|
||||
|
||||
items, total = await notification_service.list_notifications(
|
||||
db_session, user_a.id, mute_categories=["instance"]
|
||||
)
|
||||
|
||||
assert total == 1
|
||||
assert items[0].id == n2.id
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_get_unread_count_excludes_dismissed(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
) -> None:
|
||||
n = await notification_service.create_notification(
|
||||
db_session,
|
||||
user_a.id,
|
||||
category="instance",
|
||||
severity="info",
|
||||
title="Unread dismissed",
|
||||
)
|
||||
await notification_service.dismiss(db_session, n.id, user_a.id)
|
||||
|
||||
count = await notification_service.get_unread_count(db_session, user_a.id)
|
||||
|
||||
assert count == 0
|
||||
|
||||
|
||||
@pytest.mark.unit
|
||||
@pytest.mark.asyncio
|
||||
async def test_mark_all_read_affects_only_caller(
|
||||
db_session: AsyncSession,
|
||||
notification_service: NotificationService,
|
||||
user_a: User,
|
||||
user_b: User,
|
||||
) -> None:
|
||||
for i in range(3):
|
||||
await notification_service.create_notification(
|
||||
db_session, user_a.id, category="instance", severity="info", title=f"A-{i}"
|
||||
)
|
||||
for i in range(2):
|
||||
await notification_service.create_notification(
|
||||
db_session, user_b.id, category="instance", severity="info", title=f"B-{i}"
|
||||
)
|
||||
|
||||
marked = await notification_service.mark_all_read(db_session, user_a.id)
|
||||
|
||||
assert marked == 3
|
||||
count_a = await notification_service.get_unread_count(db_session, user_a.id)
|
||||
count_b = await notification_service.get_unread_count(db_session, user_b.id)
|
||||
assert count_a == 0
|
||||
assert count_b == 2
|
||||
Reference in New Issue
Block a user