ai-unit-test-generator 2.0.6 → 2.0.7

package/CHANGELOG.md

@@ -5,6 +5,15 @@ 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

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.6",
+  "version": "2.0.7",
   "description": "AI-powered unit test generator with smart priority scoring",
   "keywords": [
     "unit-test",