qat-cli 0.3.2 → 0.3.3

package/dist/index.cjs

@@ -1047,12 +1047,36 @@ import_handlebars.default.registerHelper("propTestValue", (prop) => {
   };
   return map[prop.type] || "'test-value'";
 });
+function resolveImportPath(testType, targetPath) {
+  if (!targetPath) return targetPath;
+  const testDirMap = {
+    unit: "tests/unit",
+    component: "tests/component",
+    e2e: "tests/e2e",
+    api: "tests/api",
+    visual: "tests/visual",
+    performance: "tests/e2e"
+  };
+  const testDir = testDirMap[testType];
+  if (!testDir) return targetPath;
+  if (targetPath.startsWith("../") || targetPath.startsWith("./../")) {
+    return targetPath;
+  }
+  const depth = testDir.split("/").length;
+  const prefix = "../".repeat(depth);
+  let cleanPath = targetPath.replace(/^\.\//, "");
+  if (!cleanPath.endsWith(".vue")) {
+    cleanPath = cleanPath.replace(/\.(ts|js|tsx|jsx)$/, "");
+  }
+  return `${prefix}${cleanPath}`;
+}
 function registerTemplate(type, templateContent) {
   customTemplates.set(type, templateContent);
 }
 function renderTemplate(type, context) {
   const templateContent = loadTemplate(type);
   const template = import_handlebars.default.compile(templateContent);
+  const resolvedTarget = resolveImportPath(type, context.target);
   const fullContext = {
     vueVersion: 3,
     typescript: true,
@@ -1073,6 +1097,8 @@ function renderTemplate(type, context) {
     requiredProps: [],
     optionalProps: [],
     ...context,
+    // 使用计算后的正确路径
+    target: resolvedTarget,
     framework: context.framework || "vue",
     camelName: context.camelName || toCamelCase(context.name),
     pascalName: context.pascalName || toPascalCase(context.name)
@@ -1774,14 +1800,14 @@ function buildVitestArgs(options) {
   if (options.files && options.files.length > 0) {
     args.push(...options.files);
   } else {
-    const includeMap = {
-      unit: ["tests/unit"],
-      component: ["tests/component"],
-      api: ["tests/api"]
+    const pathMap = {
+      unit: "tests/unit/**/*.test.ts",
+      component: "tests/component/**/*.test.ts",
+      api: "tests/api/**/*.test.ts"
     };
-    const includes = includeMap[options.type];
-    if (includes) {
-      args.push("--include", includes.map((d) => `${d}/**/*.test.ts`).join(","));
+    const testPattern = pathMap[options.type];
+    if (testPattern) {
+      args.push(testPattern);
     }
   }
   if (options.coverage) {
@@ -2764,8 +2790,13 @@ ${error.actual ? `\u5B9E\u9645\u503C: ${error.actual}` : ""}`;
 6. \u5982\u679C\u6709 props/emits \u4FE1\u606F\uFF0C\u52A1\u5FC5\u9488\u5BF9\u6BCF\u4E2A prop \u548C emit \u751F\u6210\u6D4B\u8BD5`;
   }
   buildGenerateTestUserPrompt(req) {
+    const importPath = this.computeTestImportPath(req.type, req.target);
     let prompt = `\u8BF7\u4E3A\u4EE5\u4E0B\u6587\u4EF6\u751F\u6210${req.type}\u6D4B\u8BD5\u4EE3\u7801:
 \u76EE\u6807\u6587\u4EF6: ${req.target}
+\u6D4B\u8BD5\u6587\u4EF6\u5C06\u653E\u5728: ${this.getTestOutputDir(req.type)}/
+\u6B63\u786E\u7684 import \u8DEF\u5F84: ${importPath}
+
+\u91CD\u8981\uFF1Aimport \u8BED\u53E5\u4E2D\u5FC5\u987B\u4F7F\u7528\u4E0A\u8FF0\u6B63\u786E\u7684\u76F8\u5BF9\u8DEF\u5F84 ${importPath}\uFF0C\u4E0D\u8981\u4F7F\u7528 ${req.target} \u6216\u5176\u4ED6\u8DEF\u5F84\uFF01
 `;
     if (req.analysis) {
       prompt += "\n\u6E90\u7801\u5206\u6790\u7ED3\u679C:\n";
@@ -2815,6 +2846,46 @@ ${req.context}
     }
     return prompt;
   }
+  /**
+   * 根据测试类型和源文件路径，计算从测试文件到源文件的正确相对导入路径
+   */
+  computeTestImportPath(testType, targetPath) {
+    if (!targetPath) return targetPath;
+    const testDirMap = {
+      unit: "tests/unit",
+      component: "tests/component",
+      e2e: "tests/e2e",
+      api: "tests/api",
+      visual: "tests/visual",
+      performance: "tests/e2e"
+    };
+    const testDir = testDirMap[testType];
+    if (!testDir) return targetPath;
+    if (targetPath.startsWith("../") || targetPath.startsWith("./../")) {
+      return targetPath;
+    }
+    const depth = testDir.split("/").length;
+    const prefix = "../".repeat(depth);
+    let cleanPath = targetPath.replace(/^\.\//, "");
+    if (!cleanPath.endsWith(".vue")) {
+      cleanPath = cleanPath.replace(/\.(ts|js|tsx|jsx)$/, "");
+    }
+    return `${prefix}${cleanPath}`;
+  }
+  /**
+   * 获取测试输出目录
+   */
+  getTestOutputDir(testType) {
+    const dirMap = {
+      unit: "tests/unit",
+      component: "tests/component",
+      e2e: "tests/e2e",
+      api: "tests/api",
+      visual: "tests/visual",
+      performance: "tests/e2e"
+    };
+    return dirMap[testType] || "tests/unit";
+  }
   parseGenerateTestResponse(content) {
     const codeBlockMatch = content.match(/```(?:typescript|ts|javascript|js)?\s*\n([\s\S]*?)```/);
     const code = codeBlockMatch ? codeBlockMatch[1].trim() : content.replace(/^(?:```[\s\S]*?\n)?/, "").replace(/\n?```$/, "").trim();
@@ -2843,10 +2914,12 @@ ISSUES: \u95EE\u9898\u5217\u8868\uFF08\u6BCF\u884C\u4E00\u4E2A\uFF0C\u683C\u5F0F
 SUGGESTIONS: \u6539\u8FDB\u5EFA\u8BAE\u5217\u8868\uFF08\u6BCF\u884C\u4E00\u4E2A\uFF0C\u683C\u5F0F "- \u5EFA\u8BAE\u63CF\u8FF0"\uFF09`;
   }
   buildReviewTestUserPrompt(req) {
+    const importPath = this.computeTestImportPath(req.testType, req.target);
     let prompt = `\u8BF7\u5BA1\u67E5\u4EE5\u4E0B\u6D4B\u8BD5\u7528\u4F8B\u662F\u5426\u4E0E\u6E90\u7801\u8D34\u5207\u4E14\u51C6\u786E\u3002
 
 \u88AB\u6D4B\u6587\u4EF6: ${req.target}
 \u6D4B\u8BD5\u7C7B\u578B: ${req.testType}
+\u6B63\u786E\u7684 import \u8DEF\u5F84\u5E94\u4E3A: ${importPath}\uFF08\u6D4B\u8BD5\u6587\u4EF6\u4F4D\u4E8E ${this.getTestOutputDir(req.testType)}/\uFF09
 
 --- \u6E90\u7801\u5185\u5BB9 ---
 \`\`\`typescript