video-shuoshan/backend/tests/test_1080p_billing.py

"""
1080P 分辨率支持的计费逻辑测试 — 严格对齐用户三原则：
1. 不兜底/静默降级
2. 钱的计算绝对准确（纯官方公式）
3. 不隐藏 bug（非法组合 fail loud）

运行方式：
    cd backend && source venv/Scripts/activate && python -m pytest tests/test_1080p_billing.py -v
或 Django test runner：
    python manage.py test tests.test_1080p_billing
"""
import os
import sys
import django

# Django setup
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'config.settings')
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))
django.setup()

import unittest
from utils.billing import (
    RESOLUTION_MAP,
    get_resolution,
    estimate_tokens,
    calculate_cost,
    calculate_base_cost,
)


class TestResolutionMap(unittest.TestCase):
    """验证 RESOLUTION_MAP 的 18 个组合像素值与官方文档一致。"""

    def test_1080p_pixels(self):
        # 来自 docs/API文档/创建视频生成任务API.md Seedance 2.0 & 2.0 fast 列
        self.assertEqual(get_resolution('16:9', '1080p'), (1920, 1080))
        self.assertEqual(get_resolution('9:16', '1080p'), (1080, 1920))
        self.assertEqual(get_resolution('4:3', '1080p'), (1664, 1248))
        self.assertEqual(get_resolution('1:1', '1080p'), (1440, 1440))
        self.assertEqual(get_resolution('3:4', '1080p'), (1248, 1664))
        # 21:9 特别注意：是 2206×946，不是 seedance 1.0 的 2176×928
        self.assertEqual(get_resolution('21:9', '1080p'), (2206, 946))

    def test_720p_pixels(self):
        self.assertEqual(get_resolution('16:9', '720p'), (1280, 720))
        self.assertEqual(get_resolution('21:9', '720p'), (1470, 630))

    def test_480p_pixels(self):
        self.assertEqual(get_resolution('16:9', '480p'), (864, 496))
        self.assertEqual(get_resolution('21:9', '480p'), (992, 432))

    def test_invalid_combo_raises(self):
        """原则 1：非法组合必须 fail loud，不静默降级。"""
        with self.assertRaises(KeyError):
            get_resolution('adaptive', '720p')  # adaptive 不在 map
        with self.assertRaises(KeyError):
            get_resolution('16:9', '4K')  # 不存在的 tier
        with self.assertRaises(KeyError):
            get_resolution('unknown', 'unknown')

    def test_tier_required(self):
        """tier 参数必填，不允许默认 720p 静默降级。"""
        with self.assertRaises(TypeError):
            get_resolution('16:9')  # type: ignore - 故意漏参数


class TestEstimateTokens(unittest.TestCase):
    """
    严格对齐官方公式：`(输入视频时长+输出时长) × 宽 × 高 × 帧率 / 1024`
    预估端不做最低 token 修正（那是火山计费侧逻辑）。
    """

    def test_formula_no_input_video(self):
        # 720P 16:9 (1280×720), 5s 输出, 24fps
        # 1280 × 720 × 24 × 5 / 1024 = 108000
        self.assertEqual(estimate_tokens(1280, 720, 5), 108000)

    def test_formula_with_input_video(self):
        # 720P 16:9, 5s 输出 + 5s 输入 = 10s 总时长
        # 1280 × 720 × 24 × 10 / 1024 = 216000
        self.assertEqual(estimate_tokens(1280, 720, 5, input_video_duration=5), 216000)

    def test_1080p_formula(self):
        # 1080P 16:9 (1920×1080), 5s 输出, 无输入视频
        # 1920 × 1080 × 24 × 5 / 1024 = 243000
        self.assertEqual(estimate_tokens(1920, 1080, 5), 243000)

    def test_1080p_with_input_video(self):
        # 1080P 16:9, 5s 输出 + 2s 输入 = 7s
        # 1920 × 1080 × 24 × 7 / 1024 = 340200
        self.assertEqual(estimate_tokens(1920, 1080, 5, input_video_duration=2), 340200)

    def test_no_silent_min_token_adjustment(self):
        """原则 2：预估端严格按公式，不做最低 token 修正。
        火山文档说 1080p 5s 输入含视频最低 437400 tokens，但那是火山计费侧的事，
        我们预估就老老实实按公式算 (5s+2s)×1920×1080×24/1024 = 340200，不擅自拉高。
        """
        # 1080p 5s 输出 + 2s 输入 = 7s 总时长
        # 公式值 340200，官方最低 437400
        # 我们应该返回公式值，不主动调到最低值
        result = estimate_tokens(1920, 1080, 5, input_video_duration=2)
        self.assertEqual(result, 340200, "预估端不应主动修正到火山最低 token")

    def test_float_input_duration(self):
        """输入视频时长可能是浮点数（前端 getMediaInfo 读取），要正确累加。"""
        # 720P 16:9, 5s 输出 + 3.5s 输入 = 8.5s
        # 1280 × 720 × 24 × 8.5 / 1024 = 183600
        self.assertEqual(estimate_tokens(1280, 720, 5, input_video_duration=3.5), 183600)


class TestGetTokenPrice(unittest.TestCase):
    """验证单价选择逻辑 — 4 种模型×视频组合 + 1080p 独立单价 + Fast+1080P fail loud。"""

    def setUp(self):
        # Mock QuotaConfig — 用官方文档默认值
        from types import SimpleNamespace
        from decimal import Decimal
        self.config = SimpleNamespace(
            base_token_price=Decimal('46'),
            base_token_price_video=Decimal('28'),
            base_token_price_fast=Decimal('37'),
            base_token_price_fast_video=Decimal('22'),
            base_token_price_1080p=Decimal('51'),
            base_token_price_1080p_video=Decimal('31'),
        )
        from apps.generation.views import _get_token_price
        self._get_token_price = _get_token_price

    def test_seedance_2_0_720p_no_video(self):
        """AirDrama 720P 不含视频 = 46 元/百万 tokens."""
        price = self._get_token_price(self.config, 'seedance_2.0', False, '720p')
        self.assertEqual(price, 46)

    def test_seedance_2_0_720p_with_video(self):
        """AirDrama 720P 含视频 = 28."""
        price = self._get_token_price(self.config, 'seedance_2.0', True, '720p')
        self.assertEqual(price, 28)

    def test_seedance_2_0_480p_same_as_720p(self):
        """480p 和 720p 共享同一单价（官方价格一致）。"""
        price_480 = self._get_token_price(self.config, 'seedance_2.0', False, '480p')
        price_720 = self._get_token_price(self.config, 'seedance_2.0', False, '720p')
        self.assertEqual(price_480, price_720)

    def test_seedance_2_0_1080p_no_video(self):
        """AirDrama 1080P 不含视频 = 51（独立单价，不是 720p 的 46）."""
        price = self._get_token_price(self.config, 'seedance_2.0', False, '1080p')
        self.assertEqual(price, 51)

    def test_seedance_2_0_1080p_with_video(self):
        """AirDrama 1080P 含视频 = 31（独立单价，不是 720p 的 28）."""
        price = self._get_token_price(self.config, 'seedance_2.0', True, '1080p')
        self.assertEqual(price, 31)

    def test_fast_720p_no_video(self):
        """Fast 720P 不含视频 = 37."""
        price = self._get_token_price(self.config, 'seedance_2.0_fast', False, '720p')
        self.assertEqual(price, 37)

    def test_fast_720p_with_video(self):
        """Fast 720P 含视频 = 22."""
        price = self._get_token_price(self.config, 'seedance_2.0_fast', True, '720p')
        self.assertEqual(price, 22)

    def test_fast_480p_uses_fast_price(self):
        """Fast 不分 480p/720p，都用 fast 单价。"""
        price_480 = self._get_token_price(self.config, 'seedance_2.0_fast', False, '480p')
        price_720 = self._get_token_price(self.config, 'seedance_2.0_fast', False, '720p')
        self.assertEqual(price_480, price_720)
        self.assertEqual(price_480, 37)

    def test_fast_1080p_raises_value_error(self):
        """原则 1 + 3：Fast + 1080P 必须 fail loud，不能静默按 720p 价（欺骗用户）。"""
        with self.assertRaises(ValueError) as ctx:
            self._get_token_price(self.config, 'seedance_2.0_fast', False, '1080p')
        self.assertIn('1080p', str(ctx.exception).lower())

        with self.assertRaises(ValueError):
            self._get_token_price(self.config, 'seedance_2.0_fast', True, '1080p')


class TestCalculateCost(unittest.TestCase):
    """验证扣费金额计算 = tokens × 单价 × (1 + 加价%)，精确到分."""

    def test_720p_cost_matches_official_example(self):
        """官方示例：720P 5s 16:9 = 4.97 元（无加价）."""
        # 720p 5s 公式值 108000 tokens
        tokens = estimate_tokens(1280, 720, 5)
        # 46 元/百万 × 108000 / 1000000 = 4.968 ≈ 4.97
        cost = calculate_cost(tokens, 46, 0)
        self.assertEqual(str(cost), '4.97')

    def test_1080p_no_video_cost(self):
        """1080P 5s 16:9 不含视频 = 1920×1080×24×5/1024 × 51 / 1000000 = 12.393 ≈ 12.39 元."""
        tokens = estimate_tokens(1920, 1080, 5)
        cost = calculate_cost(tokens, 51, 0)
        self.assertEqual(str(cost), '12.39')

    def test_markup_applied(self):
        """团队加价 20% 的情况。"""
        tokens = estimate_tokens(1280, 720, 5)  # 108000
        cost = calculate_cost(tokens, 46, 20)
        # 4.968 × 1.2 = 5.9616 → 5.96
        self.assertEqual(str(cost), '5.96')


if __name__ == '__main__':
    unittest.main(verbosity=2)