@ai-dev-tools/csharp-copilot-core 0.0.36 → 0.0.37
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/out/gen/autoFix.js +2 -2
- package/out/gen/autoFix.js.map +1 -1
- package/package.json +1 -1
package/out/gen/autoFix.js
CHANGED
|
@@ -24,7 +24,7 @@ async function autoFixTestCode(prompts, sourceCodePath, sourceCode, dependency,
|
|
|
24
24
|
const sourceCodeProjectPath = (0, getCodeStructurePath_1.getSourceFileCsprojFilePath)(sourceCodePath);
|
|
25
25
|
for (let i = 0; i <= retryTimes; i++) {
|
|
26
26
|
// write generated code, to review and improve code quality
|
|
27
|
-
|
|
27
|
+
!isVsPlugin && (0, writeGenCode_1.writeGenCode)(sourceCodePath, generatedCode, i, sourceCodeProjectPath);
|
|
28
28
|
// for case: generatedCode is invalid, just retry instead of auto fix process
|
|
29
29
|
if (!generatedCode) {
|
|
30
30
|
console.error(`Generated code is invalid, try to regenerate. \nGenerated code: \n${generatedCode}`);
|
|
@@ -76,7 +76,7 @@ async function autoFixTestCode(prompts, sourceCodePath, sourceCode, dependency,
|
|
|
76
76
|
return { success: false, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult, testResult };
|
|
77
77
|
}
|
|
78
78
|
}
|
|
79
|
-
|
|
79
|
+
!isVsPlugin && (0, writeFailedVerifyResult_1.writeFailedVerifyResult)(sourceCodePath, i, sourceCodeProjectPath, errorMessage, !testResult);
|
|
80
80
|
// keep test failed tast cases in test file automatically
|
|
81
81
|
// if (!isBuildFailure && vsTestRes.passed !== 0) {
|
|
82
82
|
// console.log("remove failed test cases in test file");
|
package/out/gen/autoFix.js.map
CHANGED
|
@@ -1 +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,WAAW,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,WAAW,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 isBenchmark && 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 isBenchmark && 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}"]}
|
|
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}"]}
|