@ai-dev-tools/csharp-copilot-core 0.0.37 → 0.0.38

package/out/gen/autoFix.d.ts

@@ -1,4 +1,4 @@
 import { GenResult } from "../types/genResult";
 export declare function autoFixTestCode(prompts: any[], sourceCodePath: string, sourceCode: string, dependency: string, testCode: string, testProjectPath: string, testFilePath: string, testFramework: string, isXapCode: boolean, isVsPlugin: boolean, testFileExist: boolean, keepFailedTests: boolean, insertedRanges?: [number, number][], retryTimes?: number, isBenchmark?: boolean, originalTestLength?: number): Promise<GenResult>;
-export declare function generateAutoFixCode(prompts: any[], sourceCodePath: string, sourceCode: string, dependency: string, generatedCode: string, errorMessage: string, isBuildFailure: boolean, testFilePath: string, testFramework: string, isXapCode: boolean, testFileExist: boolean): Promise<string>;
+export declare function generateAutoFixCode(prompts: any[], sourceCodePath: string, sourceCode: string, dependency: string, generatedCode: string, errorMessage: string, isBuildFailure: boolean, testFilePath: string, testFramework: string, isXapCode: boolean, testFileExist: boolean, buildResult?: any, testResult?: any): Promise<string>;
 export declare function autoRetryTestCode(prompts: any[], sourceCodePath: string, testCode: string, testProjectPath: string, testFilePath: string, testFramework: string, isXapCode: boolean, retryTimes?: number): Promise<GenResult>;

package/out/gen/autoFix.js

@@ -9,6 +9,7 @@ const preparePrompt_1 = require("../llm/preparePrompt");
 const verifyBuildErrorCausedFile_1 = require("../utils/verifyBuildErrorCausedFile");
 const verifyGeneratedCode_1 = require("../utils/verifyGeneratedCode");
 const removeFailedTestMethods_1 = require("../utils/removeFailedTestMethods");
+const moreUTAutoFixUtil_1 = require("../utils/moreUTAutoFixUtil");
 const writeGenCode_1 = require("../utils/writeGenCode");
 const fileUtils_1 = require("../utils/fileUtils");
 const writeFailedVerifyResult_1 = require("../utils/writeFailedVerifyResult");
@@ -92,7 +93,7 @@ async function autoFixTestCode(prompts, sourceCodePath, sourceCode, dependency,
                 console.error("!!!Verify process failed, retrying...");
                 continue;
             }
-            generatedCode = await generateAutoFixCode(prompts, sourceCodePath, sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode, testFileExist);
+            generatedCode = await generateAutoFixCode(prompts, sourceCodePath, sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode, testFileExist, buildResult, testResult);
         }
         // Try to remove failed test methods and return success if possible (only when keepFailedTests is false)
         if (!keepFailedTests) {
@@ -151,13 +152,17 @@ function verifyBuildErrorCausedLines(buildResult, insertedRanges) {
     }
     return true; // All errors are in inserted ranges
 }
-async function generateAutoFixCode(prompts, sourceCodePath, sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode, testFileExist) {
+async function generateAutoFixCode(prompts, sourceCodePath, sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode, testFileExist, buildResult, testResult) {
     console.log("start auto-fix process...");
     const autofixPrompt = testFileExist
         ? (0, preparePrompt_1.prepareMoreUtAutoFixPrompt)(sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, isXapCode)
         : (0, preparePrompt_1.prepareUtAutoFixPrompt)(prompts, generatedCode, errorMessage, isBuildFailure);
+    // Extract error test method names for the more UT auto-fix scenario
+    const errorMethodNames = testFileExist
+        ? (0, moreUTAutoFixUtil_1.extractErrorTestMethodNames)(generatedCode, isBuildFailure, buildResult, testResult)
+        : undefined;
     // get valid code from LLM response with retry
-    const llmResult = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, testFramework, isXapCode, false, testFilePath, true, 3, sourceCode);
+    const llmResult = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(autofixPrompt, testFramework, isXapCode, testFileExist, testFilePath, true, 3, sourceCode, errorMethodNames);
     const newGeneratedCode = llmResult?.testCode;
     // if generate code is valid, write test code to test project
     !!newGeneratedCode && (0, fileUtils_1.writeTestCode)(newGeneratedCode, testFilePath);

package/out/gen/autoFix.js.map

@@ -1 +1 @@
-{"version":3,"file":"autoFix.js","sourceRoot":"","sources":["../../src/gen/autoFix.ts"],"names":[],"mappings":";;AAYA,0CAsIC;AAoCD,kDA0BC;AAID,8CA+BC;AAlPD,gEAA6D;AAC7D,qEAAkE;AAClE,wDAA0F;AAC1F,oFAAiF;AACjF,sEAAmE;AACnE,8EAA2E;AAC3E,wDAAqD;AACrD,kDAAiE;AACjE,8EAA2E;AAC3E,wEAA4E;AAErE,KAAK,UAAU,eAAe,CACjC,OAAc,EACd,cAAsB,EACtB,UAAkB,EAClB,UAAkB,EAClB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,UAAmB,EACnB,aAAsB,EACtB,eAAwB,EACxB,iBAAqC,SAAS,EAC9C,aAAqB,CAAC,EACtB,cAAuB,KAAK,EAC5B,qBAA6B,CAAC;IAE9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,UAAU,CAAC;IACf,IAAI,WAAW,CAAC;IAChB,IAAI,YAAY,CAAC;IACjB,IAAI,cAAc,GAAG,IAAI,CAAC;IAC1B,IAAI,CAAC;QACD,MAAM,qBAAqB,GAAG,IAAA,kDAA2B,EAAC,cAAc,CAAC,CAAC;QAC1E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,2DAA2D;YAC3D,CAAC,UAAU,IAAI,IAAA,2BAAY,EAAC,cAAc,EAAE,aAAa,EAAE,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAErF,6EAA6E;YAC7E,IAAI,CAAC,aAAa,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,qEAAqE,aAAa,EAAE,CAAC,CAAC;gBACpG,MAAM,SAAS,GAAG,MAAM,IAAA,+CAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;gBACrI,aAAa,GAAG,SAAS,EAAE,QAAQ,CAAC;gBACpC,cAAc,GAAG,SAAS,EAAE,cAAc,CAAC;gBAC3C,SAAS;YACb,CAAC;YAED,wFAAwF;YACxF,MAAM,YAAY,GAAG,MAAM,IAAA,iCAAe,EAAC,YAAY,CAAC,CAAC;YACzD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBACvB,4EAA4E;gBAC5E,gHAAgH;gBAChH,MAAM,YAAY,GAAG,MAAM,IAAA,yCAAmB,EAAC,aAAa,EAAE,eAAe,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;gBACzG,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;gBACrC,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,cAAc,GAAG,CAAC,UAAU,CAAC;gBAE7B,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,gEAAgE;oBAChE,MAAM,WAAW,GAAG,IAAA,wBAAY,EAAC,YAAY,CAAC,CAAC;oBAC/C,IAAI,kBAAkB,GAAG,CAAC,IAAI,WAAW,IAAI,WAAW,CAAC,MAAM,IAAI,kBAAkB,EAAE,CAAC;wBACpF,OAAO,CAAC,KAAK,CAAC,2FAA2F,CAAC,CAAC;oBAC/G,CAAC;yBAAM,CAAC;wBACJ,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;wBACjD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;oBACnH,CAAC;gBACL,CAAC;gBAED,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,YAAY,CAAC,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACJ,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,EAAE,YAAY,CAAC,CAAC;YACxF,CAAC;YAED,uFAAuF;YACvF,MAAM,eAAe,GAAG,IAAA,uDAA0B,EAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,kFAAkF,CAAC,CAAC;gBAClG,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;YACpH,CAAC;YAED,qFAAqF;YACrF,0FAA0F;YAC1F,IAAI,CAAC,KAAK,CAAC,IAAI,cAAc,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzD,MAAM,sBAAsB,GAAG,2BAA2B,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;gBACxF,IAAI,CAAC,sBAAsB,EAAE,CAAC;oBAC1B,OAAO,CAAC,KAAK,CAAC,wFAAwF,CAAC,CAAC;oBACxG,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACpH,CAAC;YACL,CAAC;YAED,CAAC,UAAU,IAAI,IAAA,iDAAuB,EAAC,cAAc,EAAE,CAAC,EAAE,qBAAqB,EAAE,YAAY,EAAE,CAAC,UAAU,CAAC,CAAC;YAC5G,yDAAyD;YACzD,mDAAmD;YACnD,4DAA4D;YAC5D,gEAAgE;YAChE,yDAAyD;YACzD,IAAI;YAEJ,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACnB,OAAO,CAAC,GAAG,CAAC,wBAAwB,UAAU,sDAAsD,CAAC,CAAC;gBACtG,MAAM;YACV,CAAC;YAED,wDAAwD;YACxD,IAAI,CAAC,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9B,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBACvD,SAAS;YACb,CAAC;YAED,aAAa,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QACnM,CAAC;QAED,wGAAwG;QACxG,IAAI,CAAC,eAAe,EAAE,CAAC;YACnB,MAAM,YAAY,GAAG,MAAM,IAAA,iDAAuB,EAAC,aAAa,EAAE,YAAY,EAAE,eAAe,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,kBAAkB,CAAC,CAAC;YAC1K,IAAI,YAAY,EAAE,CAAC;gBACf,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,yDAAyD;oBACzD,MAAM,aAAa,GAAG,IAAA,wBAAY,EAAC,YAAY,CAAC,CAAC;oBACjD,MAAM,cAAc,GAAG,oCAAoC,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC;oBACtF,IAAI,CAAC,cAAc,EAAE,CAAC;wBAClB,OAAO,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;wBACxE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;oBACpH,CAAC;gBACL,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,8BAA8B,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC;gBAC5F,OAAO,YAAY,CAAC;YACxB,CAAC;QACL,CAAC;QAED,OAAO,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;QACxE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;IACpH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAChI,CAAC;YAAS,CAAC;QACP,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IAC5E,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,SAAS,2BAA2B,CAAC,WAAgB,EAAE,cAAkC;IACrF,IAAI,CAAC,WAAW,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzE,OAAO,IAAI,CAAC,CAAC,kBAAkB;IACnC,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;QACrC,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;QAC7B,IAAI,CAAC,SAAS;YAAE,SAAS;QAEzB,IAAI,iBAAiB,GAAG,KAAK,CAAC;QAC9B,KAAK,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,cAAc,EAAE,CAAC;YACxC,IAAI,SAAS,IAAI,KAAK,IAAI,SAAS,IAAI,GAAG,EAAE,CAAC;gBACzC,iBAAiB,GAAG,IAAI,CAAC;gBACzB,MAAM;YACV,CAAC;QACL,CAAC;QAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrB,uDAAuD;YACvD,OAAO,CAAC,KAAK,CAAC,uBAAuB,SAAS,kDAAkD,CAAC,CAAC;YAClG,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IAED,OAAO,IAAI,CAAC,CAAC,oCAAoC;AACrD,CAAC;AAGM,KAAK,UAAU,mBAAmB,CACrC,OAAc,EACd,cAAsB,EACtB,UAAkB,EAClB,UAAkB,EAClB,aAAqB,EACrB,YAAoB,EACpB,cAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,aAAsB;IAEtB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IACzC,MAAM,aAAa,GAAG,aAAa;QAC/B,CAAC,CAAC,IAAA,0CAA0B,EAAC,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,SAAS,CAAC;QAC5G,CAAC,CAAC,IAAA,sCAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,CAAC,CAAC;IAEnF,8CAA8C;IAC9C,MAAM,SAAS,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;IAClI,MAAM,gBAAgB,GAAG,SAAS,EAAE,QAAQ,CAAC;IAE7C,6DAA6D;IAC7D,CAAC,CAAC,gBAAgB,IAAI,IAAA,yBAAa,EAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;IAEpE,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAGD,yBAAyB;AAClB,KAAK,UAAU,iBAAiB,CACnC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,aAAqB,CAAC;IAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,SAAS,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,uCAAuC,UAAU,YAAY,CAAC,CAAC;YAC7E,MAAM;QACV,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC;QAC9B,mCAAmC;QACnC,MAAM,SAAS,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;QACnH,aAAa,GAAG,SAAS,EAAE,QAAQ,CAAC;QACpC,kCAAkC;QAClC,IAAA,yBAAa,EAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IAExE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AAClJ,CAAC","sourcesContent":["import { GenResult } from \"../types/genResult\";\r\nimport { checkCodeSyntax } from \"../analyze/checkCodeSyntax\";\r\nimport { ensureValidLLMResponse } from \"./ensureValidLLMResponse\";\r\nimport { prepareUtAutoFixPrompt, prepareMoreUtAutoFixPrompt } from \"../llm/preparePrompt\";\r\nimport { verifyBuildErrorCausedFile } from \"../utils/verifyBuildErrorCausedFile\";\r\nimport { verifyGeneratedCode } from \"../utils/verifyGeneratedCode\";\r\nimport { removeFailedTestMethods } from \"../utils/removeFailedTestMethods\";\r\nimport { writeGenCode } from \"../utils/writeGenCode\";\r\nimport { writeTestCode, readTestCode } from \"../utils/fileUtils\";\r\nimport { writeFailedVerifyResult } from \"../utils/writeFailedVerifyResult\";\r\nimport { getSourceFileCsprojFilePath } from \"../utils/getCodeStructurePath\";\r\n\r\nexport async function autoFixTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    sourceCode: string,\r\n    dependency: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    isVsPlugin: boolean,\r\n    testFileExist: boolean,\r\n    keepFailedTests: boolean,\r\n    insertedRanges: [number, number][] = undefined,\r\n    retryTimes: number = 3,\r\n    isBenchmark: boolean = false,\r\n    originalTestLength: number = 0\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let testResult;\r\n    let buildResult;\r\n    let errorMessage;\r\n    let isBuildFailure = true;\r\n    try {\r\n        const sourceCodeProjectPath = getSourceFileCsprojFilePath(sourceCodePath);\r\n        for (let i = 0; i <= retryTimes; i++) {\r\n            // write generated code, to review and improve code quality\r\n            !isVsPlugin && writeGenCode(sourceCodePath, generatedCode, i, sourceCodeProjectPath);\r\n\r\n            // for case: generatedCode is invalid, just retry instead of auto fix process\r\n            if (!generatedCode) {\r\n                console.error(`Generated code is invalid, try to regenerate. \\nGenerated code: \\n${generatedCode}`);\r\n                const llmResult = await ensureValidLLMResponse(prompts, testFramework, isXapCode, testFileExist, testFilePath, false, 3, sourceCode);\r\n                generatedCode = llmResult?.testCode;\r\n                insertedRanges = llmResult?.insertedRanges;\r\n                continue;\r\n            }\r\n\r\n            // analyze generated code issue before verify: build, because build will cost much time.\r\n            const syntaxResult = await checkCodeSyntax(testFilePath);\r\n            if (syntaxResult.success) {\r\n                // if generated code passed the syntax check, verify the generated test code\r\n                // const { success, message, buildErrors, vsTestResult } = await verifyTestCode(generatedCode, testProjectPath);\r\n                const verifyResult = await verifyGeneratedCode(generatedCode, testProjectPath, testFilePath, isVsPlugin);\r\n                errorMessage = verifyResult.message;\r\n                testResult = verifyResult.testResult;\r\n                buildResult = verifyResult.buildErrors;\r\n                isBuildFailure = !testResult;\r\n\r\n                if (verifyResult.success) {\r\n                    // Check if new code length is greater than original test length\r\n                    const currentCode = readTestCode(testFilePath);\r\n                    if (originalTestLength > 0 && currentCode && currentCode.length <= originalTestLength) {\r\n                        console.error(\"Generated code length is not greater than original test code length, treating as failure.\");\r\n                    } else {\r\n                        console.log(\"XapUt code generated successfully\");\r\n                        return { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                    }\r\n                }\r\n\r\n                console.error(\"Failed to verifyTestCode, error:\", errorMessage);\r\n            } else {\r\n                buildResult = syntaxResult.buildErrors;\r\n                errorMessage = syntaxResult.message;\r\n                console.error(\"Failed to check the generated UT code syntax, error:\", errorMessage);\r\n            }\r\n\r\n            // if errors caused by other code files instead of generate test code, abort generation\r\n            const validBuildError = verifyBuildErrorCausedFile(testFilePath, buildResult);\r\n            if (!validBuildError) {\r\n                console.error(\"Due to errors caused by other code files, abort generation, please fix it first.\");\r\n                return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n            }\r\n\r\n            // if errors are caused by original test lines (not inserted lines), abort generation\r\n            // Only check on first iteration since insertedRanges becomes stale after code is modified\r\n            if (i === 0 && insertedRanges && insertedRanges.length > 0) {\r\n                const isErrorInInsertedLines = verifyBuildErrorCausedLines(buildResult, insertedRanges);\r\n                if (!isErrorInInsertedLines) {\r\n                    console.error(\"Errors are caused by original test code lines, not by inserted code. Abort generation.\");\r\n                    return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n            }\r\n\r\n            !isVsPlugin && writeFailedVerifyResult(sourceCodePath, i, sourceCodeProjectPath, errorMessage, !testResult);\r\n            // keep test failed tast cases in test file automatically\r\n            // if (!isBuildFailure && vsTestRes.passed !== 0) {\r\n            //     console.log(\"remove failed test cases in test file\");\r\n            //     removeFailedTestCasesInTestFile(vsTestRes, testCodePath);\r\n            //     return { success: true, testCode: generatedCode };\r\n            // }\r\n\r\n            if (i === retryTimes) {\r\n                console.log(`Autofix failed after ${retryTimes} attempts, will try to remove failed test methods...`);\r\n                break;\r\n            }\r\n\r\n            // case: vs plugin no response or msbuild/vstest failed.\r\n            if (!buildResult && !testResult) {\r\n                console.error(\"!!!Verify process failed, retrying...\");\r\n                continue;\r\n            }\r\n\r\n            generatedCode = await generateAutoFixCode(prompts, sourceCodePath, sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode, testFileExist);\r\n        }\r\n\r\n        // Try to remove failed test methods and return success if possible (only when keepFailedTests is false)\r\n        if (!keepFailedTests) {\r\n            const removeResult = await removeFailedTestMethods(generatedCode, testFilePath, testProjectPath, sourceCodePath, buildResult, testResult, isVsPlugin, originalTestLength);\r\n            if (removeResult) {\r\n                if (removeResult.success) {\r\n                    // Check if there are still test methods left in the code\r\n                    const remainingCode = readTestCode(testFilePath);\r\n                    const hasTestMethods = /\\[(TestMethod|Test|Fact|Theory)\\]/i.test(remainingCode || '');\r\n                    if (!hasTestMethods) {\r\n                        console.error(\"No test methods remaining after removing failed tests.\");\r\n                        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                    }\r\n                }\r\n                console.log(`Remove failed test methods ${removeResult.success ? 'succeeded' : 'failed'}.`);\r\n                return removeResult;\r\n            }\r\n        }\r\n\r\n        console.error(\"Failed to generate valid test code after all attempts.\");\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n    } catch (error) {\r\n        console.error(\"Error during auto-fix process:\", error);\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    } finally {\r\n        console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n    }\r\n}\r\n\r\n/**\r\n * Verify if build errors are caused by the inserted lines.\r\n * Returns true if all build errors are within the inserted ranges (meaning they are caused by new code).\r\n\r\n * Returns false if any build error is outside the inserted ranges (meaning it's caused by original code).\r\n */\r\nfunction verifyBuildErrorCausedLines(buildResult: any, insertedRanges: [number, number][]): boolean {\r\n    if (!buildResult || !buildResult.errors || buildResult.errors.length === 0) {\r\n        return true; // No build errors\r\n    }\r\n\r\n    for (const error of buildResult.errors) {\r\n        const errorLine = error.line;\r\n        if (!errorLine) continue;\r\n\r\n        let isInInsertedRange = false;\r\n        for (const [start, end] of insertedRanges) {\r\n            if (errorLine >= start && errorLine <= end) {\r\n                isInInsertedRange = true;\r\n                break;\r\n            }\r\n        }\r\n\r\n        if (!isInInsertedRange) {\r\n            // This error is in original code, not in inserted code\r\n            console.error(`Build error at line ${errorLine} is in original test code, not in inserted code.`);\r\n            return false;\r\n        }\r\n    }\r\n\r\n    return true; // All errors are in inserted ranges\r\n}\r\n\r\n\r\nexport async function generateAutoFixCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    sourceCode: string,\r\n    dependency: string,\r\n    generatedCode: string,\r\n    errorMessage: string,\r\n    isBuildFailure: boolean,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    testFileExist: boolean\r\n): Promise<string> {\r\n    console.log(\"start auto-fix process...\");\r\n    const autofixPrompt = testFileExist \r\n        ? prepareMoreUtAutoFixPrompt(sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, isXapCode)\r\n        : prepareUtAutoFixPrompt(prompts, generatedCode, errorMessage, isBuildFailure);\r\n\r\n    // get valid code from LLM response with retry\r\n    const llmResult = await ensureValidLLMResponse(autofixPrompt, testFramework, isXapCode, false, testFilePath, true, 3, sourceCode);\r\n    const newGeneratedCode = llmResult?.testCode;\r\n\r\n    // if generate code is valid, write test code to test project\r\n    !!newGeneratedCode && writeTestCode(newGeneratedCode, testFilePath);\r\n\r\n    return newGeneratedCode;\r\n}\r\n\r\n\r\n// use for quick dev test\r\nexport async function autoRetryTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    retryTimes: number = 3\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let vsTestRes;\r\n    for (let i = 0; i <= retryTimes; i++) {\r\n        if (i === retryTimes) {\r\n            console.error(`Failed to generate XapUt code after ${retryTimes} attempts.`);\r\n            break;\r\n        }\r\n\r\n        const autofixPrompt = prompts;\r\n        // get valid code from LLM response\r\n        const llmResult = await ensureValidLLMResponse(autofixPrompt, testFramework, isXapCode, false, testFilePath, true);\r\n        generatedCode = llmResult?.testCode;\r\n        // write test code to test project\r\n        writeTestCode(generatedCode, testFilePath);\r\n    }\r\n\r\n    console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n    return { success: true, testFilePath: testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: vsTestRes };\r\n}"]}
+{"version":3,"file":"autoFix.js","sourceRoot":"","sources":["../../src/gen/autoFix.ts"],"names":[],"mappings":";;AAaA,0CAsIC;AAoCD,kDAiCC;AAID,8CA+BC;AA1PD,gEAA6D;AAC7D,qEAAkE;AAClE,wDAA0F;AAC1F,oFAAiF;AACjF,sEAAmE;AACnE,8EAA2E;AAC3E,kEAAyE;AACzE,wDAAqD;AACrD,kDAAiE;AACjE,8EAA2E;AAC3E,wEAA4E;AAErE,KAAK,UAAU,eAAe,CACjC,OAAc,EACd,cAAsB,EACtB,UAAkB,EAClB,UAAkB,EAClB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,UAAmB,EACnB,aAAsB,EACtB,eAAwB,EACxB,iBAAqC,SAAS,EAC9C,aAAqB,CAAC,EACtB,cAAuB,KAAK,EAC5B,qBAA6B,CAAC;IAE9B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,UAAU,CAAC;IACf,IAAI,WAAW,CAAC;IAChB,IAAI,YAAY,CAAC;IACjB,IAAI,cAAc,GAAG,IAAI,CAAC;IAC1B,IAAI,CAAC;QACD,MAAM,qBAAqB,GAAG,IAAA,kDAA2B,EAAC,cAAc,CAAC,CAAC;QAC1E,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;YACnC,2DAA2D;YAC3D,CAAC,UAAU,IAAI,IAAA,2BAAY,EAAC,cAAc,EAAE,aAAa,EAAE,CAAC,EAAE,qBAAqB,CAAC,CAAC;YAErF,6EAA6E;YAC7E,IAAI,CAAC,aAAa,EAAE,CAAC;gBACjB,OAAO,CAAC,KAAK,CAAC,qEAAqE,aAAa,EAAE,CAAC,CAAC;gBACpG,MAAM,SAAS,GAAG,MAAM,IAAA,+CAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,EAAE,UAAU,CAAC,CAAC;gBACrI,aAAa,GAAG,SAAS,EAAE,QAAQ,CAAC;gBACpC,cAAc,GAAG,SAAS,EAAE,cAAc,CAAC;gBAC3C,SAAS;YACb,CAAC;YAED,wFAAwF;YACxF,MAAM,YAAY,GAAG,MAAM,IAAA,iCAAe,EAAC,YAAY,CAAC,CAAC;YACzD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;gBACvB,4EAA4E;gBAC5E,gHAAgH;gBAChH,MAAM,YAAY,GAAG,MAAM,IAAA,yCAAmB,EAAC,aAAa,EAAE,eAAe,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;gBACzG,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,UAAU,GAAG,YAAY,CAAC,UAAU,CAAC;gBACrC,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,cAAc,GAAG,CAAC,UAAU,CAAC;gBAE7B,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,gEAAgE;oBAChE,MAAM,WAAW,GAAG,IAAA,wBAAY,EAAC,YAAY,CAAC,CAAC;oBAC/C,IAAI,kBAAkB,GAAG,CAAC,IAAI,WAAW,IAAI,WAAW,CAAC,MAAM,IAAI,kBAAkB,EAAE,CAAC;wBACpF,OAAO,CAAC,KAAK,CAAC,2FAA2F,CAAC,CAAC;oBAC/G,CAAC;yBAAM,CAAC;wBACJ,OAAO,CAAC,GAAG,CAAC,mCAAmC,CAAC,CAAC;wBACjD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;oBACnH,CAAC;gBACL,CAAC;gBAED,OAAO,CAAC,KAAK,CAAC,kCAAkC,EAAE,YAAY,CAAC,CAAC;YACpE,CAAC;iBAAM,CAAC;gBACJ,WAAW,GAAG,YAAY,CAAC,WAAW,CAAC;gBACvC,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;gBACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,EAAE,YAAY,CAAC,CAAC;YACxF,CAAC;YAED,uFAAuF;YACvF,MAAM,eAAe,GAAG,IAAA,uDAA0B,EAAC,YAAY,EAAE,WAAW,CAAC,CAAC;YAC9E,IAAI,CAAC,eAAe,EAAE,CAAC;gBACnB,OAAO,CAAC,KAAK,CAAC,kFAAkF,CAAC,CAAC;gBAClG,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;YACpH,CAAC;YAED,qFAAqF;YACrF,0FAA0F;YAC1F,IAAI,CAAC,KAAK,CAAC,IAAI,cAAc,IAAI,cAAc,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACzD,MAAM,sBAAsB,GAAG,2BAA2B,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;gBACxF,IAAI,CAAC,sBAAsB,EAAE,CAAC;oBAC1B,OAAO,CAAC,KAAK,CAAC,wFAAwF,CAAC,CAAC;oBACxG,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;gBACpH,CAAC;YACL,CAAC;YAED,CAAC,UAAU,IAAI,IAAA,iDAAuB,EAAC,cAAc,EAAE,CAAC,EAAE,qBAAqB,EAAE,YAAY,EAAE,CAAC,UAAU,CAAC,CAAC;YAC5G,yDAAyD;YACzD,mDAAmD;YACnD,4DAA4D;YAC5D,gEAAgE;YAChE,yDAAyD;YACzD,IAAI;YAEJ,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;gBACnB,OAAO,CAAC,GAAG,CAAC,wBAAwB,UAAU,sDAAsD,CAAC,CAAC;gBACtG,MAAM;YACV,CAAC;YAED,wDAAwD;YACxD,IAAI,CAAC,WAAW,IAAI,CAAC,UAAU,EAAE,CAAC;gBAC9B,OAAO,CAAC,KAAK,CAAC,uCAAuC,CAAC,CAAC;gBACvD,SAAS;YACb,CAAC;YAED,aAAa,GAAG,MAAM,mBAAmB,CAAC,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,YAAY,EAAE,aAAa,EAAE,SAAS,EAAE,aAAa,EAAE,WAAW,EAAE,UAAU,CAAC,CAAC;QAC5N,CAAC;QAED,wGAAwG;QACxG,IAAI,CAAC,eAAe,EAAE,CAAC;YACnB,MAAM,YAAY,GAAG,MAAM,IAAA,iDAAuB,EAAC,aAAa,EAAE,YAAY,EAAE,eAAe,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,UAAU,EAAE,kBAAkB,CAAC,CAAC;YAC1K,IAAI,YAAY,EAAE,CAAC;gBACf,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,yDAAyD;oBACzD,MAAM,aAAa,GAAG,IAAA,wBAAY,EAAC,YAAY,CAAC,CAAC;oBACjD,MAAM,cAAc,GAAG,oCAAoC,CAAC,IAAI,CAAC,aAAa,IAAI,EAAE,CAAC,CAAC;oBACtF,IAAI,CAAC,cAAc,EAAE,CAAC;wBAClB,OAAO,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;wBACxE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;oBACpH,CAAC;gBACL,CAAC;gBACD,OAAO,CAAC,GAAG,CAAC,8BAA8B,YAAY,CAAC,OAAO,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,QAAQ,GAAG,CAAC,CAAC;gBAC5F,OAAO,YAAY,CAAC;YACxB,CAAC;QACL,CAAC;QAED,OAAO,CAAC,KAAK,CAAC,wDAAwD,CAAC,CAAC;QACxE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,EAAE,CAAC;IACpH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACb,OAAO,CAAC,KAAK,CAAC,gCAAgC,EAAE,KAAK,CAAC,CAAC;QACvD,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IAChI,CAAC;YAAS,CAAC;QACP,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IAC5E,CAAC;AACL,CAAC;AAED;;;;;GAKG;AACH,SAAS,2BAA2B,CAAC,WAAgB,EAAE,cAAkC;IACrF,IAAI,CAAC,WAAW,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,WAAW,CAAC,MAAM,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QACzE,OAAO,IAAI,CAAC,CAAC,kBAAkB;IACnC,CAAC;IAED,KAAK,MAAM,KAAK,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;QACrC,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,CAAC;QAC7B,IAAI,CAAC,SAAS;YAAE,SAAS;QAEzB,IAAI,iBAAiB,GAAG,KAAK,CAAC;QAC9B,KAAK,MAAM,CAAC,KAAK,EAAE,GAAG,CAAC,IAAI,cAAc,EAAE,CAAC;YACxC,IAAI,SAAS,IAAI,KAAK,IAAI,SAAS,IAAI,GAAG,EAAE,CAAC;gBACzC,iBAAiB,GAAG,IAAI,CAAC;gBACzB,MAAM;YACV,CAAC;QACL,CAAC;QAED,IAAI,CAAC,iBAAiB,EAAE,CAAC;YACrB,uDAAuD;YACvD,OAAO,CAAC,KAAK,CAAC,uBAAuB,SAAS,kDAAkD,CAAC,CAAC;YAClG,OAAO,KAAK,CAAC;QACjB,CAAC;IACL,CAAC;IAED,OAAO,IAAI,CAAC,CAAC,oCAAoC;AACrD,CAAC;AAGM,KAAK,UAAU,mBAAmB,CACrC,OAAc,EACd,cAAsB,EACtB,UAAkB,EAClB,UAAkB,EAClB,aAAqB,EACrB,YAAoB,EACpB,cAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,aAAsB,EACtB,WAAiB,EACjB,UAAgB;IAEhB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;IACzC,MAAM,aAAa,GAAG,aAAa;QAC/B,CAAC,CAAC,IAAA,0CAA0B,EAAC,UAAU,EAAE,UAAU,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,EAAE,SAAS,CAAC;QAC5G,CAAC,CAAC,IAAA,sCAAsB,EAAC,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,cAAc,CAAC,CAAC;IAEnF,oEAAoE;IACpE,MAAM,gBAAgB,GAAG,aAAa;QAClC,CAAC,CAAC,IAAA,+CAA2B,EAAC,aAAa,EAAE,cAAc,EAAE,WAAW,EAAE,UAAU,CAAC;QACrF,CAAC,CAAC,SAAS,CAAC;IAEhB,8CAA8C;IAC9C,MAAM,SAAS,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,aAAa,EAAE,YAAY,EAAE,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,gBAAgB,CAAC,CAAC;IAC5J,MAAM,gBAAgB,GAAG,SAAS,EAAE,QAAQ,CAAC;IAE7C,6DAA6D;IAC7D,CAAC,CAAC,gBAAgB,IAAI,IAAA,yBAAa,EAAC,gBAAgB,EAAE,YAAY,CAAC,CAAC;IAEpE,OAAO,gBAAgB,CAAC;AAC5B,CAAC;AAGD,yBAAyB;AAClB,KAAK,UAAU,iBAAiB,CACnC,OAAc,EACd,cAAsB,EACtB,QAAgB,EAChB,eAAuB,EACvB,YAAoB,EACpB,aAAqB,EACrB,SAAkB,EAClB,aAAqB,CAAC;IAEtB,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,IAAI,aAAa,GAAG,QAAQ,CAAC;IAC7B,IAAI,SAAS,CAAC;IACd,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,IAAI,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QACnC,IAAI,CAAC,KAAK,UAAU,EAAE,CAAC;YACnB,OAAO,CAAC,KAAK,CAAC,uCAAuC,UAAU,YAAY,CAAC,CAAC;YAC7E,MAAM;QACV,CAAC;QAED,MAAM,aAAa,GAAG,OAAO,CAAC;QAC9B,mCAAmC;QACnC,MAAM,SAAS,GAAG,MAAM,IAAA,+CAAsB,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,IAAI,CAAC,CAAC;QACnH,aAAa,GAAG,SAAS,EAAE,QAAQ,CAAC;QACpC,kCAAkC;QAClC,IAAA,yBAAa,EAAC,aAAa,EAAE,YAAY,CAAC,CAAC;IAC/C,CAAC;IAED,OAAO,CAAC,GAAG,CAAC,iBAAiB,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IAExE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;AAClJ,CAAC","sourcesContent":["import { GenResult } from \"../types/genResult\";\r\nimport { checkCodeSyntax } from \"../analyze/checkCodeSyntax\";\r\nimport { ensureValidLLMResponse } from \"./ensureValidLLMResponse\";\r\nimport { prepareUtAutoFixPrompt, prepareMoreUtAutoFixPrompt } from \"../llm/preparePrompt\";\r\nimport { verifyBuildErrorCausedFile } from \"../utils/verifyBuildErrorCausedFile\";\r\nimport { verifyGeneratedCode } from \"../utils/verifyGeneratedCode\";\r\nimport { removeFailedTestMethods } from \"../utils/removeFailedTestMethods\";\r\nimport { extractErrorTestMethodNames } from \"../utils/moreUTAutoFixUtil\";\r\nimport { writeGenCode } from \"../utils/writeGenCode\";\r\nimport { writeTestCode, readTestCode } from \"../utils/fileUtils\";\r\nimport { writeFailedVerifyResult } from \"../utils/writeFailedVerifyResult\";\r\nimport { getSourceFileCsprojFilePath } from \"../utils/getCodeStructurePath\";\r\n\r\nexport async function autoFixTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    sourceCode: string,\r\n    dependency: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    isVsPlugin: boolean,\r\n    testFileExist: boolean,\r\n    keepFailedTests: boolean,\r\n    insertedRanges: [number, number][] = undefined,\r\n    retryTimes: number = 3,\r\n    isBenchmark: boolean = false,\r\n    originalTestLength: number = 0\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let testResult;\r\n    let buildResult;\r\n    let errorMessage;\r\n    let isBuildFailure = true;\r\n    try {\r\n        const sourceCodeProjectPath = getSourceFileCsprojFilePath(sourceCodePath);\r\n        for (let i = 0; i <= retryTimes; i++) {\r\n            // write generated code, to review and improve code quality\r\n            !isVsPlugin && writeGenCode(sourceCodePath, generatedCode, i, sourceCodeProjectPath);\r\n\r\n            // for case: generatedCode is invalid, just retry instead of auto fix process\r\n            if (!generatedCode) {\r\n                console.error(`Generated code is invalid, try to regenerate. \\nGenerated code: \\n${generatedCode}`);\r\n                const llmResult = await ensureValidLLMResponse(prompts, testFramework, isXapCode, testFileExist, testFilePath, false, 3, sourceCode);\r\n                generatedCode = llmResult?.testCode;\r\n                insertedRanges = llmResult?.insertedRanges;\r\n                continue;\r\n            }\r\n\r\n            // analyze generated code issue before verify: build, because build will cost much time.\r\n            const syntaxResult = await checkCodeSyntax(testFilePath);\r\n            if (syntaxResult.success) {\r\n                // if generated code passed the syntax check, verify the generated test code\r\n                // const { success, message, buildErrors, vsTestResult } = await verifyTestCode(generatedCode, testProjectPath);\r\n                const verifyResult = await verifyGeneratedCode(generatedCode, testProjectPath, testFilePath, isVsPlugin);\r\n                errorMessage = verifyResult.message;\r\n                testResult = verifyResult.testResult;\r\n                buildResult = verifyResult.buildErrors;\r\n                isBuildFailure = !testResult;\r\n\r\n                if (verifyResult.success) {\r\n                    // Check if new code length is greater than original test length\r\n                    const currentCode = readTestCode(testFilePath);\r\n                    if (originalTestLength > 0 && currentCode && currentCode.length <= originalTestLength) {\r\n                        console.error(\"Generated code length is not greater than original test code length, treating as failure.\");\r\n                    } else {\r\n                        console.log(\"XapUt code generated successfully\");\r\n                        return { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                    }\r\n                }\r\n\r\n                console.error(\"Failed to verifyTestCode, error:\", errorMessage);\r\n            } else {\r\n                buildResult = syntaxResult.buildErrors;\r\n                errorMessage = syntaxResult.message;\r\n                console.error(\"Failed to check the generated UT code syntax, error:\", errorMessage);\r\n            }\r\n\r\n            // if errors caused by other code files instead of generate test code, abort generation\r\n            const validBuildError = verifyBuildErrorCausedFile(testFilePath, buildResult);\r\n            if (!validBuildError) {\r\n                console.error(\"Due to errors caused by other code files, abort generation, please fix it first.\");\r\n                return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n            }\r\n\r\n            // if errors are caused by original test lines (not inserted lines), abort generation\r\n            // Only check on first iteration since insertedRanges becomes stale after code is modified\r\n            if (i === 0 && insertedRanges && insertedRanges.length > 0) {\r\n                const isErrorInInsertedLines = verifyBuildErrorCausedLines(buildResult, insertedRanges);\r\n                if (!isErrorInInsertedLines) {\r\n                    console.error(\"Errors are caused by original test code lines, not by inserted code. Abort generation.\");\r\n                    return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                }\r\n            }\r\n\r\n            !isVsPlugin && writeFailedVerifyResult(sourceCodePath, i, sourceCodeProjectPath, errorMessage, !testResult);\r\n            // keep test failed tast cases in test file automatically\r\n            // if (!isBuildFailure && vsTestRes.passed !== 0) {\r\n            //     console.log(\"remove failed test cases in test file\");\r\n            //     removeFailedTestCasesInTestFile(vsTestRes, testCodePath);\r\n            //     return { success: true, testCode: generatedCode };\r\n            // }\r\n\r\n            if (i === retryTimes) {\r\n                console.log(`Autofix failed after ${retryTimes} attempts, will try to remove failed test methods...`);\r\n                break;\r\n            }\r\n\r\n            // case: vs plugin no response or msbuild/vstest failed.\r\n            if (!buildResult && !testResult) {\r\n                console.error(\"!!!Verify process failed, retrying...\");\r\n                continue;\r\n            }\r\n\r\n            generatedCode = await generateAutoFixCode(prompts, sourceCodePath, sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, testFilePath, testFramework, isXapCode, testFileExist, buildResult, testResult);\r\n        }\r\n\r\n        // Try to remove failed test methods and return success if possible (only when keepFailedTests is false)\r\n        if (!keepFailedTests) {\r\n            const removeResult = await removeFailedTestMethods(generatedCode, testFilePath, testProjectPath, sourceCodePath, buildResult, testResult, isVsPlugin, originalTestLength);\r\n            if (removeResult) {\r\n                if (removeResult.success) {\r\n                    // Check if there are still test methods left in the code\r\n                    const remainingCode = readTestCode(testFilePath);\r\n                    const hasTestMethods = /\\[(TestMethod|Test|Fact|Theory)\\]/i.test(remainingCode || '');\r\n                    if (!hasTestMethods) {\r\n                        console.error(\"No test methods remaining after removing failed tests.\");\r\n                        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n                    }\r\n                }\r\n                console.log(`Remove failed test methods ${removeResult.success ? 'succeeded' : 'failed'}.`);\r\n                return removeResult;\r\n            }\r\n        }\r\n\r\n        console.error(\"Failed to generate valid test code after all attempts.\");\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };\r\n    } catch (error) {\r\n        console.error(\"Error during auto-fix process:\", error);\r\n        return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    } finally {\r\n        console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n    }\r\n}\r\n\r\n/**\r\n * Verify if build errors are caused by the inserted lines.\r\n * Returns true if all build errors are within the inserted ranges (meaning they are caused by new code).\r\n\r\n * Returns false if any build error is outside the inserted ranges (meaning it's caused by original code).\r\n */\r\nfunction verifyBuildErrorCausedLines(buildResult: any, insertedRanges: [number, number][]): boolean {\r\n    if (!buildResult || !buildResult.errors || buildResult.errors.length === 0) {\r\n        return true; // No build errors\r\n    }\r\n\r\n    for (const error of buildResult.errors) {\r\n        const errorLine = error.line;\r\n        if (!errorLine) continue;\r\n\r\n        let isInInsertedRange = false;\r\n        for (const [start, end] of insertedRanges) {\r\n            if (errorLine >= start && errorLine <= end) {\r\n                isInInsertedRange = true;\r\n                break;\r\n            }\r\n        }\r\n\r\n        if (!isInInsertedRange) {\r\n            // This error is in original code, not in inserted code\r\n            console.error(`Build error at line ${errorLine} is in original test code, not in inserted code.`);\r\n            return false;\r\n        }\r\n    }\r\n\r\n    return true; // All errors are in inserted ranges\r\n}\r\n\r\n\r\nexport async function generateAutoFixCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    sourceCode: string,\r\n    dependency: string,\r\n    generatedCode: string,\r\n    errorMessage: string,\r\n    isBuildFailure: boolean,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    testFileExist: boolean,\r\n    buildResult?: any,\r\n    testResult?: any\r\n): Promise<string> {\r\n    console.log(\"start auto-fix process...\");\r\n    const autofixPrompt = testFileExist \r\n        ? prepareMoreUtAutoFixPrompt(sourceCode, dependency, generatedCode, errorMessage, isBuildFailure, isXapCode)\r\n        : prepareUtAutoFixPrompt(prompts, generatedCode, errorMessage, isBuildFailure);\r\n\r\n    // Extract error test method names for the more UT auto-fix scenario\r\n    const errorMethodNames = testFileExist\r\n        ? extractErrorTestMethodNames(generatedCode, isBuildFailure, buildResult, testResult)\r\n        : undefined;\r\n\r\n    // get valid code from LLM response with retry\r\n    const llmResult = await ensureValidLLMResponse(autofixPrompt, testFramework, isXapCode, testFileExist, testFilePath, true, 3, sourceCode, errorMethodNames);\r\n    const newGeneratedCode = llmResult?.testCode;\r\n\r\n    // if generate code is valid, write test code to test project\r\n    !!newGeneratedCode && writeTestCode(newGeneratedCode, testFilePath);\r\n\r\n    return newGeneratedCode;\r\n}\r\n\r\n\r\n// use for quick dev test\r\nexport async function autoRetryTestCode(\r\n    prompts: any[],\r\n    sourceCodePath: string,\r\n    testCode: string,\r\n    testProjectPath: string,\r\n    testFilePath: string,\r\n    testFramework: string,\r\n    isXapCode: boolean,\r\n    retryTimes: number = 3\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    let generatedCode = testCode;\r\n    let vsTestRes;\r\n    for (let i = 0; i <= retryTimes; i++) {\r\n        if (i === retryTimes) {\r\n            console.error(`Failed to generate XapUt code after ${retryTimes} attempts.`);\r\n            break;\r\n        }\r\n\r\n        const autofixPrompt = prompts;\r\n        // get valid code from LLM response\r\n        const llmResult = await ensureValidLLMResponse(autofixPrompt, testFramework, isXapCode, false, testFilePath, true);\r\n        generatedCode = llmResult?.testCode;\r\n        // write test code to test project\r\n        writeTestCode(generatedCode, testFilePath);\r\n    }\r\n\r\n    console.log(`Autofix time: ${(Date.now() - startTime) / 1000} seconds`);\r\n\r\n    return { success: true, testFilePath: testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: vsTestRes };\r\n}"]}

package/out/gen/ensureValidLLMResponse.d.ts

@@ -2,4 +2,4 @@ export interface LLMResponseResult {
     testCode: string;
     insertedRanges?: [number, number][];
 }
-export declare function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, testFileExist: boolean, testFilePath: string, isAutoFix?: boolean, retryTimes?: number, sourceCode?: string): Promise<LLMResponseResult>;
+export declare function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, isMoreUTScenario: boolean, testFilePath: string, isAutoFix?: boolean, retryTimes?: number, sourceCode?: string, errorMethodNames?: string[]): Promise<LLMResponseResult>;

package/out/gen/ensureValidLLMResponse.js

@@ -3,21 +3,25 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.ensureValidLLMResponse = ensureValidLLMResponse;
 const postGenProcess_1 = require("./postGen/postGenProcess");
 const postGenMoreUTProcess_1 = require("./postGen/postGenMoreUTProcess");
+const postGenMoreUTAutofixProcess_1 = require("./postGen/postGenMoreUTAutofixProcess");
 const sendRequestToLLM_1 = require("../llm/sendRequestToLLM");
-async function ensureValidLLMResponse(prompt, testFramework, isXapCode, testFileExist, testFilePath, isAutoFix = false, retryTimes = 3, sourceCode) {
+async function ensureValidLLMResponse(prompt, testFramework, isXapCode, isMoreUTScenario, testFilePath, isAutoFix = false, retryTimes = 3, sourceCode, errorMethodNames) {
     for (let i = 0; i < retryTimes; i++) {
         // call LLM service to gen code
         const { content: generatedCode } = await (0, sendRequestToLLM_1.sendRequestToLLM)(prompt);
         // extract valid code from LLM response
         let testCode;
         let insertedRanges = undefined;
-        if (testFileExist && !isAutoFix) {
+        if (isMoreUTScenario && !isAutoFix) {
             const result = (0, postGenMoreUTProcess_1.postGenMoreUTProcess)(generatedCode, testFilePath, isXapCode);
             if (result) {
                 testCode = result.testCode;
                 insertedRanges = result.insertedRanges;
             }
         }
+        else if (isMoreUTScenario && isAutoFix) {
+            testCode = (0, postGenMoreUTAutofixProcess_1.postGenMoreUTAutofixProcess)(generatedCode, testFramework, isXapCode, sourceCode, testFilePath, errorMethodNames);
+        }
         else {
             testCode = (0, postGenProcess_1.postGenProcess)(generatedCode, testFramework, isXapCode, sourceCode);
         }

package/out/gen/ensureValidLLMResponse.js.map

@@ -1 +1 @@
-{"version":3,"file":"ensureValidLLMResponse.js","sourceRoot":"","sources":["../../src/gen/ensureValidLLMResponse.ts"],"names":[],"mappings":";;AASA,wDAwBC;AAjCD,6DAA0D;AAC1D,yEAAsE;AACtE,8DAA2D;AAOpD,KAAK,UAAU,sBAAsB,CAAC,MAAa,EAAE,aAAqB,EAAE,SAAkB,EAAE,aAAsB,EAAE,YAAoB,EAAE,YAAqB,KAAK,EAAE,aAAqB,CAAC,EAAE,UAAmB;IACxN,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,+BAA+B;QAC/B,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,IAAA,mCAAgB,EAAC,MAAM,CAAC,CAAC;QAElE,uCAAuC;QACvC,IAAI,QAAgB,CAAC;QACrB,IAAI,cAAc,GAAuB,SAAS,CAAC;QACnD,IAAI,aAAa,IAAI,CAAC,SAAS,EAAE,CAAC;YAC9B,MAAM,MAAM,GAAG,IAAA,2CAAoB,EAAC,aAAa,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;YAC5E,IAAI,MAAM,EAAE,CAAC;gBACT,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;gBAC3B,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;YAC3C,CAAC;QACL,CAAC;aAAM,CAAC;YACJ,QAAQ,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QACnF,CAAC;QACD,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;QACxC,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC;IAC5F,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC","sourcesContent":["import { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { postGenMoreUTProcess } from \"./postGen/postGenMoreUTProcess\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\n\r\nexport interface LLMResponseResult {\r\n    testCode: string;\r\n    insertedRanges?: [number, number][];\r\n}\r\n\r\nexport async function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, testFileExist: boolean, testFilePath: string, isAutoFix: boolean = false, retryTimes: number = 3, sourceCode?: string): Promise<LLMResponseResult> {\r\n    for (let i = 0; i < retryTimes; i++) {\r\n        // call LLM service to gen code\r\n        const { content: generatedCode } = await sendRequestToLLM(prompt);\r\n    \r\n        // extract valid code from LLM response\r\n        let testCode: string;\r\n        let insertedRanges: [number, number][] = undefined;\r\n        if (testFileExist && !isAutoFix) {\r\n            const result = postGenMoreUTProcess(generatedCode, testFilePath, isXapCode);\r\n            if (result) {\r\n                testCode = result.testCode;\r\n                insertedRanges = result.insertedRanges;\r\n            }\r\n        } else {\r\n            testCode = postGenProcess(generatedCode, testFramework, isXapCode, sourceCode);\r\n        }\r\n        if (!!testCode) {\r\n            return { testCode, insertedRanges };\r\n        }\r\n        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);\r\n    }\r\n\r\n    return undefined;\r\n}"]}
+{"version":3,"file":"ensureValidLLMResponse.js","sourceRoot":"","sources":["../../src/gen/ensureValidLLMResponse.ts"],"names":[],"mappings":";;AAUA,wDA0BC;AApCD,6DAA0D;AAC1D,yEAAsE;AACtE,uFAAoF;AACpF,8DAA2D;AAOpD,KAAK,UAAU,sBAAsB,CAAC,MAAa,EAAE,aAAqB,EAAE,SAAkB,EAAE,gBAAyB,EAAE,YAAoB,EAAE,YAAqB,KAAK,EAAE,aAAqB,CAAC,EAAE,UAAmB,EAAE,gBAA2B;IACxP,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,+BAA+B;QAC/B,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,IAAA,mCAAgB,EAAC,MAAM,CAAC,CAAC;QAElE,uCAAuC;QACvC,IAAI,QAAgB,CAAC;QACrB,IAAI,cAAc,GAAuB,SAAS,CAAC;QACnD,IAAI,gBAAgB,IAAI,CAAC,SAAS,EAAE,CAAC;YACjC,MAAM,MAAM,GAAG,IAAA,2CAAoB,EAAC,aAAa,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;YAC5E,IAAI,MAAM,EAAE,CAAC;gBACT,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC;gBAC3B,cAAc,GAAG,MAAM,CAAC,cAAc,CAAC;YAC3C,CAAC;QACL,CAAC;aAAM,IAAI,gBAAgB,IAAI,SAAS,EAAE,CAAC;YACvC,QAAQ,GAAG,IAAA,yDAA2B,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,EAAE,YAAY,EAAE,gBAAgB,CAAC,CAAC;QAChI,CAAC;aAAM,CAAC;YACJ,QAAQ,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,aAAa,EAAE,SAAS,EAAE,UAAU,CAAC,CAAC;QACnF,CAAC;QACD,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,CAAC;QACxC,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC;IAC5F,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC","sourcesContent":["import { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { postGenMoreUTProcess } from \"./postGen/postGenMoreUTProcess\";\r\nimport { postGenMoreUTAutofixProcess } from \"./postGen/postGenMoreUTAutofixProcess\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\n\r\nexport interface LLMResponseResult {\r\n    testCode: string;\r\n    insertedRanges?: [number, number][];\r\n}\r\n\r\nexport async function ensureValidLLMResponse(prompt: any[], testFramework: string, isXapCode: boolean, isMoreUTScenario: boolean, testFilePath: string, isAutoFix: boolean = false, retryTimes: number = 3, sourceCode?: string, errorMethodNames?: string[]): Promise<LLMResponseResult> {\r\n    for (let i = 0; i < retryTimes; i++) {\r\n        // call LLM service to gen code\r\n        const { content: generatedCode } = await sendRequestToLLM(prompt);\r\n    \r\n        // extract valid code from LLM response\r\n        let testCode: string;\r\n        let insertedRanges: [number, number][] = undefined;\r\n        if (isMoreUTScenario && !isAutoFix) {\r\n            const result = postGenMoreUTProcess(generatedCode, testFilePath, isXapCode);\r\n            if (result) {\r\n                testCode = result.testCode;\r\n                insertedRanges = result.insertedRanges;\r\n            }\r\n        } else if (isMoreUTScenario && isAutoFix) {\r\n            testCode = postGenMoreUTAutofixProcess(generatedCode, testFramework, isXapCode, sourceCode, testFilePath, errorMethodNames);\r\n        } else {\r\n            testCode = postGenProcess(generatedCode, testFramework, isXapCode, sourceCode);\r\n        }\r\n        if (!!testCode) {\r\n            return { testCode, insertedRanges };\r\n        }\r\n        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);\r\n    }\r\n\r\n    return undefined;\r\n}"]}

package/out/gen/postGen/postGenMoreUTAutofixProcess.d.ts

@@ -0,0 +1,17 @@
+/**
+ * Post-process LLM response for the "more UT auto-fix" scenario.
+ *
+ * The LLM is asked to return the **complete** fixed test file, but it sometimes
+ * omits unchanged test methods.  This function performs a *surgical merge*:
+ *
+ *   1. Extract & clean the AI code from the markdown response.
+ *   2. Replace **only** the error test methods in the original file with the
+ *      AI-fixed versions.
+ *   3. Append any **new** test methods or helper methods that the AI added.
+ *   4. Replace the `using` block wholesale with the AI's version.
+ *   5. Apply standard namespace / Xap repairs.
+ *
+ * If `errorMethodNames` is empty or `testFilePath` is unavailable the function
+ * falls back to standard `postGenProcess`-style processing.
+ */
+export declare function postGenMoreUTAutofixProcess(generatedCode: string, testFramework: string, isXapCode: boolean, sourceCode?: string, testFilePath?: string, errorMethodNames?: string[]): string;