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

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 {
@@ -667,8 +703,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 +875,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 +902,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": "1.0.3",
   "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",