ai-unit-test-generator 2.0.5 → 2.0.7

package/CHANGELOG.md

@@ -5,6 +5,24 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [2.0.7] - 2025-01-11
+
+### 🐛 Hotfix
+- **Fixed**: `ai-test generate` workflow - `batch.mjs` now properly captures `prompt-builder.mjs` stdout and writes to `prompt.txt`
+- Modified `sh()` helper to support stdout capture with `captureStdout` option
+- Resolves missing `prompt.txt` file issue in generation workflow
+
+---
+
+## [2.0.6] - 2025-01-11
+
+### 🐛 Hotfix
+- **Fixed**: `ai-test generate` command - corrected `client.mjs` argument parsing
+- Now supports `--prompt`, `--prompt-file`, and `--promptFile` for compatibility
+- Resolves `Prompt file not found: prompt.txt` error in batch generation workflow
+
+---
+
 ## [2.0.5] - 2025-01-11
 
 ### ✨ Feature: Simplified AI Suggestion Review

package/lib/ai/client.mjs

@@ -60,7 +60,8 @@ export async function runOnce({ prompt, promptFile, out = 'reports/ai_response.t
 
 export async function runCLI(argv = process.argv) {
   const args = parseArgs(argv)
-  const promptFile = args['prompt'] || args['prompt-file'] || null
+  // 支持 --prompt, --prompt-file, --promptFile 三种形式
+  const promptFile = args['prompt'] || args['prompt-file'] || args['promptFile'] || null
   const out = args['out'] || 'reports/ai_response.txt'
   const model = args['model'] || null
   const timeoutSec = args['timeout'] ? Number(args['timeout']) : 600

package/lib/workflows/batch.mjs

@@ -12,10 +12,24 @@ const __filename = fileURLToPath(import.meta.url)
 const __dirname = dirname(__filename)
 const pkgRoot = join(__dirname, '../..')
 
-function sh(cmd, args = []) {
+function sh(cmd, args = [], options = {}) {
   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}`)))
+    const stdio = options.captureStdout ? ['inherit', 'pipe', 'inherit'] : 'inherit'
+    const child = spawn(cmd, args, { stdio, cwd: process.cwd() })
+    
+    const chunks = []
+    if (options.captureStdout) {
+      child.stdout.on('data', d => chunks.push(Buffer.from(d)))
+    }
+    
+    child.on('close', code => {
+      if (code === 0) {
+        const output = options.captureStdout ? Buffer.concat(chunks).toString('utf8') : null
+        resolve(output)
+      } else {
+        reject(new Error(`${cmd} exited ${code}`))
+      }
+    })
     child.on('error', reject)
   })
 }
@@ -112,11 +126,16 @@ async function main(argv = process.argv) {
     '--skip', String(skip),
     '--only-todo' // 新增：只处理 TODO 状态
   ]
+  
+  let promptText
   try {
-    await sh('node', [...promptArgs, '--hints-file', 'reports/hints.txt'])
+    promptText = await sh('node', [...promptArgs, '--hints-file', 'reports/hints.txt'], { captureStdout: true })
   } catch {
-    await sh('node', promptArgs)
+    promptText = await sh('node', promptArgs, { captureStdout: true })
   }
+  
+  // 写入 prompt.txt
+  writeFileSync('prompt.txt', promptText, 'utf-8')
 
   // 2) 调用 AI
   console.log('\n🤖 Calling AI...')

package/package.json

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