feat(cross_end): add driver-to-settlement E2E test with SQL retry

- Add DriverFreightFlowPage (Android): 28-step flow covering accept/load/unload - Add FreightAccountingPage (Web): accounting/audit/finance-payment/shipper-payment - Add db_helper: update_risk_control_status via pymysql - Fix ensure_logged_in: coordinate tap on 'mine' tab + login check - Add test_driver_to_settlement.py: full cross-end E2E with SQL retry Co-authored-by: xst <xst@ntocc.com>
2026-07-22 17:18:52 +08:00
parent 6f5668b7ec
commit f75a6f5ec9
6 changed files with 763 additions and 22 deletions
@@ -0,0 +1,72 @@
+"""MySQL 数据库操作工具 — 测试中的数据库校验/数据修改。
+
+Usage:
+    from tests.utils.db_helper import update_risk_control_status
+
+    affected = update_risk_control_status("RHJC040026071600021457570057", status=1)
+"""
+
+import pymysql
+
+DEFAULT_DB_CONFIG = {
+    "host": "36.137.133.24",
+    "port": 3306,
+    "user": "ntocc",
+    "password": "Rgxi3!kD9jjXpgzx",
+    "database": "ntocc",
+    "charset": "utf8mb4",
+}
+
+
+def get_connection(config: dict | None = None):
+    """获取数据库连接。"""
+    cfg = config or DEFAULT_DB_CONFIG
+    return pymysql.connect(**cfg)
+
+
+def execute_sql(sql: str, params: tuple | None = None, config: dict | None = None) -> int:
+    """执行 SQL（INSERT/UPDATE/DELETE），返回影响行数。"""
+    conn = get_connection(config)
+    try:
+        with conn.cursor() as cursor:
+            affected = cursor.execute(sql, params)
+            conn.commit()
+            return affected
+    finally:
+        conn.close()
+
+
+def query_sql(sql: str, params: tuple | None = None, config: dict | None = None) -> list[dict]:
+    """执行查询 SQL，返回结果列表。"""
+    conn = get_connection(config)
+    try:
+        with conn.cursor(pymysql.cursors.DictCursor) as cursor:
+            cursor.execute(sql, params)
+            return cursor.fetchall()
+    finally:
+        conn.close()
+
+
+def update_risk_control_status(order_code: str, status: int = 1, config: dict | None = None) -> int:
+    """更新运单风控状态。
+
+    执行 SQL:
+        UPDATE tms_order_detail_ext SET risk_control_status = <status>
+        WHERE order_detail_code IN (
+            SELECT `code` FROM `tms_order_detail` WHERE order_code = '<order_code>'
+        )
+
+    Args:
+        order_code: 运输单号（对应 tms_order_detail.order_code）
+        status: 风控状态值，默认 1
+        config: 数据库连接配置
+    Returns:
+        影响行数
+    """
+    sql = (
+        "UPDATE tms_order_detail_ext SET risk_control_status = %s "
+        "WHERE order_detail_code IN ("
+        "  SELECT `code` FROM `tms_order_detail` WHERE order_code = %s"
+        ")"
+    )
+    return execute_sql(sql, (status, order_code), config)