shaguabijia-app-server/tests/test_admin_ad_revenue_scope.py

"""广告收益报表的环境与业务代码位口径。"""
from __future__ import annotations

from datetime import UTC, datetime

from sqlalchemy import delete

from app.admin.repositories import ad_revenue
from app.db.session import SessionLocal
from app.models.ad_ecpm import AdEcpmRecord
from app.models.ad_pangle_revenue import AdPangleDailyRevenue
from app.models.ad_reward import AdRewardRecord
from app.models.user import User

REPORT_DATE = "2040-02-03"
PLAYBACK_DATE = "2040-02-04"


def test_business_scope_filters_client_and_pangle_by_env_and_code(monkeypatch) -> None:
    db = SessionLocal()
    try:
        user = User(
            phone="18800009991",
            username="29999999991",
            register_channel="sms",
        )
        db.add(user)
        db.flush()
        db.add_all([
            AdEcpmRecord(
                user_id=user.id, ad_type="reward_video", ad_session_id="scope-prod-business",
                app_env="prod", our_code_id="prod-reward", ecpm_raw="10000",
                report_date=REPORT_DATE, created_at=datetime(2040, 2, 3, tzinfo=UTC),
            ),
            AdEcpmRecord(
                user_id=user.id, ad_type="draw", ad_session_id="scope-prod-demo",
                app_env="prod", our_code_id="prod-demo", ecpm_raw="20000",
                report_date=REPORT_DATE, created_at=datetime(2040, 2, 3, tzinfo=UTC),
            ),
            AdEcpmRecord(
                user_id=user.id, ad_type="draw", ad_session_id="scope-prod-known-business",
                app_env="prod", our_code_id="104098712", ecpm_raw="40000",
                report_date=REPORT_DATE, created_at=datetime(2040, 2, 3, tzinfo=UTC),
            ),
            AdEcpmRecord(
                user_id=user.id, ad_type="reward_video", ad_session_id="scope-test-business",
                app_env="test", our_code_id="104127529", ecpm_raw="30000",
                report_date=REPORT_DATE, created_at=datetime(2040, 2, 3, tzinfo=UTC),
            ),
            AdPangleDailyRevenue(
                report_date=REPORT_DATE, app_env="prod", our_code_id="prod-reward",
                adn="", revenue_yuan=1.5, api_revenue_yuan=1.2, impressions=10,
            ),
            AdPangleDailyRevenue(
                report_date=REPORT_DATE, app_env="prod", our_code_id="prod-demo",
                adn="", revenue_yuan=8.0, api_revenue_yuan=7.0, impressions=40,
            ),
            AdPangleDailyRevenue(
                report_date=REPORT_DATE, app_env="prod", our_code_id="104098712",
                adn="", revenue_yuan=2.5, api_revenue_yuan=2.0, impressions=20,
            ),
            AdPangleDailyRevenue(
                report_date=REPORT_DATE, app_env="test", our_code_id="104127529",
                adn="", revenue_yuan=9.0, api_revenue_yuan=8.0, impressions=50,
            ),
        ])
        db.commit()
        monkeypatch.setattr(
            ad_revenue.app_config,
            "get_ad_config",
            lambda _db: {
                "reward_code_id": "prod-reward",
                "compare_draw_code_id": "prod-draw",
                "coupon_draw_code_id": "prod-draw",
            },
        )

        business = ad_revenue.ad_revenue_report(
            db,
            date_from=REPORT_DATE,
            date_to=REPORT_DATE,
            app_env="prod",
            revenue_scope="business",
        )
        assert business["total_impressions"] == 2
        assert business["total_revenue_yuan"] == 0.5
        assert business["total_pangle_revenue_yuan"] == 4.0
        assert business["total_pangle_api_revenue_yuan"] == 3.2

        all_codes = ad_revenue.ad_revenue_report(
            db,
            date_from=REPORT_DATE,
            date_to=REPORT_DATE,
            app_env="prod",
            revenue_scope="all",
        )
        assert all_codes["total_impressions"] == 3
        assert all_codes["total_revenue_yuan"] == 0.7
        assert all_codes["total_pangle_revenue_yuan"] == 12.0
        assert all_codes["total_pangle_api_revenue_yuan"] == 10.2

        test_business = ad_revenue.ad_revenue_report(
            db,
            date_from=REPORT_DATE,
            date_to=REPORT_DATE,
            app_env="test",
            revenue_scope="business",
        )
        assert test_business["total_impressions"] == 1
        assert test_business["total_revenue_yuan"] == 0.3
        assert test_business["total_pangle_revenue_yuan"] == 9.0
        assert test_business["total_pangle_api_revenue_yuan"] == 8.0

        all_env_business = ad_revenue.ad_revenue_report(
            db,
            date_from=REPORT_DATE,
            date_to=REPORT_DATE,
            app_env=None,
            revenue_scope="business",
        )
        assert all_env_business["total_impressions"] == 3
        assert all_env_business["total_revenue_yuan"] == 0.8
        assert all_env_business["total_pangle_revenue_yuan"] == 13.0
        assert all_env_business["total_pangle_api_revenue_yuan"] == 11.2
    finally:
        db.rollback()
        db.execute(delete(AdPangleDailyRevenue).where(AdPangleDailyRevenue.report_date == REPORT_DATE))
        db.execute(delete(AdEcpmRecord).where(AdEcpmRecord.report_date == REPORT_DATE))
        db.execute(delete(User).where(User.phone == "18800009991"))
        db.commit()
        db.close()


def test_reward_video_incomplete_playback_has_zero_revenue() -> None:
    db = SessionLocal()
    phone = "18800009992"
    sessions = {
        "closed_early": "rv-zero-closed",
        "too_short": "rv-zero-short",
        "capped": "rv-keep-capped",
        "granted": "rv-keep-granted",
    }
    try:
        user = User(phone=phone, username="29999999992", register_channel="sms")
        db.add(user)
        db.flush()

        for index, (status, session_id) in enumerate(sessions.items(), start=1):
            created_at = datetime(2040, 2, 4, index, tzinfo=UTC)
            db.add(AdEcpmRecord(
                user_id=user.id,
                ad_type="reward_video",
                ad_session_id=session_id,
                app_env="prod",
                our_code_id="prod-reward",
                ecpm_raw="10000",
                report_date=PLAYBACK_DATE,
                created_at=created_at,
            ))
            db.add(AdRewardRecord(
                trans_id=f"{session_id}-trans",
                user_id=user.id,
                coin=0,
                status=status,
                reward_scene="reward_video",
                ad_session_id=session_id,
                app_env="prod",
                our_code_id="prod-reward",
                ecpm_raw="10000",
                reward_date=PLAYBACK_DATE,
                created_at=created_at,
            ))
        db.commit()

        result = ad_revenue.ad_revenue_report(
            db,
            date_from=PLAYBACK_DATE,
            date_to=PLAYBACK_DATE,
            user_id=user.id,
            ad_type="reward_video",
            app_env="prod",
            revenue_scope="all",
            granularity="hour",
        )

        revenue_by_status = {row["status"]: row["revenue_yuan"] for row in result["items"]}
        assert revenue_by_status == {
            "closed_early": 0.0,
            "too_short": 0.0,
            "capped": 0.1,
            "granted": 0.1,
        }
        assert result["total_impressions"] == 4
        assert result["total_revenue_yuan"] == 0.2
        assert len(result["daily"]) == 1
        assert result["daily"][0]["date"] == PLAYBACK_DATE
        assert result["daily"][0]["impressions"] == 4
        assert result["daily"][0]["revenue_yuan"] == 0.2
        assert sum(row["revenue_yuan"] for row in result["hourly"]) == 0.2
        assert result["type_stats"]["reward_video"] == {
            "impressions": 4,
            "revenue_yuan": 0.2,
        }
    finally:
        db.rollback()
        db.execute(delete(AdRewardRecord).where(AdRewardRecord.reward_date == PLAYBACK_DATE))
        db.execute(delete(AdEcpmRecord).where(AdEcpmRecord.report_date == PLAYBACK_DATE))
        db.execute(delete(User).where(User.phone == phone))
        db.commit()
        db.close()