baodan/tests/ppt_poster_optimization_test.py

"""PPT/海报优化规则回归测试。"""
import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1] / "api"))


def test_currency_aliases_and_unknown_are_not_fabricated():
    from insurance.ppt.normalizer import _normalize_currency

    assert _normalize_currency("USB") == "USD"
    assert _normalize_currency("RMB") == "CNY"
    assert _normalize_currency("￥") == "CNY"
    assert _normalize_currency("XYZ") is None
    assert _normalize_currency(None) is None


def test_optional_review_fields_are_normalized_without_becoming_required():
    from insurance.ppt.normalizer import normalize_savings_plan

    plan = normalize_savings_plan({
        "product_name": "储蓄计划",
        "insured": {"age": 35, "gender": "女", "smoker": "否"},
        "policy": {
            "currency": "RMB",
            "annual_premium": 10000,
            "premium_payment_period": 5,
            "basic_plan_annual_premium": 9000,
            "basic_sum_insured": 500000,
            "first_year_amount_due": 9800,
        },
    })

    assert plan["insured"]["smoker"] == "no"
    assert plan["policy"]["currency"] == "CNY"
    assert plan["policy"]["basicPlanAnnualPremium"] == 9000
    assert plan["policy"]["basicSumInsured"] == 500000
    assert plan["policy"]["firstYearAmountDue"] == 9800


def test_benefit_and_withdrawal_issues_never_block_savings_generation():
    from insurance.ppt.validator import validate_formal_savings_plan

    issues = validate_formal_savings_plan({
        "productName": "储蓄计划",
        "insured": {"age": 35, "smoker": "no"},
        "policy": {"currency": "USD", "annualPremium": 10000, "payYears": 5},
        "benefitRows": [
            {
                "policyYear": 1,
                "guaranteedCashValue": 100,
                "totalSurrenderValue": 50,
            }
        ],
        "withdrawalRows": [{"policyYear": 1}, {"policyYear": 3}],
        "source": {"pdfHash": "hash"},
    })

    affected = [
        issue for issue in issues
        if issue.section in ("benefitRows", "withdrawalRows")
    ]
    assert affected
    assert all(issue.level == "warn" for issue in affected)


def test_brand_policy_controls_company_product_and_logo_independently():
    from insurance.ppt.masking import apply_brand_policy, build_brand_policy

    company = {
        "id": "c1",
        "displayName": "真实保司",
        "maskedDisplayName": "保X",
        "maskingEnabled": True,
        "logoEnabled": False,
        "logoUrl": "/logo.png",
    }
    product = {
        "id": "p1",
        "displayName": "真实产品",
        "maskedDisplayName": "产X",
        "maskingEnabled": False,
    }
    policy = build_brand_policy(company, [product])
    masked_company, masked_product = apply_brand_policy(company, product, policy)

    assert masked_company["displayName"] == "保X"
    assert masked_company["logoUrl"] == ""
    assert masked_product["displayName"] == "真实产品"


def test_all_six_core_poster_profiles_have_distinct_content_budgets():
    from insurance.poster.field_profiles import get_field_profile

    profiles = {
        (plan_type, output_mode): get_field_profile(plan_type, output_mode)
        for plan_type in ("savings", "ci", "iul")
        for output_mode in ("single", "long")
    }
    assert len(profiles) == 6
    for plan_type in ("savings", "ci", "iul"):
        assert profiles[(plan_type, "single")]["maxFeatureCount"] == 3
        assert profiles[(plan_type, "long")]["maxFeatureCount"] == 6


def test_single_poster_never_contains_full_benefit_table():
    from insurance.poster.content_builder import build_poster_content

    rows = [
        {"policy_year": year, "total_surrender": year * 1000}
        for year in range(1, 31)
    ]
    content = build_poster_content(
        {"benefit_illustration": rows},
        {"features": [{"title": str(index)} for index in range(8)]},
        output_mode="single",
        plan_type="savings",
    )
    assert len(content["benefit_table"]) <= 1
    assert len(content["features"]) <= 3


def test_custom_scenario_is_compatible_only_with_its_calculation_mode(monkeypatch):
    from insurance.ppt import scenarios

    monkeypatch.setattr(scenarios, "get_scenario_config", lambda _code: {
        "generationMode": "single",
        "baseScenario": None,
    })
    assert scenarios.template_scenario_compatible("retirement", "generic_single")
    assert not scenarios.template_scenario_compatible("retirement", "generic_compare")