"""广告收益报表的环境与业务代码位口径。""" 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()