ai-unit-test-generator 1.4.5 → 2.0.1

package/lib/workflows/generate.mjs

@@ -0,0 +1,98 @@
+/**
+ * Generate 工作流：生成单元测试
+ */
+
+import { spawn } from 'node:child_process'
+import { existsSync, readFileSync } from 'node:fs'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const PKG_ROOT = join(__dirname, '../..')
+
+/**
+ * 生成单批测试
+ */
+async function generateBatch(priority, count, skip, report) {
+  const batchScript = join(PKG_ROOT, 'lib/workflows/batch.mjs')
+  
+  return new Promise((resolve, reject) => {
+    const child = spawn('node', [batchScript, priority, String(count), String(skip), report], {
+      stdio: 'inherit',
+      cwd: process.cwd()
+    })
+    
+    child.on('close', (code) => {
+      if (code === 0) {
+        resolve({ generated: count, passed: count })
+      } else {
+        reject(new Error(`Batch generation failed with code ${code}`))
+      }
+    })
+    
+    child.on('error', reject)
+  })
+}
+
+/**
+ * Generate 工作流
+ */
+export async function generate(options) {
+  const { count, priority, all, report } = options
+  
+  // 1. 检查报告是否存在
+  if (!existsSync(report)) {
+    console.error(`❌ Report not found: ${report}`)
+    console.log(`   Run: ai-test scan`)
+    process.exit(1)
+  }
+  
+  if (all) {
+    // 2. 持续生成直到所有 TODO 完成
+    console.log(`🚀 Generating all ${priority} TODO functions...\n`)
+    
+    let batchNum = 1
+    let totalGenerated = 0
+    let totalPassed = 0
+    
+    while (true) {
+      // 检查还有多少 TODO
+      const content = readFileSync(report, 'utf-8')
+      const lines = content.split('\n')
+      const todoLines = lines.filter(line => 
+        line.includes('| TODO |') && line.includes(`| ${priority} |`)
+      )
+      
+      if (todoLines.length === 0) {
+        console.log(`\n✅ All ${priority} functions completed!`)
+        console.log(`   Total generated: ${totalGenerated}`)
+        console.log(`   Total passed: ${totalPassed}`)
+        break
+      }
+      
+      console.log(`\n━━━━ Batch ${batchNum} (${todoLines.length} TODO remaining) ━━━━`)
+      
+      try {
+        const result = await generateBatch(priority, Math.min(count, todoLines.length), 0, report)
+        totalGenerated += result.generated
+        totalPassed += result.passed
+        batchNum++
+      } catch (err) {
+        console.error(`❌ Batch ${batchNum} failed:`, err.message)
+        break
+      }
+    }
+  } else {
+    // 3. 生成指定数量
+    console.log(`🚀 Generating ${count} ${priority} functions...\n`)
+    
+    try {
+      await generateBatch(priority, count, 0, report)
+    } catch (err) {
+      console.error('❌ Generation failed:', err.message)
+      process.exit(1)
+    }
+  }
+}
+

package/lib/workflows/index.mjs

@@ -5,6 +5,13 @@
  * 组合 Core、AI 和 Testing 模块实现端到端工作流
  */
 
+// 主要工作流
+export * from './init.mjs'
+export * from './analyze.mjs'
+export * from './scan.mjs'
+export * from './generate.mjs'
+
+// 批处理工作流（内部使用）
 export { main as runBatch } from './batch.mjs'
 export { main as runAll } from './all.mjs'
 

package/lib/workflows/init.mjs

@@ -0,0 +1,45 @@
+/**
+ * Init 工作流：初始化配置文件
+ */
+
+import { existsSync, copyFileSync } from 'node:fs'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const PKG_ROOT = join(__dirname, '../..')
+
+/**
+ * 初始化配置文件
+ */
+export async function init(options) {
+  const { config } = options
+  const configPath = config || 'ai-test.config.jsonc'
+  
+  // 检查配置文件是否已存在
+  if (existsSync(configPath)) {
+    console.log(`⚠️  Config already exists: ${configPath}`)
+    console.log('   Delete it first or use a different path with -c option')
+    process.exit(1)
+  }
+  
+  // 复制模板
+  console.log('⚙️  Creating default config...')
+  const templatePath = join(PKG_ROOT, 'templates', 'default.config.jsonc')
+  
+  try {
+    copyFileSync(templatePath, configPath)
+    console.log(`✅ Config created: ${configPath}`)
+    
+    // 显示下一步建议
+    console.log('\n💡 Next steps:')
+    console.log('   1. Review and customize the config (optional)')
+    console.log('   2. Run `ai-test analyze` to let AI analyze your codebase')
+    console.log('   3. Or run `ai-test scan` to start scoring immediately')
+  } catch (err) {
+    console.error(`❌ Failed to create config: ${err.message}`)
+    process.exit(1)
+  }
+}
+

package/lib/workflows/scan.mjs

@@ -0,0 +1,144 @@
+/**
+ * Scan 工作流：扫描代码 + 打分 + 生成报告
+ */
+
+import { spawn } from 'node:child_process'
+import { existsSync, mkdirSync, readFileSync } from 'node:fs'
+import { join, dirname } from 'node:path'
+import { fileURLToPath } from 'node:url'
+import { detectConfig, readConfig } from '../utils/config-manager.mjs'
+
+const __filename = fileURLToPath(import.meta.url)
+const __dirname = dirname(__filename)
+const PKG_ROOT = join(__dirname, '../..')
+
+/**
+ * 移除 JSON 注释
+ */
+function stripJsonComments(str) {
+  return String(str)
+    .replace(/\/\*[\s\S]*?\*\//g, '')
+    .replace(/(^|\s)\/\/.*$/gm, '')
+}
+
+/**
+ * 运行脚本
+ */
+function runScript(scriptPath, args) {
+  return new Promise((resolve, reject) => {
+    const fullPath = join(PKG_ROOT, 'lib', scriptPath)
+    const child = spawn('node', [fullPath, ...args], {
+      stdio: 'inherit',
+      cwd: process.cwd()
+    })
+
+    child.on('close', (code) => {
+      if (code === 0) {
+        resolve()
+      } else {
+        reject(new Error(`Script ${scriptPath} exited with code ${code}`))
+      }
+    })
+
+    child.on('error', reject)
+  })
+}
+
+/**
+ * Scan 工作流
+ */
+export async function scan(options) {
+  let { config, output, skipGit } = options
+  
+  // 1. 检查配置
+  console.log('🔍 Step 1: Checking configuration...')
+  const configPath = detectConfig(config)
+  
+  if (!configPath) {
+    console.error('❌ Config not found. Run `ai-test init` first.')
+    process.exit(1)
+  }
+  
+  config = configPath
+  console.log(`   Using config: ${config}\n`)
+  
+  // 2. 创建输出目录
+  if (!existsSync(output)) {
+    mkdirSync(output, { recursive: true })
+  }
+  
+  // 3. 可选：运行覆盖率
+  try {
+    const cfgText = existsSync(config) ? readFileSync(config, 'utf-8') : '{}'
+    const cfg = JSON.parse(stripJsonComments(cfgText))
+    const covCfg = cfg?.coverage || { runBeforeScan: false }
+    
+    if (covCfg.runBeforeScan) {
+      console.log('🧪 Running coverage before scan...')
+      await new Promise((resolve) => {
+        const cmd = covCfg.command || 'npx jest --coverage --silent'
+        const child = spawn(cmd, { stdio: 'inherit', shell: true, cwd: process.cwd() })
+        child.on('close', () => resolve())
+        child.on('error', () => resolve())
+      })
+      console.log('✅ Coverage completed.\n')
+    }
+  } catch (err) {
+    console.warn('⚠️  Coverage step failed. Continuing scan.')
+  }
+  
+  console.log('🚀 Starting code scan...\n')
+  
+  try {
+    // 4. 扫描 AST + 复杂度
+    console.log('📋 Step 2: Scanning targets...')
+    await runScript('core/scanner.mjs', [
+      '--config', config,
+      '--out', join(output, 'targets.json')
+    ])
+    
+    // 5. Git 信号（可选）
+    if (!skipGit) {
+      console.log('\n📊 Step 3: Analyzing Git history...')
+      await runScript('core/git-analyzer.mjs', [
+        '--targets', join(output, 'targets.json'),
+        '--out', join(output, 'git_signals.json')
+      ])
+    }
+    
+    // 6. 打分
+    console.log('\n🎯 Step 4: Scoring targets...')
+    const scoreArgs = [
+      '--targets', join(output, 'targets.json'),
+      '--config', config,
+      '--out-md', join(output, 'ut_scores.md'),
+      '--out-csv', join(output, 'ut_scores.csv')
+    ]
+    
+    if (!skipGit && existsSync(join(output, 'git_signals.json'))) {
+      scoreArgs.push('--git', join(output, 'git_signals.json'))
+    }
+    
+    await runScript('core/scorer.mjs', scoreArgs)
+    
+    // 7. 统计结果
+    const reportPath = join(output, 'ut_scores.md')
+    if (existsSync(reportPath)) {
+      const content = readFileSync(reportPath, 'utf-8')
+      const todoCount = (content.match(/\| TODO \|/g) || []).length
+      const doneCount = (content.match(/\| DONE \|/g) || []).length
+      
+      console.log('\n✅ Scan completed!')
+      console.log(`\n📊 Status:`)
+      console.log(`   TODO: ${todoCount}`)
+      console.log(`   DONE: ${doneCount}`)
+      console.log(`   Total: ${todoCount + doneCount}`)
+      console.log(`\n📄 Report: ${reportPath}`)
+      console.log(`\n💡 Next: ai-test generate`)
+    }
+  } catch (err) {
+    console.error('❌ Scan failed:', err.message)
+    process.exit(1)
+  }
+}
+

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "ai-unit-test-generator",
-  "version": "1.4.5",
+  "version": "2.0.1",
   "description": "AI-powered unit test generator with smart priority scoring",
   "keywords": [
     "unit-test",

package/templates/default.config.jsonc

@@ -310,5 +310,63 @@
     "CC": 6,           // 复杂度默认值
     "ERLikelihood": 3, // 错误可能性默认值
     "Testability": 6   // 可测试性默认值
+  },
+  
+  // ========== AI 增强配置（AI 唯一可写区域）==========
+  
+  "aiEnhancement": {
+    "enabled": true,        // 是否启用 AI 增强
+    "analyzed": false,      // 是否已完成 AI 分析
+    "analyzedAt": null,     // 分析时间戳
+    
+    // ✅ 业务实体关键词（用于 AST 扫描识别业务对象）
+    "entityKeywords": [
+      "Payment", "Order", "Booking", "User", "Hotel", "Room", 
+      "Cart", "Price", "Guest", "Request", "Response"
+    ],
+    
+    // AI 建议（由 ai-test analyze 填充）
+    "suggestions": {
+      "businessCriticalPaths": [
+        // 示例：
+        // {
+        //   "pattern": "services/payment/**",
+        //   "confidence": 0.95,
+        //   "reason": "Handles Stripe payment processing",
+        //   "suggestedBC": 10,
+        //   "evidence": [
+        //     "Contains processPayment with Stripe API",
+        //     "Referenced by checkout flow",
+        //     "Handles money transactions"
+        //   ]
+        // }
+      ],
+      "highRiskModules": [
+        // 示例：
+        // {
+        //   "pattern": "utils/date/**",
+        //   "confidence": 0.88,
+        //   "reason": "Complex timezone calculations",
+        //   "suggestedER": 8,
+        //   "evidence": [
+        //     "Multiple timezone conversions",
+        //     "Handles DST edge cases"
+        //   ]
+        // }
+      ],
+      "testabilityAdjustments": [
+        // 示例：
+        // {
+        //   "pattern": "utils/**",
+        //   "confidence": 0.92,
+        //   "reason": "Pure functions, easy to test",
+        //   "adjustment": "+1",
+        //   "evidence": [
+        //     "No side effects",
+        //     "No external dependencies"
+        //   ]
+        // }
+      ]
+    }
   }
 }