mirror of
https://gitcode.com/JianFeeeee/TrulyMEM-TrueHumanMEM.git
synced 2026-10-30 13:59:11 +00:00
test: 添加UI层历史加载测试,更新方法名称
This commit is contained in:
@ -180,18 +180,40 @@ class TestUIWithBackendClient:
|
|||||||
app = GraphMemoryApp(backend_server=server)
|
app = GraphMemoryApp(backend_server=server)
|
||||||
client = app._backend_client
|
client = app._backend_client
|
||||||
|
|
||||||
# 测试 get_status
|
|
||||||
status = client.get_status()
|
status = client.get_status()
|
||||||
assert status.get("success") is True
|
assert status.get("success") is True
|
||||||
|
|
||||||
# 测试 update_config
|
result = client.update_settings(
|
||||||
result = client.update_config(api_key="sk-test", base_url="https://api.deepseek.com")
|
api_config={"api_key": "sk-test", "base_url": "https://api.deepseek.com", "model": "deepseek-chat"},
|
||||||
|
tool_limits={"persona_query_max": 1}
|
||||||
|
)
|
||||||
assert result.get("success") is True
|
assert result.get("success") is True
|
||||||
|
|
||||||
server.shutdown()
|
server.shutdown()
|
||||||
finally:
|
finally:
|
||||||
if os.path.exists(db_path):
|
if os.path.exists(db_path):
|
||||||
os.unlink(db_path)
|
os.unlink(db_path)
|
||||||
|
|
||||||
|
def test_ui_get_history(self):
|
||||||
|
from ui import GraphMemoryApp
|
||||||
|
from core import BackendServer
|
||||||
|
|
||||||
|
with tempfile.NamedTemporaryFile(suffix=".db", delete=False) as f:
|
||||||
|
db_path = f.name
|
||||||
|
try:
|
||||||
|
server = BackendServer(db_path=db_path)
|
||||||
|
server.start(api_key="")
|
||||||
|
|
||||||
|
app = GraphMemoryApp(backend_server=server)
|
||||||
|
client = app._backend_client
|
||||||
|
|
||||||
|
history = client.get_history()
|
||||||
|
assert isinstance(history, list)
|
||||||
|
|
||||||
|
server.shutdown()
|
||||||
|
finally:
|
||||||
|
if os.path.exists(db_path):
|
||||||
|
os.unlink(db_path)
|
||||||
|
|
||||||
def test_ui_only_uses_backend_client(self):
|
def test_ui_only_uses_backend_client(self):
|
||||||
from ui import GraphMemoryApp
|
from ui import GraphMemoryApp
|
||||||
|
|||||||
Reference in New Issue
Block a user