TelegramBotAdmin/tests/test_giveaway_points.py

from __future__ import annotations

import asyncio
from datetime import UTC, datetime, timedelta

import pytest


async def _seed_points(points, *, chat_id: int, user_id: int, amount: int) -> None:
    await points.adjust_points(
        chat_id=chat_id,
        user_id=user_id,
        delta=amount,
        source=points.SOURCE_ADMIN,
        idempotency_key=f"seed:{chat_id}:{user_id}",
        reason="test seed",
    )


async def test_entry_cost_duplicate_click_and_cancel_refund(app_modules):
    points = app_modules.load("wbb.utils.dbpoints")
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    await _seed_points(points, chat_id=-100, user_id=10, amount=100)
    giveaway = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Launch",
        description="",
        prizes=[{"name": "First", "count": 1, "points_reward": 5}],
        ends_at=datetime.now(UTC) + timedelta(hours=1),
        minimum_points=50,
        entry_cost=10,
        participation_reward=2,
    )

    first, _ = await service.join_giveaway(
        giveaway["giveaway_id"],
        user_id=10,
        username="alice",
        first_name="Alice",
        display_name="Alice Example",
    )
    duplicate, _ = await service.join_giveaway(
        giveaway["giveaway_id"],
        user_id=10,
        username="alice",
        first_name="Alice",
        display_name="Alice Example",
    )
    assert (first, duplicate) == ("ok", "duplicate")
    account = await points.get_point_account(-100, 10)
    assert account["balance"] == 90
    assert account["display_name"] == "Alice Example"
    participant = await giveaway_db.get_participant(giveaway["giveaway_id"], 10)
    assert participant["display_name"] == "Alice Example"
    assert "Alice Example" in service._participant_mention(participant)
    assert "@alice" not in service._participant_mention(participant)

    ok, _, _ = await service.cancel_and_refund_giveaway(
        giveaway["giveaway_id"], chat_id=-100, publish=False
    )
    again, _, _ = await service.cancel_and_refund_giveaway(
        giveaway["giveaway_id"], chat_id=-100, publish=False
    )
    assert ok is True
    assert again is False
    assert (await points.get_point_account(-100, 10))["balance"] == 100


async def test_registration_window_blocks_early_entry_and_refreshes_count(app_modules):
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    now = datetime.now(UTC)
    future = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Scheduled",
        description="",
        prizes=[{"name": "Winner", "count": 1}],
        starts_at=now + timedelta(hours=1),
        ends_at=now + timedelta(hours=2),
    )

    status, _ = await service.join_giveaway(
        future["giveaway_id"],
        user_id=10,
        username=None,
        first_name="Alice",
    )
    assert status == "not_started"
    assert await giveaway_db.count_participants(future["giveaway_id"]) == 0
    assert await giveaway_db.list_unannounced_started_giveaways() == []

    await giveaway_db.giveawaysdb.update_one(
        {"giveaway_id": future["giveaway_id"]},
        {"$set": {"starts_at": now - timedelta(minutes=1)}},
    )
    started = await giveaway_db.list_unannounced_started_giveaways()
    assert [item["giveaway_id"] for item in started] == [future["giveaway_id"]]
    await giveaway_db.mark_giveaway_start_announced(future["giveaway_id"])
    assert await giveaway_db.list_unannounced_started_giveaways() == []

    current = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Open",
        description="",
        prizes=[{"name": "Winner", "count": 1}],
        starts_at=now - timedelta(minutes=1),
        ends_at=now + timedelta(hours=1),
    )
    await giveaway_db.attach_giveaway_message(current["giveaway_id"], -100, 99)
    status, _ = await service.join_giveaway(
        current["giveaway_id"],
        user_id=10,
        username="alice",
        first_name="Alice",
    )

    assert status == "ok"
    assert len(app_modules.app.edited_messages) == 1
    assert app_modules.app.edited_messages[0][:2] == (-100, 99)
    assert "<b>参与人数：</b>1" in app_modules.app.edited_messages[0][2]


async def test_registration_start_must_precede_draw_time(app_modules):
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    now = datetime.now(UTC)
    with pytest.raises(ValueError, match="报名开始时间必须早于开奖时间"):
        await giveaway_db.create_giveaway(
            chat_id=-100,
            creator_id=1,
            creator_name="Admin",
            title="Invalid schedule",
            description="",
            prizes=[{"name": "Winner", "count": 1}],
            starts_at=now + timedelta(hours=2),
            ends_at=now + timedelta(hours=1),
        )


async def test_running_giveaway_update_refreshes_message_and_rejects_stale_write(
    app_modules,
):
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    now = datetime.now(UTC)
    giveaway = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Original title",
        description="Original description",
        prizes=[{"name": "Winner", "count": 1, "points_reward": 5}],
        starts_at=now + timedelta(hours=1),
        ends_at=now + timedelta(hours=2),
        minimum_points=20,
        entry_cost=5,
        participation_reward=2,
    )
    await giveaway_db.attach_giveaway_message(giveaway["giveaway_id"], -100, 99)
    current = await giveaway_db.get_giveaway(giveaway["giveaway_id"])
    new_starts_at = now + timedelta(hours=2)
    new_ends_at = now + timedelta(hours=3)

    updated = await service.update_and_refresh_giveaway(
        giveaway["giveaway_id"],
        title="Updated title",
        description="Valid for one month",
        starts_at=new_starts_at,
        ends_at=new_ends_at,
        expected_updated_at=current["updated_at"],
    )

    assert updated["title"] == "Updated title"
    assert updated["description"] == "Valid for one month"
    assert int(giveaway_db.as_utc(updated["starts_at"]).timestamp() * 1000) == int(
        new_starts_at.timestamp() * 1000
    )
    assert int(giveaway_db.as_utc(updated["ends_at"]).timestamp() * 1000) == int(
        new_ends_at.timestamp() * 1000
    )
    assert updated["prizes"] == giveaway["prizes"]
    assert updated["minimum_points"] == 20
    assert updated["entry_cost"] == 5
    assert updated["participation_reward"] == 2
    assert app_modules.app.edited_messages[-1][:2] == (-100, 99)
    assert "Updated title" in app_modules.app.edited_messages[-1][2]
    assert "Valid for one month" in app_modules.app.edited_messages[-1][2]

    with pytest.raises(service.GiveawayServiceError) as error:
        await service.update_and_refresh_giveaway(
            giveaway["giveaway_id"],
            title="Stale title",
            description="Stale description",
            starts_at=new_starts_at,
            ends_at=new_ends_at,
            expected_updated_at=current["updated_at"],
        )
    assert error.value.code == "giveaway_stale"


async def test_running_giveaway_cannot_move_registration_start_after_it_begins(
    app_modules,
):
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    now = datetime.now(UTC)
    giveaway = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Open giveaway",
        description="",
        prizes=[{"name": "Winner", "count": 1}],
        starts_at=now - timedelta(minutes=1),
        ends_at=now + timedelta(hours=1),
    )

    with pytest.raises(service.GiveawayServiceError) as error:
        await service.update_and_refresh_giveaway(
            giveaway["giveaway_id"],
            title="Open giveaway",
            description="Updated",
            starts_at=now + timedelta(minutes=1),
            ends_at=now + timedelta(hours=2),
            expected_updated_at=giveaway["updated_at"],
        )
    assert error.value.code == "giveaway_registration_started"


async def test_draw_rewards_are_idempotent_under_competing_finish_calls(app_modules):
    points = app_modules.load("wbb.utils.dbpoints")
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    for user_id in (10, 11):
        await _seed_points(points, chat_id=-100, user_id=user_id, amount=20)
    giveaway = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Draw",
        description="",
        prizes=[{"name": "Winner", "count": 1, "points_reward": 5}],
        ends_at=datetime.now(UTC) + timedelta(hours=1),
        entry_cost=3,
        participation_reward=2,
    )
    for user_id in (10, 11):
        status, _ = await service.join_giveaway(
            giveaway["giveaway_id"],
            user_id=user_id,
            username=None,
            first_name=str(user_id),
        )
        assert status == "ok"

    results = await asyncio.gather(
        service.finish_and_publish_giveaway(giveaway["giveaway_id"], publish=False),
        service.finish_and_publish_giveaway(giveaway["giveaway_id"], publish=False),
    )
    assert sum(1 for result in results if result[0]) == 1
    finished = await giveaway_db.get_giveaway(giveaway["giveaway_id"])
    assert finished["status"] == giveaway_db.STATUS_FINISHED
    winner_id = int(finished["winners"][0]["user_id"])
    balances = {
        user_id: (await points.get_point_account(-100, user_id))["balance"]
        for user_id in (10, 11)
    }
    assert balances[winner_id] == 24
    assert balances[10 if winner_id == 11 else 11] == 19


async def test_draw_pins_result_and_deletes_original_giveaway(app_modules):
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    giveaway = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Pinned draw",
        description="",
        prizes=[{"name": "Winner", "count": 1}],
        ends_at=datetime.now(UTC) + timedelta(hours=1),
    )
    await giveaway_db.attach_giveaway_message(
        giveaway["giveaway_id"], -100, 99
    )
    await giveaway_db.add_participant(
        giveaway_id=giveaway["giveaway_id"],
        user_id=10,
        username="alice",
        first_name="Alice",
    )

    ok, _, _ = await service.finish_and_publish_giveaway(
        giveaway["giveaway_id"]
    )

    assert ok is True
    assert app_modules.app.pinned_messages == [(-100, 1)]
    assert app_modules.app.deleted_messages == [(-100, 99)]
    assert app_modules.app.unpinned_messages == []
    assert "抽奖结果" in app_modules.app.sent_messages[0][1]


async def test_participant_write_failure_creates_compensating_refund(
    app_modules, monkeypatch
):
    points = app_modules.load("wbb.utils.dbpoints")
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    await _seed_points(points, chat_id=-100, user_id=10, amount=20)
    giveaway = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Failure",
        description="",
        prizes=[{"name": "Winner", "count": 1}],
        ends_at=datetime.now(UTC) + timedelta(hours=1),
        entry_cost=5,
    )
    original_add_participant = service.add_participant

    async def fail_insert(**_kwargs):
        raise RuntimeError("simulated participant write failure")

    monkeypatch.setattr(service, "add_participant", fail_insert)
    with pytest.raises(RuntimeError):
        await service.join_giveaway(
            giveaway["giveaway_id"],
            user_id=10,
            username=None,
            first_name="Alice",
        )
    assert (await points.get_point_account(-100, 10))["balance"] == 20
    transactions, _ = await points.list_point_transactions(
        chat_id=-100, user_id=10, page_size=20
    )
    assert {item["source"] for item in transactions} >= {
        points.SOURCE_GIVEAWAY_ENTRY,
        points.SOURCE_GIVEAWAY_REFUND,
    }

    monkeypatch.setattr(service, "add_participant", original_add_participant)
    status, _ = await service.join_giveaway(
        giveaway["giveaway_id"],
        user_id=10,
        username=None,
        first_name="Alice",
    )
    assert status == "ok"
    assert (await points.get_point_account(-100, 10))["balance"] == 15


async def test_retry_reuses_uncompensated_entry_debit(app_modules):
    points = app_modules.load("wbb.utils.dbpoints")
    giveaway_db = app_modules.load("wbb.utils.dbgiveaway")
    service = app_modules.load("wbb.services.giveaways")
    await _seed_points(points, chat_id=-100, user_id=10, amount=20)
    giveaway = await giveaway_db.create_giveaway(
        chat_id=-100,
        creator_id=1,
        creator_name="Admin",
        title="Crash recovery",
        description="",
        prizes=[{"name": "Winner", "count": 1}],
        ends_at=datetime.now(UTC) + timedelta(hours=1),
        minimum_points=20,
        entry_cost=5,
    )
    await points.adjust_points(
        chat_id=-100,
        user_id=10,
        delta=-5,
        source=points.SOURCE_GIVEAWAY_ENTRY,
        idempotency_key=f"giveaway-entry:{giveaway['giveaway_id']}:10",
        reference_id=giveaway["giveaway_id"],
        reason="simulated crash after debit",
    )

    status, _ = await service.join_giveaway(
        giveaway["giveaway_id"],
        user_id=10,
        username=None,
        first_name="Alice",
    )

    assert status == "ok"
    assert (await points.get_point_account(-100, 10))["balance"] == 15
    transactions, _ = await points.list_point_transactions(
        chat_id=-100, user_id=10, source=points.SOURCE_GIVEAWAY_ENTRY
    )
    assert len(transactions) == 1