PolyWeather/tests/test_weekly_points_display.py

from datetime import datetime

from src.bot.io_layer import BotIOLayer
from src.database.db_manager import DBManager


class _DummyBot:
    def send_message(self, *args, **kwargs):
        return None


class _User:
    def __init__(self, user_id: int, username: str):
        self.id = user_id
        self.username = username
        self.first_name = username


def test_points_rank_display_uses_invite_points_copy(tmp_path):
    db = DBManager(str(tmp_path / "test.db"))
    io = BotIOLayer(_DummyBot(), db)
    user = _User(1001, "yuan")

    db.upsert_user(user.id, user.username)
    now = datetime.now()
    week_key = f"{now.isocalendar()[0]}-W{now.isocalendar()[1]:02d}"
    with db._get_connection() as conn:  # noqa: SLF001
        conn.execute(
            """
            UPDATE users
            SET points = ?, message_count = ?, daily_points = ?, daily_points_date = ?, weekly_points = ?, weekly_points_week = ?
            WHERE telegram_id = ?
            """,
            (206, 15, 0, now.strftime("%Y-%m-%d"), 0, "", user.id),
        )
        conn.execute(
            """
            INSERT INTO weekly_points_archive (telegram_id, week_key, points, updated_at)
            VALUES (?, ?, ?, ?)
            ON CONFLICT(telegram_id, week_key) DO UPDATE SET
                points = excluded.points,
                updated_at = excluded.updated_at
            """,
            (user.id, week_key, 192, now.isoformat()),
        )
        conn.commit()

    text = io.build_points_rank_text(user)
    assert "用户积分排行" in text
    assert "累计积分: <code>206</code>" in text
    assert "积分获取: <code>邀请付费用户</code>" in text
    assert "本周发言积分" not in text
    assert "今日发言积分" not in text