@cqsjjb/meter-sphere-mcp-server 1.0.2 → 2026.3.17

package/mcp-server.mjs

@@ -143,7 +143,29 @@ function loadProgress() {
   try {
     if (fs.existsSync(progressPath)) {
       const content = fs.readFileSync(progressPath, 'utf8');
-      return JSON.parse(content);
+      const progress = JSON.parse(content);
+      // 向后兼容：如果旧进度文件没有 knownTestCaseIds 字段，初始化它
+      if (!progress.knownTestCaseIds) {
+        progress.knownTestCaseIds = [];
+        // 如果已有已完成用例，将它们添加到 knownTestCaseIds 中
+        if (progress.completed && Array.isArray(progress.completed)) {
+          progress.completed.forEach(item => {
+            if (item.testCaseId && !progress.knownTestCaseIds.includes(item.testCaseId)) {
+              progress.knownTestCaseIds.push(item.testCaseId);
+            }
+          });
+        }
+        // 更新 total 为 knownTestCaseIds 的长度
+        progress.total = progress.knownTestCaseIds.length;
+        // 保存更新后的进度，确保向后兼容的修改被持久化
+        try {
+          saveProgress(progress);
+        } catch (error) {
+          // 如果保存失败，不影响加载，只记录错误
+          console.error('保存向后兼容更新失败:', error);
+        }
+      }
+      return progress;
     }
   } catch (error) {
     console.error('加载进度文件失败:', error);
@@ -152,7 +174,8 @@ function loadProgress() {
     lastUpdate: null,
     completed: [],
     total: 0,
-    completedCount: 0
+    completedCount: 0,
+    knownTestCaseIds: [] // 记录所有已知的用例ID（用于准确计算总数）
   };
 }
 
@@ -188,6 +211,18 @@ function markCompleted(testCaseId, priority, progress) {
       completedAt: new Date().toISOString()
     });
     progress.completedCount = progress.completed.length;
+    
+    // 确保 knownTestCaseIds 存在
+    if (!progress.knownTestCaseIds) {
+      progress.knownTestCaseIds = [];
+    }
+    // 将已完成的用例ID添加到已知集合中（如果不存在）
+    if (!progress.knownTestCaseIds.includes(testCaseId)) {
+      progress.knownTestCaseIds.push(testCaseId);
+    }
+    // 更新总数
+    progress.total = progress.knownTestCaseIds.length;
+    
     saveProgress(progress);
   }
 }
@@ -200,7 +235,8 @@ function resetProgress() {
     lastUpdate: null,
     completed: [],
     total: 0,
-    completedCount: 0
+    completedCount: 0,
+    knownTestCaseIds: []
   };
   
   try {
@@ -510,6 +546,11 @@ class MeterSphereMCPServer {
                   type: 'boolean',
                   description: '是否排除已完成的测试用例。默认为false（显示所有用例）。当用户要求继续测试、跳过已完成用例时，必须设置为true。设置为true后，返回的列表将只包含未完成的测试用例，方便继续测试工作。',
                 },
+                moduleIds: {
+                  type: 'array',
+                  items: { type: 'string' },
+                  description: '模块ID列表，用于按模块筛选测试用例。可选，未传入时使用环境变量 PLATFORM_MODULE_IDS 的配置。',
+                },
               },
               required: [],
             },
@@ -605,7 +646,7 @@ class MeterSphereMCPServer {
   }
 
   async handleGetTestList(args) {
-    const { excludeCompleted = false } = args || {};
+    const { excludeCompleted = false, moduleIds } = args || {};
     
     // 从环境变量读取配置
     const platformConfig = getPlatformConfig();
@@ -634,10 +675,10 @@ class MeterSphereMCPServer {
       projectId: platformConfig.project,
     };
 
-    
-    // 如果配置了 MODULE_IDS，添加到请求参数中
-    if (platformConfig.moduleIds && platformConfig.moduleIds.length > 0) {
-      requestData.moduleIds = platformConfig.moduleIds;
+    // 优先从 args 获取 moduleIds，否则从 platformConfig 获取
+    const effectiveModuleIds = moduleIds || platformConfig.moduleIds;
+    if (effectiveModuleIds && effectiveModuleIds.length > 0) {
+      requestData.moduleIds = effectiveModuleIds;
     }
 
     const response = await httpPost(targetUrl, headers, requestData);
@@ -667,8 +708,19 @@ class MeterSphereMCPServer {
         return getPriorityLevel(a.priority) - getPriorityLevel(b.priority);
       });
 
-    // 更新进度中的总数
-    progress.total = processedTestCases.length;
+    // 更新已知用例ID集合（合并当前查询到的用例ID）
+    if (!progress.knownTestCaseIds) {
+      progress.knownTestCaseIds = [];
+    }
+    const currentTestCaseIds = processedTestCases.map(tc => tc.id);
+    // 合并去重：将当前查询到的用例ID添加到已知集合中
+    currentTestCaseIds.forEach(id => {
+      if (!progress.knownTestCaseIds.includes(id)) {
+        progress.knownTestCaseIds.push(id);
+      }
+    });
+    // 基于已知用例ID集合计算总数
+    progress.total = progress.knownTestCaseIds.length;
     saveProgress(progress);
 
     // 如果excludeCompleted为true，过滤已完成的用例
@@ -828,7 +880,23 @@ class MeterSphereMCPServer {
   async handleGetTestProgress(args) {
     const progress = loadProgress();
 
-    if (progress.total === 0) {
+    // 确保 knownTestCaseIds 存在
+    if (!progress.knownTestCaseIds) {
+      progress.knownTestCaseIds = [];
+    }
+
+    const completedCount = progress.completedCount;
+    // 确保 total 至少等于 completedCount（避免出现负数）
+    // 如果 knownTestCaseIds 存在，使用它的长度；否则使用 completedCount
+    let total = progress.total;
+    if (progress.knownTestCaseIds && progress.knownTestCaseIds.length > 0) {
+      total = progress.knownTestCaseIds.length;
+    } else if (total < completedCount) {
+      // 如果 total 小于已完成数，说明数据不一致，使用已完成数作为总数
+      total = completedCount;
+    }
+
+    if (total === 0 && completedCount === 0) {
       return {
         content: [
           {
@@ -839,8 +907,6 @@ class MeterSphereMCPServer {
       };
     }
 
-    const completedCount = progress.completedCount;
-    const total = progress.total;
     const remaining = total - completedCount;
     const percentage = total > 0 ? Math.round((completedCount / total) * 100) : 0;
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@cqsjjb/meter-sphere-mcp-server",
-  "version": "1.0.2",
+  "version": "2026.3.17",
   "description": "MCP server for MeterSphere test cases platform - Get test cases, generate AI test prompts, track testing progress, and execute browser automation tests",
   "main": "mcp-server.mjs",
   "type": "module",