ai-unit-test-generator 1.3.0 → 1.4.1

package/README.md

@@ -319,6 +319,41 @@ Notes:
 
 ## 📈 Metrics Explained
 
+### Coverage-aware Scoring (New)
+
+We incorporate code coverage into prioritization, balancing both incremental (new code) and stock (existing code) scenarios.
+
+- Coverage Score (CS) mapping:
+  - 0% → 10 (highest)
+  - 1-40% → 8
+  - 41-70% → 6
+  - 71-90% → 3
+  - 91-100% → 1 (lowest)
+  - N/A → 5 (no data)
+- Default weights (legacy mode): `coverage: 0.20`
+- Layered weights: each layer includes a `coverage` weight (see default config)
+- Optional `coverageBoost` provides a small additive boost for low-coverage files to break ties.
+
+Coverage is parsed from `coverage/coverage-summary.json` if present. You can run coverage before `scan` to leverage coverage-aware prioritization.
+
+Configuration snippet:
+
+```json
+{
+  "coverageScoring": {
+    "naScore": 5,
+    "mapping": [
+      { "lte": 0, "score": 10 },
+      { "lte": 40, "score": 8 },
+      { "lte": 70, "score": 6 },
+      { "lte": 90, "score": 3 },
+      { "lte": 100, "score": 1 }
+    ]
+  },
+  "coverageBoost": { "enable": true, "threshold": 60, "scale": 0.5, "maxBoost": 0.5 }
+}
+```
+
 ### Testability (50% weight in Foundation layer)
 - Pure functions: 10/10
 - Simple mocks: 8-9/10

package/bin/cli.js

@@ -43,6 +43,25 @@ program
       mkdirSync(output, { recursive: true })
     }
 
+    // 可选：在扫描前自动运行覆盖率（由配置控制）
+    try {
+      const cfgText = existsSync(config) ? readFileSync(config, 'utf-8') : '{}'
+      const cfg = JSON.parse(cfgText)
+      const covCfg = cfg?.coverage || { runBeforeScan: false }
+      if (covCfg.runBeforeScan) {
+        console.log('🧪 Running coverage before scan...')
+        await new Promise((resolve, reject) => {
+          const cmd = covCfg.command || 'npx jest --coverage --silent'
+          const child = spawn(cmd, { stdio: 'inherit', shell: true, cwd: process.cwd() })
+          child.on('close', code => code === 0 ? resolve(0) : reject(new Error(`coverage exited ${code}`)))
+          child.on('error', reject)
+        })
+      }
+    } catch (err) {
+      console.warn('⚠️  Coverage step failed or Jest not installed. Skipping coverage and continuing scan.')
+      console.warn('    - npm i -D jest@29 ts-jest@29 @types/jest@29 jest-environment-jsdom@29 --legacy-peer-deps')
+    }
+
     console.log('🚀 Starting code scan...\n')
 
     try {

package/lib/core/scorer.mjs

@@ -191,8 +191,12 @@ function mapDependencyCount(depGraphData, cfg) {
   return 2
 }
 // 传统评分（兼容旧配置）
-function computeScoreLegacy({ BC, CC, ER, ROI }, cfg) { 
-  const s = BC * pickWeight(cfg,'BC',0.4) + CC * pickWeight(cfg,'CC',0.3) + ER * pickWeight(cfg,'ER',0.2) + ROI * pickWeight(cfg,'ROI',0.1)
+function computeScoreLegacy({ BC, CC, ER, ROI, coverageScore }, cfg) { 
+  const s = BC * pickWeight(cfg,'BC',0.4)
+    + CC * pickWeight(cfg,'CC',0.3)
+    + ER * pickWeight(cfg,'ER',0.2)
+    + ROI * pickWeight(cfg,'ROI',0.1)
+    + (typeof coverageScore === 'number' ? coverageScore * pickWeight(cfg,'coverage',0) : 0)
   const score = toFixedDown(s, (cfg?.round?.digits ?? 2))
   const P0 = pickThreshold(cfg,'P0',8.5), P1 = pickThreshold(cfg,'P1',6.5), P2 = pickThreshold(cfg,'P2',4.5)
   let priority = 'P3'
@@ -203,7 +207,7 @@ function computeScoreLegacy({ BC, CC, ER, ROI }, cfg) {
 }
 
 // 分层评分（新方法）
-function computeScoreLayered({ BC, CC, ER, testability, dependencyCount }, target, cfg) {
+function computeScoreLayered({ BC, CC, ER, testability, dependencyCount, coverageScore }, target, cfg) {
   const layer = target.layer || 'unknown'
   const layerDef = cfg?.layers?.[layer]
   
@@ -221,6 +225,7 @@ function computeScoreLayered({ BC, CC, ER, testability, dependencyCount }, targe
   if (weights.errorRisk !== undefined) score += ER * weights.errorRisk
   if (weights.testability !== undefined) score += testability * weights.testability
   if (weights.dependencyCount !== undefined) score += dependencyCount * weights.dependencyCount
+  if (weights.coverage !== undefined && typeof coverageScore === 'number') score += coverageScore * weights.coverage
   
   score = toFixedDown(score, cfg?.round?.digits ?? 2)
   
@@ -245,6 +250,25 @@ function computeScore(metrics, target, cfg) {
   }
 }
 
+// 覆盖率百分比到覆盖率分数（Coverage Score）映射
+function mapCoverageScore(pct, cfg) {
+  if (pct === null || pct === undefined || Number.isNaN(pct)) {
+    return cfg?.coverageScoring?.naScore ?? 5
+  }
+  const mapping = cfg?.coverageScoring?.mapping
+  if (Array.isArray(mapping) && mapping.length) {
+    const ordered = mapping.slice().sort((a,b)=> (a.lte ?? 1e9) - (b.lte ?? 1e9))
+    for (const rule of ordered) {
+      if (typeof rule.lte === 'number' && pct <= rule.lte) return rule.score
+    }
+  }
+  if (pct <= 0) return 10
+  if (pct <= 40) return 8
+  if (pct <= 70) return 6
+  if (pct <= 90) return 3
+  return 1
+}
+
 function defaultMd(rows, statusMap = new Map()) {
   // 按总分降序排序（高分在前）
   const sorted = [...rows].sort((a, b) => b.score - a.score)
@@ -252,14 +276,14 @@ function defaultMd(rows, statusMap = new Map()) {
   let md = '<!-- UT Priority Scoring Report -->\n'
   md += '<!-- Format: Status can be "TODO" | "DONE" | "SKIP" -->\n'
   md += '<!-- You can mark status by replacing TODO with DONE or SKIP -->\n\n'
-  md += '| Status | Score | Priority | Name | Type | Layer | Path | BC | CC | ER | Testability | DepCount | Notes |\n'
-  md += '|--------|-------|----------|------|------|-------|------|----|----|----|-----------|---------| ------- |\n'
+  md += '| Status | Score | Priority | Name | Type | Layer | Path | Coverage | CS | BC | CC | ER | Testability | DepCount | Notes |\n'
+  md += '|--------|-------|----------|------|------|-------|------|----------|----|----|----|----|-----------|---------| ------- |\n'
   
   sorted.forEach(r => {
     // 从状态映射中查找现有状态，否则默认为 TODO
     const key = `${r.path}#${r.name}`
     const status = statusMap.get(key) || 'TODO'
-    md += `| ${status} | ${r.score} | ${r.priority} | ${r.name} | ${r.type} | ${r.layerName || r.layer} | ${r.path} | ${r.BC} | ${r.CC} | ${r.ER} | ${r.testability || r.ROI} | ${r.dependencyCount || 'N/A'} | ${r.notes} |\n` 
+    md += `| ${status} | ${r.score} | ${r.priority} | ${r.name} | ${r.type} | ${r.layerName || r.layer} | ${r.path} | ${typeof r.coveragePct === 'number' ? r.coveragePct.toFixed(1) + '%':'N/A'} | ${r.coverageScore ?? 'N/A'} | ${r.BC} | ${r.CC} | ${r.ER} | ${r.testability || r.ROI} | ${r.dependencyCount || 'N/A'} | ${r.notes} |\n` 
   })
   
   // 添加统计信息
@@ -291,7 +315,7 @@ function defaultCsv(rows) {
   // 按总分降序排序（高分在前）
   const sorted = [...rows].sort((a, b) => b.score - a.score)
   
-  const head = ['status','score','priority','name','path','type','layer','layerName','BC','CC','ER','testability','dependencyCount','notes'].join(',')
+  const head = ['status','score','priority','name','path','type','layer','layerName','coveragePct','coverageScore','BC','CC','ER','testability','dependencyCount','notes'].join(',')
   const body = sorted.map(r => [
     'TODO',
     r.score,
@@ -301,6 +325,8 @@ function defaultCsv(rows) {
     r.type,
     r.layer || 'N/A',
     r.layerName || 'N/A',
+    (typeof r.coveragePct === 'number' ? r.coveragePct.toFixed(1) + '%' : 'N/A'),
+    (r.coverageScore ?? 'N/A'),
     r.BC,
     r.CC,
     r.ER,
@@ -483,7 +509,7 @@ async function main() {
   // P0-2: 构建依赖图
   const depGraph = cfg?.depGraph?.enable ? buildDepGraph(funcProvider.project, cfg) : null
 
-  // 可选覆盖率引导：读取 coverage-summary.json
+  // 读取覆盖率汇总（如果存在）
   let coverageSummary = null
   if (existsSync(coverageJsonPath)) {
     try { coverageSummary = JSON.parse(readFileSync(coverageJsonPath, 'utf8')) } catch {}
@@ -524,19 +550,27 @@ async function main() {
     const testability = mapTestabilityByConfig(roiHint, cfg, localROI, overrides, t)
     const dependencyCount = mapDependencyCount(depGraphData, cfg)
 
-    let { score, priority, layer, layerName } = computeScore({ BC, CC: CCFinal, ER, ROI, testability, dependencyCount }, t, cfg)
-
-    // 覆盖率加权（可选）：对低覆盖模块提升优先级或分数
+    // 计算覆盖率分数（用于打分）
+    let coveragePct = null
+    let coverageScore = null
     if (coverageSummary) {
       const fileKey = Object.keys(coverageSummary).find(k => k.endsWith(path))
       const fileCov = fileKey ? coverageSummary[fileKey] : null
       const linesPct = fileCov?.lines?.pct
+      if (typeof linesPct === 'number') coveragePct = linesPct
+      coverageScore = mapCoverageScore(linesPct, cfg)
+    }
+
+    let { score, priority, layer, layerName } = computeScore({ BC, CC: CCFinal, ER, ROI, testability, dependencyCount, coverageScore }, t, cfg)
+
+    // 覆盖率加权（可选）：对低覆盖文件小幅加分
+    if (coverageSummary) {
       const cfgBoost = cfg?.coverageBoost || { enable: false }
-      if (cfgBoost.enable && typeof linesPct === 'number') {
+      if (cfgBoost.enable && typeof coveragePct === 'number') {
         const threshold = cfgBoost.threshold ?? 60
         const maxBoost = cfgBoost.maxBoost ?? 0.5
-        if (linesPct < threshold) {
-          const ratio = (threshold - linesPct) / Math.max(threshold, 1)
+        if (coveragePct < threshold) {
+          const ratio = (threshold - coveragePct) / Math.max(threshold, 1)
           const delta = toFixedDown(Math.min(maxBoost, ratio * (cfgBoost.scale ?? 0.5)), 2)
           score = toFixedDown(score + delta, cfg?.round?.digits ?? 2)
           // 重新判定优先级
@@ -583,6 +617,8 @@ async function main() {
       ROI, 
       testability, 
       dependencyCount,
+      coveragePct: coveragePct ?? 'N/A',
+      coverageScore: coverageScore ?? 'N/A',
       score, 
       priority, 
       notes: notesParts.join('; ') 

package/lib/extract-tests.mjs

@@ -0,0 +1,199 @@
+#!/usr/bin/env node
+/**
+ * 从 AI 回复中提取测试文件并自动创建
+ */
+
+import { readFileSync, writeFileSync, mkdirSync, existsSync } from 'fs';
+import { dirname } from 'path';
+
+/**
+ * 从 AI 响应中提取测试文件
+ */
+export function extractTests(content, options = {}) {
+  const { overwrite = false, dryRun = false } = options;
+
+  // 先尝试解析 JSON Manifest（优先）
+  // 形如：```json { version: 1, files: [{ path, source, ... }] }
+  let manifest = null;
+  const manifestRegex = /```json\s*\n([\s\S]*?)\n```/gi;
+  let m;
+  while ((m = manifestRegex.exec(content)) !== null) {
+    const jsonStr = m[1].trim();
+    try {
+      const obj = JSON.parse(jsonStr);
+      if (obj && Array.isArray(obj.files)) {
+        manifest = obj;
+        break;
+      }
+    } catch {}
+  }
+  const manifestPaths = manifest?.files?.map(f => String(f.path).trim()) ?? [];
+
+  // 再匹配多种文件代码块格式（回退/兼容，增强鲁棒性）
+  const patterns = [
+    // 格式1: ### 测试文件: path (有/无反引号)
+    /###\s*测试文件\s*[:：]\s*`?([^\n`]+?)`?\s*\n```(?:typescript|ts|tsx|javascript|js|jsx|json|text)?\s*\n([\s\S]*?)\n```/gi,
+    // 格式2: **测试文件**: path
+    /\*\*测试文件\*\*\s*[:：]\s*`?([^\n`]+)`?\s*\n```(?:typescript|ts|tsx|javascript|js|jsx|json|text)?\s*\n([\s\S]*?)\n```/gi,
+    // 格式3: 文件路径: path
+    /文件路径\s*[:：]\s*`?([^\n`]+)`?\s*\n```(?:typescript|ts|tsx|javascript|js|jsx|json|text)?\s*\n([\s\S]*?)\n```/gi,
+    // 格式4: # path.test.ts (仅文件名标题)
+    /^#+\s+([^\n]+\.test\.[jt]sx?)\s*\n```(?:typescript|ts|tsx|javascript|js|jsx)?\s*\n([\s\S]*?)\n```/gim,
+    // 格式5: 更宽松的匹配（任意"path"后接代码块，路径必须含 .test.）
+    /(?:path|文件|file)\s*[:：]?\s*`?([^\n`]*\.test\.[jt]sx?[^\n`]*?)`?\s*\n```(?:typescript|ts|tsx|javascript|js|jsx)?\s*\n([\s\S]*?)\n```/gi,
+  ];
+
+  const created = [];
+  const skipped = [];
+  const errors = [];
+
+  patterns.forEach(fileRegex => {
+    let match;
+    while ((match = fileRegex.exec(content)) !== null) {
+      const [, filePath, testCode] = match;
+      const cleanPath = filePath.trim();
+
+      // 若存在 Manifest，则只允许清单内的路径
+      if (manifestPaths.length > 0 && !manifestPaths.includes(cleanPath)) {
+        continue;
+      }
+
+      // 跳过已处理的文件
+      if (created.some(f => f.path === cleanPath) || 
+          skipped.some(f => f.path === cleanPath)) {
+        continue;
+      }
+
+      // 检查文件是否已存在
+      if (existsSync(cleanPath) && !overwrite) {
+        skipped.push({ path: cleanPath, reason: 'exists' });
+        continue;
+      }
+
+      if (dryRun) {
+        created.push({ path: cleanPath, code: testCode.trim(), dryRun: true });
+        continue;
+      }
+
+      try {
+        // 确保目录存在
+        mkdirSync(dirname(cleanPath), { recursive: true });
+
+        // 写入测试文件
+        const cleanCode = testCode.trim();
+        writeFileSync(cleanPath, cleanCode + '\n');
+        created.push({ path: cleanPath, code: cleanCode });
+      } catch (err) {
+        errors.push({ path: cleanPath, error: err.message });
+      }
+    }
+  });
+
+  // 若有 Manifest 但未生成任何文件，则报告可能缺失代码块
+  if (manifestPaths.length > 0) {
+    const createdPaths = new Set(created.map(f => f.path));
+    const missing = manifestPaths.filter(p => !createdPaths.has(p));
+    missing.forEach(p => errors.push({ path: p, error: 'missing code block for manifest entry' }));
+  }
+
+  return { created, skipped, errors };
+}
+
+/**
+ * CLI 入口
+ */
+export function runCLI(argv = process.argv) {
+  const args = argv.slice(2);
+  
+  if (args.length === 0) {
+    console.error('❌ 缺少参数\n');
+    console.error('用法: ai-test extract-tests <response-file> [options]\n');
+    console.error('选项:');
+    console.error('  --overwrite    覆盖已存在的文件');
+    console.error('  --dry-run      仅显示将要创建的文件，不实际写入\n');
+    console.error('示例:');
+    console.error('  ai-test extract-tests response.txt');
+    console.error('  ai-test extract-tests response.txt --overwrite');
+    process.exit(1);
+  }
+  
+  const responseFile = args[0];
+  const options = {
+    overwrite: args.includes('--overwrite'),
+    dryRun: args.includes('--dry-run')
+  };
+  
+  if (!existsSync(responseFile)) {
+    console.error(`❌ 文件不存在: ${responseFile}`);
+    process.exit(1);
+  }
+  
+  try {
+    const content = readFileSync(responseFile, 'utf8');
+    const { created, skipped, errors } = extractTests(content, options);
+    
+    // 输出结果
+    created.forEach(f => {
+      if (f.dryRun) {
+        console.log(`[DRY-RUN] 将创建: ${f.path}`);
+      } else {
+        console.log(`✅ 创建测试文件: ${f.path}`);
+      }
+    });
+    
+    skipped.forEach(f => {
+      if (f.reason === 'exists') {
+        console.log(`⚠️  跳过（已存在）: ${f.path}`);
+      }
+    });
+    
+    errors.forEach(f => {
+      console.error(`❌ 创建失败: ${f.path} - ${f.error}`);
+    });
+    
+    console.log('');
+    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
+    console.log(`✨ 总共创建 ${created.length} 个测试文件`);
+    if (skipped.length > 0) {
+      console.log(`⚠️  跳过 ${skipped.length} 个已存在的文件`);
+    }
+    if (errors.length > 0) {
+      console.log(`❌ ${errors.length} 个文件创建失败`);
+    }
+    console.log('━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━');
+    console.log('');
+    
+    if (created.length > 0 && !options.dryRun) {
+      console.log('🧪 运行测试:');
+      console.log('   npm test');
+      console.log('');
+      console.log('📝 标记完成（如需）:');
+      const functionNames = created
+        .map(f => f.path.match(/\/([^/]+)\.test\./)?.[1])
+        .filter(Boolean)
+        .join(',');
+      if (functionNames) {
+        console.log(`   ai-test mark-done "${functionNames}"`);
+      }
+    } else if (created.length === 0) {
+      console.log('❌ 未找到任何测试文件');
+      console.log('');
+      console.log('请检查 AI 回复格式是否正确。预期格式：');
+      console.log('  ### 测试文件: src/utils/xxx.test.ts');
+      console.log('  ```typescript');
+      console.log('  // 测试代码');
+      console.log('  ```');
+    }
+    
+    process.exit(errors.length > 0 ? 1 : 0);
+  } catch (err) {
+    console.error(`❌ 错误: ${err.message}`);
+    process.exit(1);
+  }
+}
+
+// 作为脚本直接运行时
+if (import.meta.url === `file://${process.argv[1]}`) {
+  runCLI();
+}
+

package/lib/run-batch.mjs

@@ -0,0 +1,81 @@
+#!/usr/bin/env node
+/**
+ * 单批次：生成 prompt → 调用 AI → 提取测试 → 运行 Jest → 输出失败摘要
+ */
+
+import { spawn } from 'child_process'
+import { existsSync, readFileSync } from 'fs'
+
+function sh(cmd, args = []) {
+  return new Promise((resolve, reject) => {
+    const child = spawn(cmd, args, { stdio: 'inherit', cwd: process.cwd() })
+    child.on('close', code => code === 0 ? resolve(0) : reject(new Error(`${cmd} exited ${code}`)))
+    child.on('error', reject)
+  })
+}
+
+function readCoverageSummary() {
+  const path = 'coverage/coverage-summary.json'
+  if (!existsSync(path)) return null
+  try { return JSON.parse(readFileSync(path, 'utf8')) } catch { return null }
+}
+
+function getCoveragePercent(summary) {
+  if (!summary || !summary.total) return 0
+  return summary.total.lines?.pct ?? 0
+}
+
+async function main(argv = process.argv) {
+  const args = argv.slice(2)
+  const priority = args[0] || 'P0'
+  const limit = Number(args[1] || 10)
+  const skip = Number(args[2] || 0)
+  const minCovDelta = priority === 'P0' ? 2 : (priority === 'P1' ? 1 : 0)
+
+  // 记录初始覆盖率
+  const beforeCov = getCoveragePercent(readCoverageSummary())
+
+  // 1) 生成 Prompt（加入上一轮失败提示 hints.txt 如存在）
+  const promptArgs = ['ai-test-generator/lib/gen-test-prompt.mjs', '--report', 'reports/ut_scores.md', '-p', priority, '-n', String(limit), '--skip', String(skip)]
+  try { await sh('node', [...promptArgs, '--hints-file', 'reports/hints.txt']) } catch { await sh('node', promptArgs) }
+
+  // 2) 调用 AI
+  await sh('node', ['ai-test-generator/lib/ai-generate.mjs', '--prompt', 'prompt.txt', '--out', 'reports/ai_response.txt'])
+
+  // 3) 提取测试
+  await sh('node', ['ai-test-generator/lib/extract-tests.mjs', 'reports/ai_response.txt', '--overwrite'])
+
+  // 4) 运行 Jest（按优先级自适应重跑与覆盖率增量）
+  const reruns = priority === 'P0' ? 1 : (priority === 'P1' ? 0 : 0)
+  for (let i = 0; i < Math.max(1, reruns + 1); i++) {
+    try { await sh('node', ['ai-test-generator/lib/jest-runner.mjs']) } catch { /* 继续做失败分析 */ }
+  }
+
+  // 校验覆盖率增量（P0/P1）
+  if (minCovDelta > 0) {
+    const afterCov = getCoveragePercent(readCoverageSummary())
+    const delta = afterCov - beforeCov
+    if (delta < minCovDelta) {
+      console.warn(`⚠️  Coverage delta ${delta.toFixed(2)}% < required ${minCovDelta}% (before: ${beforeCov.toFixed(2)}%, after: ${afterCov.toFixed(2)}%)`)
+    } else {
+      console.log(`✅ Coverage improved: ${beforeCov.toFixed(2)}% → ${afterCov.toFixed(2)}% (+${delta.toFixed(2)}%)`)
+    }
+  }
+
+  // 5) 失败分析并落盘 hints
+  const { spawn } = await import('child_process')
+  const { writeFileSync } = await import('fs')
+  await new Promise((resolve) => {
+    const child = spawn('node', ['ai-test-generator/lib/jest-failure-analyzer.mjs'], { stdio: ['inherit','pipe','inherit'] })
+    const chunks = []
+    child.stdout.on('data', d => chunks.push(Buffer.from(d)))
+    child.on('close', () => {
+      try { const obj = JSON.parse(Buffer.concat(chunks).toString('utf8')); writeFileSync('reports/hints.txt', obj.hints?.length ? `# 上一轮失败修复建议\n- ${obj.hints.join('\n- ')}` : '') } catch {}
+      resolve()
+    })
+  })
+}
+
+if (import.meta.url === `file://${process.argv[1]}`) main()
+
+

package/lib/workflows/all.mjs

@@ -5,6 +5,12 @@
 
 import { readFileSync, existsSync } from 'fs'
 import { spawn } from 'child_process'
+import { fileURLToPath } from 'url'
+import { dirname, join } from 'path'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const pkgRoot = join(__dirname, '../..')
 
 function sh(cmd, args = []) { return new Promise((resolve, reject) => {
   const child = spawn(cmd, args, { stdio: 'inherit', cwd: process.cwd() })
@@ -37,7 +43,7 @@ Remaining TODO: ${remain}
 ━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━━`)
 
     try {
-      await sh('node', ['ai-test-generator/lib/run-batch.mjs', priority, String(batchSize), String(skip)])
+      await sh('node', [join(pkgRoot, 'lib/workflows/batch.mjs'), priority, String(batchSize), String(skip)])
     } catch (err) {
       console.error('Batch failed:', err.message)
     }

package/lib/workflows/batch.mjs

@@ -5,6 +5,12 @@
 
 import { spawn } from 'child_process'
 import { existsSync, readFileSync, writeFileSync } from 'fs'
+import { fileURLToPath } from 'url'
+import { dirname, join } from 'path'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const pkgRoot = join(__dirname, '../..')
 
 function sh(cmd, args = []) {
   return new Promise((resolve, reject) => {
@@ -99,7 +105,7 @@ async function main(argv = process.argv) {
 
   // 1) 生成 Prompt（只针对 TODO 函数，加入上一轮失败提示）
   const promptArgs = [
-    'ai-test-generator/lib/ai/prompt-builder.mjs',
+    join(pkgRoot, 'lib/ai/prompt-builder.mjs'),
     '--report', reportPath,
     '-p', priority,
     '-n', String(limit),
@@ -114,11 +120,11 @@ async function main(argv = process.argv) {
 
   // 2) 调用 AI
   console.log('\n🤖 Calling AI...')
-  await sh('node', ['ai-test-generator/lib/ai/client.mjs', '--prompt', 'prompt.txt', '--out', 'reports/ai_response.txt'])
+  await sh('node', [join(pkgRoot, 'lib/ai/client.mjs'), '--prompt', 'prompt.txt', '--out', 'reports/ai_response.txt'])
 
   // 3) 提取测试
   console.log('\n📦 Extracting tests...')
-  await sh('node', ['ai-test-generator/lib/ai/extractor.mjs', 'reports/ai_response.txt', '--overwrite'])
+  await sh('node', [join(pkgRoot, 'lib/ai/extractor.mjs'), 'reports/ai_response.txt', '--overwrite'])
 
   // 4) 运行 Jest（按优先级自适应重跑）
   console.log('\n🧪 Running tests...')
@@ -127,7 +133,7 @@ async function main(argv = process.argv) {
   
   for (let i = 0; i < Math.max(1, reruns + 1); i++) {
     try {
-      await sh('node', ['ai-test-generator/lib/testing/runner.mjs'])
+      await sh('node', [join(pkgRoot, 'lib/testing/runner.mjs')])
       testsPassed = true
       break
     } catch {
@@ -163,7 +169,7 @@ async function main(argv = process.argv) {
   const { spawn: spawnLocal } = await import('child_process')
   const { writeFileSync: writeFileSyncLocal } = await import('fs')
   await new Promise((resolve) => {
-    const child = spawnLocal('node', ['ai-test-generator/lib/testing/analyzer.mjs'], { stdio: ['inherit','pipe','inherit'] })
+    const child = spawnLocal('node', [join(pkgRoot, 'lib/testing/analyzer.mjs')], { stdio: ['inherit','pipe','inherit'] })
     const chunks = []
     child.stdout.on('data', d => chunks.push(Buffer.from(d)))
     child.on('close', () => {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-unit-test-generator",
-  "version": "1.3.0",
+  "version": "1.4.1",
   "description": "AI-powered unit test generator with smart priority scoring",
   "keywords": [
     "unit-test",
@@ -66,4 +66,3 @@
   },
   "homepage": "https://github.com/YuhengZhou/ai-unit-test-generator#readme"
 }
-

package/templates/default.config.json

@@ -22,9 +22,10 @@
         "multipleReferences": true
       },
       "weights": {
-        "testability": 0.50,
-        "dependencyCount": 0.30,
-        "complexity": 0.20
+        "testability": 0.45,
+        "dependencyCount": 0.25,
+        "complexity": 0.20,
+        "coverage": 0.10
       },
       "thresholds": { "P0": 7.5, "P1": 6.0, "P2": 4.0 },
       "coverageTarget": 100,
@@ -41,9 +42,10 @@
         "mayDependOnUtils": true
       },
       "weights": {
-        "businessCriticality": 0.40,
-        "complexity": 0.30,
-        "errorRisk": 0.30
+        "businessCriticality": 0.35,
+        "complexity": 0.25,
+        "errorRisk": 0.25,
+        "coverage": 0.15
       },
       "thresholds": { "P0": 8.0, "P1": 6.5, "P2": 4.5 },
       "coverageTarget": 80,
@@ -58,9 +60,10 @@
         "connectsLogicAndUI": true
       },
       "weights": {
-        "businessCriticality": 0.50,
-        "complexity": 0.30,
-        "errorRisk": 0.20
+        "businessCriticality": 0.45,
+        "complexity": 0.25,
+        "errorRisk": 0.20,
+        "coverage": 0.10
       },
       "thresholds": { "P0": 8.0, "P1": 6.5, "P2": 4.5 },
       "coverageTarget": 70,
@@ -75,15 +78,40 @@
         "hasInteraction": true
       },
       "weights": {
-        "businessCriticality": 0.40,
-        "complexity": 0.30,
-        "testability": 0.30
+        "businessCriticality": 0.35,
+        "complexity": 0.25,
+        "testability": 0.25,
+        "coverage": 0.15
       },
       "thresholds": { "P0": 8.5, "P1": 7.0, "P2": 5.0 },
       "coverageTarget": 50,
       "notes": "UI层，更适合集成测试"
     }
   },
+  "coverageScoring": {
+    "naScore": 5,
+    "mapping": [
+      { "lte": 0, "score": 10 },
+      { "lte": 40, "score": 8 },
+      { "lte": 70, "score": 6 },
+      { "lte": 90, "score": 3 },
+      { "lte": 100, "score": 1 }
+    ]
+  },
+  "coverage": {
+    "runBeforeScan": true,
+    "command": "npx jest --coverage --silent"
+  },
+  "weights": {
+    "BC": 0.25,
+    "CC": 0.15,
+    "ER": 0.15,
+    "ROI": 0.15,
+    "dependencyCount": 0.10,
+    "coverage": 0.20
+  },
+  "thresholds": { "P0": 8.0, "P1": 6.5, "P2": 4.5 },
+  "coverageBoost": { "enable": true, "threshold": 60, "scale": 0.5, "maxBoost": 0.5 },
   "ccFusion": {
     "useCognitive": true,
     "cyclomaticT": 15,