update tools

Author: LuckyYC

backend/tests/conftest.py

@@ -5,13 +5,20 @@
 
 import asyncio
 import os
+import sys
+from pathlib import Path
 from unittest.mock import AsyncMock, MagicMock, Mock, patch
 
 import pytest
 
 # Set testing environment variables before any imports
 os.environ["TESTING"] = "1"
 
+# Ensure backend package is importable when tests run from project root
+BACKEND_ROOT = Path(__file__).resolve().parents[1]
+if str(BACKEND_ROOT) not in sys.path:
+    sys.path.insert(0, str(BACKEND_ROOT))
+
 
 # Setup async test event loop
 @pytest.fixture(scope="session")
@@ -27,12 +34,24 @@ def mock_db_session():
     """Mock database session"""
     mock_session = AsyncMock()
     mock_session.execute = AsyncMock()
+    mock_session.execute.return_value = Mock()
+    mock_session.execute.return_value.scalars.return_value.all.return_value = []
+    mock_session.execute.return_value.scalar_one_or_none.return_value = None
     mock_session.scalar = AsyncMock()
     mock_session.add = Mock()
     mock_session.flush = AsyncMock()
     mock_session.commit = AsyncMock()
     mock_session.rollback = AsyncMock()
     mock_session.close = AsyncMock()
+
+    class _AsyncTransaction:
+        async def __aenter__(self):
+            return None
+
+        async def __aexit__(self, exc_type, exc, tb):
+            return False
+
+    mock_session.begin = Mock(return_value=_AsyncTransaction())
     return mock_session
 
 

backend/tests/test_system_config.py

@@ -2,12 +2,14 @@
 Test cases for system configuration batch operations.
 """
 
+from contextlib import asynccontextmanager
 from unittest.mock import AsyncMock, Mock, patch
 
 import pytest
 from fastapi.testclient import TestClient
 
 from app import app
+from utils.tools import mask_config_value
 
 
 @pytest.mark.asyncio
@@ -35,8 +37,13 @@ async def test_batch_upsert_system_configs(mock_db_session, mock_request):
         },
     ]
 
-    # Mock the database operations
-    with patch("service.system_service.get_db_session", return_value=mock_db_session):
+    @asynccontextmanager
+    async def mock_session_factory():
+        yield mock_db_session
+
+    # Mock the database session factory used by middleware
+    with patch("middleware.db_middleware.async_session_factory") as mock_factory:
+        mock_factory.return_value = mock_session_factory()
         with TestClient(app) as client:
             response = client.post("/api/system/batch", json={"configs": test_configs})
             assert response.status_code == 200
@@ -53,7 +60,10 @@ async def test_batch_upsert_system_configs(mock_db_session, mock_request):
                 ]
                 # Sensitive values should be masked
                 if config["config_key"] == "test_api_key":
-                    assert config["config_value"] == "••••••••••••••••"
+                    expected_value = mask_config_value(
+                        config["config_key"], "sk-test-key"
+                    )
+                    assert config["config_value"] == expected_value
 
 
 @pytest.mark.asyncio
@@ -76,8 +86,12 @@ async def test_batch_upsert_mixed_operations(mock_db_session, mock_request):
         },
     ]
 
-    # Mock the database operations
-    with patch("service.system_service.get_db_session", return_value=mock_db_session):
+    @asynccontextmanager
+    async def mock_session_factory():
+        yield mock_db_session
+
+    with patch("middleware.db_middleware.async_session_factory") as mock_factory:
+        mock_factory.return_value = mock_session_factory()
         with TestClient(app) as client:
             response = client.post("/api/system/batch", json={"configs": mixed_configs})
             assert response.status_code == 200

data/prompts/ShareGPT_V3_partial.json data/ShareGPT_V3_partial.jsondata/prompts/ShareGPT_V3_partial.json renamed to data/ShareGPT_V3_partial.json

@@ -401,7 +401,7 @@
       "selectDatasetFile": "Select Dataset File",
       "jsonlFormatDescription": "Required format: .jsonl file with each line containing {\"id\": \"...\", \"prompt\": \"...\"}",
       "datasetFileFormatDescription": "Supports JSON (ShareGPT format) and JSONL formats. ShareGPT: [{\"id\": \"...\", \"conversations\": [...]}], JSONL: one JSON object per line {\"id\": \"...\", \"prompt\": \"...\"}",
-      "datasetImageMountWarning": "⚠️ If dataset contains images, ensure image files are mounted to the container, see deployment docs",
+      "datasetImageMountWarning": "⚠️ If the dataset contains images, ensure image files are mounted to the container. See DATASET_GUIDE for details.",
       "jsonlData": "JSONL Data",
       "jsonlDataTooltip": "Each line must be a valid JSON object with \"id\" and \"prompt\" fields.",
       "testDuration": "Test Duration (seconds)",

data/prompts/text_self-built.jsonl data/text_self-built.jsonldata/prompts/text_self-built.jsonl renamed to data/text_self-built.jsonl

@@ -401,7 +401,7 @@
       "selectDatasetFile": "选择数据集文件",
       "jsonlFormatDescription": "必需格式：.jsonl文件，每行包含{\"id\": \"...\", \"prompt\": \"...\"}",
       "datasetFileFormatDescription": "支持 JSON (ShareGPT格式) 和 JSONL 格式。ShareGPT格式: [{\"id\": \"...\", \"conversations\": [...]}]，JSONL格式: 每行一个JSON对象 {\"id\": \"...\", \"prompt\": \"...\"}",
-      "datasetImageMountWarning": "⚠️ 若数据集包含图片，请确保图片文件已挂载到容器，详见部署文档",
+      "datasetImageMountWarning": "⚠️ 若数据集包含图片，请确保图片文件已挂载到容器对应目录下，详见 DATASET_GUIDE。",
       "jsonlData": "JSONL数据",
       "jsonlDataTooltip": "每行必须是具有id和prompt字段的有效JSON对象。",
       "testDuration": "测试持续时间（秒）",

data/prompts/vision_self-built.jsonl data/vision_self-built.jsonldata/prompts/vision_self-built.jsonl renamed to data/vision_self-built.jsonl

@@ -20,7 +20,6 @@
     "LOG_DIR",
     "LOG_TASK_DIR",
     "DATA_DIR",
-    "PROMPTS_DIR",
     "IMAGES_DIR",
     "UPLOAD_FOLDER",
     "HTTP_OK",

frontend/public/locales/en/translation.json

@@ -25,8 +25,7 @@
 
 # === LOG PATHS ===
 LOG_TASK_DIR = os.path.join(LOG_DIR, "task")
-# === DATA PATHS ===
-PROMPTS_DIR = os.path.join(DATA_DIR, "prompts")
+# === IMAGES PATHS ===
 IMAGES_DIR = os.path.join(DATA_DIR, "pic")
 
 # === HTTP CONSTANTS ===
@@ -57,7 +56,6 @@
     "LOG_DIR",
     "LOG_TASK_DIR",
     "DATA_DIR",
-    "PROMPTS_DIR",
     "IMAGES_DIR",
     "UPLOAD_FOLDER",
     # http

frontend/public/locales/zh/translation.json

@@ -0,0 +1,15 @@
+"""
+Shared pytest configuration for st_engine tests.
+"""
+
+import os
+import sys
+from pathlib import Path
+
+# Ensure tests run in test mode
+os.environ.setdefault("TESTING", "1")
+
+# Make the st_engine package importable when tests run from repository root
+ST_ENGINE_ROOT = Path(__file__).resolve().parents[1]
+if str(ST_ENGINE_ROOT) not in sys.path:
+    sys.path.insert(0, str(ST_ENGINE_ROOT))