@nhonh/qabot 0.5.0 → 0.5.1

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@nhonh/qabot",
-  "version": "0.5.0",
+  "version": "0.5.1",
   "description": "AI-powered universal QA automation tool. Import any project, AI analyzes and runs tests across all layers.",
   "main": "src/index.js",
   "bin": {

package/src/ai/ai-engine.js

@@ -94,9 +94,13 @@ export class AIEngine {
         if (!isFirst) {
           batchContext.existingTestCode = "";
           batchContext.sourceCode = "";
+          batchContext.isContinuation = true;
         }
         const prompt = buildGenerationPrompt(batches[i], batchContext);
-        const code = await this.complete(prompt);
+        let code = await this.complete(prompt);
+        if (!isFirst) {
+          code = stripDuplicateImports(code);
+        }
         fullCode += (isFirst ? "" : "\n\n") + code;
       }
     } finally {
@@ -298,6 +302,36 @@ export class AIEngine {
   }
 }
 
+function stripDuplicateImports(code) {
+  const lines = code.split("\n");
+  let inMockBlock = false;
+  let braceDepth = 0;
+  const filtered = [];
+
+  for (const line of lines) {
+    const trimmed = line.trim();
+
+    if (trimmed.startsWith("import ")) continue;
+    if (trimmed.startsWith("jest.mock(")) {
+      inMockBlock = true;
+      braceDepth = 0;
+    }
+
+    if (inMockBlock) {
+      braceDepth += (line.match(/\{/g) || []).length;
+      braceDepth -= (line.match(/\}/g) || []).length;
+      if (braceDepth <= 0 && trimmed.endsWith(");")) {
+        inMockBlock = false;
+      }
+      continue;
+    }
+
+    filtered.push(line);
+  }
+
+  return filtered.join("\n").trim();
+}
+
 function resolveApiKey(config) {
   if (config.apiKey) return config.apiKey;
   if (config.apiKeyEnv) return process.env[config.apiKeyEnv] || "";

package/src/ai/prompt-builder.js

@@ -44,6 +44,11 @@ export function buildGenerationPrompt(testCases, context) {
     : "";
 
   const importPath = context.importPath || "../index";
+  const isContinuation = context.isContinuation || false;
+
+  const continuationRule = isContinuation
+    ? `\nIMPORTANT: This is a CONTINUATION batch. Do NOT include any import statements, require() calls, or jest.mock() calls. Write ONLY describe() and it() blocks. The imports and mocks are already defined in a previous batch.\n`
+    : "";
 
   return `You are a senior test engineer writing production-quality test code.
 
@@ -52,16 +57,15 @@ export function buildGenerationPrompt(testCases, context) {
 - Test Runner: ${context.runner || "jest"}
 - Language: JavaScript/JSX
 - Import path for module under test: "${importPath}"
-${sourceSection}${existingTestSection}
+${sourceSection}${existingTestSection}${continuationRule}
 ## Test Cases to Implement
 ${JSON.stringify(testCases, null, 2)}
 
 ## Rules
-1. Write a COMPLETE, RUNNABLE test file with all imports
+1. ${isContinuation ? "Write ONLY describe/it blocks — NO imports, NO jest.mock" : "Write a COMPLETE, RUNNABLE test file with all imports"}
 2. Use @testing-library/react for rendering components
 3. Use @testing-library/user-event for user interactions
-4. Use jest.mock() for module mocking
-5. Use screen queries: getByRole, getByText, getByTestId
+${isContinuation ? "" : "4. Use jest.mock() for module mocking\n"}5. Use screen queries: getByRole, getByText, getByTestId
 6. Use waitFor and findBy* for async assertions
 7. Each test must be independent — no shared mutable state
 8. Use descriptive it() names matching the test case names above

package/src/core/constants.js

@@ -1,4 +1,4 @@
-export const VERSION = "0.5.0";
+export const VERSION = "0.5.1";
 export const TOOL_NAME = "qabot";
 
 export const PROJECT_TYPES = [