from datetime import datetime

from app.service.risk_natural_language import SHANGHAI, parse_risk_alert_filters


def test_parse_customer_product_rule_and_recent_days() -> None:
    now = datetime(2026, 9, 10, 10, 0, tzinfo=SHANGHAI)

    filters = parse_risk_alert_filters(
        "查询客户编号 CUST-001、产品代码 P-R5、规则 RW-007、"
        "最近 3 天的中风险预警",
        now=now,
    )

    assert filters == {
        "customer_no": "CUST-001",
        "product_code": "P-R5",
        "risk_level": "中",
        "rule_code": "RW-007",
        "start_time": "2026-09-07T02:00:00",
        "end_time": "2026-09-10T02:00:00",
    }


def test_parse_explicit_date_range_as_utc() -> None:
    filters = parse_risk_alert_filters(
        "查看 2026-09-01 至 2026-09-03 的高风险预警",
        now=datetime(2026, 9, 10, 10, 0, tzinfo=SHANGHAI),
    )

    assert filters["start_time"] == "2026-08-31T16:00:00"
    assert filters["end_time"] == "2026-09-03T15:59:59.999999"


def test_parse_product_name_and_this_month() -> None:
    now = datetime(2026, 9, 10, 10, 0, tzinfo=SHANGHAI)

    filters = parse_risk_alert_filters(
        "本月产品名称 成长精选 的低风险预警",
        now=now,
    )

    assert filters["product_name"] == "成长精选"
    assert filters["risk_level"] == "低"
    assert filters["start_time"] == "2026-08-31T16:00:00"
    assert filters["end_time"] == "2026-09-10T02:00:00"