47 lines
1.4 KiB
Python
47 lines
1.4 KiB
Python
|
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"
|