feat: AI 余额监控服务（平台/账号 CRUD、并发轮询、Telegram 阈值提醒）

tests/test_fetcher.py

@@ -0,0 +1,174 @@
+"""fetcher 核心逻辑测试：路径提取、模板渲染、请求/重试策略。"""
+
+import pytest
+
+from app.fetcher import extract_balance, fetch_balance, render_template
+
+
+# ---------- extract_balance ----------
+
+class TestExtractBalance:
+    def test_dot_path(self):
+        assert extract_balance({"data": {"balance": 12.5}}, "data.balance") == 12.5
+
+    def test_array_index(self):
+        assert extract_balance({"data": [{"balance": 3.0}]}, "data[0].balance") == 3.0
+
+    def test_dollar_prefix(self):
+        assert extract_balance({"a": {"b": 7}}, "$.a.b") == 7
+
+    def test_root_key(self):
+        assert extract_balance({"balance": 1}, "balance") == 1
+
+    def test_nested_array(self):
+        data = {"list": [[1, 2], [3, 4]]}
+        assert extract_balance(data, "list[1][0]") == 3
+
+    def test_string_number(self):
+        assert extract_balance({"data": {"balance": "9.99"}}, "data.balance") == 9.99
+
+    def test_missing_key_raises(self):
+        with pytest.raises(ValueError):
+            extract_balance({"data": {}}, "data.balance")
+
+    def test_index_out_of_range_raises(self):
+        with pytest.raises(ValueError):
+            extract_balance({"data": []}, "data[0].balance")
+
+    def test_non_numeric_raises(self):
+        with pytest.raises(ValueError):
+            extract_balance({"data": {"balance": "abc"}}, "data.balance")
+
+    def test_bool_rejected(self):
+        with pytest.raises(ValueError):
+            extract_balance({"data": {"balance": True}}, "data.balance")
+
+
+# ---------- render_template ----------
+
+class TestRenderTemplate:
+    def test_replace(self):
+        assert render_template("https://x?key={{apiKey}}&a=1", "sk-123") == "https://x?key=sk-123&a=1"
+
+    def test_multiple(self):
+        assert render_template("{{apiKey}}/{{apiKey}}", "k") == "k/k"
+
+
+# ---------- fetch_balance（mock requests） ----------
+
+class FakeResponse:
+    def __init__(self, status_code=200, text="", json_data=None):
+        self.status_code = status_code
+        self.text = text
+        self._json = json_data
+
+    def json(self):
+        if self._json is None:
+            raise ValueError("No JSON object could be decoded")
+        return self._json
+
+
+PLATFORM_GET = {
+    "method": "GET",
+    "url": "https://x.test/api?key={{apiKey}}",
+    "headers": {"Authorization": "Bearer {{apiKey}}"},
+    "body": "",
+    "balance_path": "data.balance",
+}
+
+
+class TestFetchBalance:
+    def test_get_success(self, monkeypatch):
+        calls = []
+
+        def fake_get(url, headers=None, timeout=10):
+            calls.append((url, headers))
+            assert url == "https://x.test/api?key=sk-1"
+            assert headers["Authorization"] == "Bearer sk-1"
+            return FakeResponse(200, json_data={"data": {"balance": 42.5}})
+
+        monkeypatch.setattr("requests.get", fake_get)
+        result = fetch_balance(PLATFORM_GET, "sk-1", retry_count=2, timeout=10)
+        assert result.ok and result.balance == 42.5
+        assert len(calls) == 1
+
+    def test_post_with_body(self, monkeypatch):
+        platform = dict(PLATFORM_GET, method="POST", body='{"api_key": "{{apiKey}}"}')
+        captured = {}
+
+        def fake_post(url, headers=None, data=None, timeout=10):
+            captured["data"] = data
+            return FakeResponse(200, json_data={"data": {"balance": 1}})
+
+        monkeypatch.setattr("requests.post", fake_post)
+        result = fetch_balance(platform, "sk-2", retry_count=2, timeout=10)
+        assert result.ok
+        assert captured["data"] == '{"api_key": "sk-2"}'
+
+    def test_401_is_auth_error_no_retry(self, monkeypatch):
+        calls = []
+
+        def fake_get(url, headers=None, timeout=10):
+            calls.append(1)
+            return FakeResponse(401, text="unauthorized")
+
+        monkeypatch.setattr("requests.get", fake_get)
+        result = fetch_balance(PLATFORM_GET, "bad", retry_count=3, timeout=10)
+        assert not result.ok and result.auth_error
+        assert len(calls) == 1
+
+    def test_5xx_retries_then_success(self, monkeypatch):
+        calls = []
+
+        def fake_get(url, headers=None, timeout=10):
+            calls.append(1)
+            return FakeResponse(500) if len(calls) < 3 else FakeResponse(200, json_data={"data": {"balance": 5}})
+
+        monkeypatch.setattr("requests.get", fake_get)
+        result = fetch_balance(PLATFORM_GET, "k", retry_count=2, timeout=10)
+        assert result.ok and result.balance == 5
+        assert len(calls) == 3
+
+    def test_5xx_all_fail(self, monkeypatch):
+        monkeypatch.setattr("requests.get", lambda *a, **k: FakeResponse(503))
+        result = fetch_balance(PLATFORM_GET, "k", retry_count=2, timeout=10)
+        assert not result.ok and not result.auth_error
+
+    def test_network_error_retries(self, monkeypatch):
+        calls = []
+
+        def fake_get(url, headers=None, timeout=10):
+            calls.append(1)
+            import requests
+            if len(calls) < 3:
+                raise requests.ConnectionError("boom")
+            return FakeResponse(200, json_data={"data": {"balance": 8}})
+
+        monkeypatch.setattr("requests.get", fake_get)
+        result = fetch_balance(PLATFORM_GET, "k", retry_count=2, timeout=10)
+        assert result.ok and result.balance == 8
+
+    def test_other_4xx_no_retry(self, monkeypatch):
+        calls = []
+
+        def fake_get(url, headers=None, timeout=10):
+            calls.append(1)
+            return FakeResponse(404, text="not found")
+
+        monkeypatch.setattr("requests.get", fake_get)
+        result = fetch_balance(PLATFORM_GET, "k", retry_count=2, timeout=10)
+        assert not result.ok and not result.auth_error
+        assert "404" in result.error
+        assert len(calls) == 1
+
+    def test_bad_json_path(self, monkeypatch):
+        monkeypatch.setattr("requests.get", lambda *a, **k: FakeResponse(200, json_data={"x": 1}))
+        result = fetch_balance(PLATFORM_GET, "k", retry_count=0, timeout=10)
+        assert not result.ok
+        assert "路径不存在" in result.error
+
+    def test_invalid_json_body(self, monkeypatch):
+        monkeypatch.setattr("requests.get", lambda *a, **k: FakeResponse(200, text="<html>"))
+        result = fetch_balance(PLATFORM_GET, "k", retry_count=0, timeout=10)
+        assert not result.ok
+        assert "JSON" in result.error