fix: 绿联接口支持可配置SSL证书校验

tests/test_ugreen_api.py

@@ -0,0 +1,113 @@
+import unittest
+from types import SimpleNamespace
+from unittest.mock import patch
+
+from app.modules.ugreen.api import Api
+
+
+class _FakeResponse:
+    def __init__(self, payload: dict, headers: dict | None = None):
+        self._payload = payload
+        self.headers = headers or {}
+
+    def json(self):
+        return self._payload
+
+
+class _FakeSession:
+    def __init__(self, get_responses=None, post_responses=None):
+        self._get_responses = list(get_responses or [])
+        self._post_responses = list(post_responses or [])
+        self.calls: list[tuple[str, dict]] = []
+        self.cookies = SimpleNamespace(
+            get_dict=lambda: {},
+            update=lambda *_args, **_kwargs: None,
+        )
+
+    def get(self, *args, **kwargs):
+        if args:
+            kwargs = {"url": args[0], **kwargs}
+        self.calls.append(("GET", kwargs))
+        return self._get_responses.pop(0) if self._get_responses else _FakeResponse({})
+
+    def post(self, *args, **kwargs):
+        if args:
+            kwargs = {"url": args[0], **kwargs}
+        self.calls.append(("POST", kwargs))
+        return self._post_responses.pop(0) if self._post_responses else _FakeResponse({})
+
+    @staticmethod
+    def close():
+        return None
+
+
+class _FakeCrypto:
+    def __init__(self, *args, **kwargs):
+        pass
+
+    @staticmethod
+    def rsa_encrypt_long(raw: str) -> str:
+        return f"enc:{raw}"
+
+    @staticmethod
+    def build_encrypted_request(url: str, method: str = "GET", params=None, **kwargs):
+        return SimpleNamespace(url=url, headers={}, params=params or {}, json=None, aes_key="k")
+
+    @staticmethod
+    def decrypt_response(payload, aes_key):
+        return payload
+
+
+class UgreenApiVerifySslTest(unittest.TestCase):
+    def test_request_json_default_verify_ssl_true(self):
+        api = Api(host="https://example.com")
+        fake_session = _FakeSession(
+            get_responses=[_FakeResponse({"code": 200})],
+            post_responses=[_FakeResponse({"code": 200})],
+        )
+        api._session = fake_session
+
+        api._request_json(url="https://example.com/a", method="GET")
+        api._request_json(url="https://example.com/b", method="POST", json_data={"x": 1})
+
+        self.assertEqual(fake_session.calls[0][1].get("verify"), True)
+        self.assertEqual(fake_session.calls[1][1].get("verify"), True)
+
+    def test_login_logout_follow_verify_ssl_flag(self):
+        api = Api(host="https://example.com", verify_ssl=False)
+        fake_session = _FakeSession(
+            get_responses=[_FakeResponse({})],
+            post_responses=[
+                _FakeResponse({"code": 200, "msg": "ok", "data": {}}, headers={"x-rsa-token": "BEGIN TEST"}),
+                _FakeResponse(
+                    {
+                        "code": 200,
+                        "msg": "ok",
+                        "data": {
+                            "token": "token-value",
+                            "public_key": "BEGIN LOGIN KEY",
+                            "static_token": "static-token",
+                            "is_ugk": False,
+                        },
+                    }
+                ),
+            ],
+        )
+        api._session = fake_session
+
+        with patch("app.modules.ugreen.api.UgreenCrypto", _FakeCrypto):
+            token = api.login("tester", "pwd")
+            self.assertEqual(token, "token-value")
+            api.logout()
+
+        self.assertEqual(len(fake_session.calls), 3)
+        self.assertEqual(fake_session.calls[0][0], "POST")
+        self.assertEqual(fake_session.calls[1][0], "POST")
+        self.assertEqual(fake_session.calls[2][0], "GET")
+        self.assertEqual(fake_session.calls[0][1].get("verify"), False)
+        self.assertEqual(fake_session.calls[1][1].get("verify"), False)
+        self.assertEqual(fake_session.calls[2][1].get("verify"), False)
+
+
+if __name__ == "__main__":
+    unittest.main()