提交

backend/tests/test_sync_utils.py

@@ -0,0 +1,85 @@
+"""sync_utils 纯函数：取值清洗、日期格式化、限频重试语义。"""
+from __future__ import annotations
+
+import math
+from datetime import timedelta
+
+import pytest
+
+from app.data import sync_utils
+
+
+def test_s_clean():
+    assert sync_utils.s_clean("  平安银行 ") == "平安银行"
+    assert sync_utils.s_clean("") is None
+    assert sync_utils.s_clean("   ") is None
+    assert sync_utils.s_clean(None) is None
+    assert sync_utils.s_clean(float("nan")) is None  # pandas NaN
+
+
+def test_f_clean():
+    assert sync_utils.f_clean("3.14") == 3.14
+    assert sync_utils.f_clean(2) == 2.0
+    assert sync_utils.f_clean(float("nan")) is None
+    assert sync_utils.f_clean(None) is None
+    assert sync_utils.f_clean("abc") is None
+    assert sync_utils.f_clean(math.inf) == math.inf  # inf 非 NaN，原样保留
+
+
+def test_d8_iso():
+    assert sync_utils.d8_iso("20240102") == "2024-01-02"
+    assert sync_utils.d8_iso(20240102) == "2024-01-02"
+    assert sync_utils.d8_iso(None) is None
+    assert sync_utils.d8_iso("") is None
+
+
+def test_fresh():
+    now = sync_utils.utcnow()
+    assert sync_utils.fresh(now, days=7) is True
+    assert sync_utils.fresh(now - timedelta(days=8), days=7) is False
+    assert sync_utils.fresh(None, days=7) is False
+
+
+def test_call_retry_passes_through_args():
+    calls = []
+
+    def fn(a, b=0):
+        calls.append((a, b))
+        return a + b
+
+    assert sync_utils.call_retry(fn, 1, b=2) == 3
+    assert calls == [(1, 2)]
+
+
+def test_call_retry_rate_limit_retries_once(monkeypatch):
+    """「每分钟」级频率超限等 62s 重试一次；重试成功则返回结果。"""
+    monkeypatch.setattr(sync_utils.time, "sleep", lambda s: None)
+    calls = []
+
+    def fn():
+        calls.append(1)
+        if len(calls) == 1:
+            raise RuntimeError("抱歉，您每分钟最多访问该接口5次")
+        return "ok"
+
+    assert sync_utils.call_retry(fn) == "ok"
+    assert len(calls) == 2
+
+
+def test_call_retry_hourly_limit_raises(monkeypatch):
+    """小时级限频不重试，直接抛出。"""
+    monkeypatch.setattr(sync_utils.time, "sleep", lambda s: None)
+
+    def fn():
+        raise RuntimeError("您每小时最多访问该接口10次")
+
+    with pytest.raises(RuntimeError):
+        sync_utils.call_retry(fn)
+
+
+def test_call_retry_other_errors_raise():
+    def fn():
+        raise ValueError("数据源故障")
+
+    with pytest.raises(ValueError):
+        sync_utils.call_retry(fn)