feat: add FastAPI manager API compatibility baseline

2026-07-22 15:13:55 +08:00 · 2026-07-20 17:00:13 +08:00
parent 7c58fa37b2
commit 804ddb51f2
140 changed files with 47169 additions and 1 deletions
@@ -0,0 +1,136 @@
+from __future__ import annotations
+
+import json
+import subprocess
+from datetime import datetime
+from pathlib import Path
+from typing import Any
+from zoneinfo import ZoneInfo
+
+import pytest
+
+from app.core.redis import JavaRedisCodec
+
+TARGET_ROOT = Path(__file__).resolve().parents[1]
+REPOSITORY_ROOT = TARGET_ROOT.parents[1]
+JAVA_PROJECT = REPOSITORY_ROOT / "main" / "manager-api"
+MAVEN = REPOSITORY_ROOT / ".runtime" / "maven" / "bin" / "mvn"
+MAVEN_REPOSITORY = REPOSITORY_ROOT / ".runtime" / "m2"
+JAVA = REPOSITORY_ROOT / ".runtime" / "jdk" / "bin"
+
+
+@pytest.fixture(scope="module")
+def java_redis_vectors(tmp_path_factory: pytest.TempPathFactory) -> dict[str, Any]:
+    work = tmp_path_factory.mktemp("redis-java")
+    classpath_file = work / "classpath.txt"
+    subprocess.run(  # noqa: S603
+        [
+            str(MAVEN),
+            "-o",
+            "-q",
+            f"-Dmaven.repo.local={MAVEN_REPOSITORY}",
+            "-DskipTests",
+            "compile",
+            "dependency:build-classpath",
+            f"-Dmdep.outputFile={classpath_file}",
+        ],
+        cwd=JAVA_PROJECT,
+        check=True,
+        capture_output=True,
+        text=True,
+    )
+    classpath = f"{JAVA_PROJECT / 'target' / 'classes'}:{classpath_file.read_text(encoding='utf-8').strip()}"
+    subprocess.run(  # noqa: S603
+        [
+            str(JAVA / "javac"),
+            "-cp",
+            classpath,
+            "-d",
+            str(work),
+            str(TARGET_ROOT / "tests" / "java" / "RedisCodecVectors.java"),
+        ],
+        check=True,
+        capture_output=True,
+        text=True,
+    )
+    result = subprocess.run(  # noqa: S603
+        [str(JAVA / "java"), "-cp", f"{work}:{classpath}", "RedisCodecVectors"],
+        check=True,
+        capture_output=True,
+        text=True,
+    )
+    vectors: dict[str, Any] = {}
+    for line in result.stdout.splitlines():
+        name, wire, _decoded_class = line.split("\t", 2)
+        vectors[name] = json.loads(wire)
+    return vectors
+
+
+def _python_wire(value: Any, **kwargs: Any) -> Any:
+    return json.loads(JavaRedisCodec.encode(value, **kwargs))
+
+
+def test_spring_redis_serializer_core_wire_format_matches_python(java_redis_vectors: dict[str, Any]) -> None:
+    assert java_redis_vectors["string"] == _python_wire("hello")
+    assert java_redis_vectors["integer"] == _python_wire(7)
+    assert java_redis_vectors["long"] == _python_wire(2_147_483_648)
+    assert java_redis_vectors["boolean"] == _python_wire(True)
+    nested = {"enabled": True}
+    assert java_redis_vectors["map"] == _python_wire(
+        {
+            "text": "hello",
+            "number": 2_147_483_648,
+            "list": ["a", "b"],
+            "nested": nested,
+        }
+    )
+    assert java_redis_vectors["list"] == _python_wire(["a", 2_147_483_648, nested])
+    epoch_in_shanghai = datetime.fromtimestamp(0, ZoneInfo("Asia/Shanghai")).replace(tzinfo=None)
+    assert java_redis_vectors["date"] == _python_wire(epoch_in_shanghai)
+
+
+def test_spring_redis_serializer_domain_pojo_wire_format_matches_python(java_redis_vectors: dict[str, Any]) -> None:
+    assert java_redis_vectors["pojo"] == _python_wire(
+        {
+            "id": "voice-1",
+            "languages": None,
+            "name": "sample",
+            "remark": None,
+            "reference_audio": None,
+            "reference_text": None,
+            "sort": 2,
+            "tts_model_id": None,
+            "tts_voice": None,
+            "voice_demo": None,
+        },
+        java_type="xiaozhi.modules.timbre.vo.TimbreDetailsVO",
+        field_java_types={"sort": "java.lang.Long"},
+    )
+    assert java_redis_vectors["model-pojo"] == _python_wire(
+        {
+            "id": "model-1",
+            "model_type": "LLM",
+            "model_code": None,
+            "model_name": None,
+            "is_default": None,
+            "is_enabled": None,
+            "config_json": {"type": "openai", "api_key": "secret"},
+            "doc_link": None,
+            "remark": None,
+            "sort": None,
+            "updater": None,
+            "update_date": None,
+            "creator": None,
+            "create_date": None,
+        },
+        java_type="xiaozhi.modules.model.entity.ModelConfigEntity",
+        field_java_types={
+            "configJson": "cn.hutool.json.JSONObject",
+            "creator": "java.lang.Long",
+            "updater": "java.lang.Long",
+        },
+    )
+    assert java_redis_vectors["dict-list"] == _python_wire(
+        [{"name": "China", "key": "+86"}],
+        item_java_type="xiaozhi.modules.sys.vo.SysDictDataItem",
+    )