feat: 迁移奶龙风控业务模块与演示文档
This commit is contained in:
@@ -0,0 +1,46 @@
|
||||
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"
|
||||
Reference in New Issue
Block a user