mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-07-06 22:12:12 +02:00
ca9bd28934
Apply the same rename to surfsense_backend/tests: workspace_id fields/vars, Workspace* classes/schemas, table name searchspaces -> workspaces in raw SQL, and the API URL spellings -> /workspaces. Preserves the carve-out wire literals tests assert (Celery task names, OTel key "search_space.id").
64 lines
2.1 KiB
Python
64 lines
2.1 KiB
Python
"""Behavior guard for the @mention notification handler."""
|
|
|
|
from __future__ import annotations
|
|
|
|
import pytest
|
|
from sqlalchemy.ext.asyncio import AsyncSession
|
|
|
|
from app.db import Workspace, User
|
|
from app.notifications.service import NotificationService
|
|
|
|
pytestmark = pytest.mark.integration
|
|
|
|
handler = NotificationService.mention
|
|
|
|
|
|
async def _notify(db_session, db_user, db_workspace, *, mention_id=1, preview="hi"):
|
|
"""Raise an @mention notification for the assertions in the tests below."""
|
|
return await handler.notify_new_mention(
|
|
session=db_session,
|
|
mentioned_user_id=db_user.id,
|
|
mention_id=mention_id,
|
|
comment_id=10,
|
|
message_id=20,
|
|
thread_id=30,
|
|
thread_title="Thread",
|
|
author_id="author-1",
|
|
author_name="Alice",
|
|
author_avatar_url=None,
|
|
author_email="alice@surfsense.net",
|
|
content_preview=preview,
|
|
workspace_id=db_workspace.id,
|
|
)
|
|
|
|
|
|
async def test_new_mention_title_and_message(
|
|
db_session: AsyncSession, db_user: User, db_workspace: Workspace
|
|
):
|
|
"""A mention notification names the author and carries the comment preview."""
|
|
notification = await _notify(db_session, db_user, db_workspace, preview="hello")
|
|
|
|
assert notification.type == "new_mention"
|
|
assert notification.title == "Alice mentioned you"
|
|
assert notification.message == "hello"
|
|
|
|
|
|
async def test_new_mention_truncates_long_preview(
|
|
db_session: AsyncSession, db_user: User, db_workspace: Workspace
|
|
):
|
|
"""A long comment preview is truncated in the mention message."""
|
|
notification = await _notify(
|
|
db_session, db_user, db_workspace, preview="x" * 150
|
|
)
|
|
|
|
assert notification.message == "x" * 100 + "..."
|
|
|
|
|
|
async def test_new_mention_is_idempotent(
|
|
db_session: AsyncSession, db_user: User, db_workspace: Workspace
|
|
):
|
|
"""Re-notifying the same mention id reuses the existing notification row."""
|
|
first = await _notify(db_session, db_user, db_workspace, mention_id=7)
|
|
second = await _notify(db_session, db_user, db_workspace, mention_id=7)
|
|
|
|
assert second.id == first.id
|