@ai-dev-tools/csharp-copilot-core 0.0.33 → 0.0.34

package/out/gen/postGen/postGenProcess.js

@@ -3,22 +3,24 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.postGenProcess = postGenProcess;
 const extractCodeFromResponse_1 = require("./extractCodeFromResponse");
 const removeSingleLines_1 = require("./removeSingleLines");
-const removeComments_1 = require("./removeComments");
 const repairRequiredNameSpaces_1 = require("./repairRequiredNameSpaces");
 const validateTestCode_1 = require("./validateTestCode");
-const addAiAnnotation_1 = require("./addAiAnnotation");
-function postGenProcess(generatedCode, testFramework, isXapCode) {
+function postGenProcess(generatedCode, testFramework, isXapCode, sourceCode) {
     if (!isValidTestCode(generatedCode)) {
         return undefined;
     }
     let testCode = (0, extractCodeFromResponse_1.extractCodeFromResponse)(generatedCode);
-    testCode = (0, removeComments_1.removeTestCodeComments)(testCode);
+    // testCode = removeTestCodeComments(testCode);
     testCode = (0, removeSingleLines_1.removeCSharpOnlyLines)(testCode);
     testCode = (0, repairRequiredNameSpaces_1.repairRequiredNameSpaces)(testCode, testFramework);
     if (isXapCode) {
         testCode = xapCodePostGenProcess(testCode);
     }
-    testCode = (0, addAiAnnotation_1.addAiAnnotation)(testCode);
+    // Repair using aliases from source code if provided
+    if (sourceCode) {
+        testCode = (0, repairRequiredNameSpaces_1.repairUsingStatements)(testCode, sourceCode);
+    }
+    // testCode = addAiAnnotation(testCode);
     console.log(`Post generation process completed, code length: ${testCode.length}`);
     return testCode;
 }
@@ -34,6 +36,4 @@ function isValidTestCode(testCode) {
     const isValid = (0, validateTestCode_1.validateTestCode)(testCode);
     return isValid;
 }
-// const code = fs.readFileSync("D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Tests\\Workflows\\Overview\\HttpRequestParser\\Utils\\SqmidHelperTests.cs", 'utf-8');
-// postGenProcess(code, false);
 //# sourceMappingURL=postGenProcess.js.map

package/out/gen/postGen/postGenProcess.js.map

@@ -1 +1 @@
-{"version":3,"file":"postGenProcess.js","sourceRoot":"","sources":["../../../src/gen/postGen/postGenProcess.ts"],"names":[],"mappings":";;AAOA,wCAiBC;AAxBD,uEAAoE;AACpE,2DAA4D;AAC5D,qDAA0D;AAC1D,yEAAmG;AACnG,yDAAsD;AACtD,uDAAoD;AAEpD,SAAgB,cAAc,CAAC,aAAqB,EAAE,aAAqB,EAAE,SAAkB;IAC3F,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,IAAI,QAAQ,GAAG,IAAA,iDAAuB,EAAC,aAAa,CAAC,CAAC;IACtD,QAAQ,GAAG,IAAA,uCAAsB,EAAC,QAAQ,CAAC,CAAC;IAC5C,QAAQ,GAAG,IAAA,yCAAqB,EAAC,QAAQ,CAAC,CAAC;IAC3C,QAAQ,GAAG,IAAA,mDAAwB,EAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAC7D,IAAI,SAAS,EAAE,CAAC;QACZ,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,QAAQ,GAAG,IAAA,iCAAe,EAAC,QAAQ,CAAC,CAAC;IAErC,OAAO,CAAC,GAAG,CAAC,mDAAmD,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAClF,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAgB;IAC3C,MAAM,aAAa,GAAG,IAAA,sDAA2B,EAAC,QAAQ,CAAC,CAAC;IAC5D,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACrC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACtC,OAAO,KAAK,CAAC,CAAC,0BAA0B;IAC5C,CAAC;IACD,0EAA0E;IAC1E,MAAM,OAAO,GAAG,IAAA,mCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3C,OAAO,OAAO,CAAC;AACnB,CAAC;AAED,uMAAuM;AACvM,+BAA+B","sourcesContent":["import { extractCodeFromResponse } from \"./extractCodeFromResponse\";\r\nimport { removeCSharpOnlyLines } from \"./removeSingleLines\";\r\nimport { removeTestCodeComments } from \"./removeComments\";\r\nimport { repairRequiredNameSpaces, repairXapRequiredNameSpaces } from \"./repairRequiredNameSpaces\";\r\nimport { validateTestCode } from \"./validateTestCode\";\r\nimport { addAiAnnotation } from \"./addAiAnnotation\";\r\n\r\nexport function postGenProcess(generatedCode: string, testFramework: string, isXapCode: boolean): string {\r\n    if (!isValidTestCode(generatedCode)) {\r\n        return undefined;\r\n    }\r\n\r\n    let testCode = extractCodeFromResponse(generatedCode);\r\n    testCode = removeTestCodeComments(testCode);\r\n    testCode = removeCSharpOnlyLines(testCode);\r\n    testCode = repairRequiredNameSpaces(testCode, testFramework);\r\n    if (isXapCode) {\r\n        testCode = xapCodePostGenProcess(testCode);\r\n    }\r\n\r\n    testCode = addAiAnnotation(testCode);\r\n\r\n    console.log(`Post generation process completed, code length: ${testCode.length}`);\r\n    return testCode;\r\n}\r\n\r\nfunction xapCodePostGenProcess(testCode: string): string {\r\n    const processedCode = repairXapRequiredNameSpaces(testCode);\r\n    return processedCode;\r\n}\r\n\r\nfunction isValidTestCode(testCode: string): boolean {\r\n    if (!testCode || testCode.trim() === '') {\r\n        return false; // Empty code is not valid\r\n    }\r\n    // Check if the test code contains any test attributes or using statements\r\n    const isValid = validateTestCode(testCode);\r\n    return isValid;\r\n}\r\n\r\n// const code = fs.readFileSync(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Tests\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelperTests.cs\", 'utf-8');\r\n// postGenProcess(code, false);\r\n"]}
+{"version":3,"file":"postGenProcess.js","sourceRoot":"","sources":["../../../src/gen/postGen/postGenProcess.ts"],"names":[],"mappings":";;AAKA,wCAsBC;AA3BD,uEAAoE;AACpE,2DAA4D;AAC5D,yEAA0H;AAC1H,yDAAsD;AAEtD,SAAgB,cAAc,CAAC,aAAqB,EAAE,aAAqB,EAAE,SAAkB,EAAE,UAAmB;IAChH,IAAI,CAAC,eAAe,CAAC,aAAa,CAAC,EAAE,CAAC;QAClC,OAAO,SAAS,CAAC;IACrB,CAAC;IAED,IAAI,QAAQ,GAAG,IAAA,iDAAuB,EAAC,aAAa,CAAC,CAAC;IACtD,+CAA+C;IAC/C,QAAQ,GAAG,IAAA,yCAAqB,EAAC,QAAQ,CAAC,CAAC;IAC3C,QAAQ,GAAG,IAAA,mDAAwB,EAAC,QAAQ,EAAE,aAAa,CAAC,CAAC;IAC7D,IAAI,SAAS,EAAE,CAAC;QACZ,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IAED,oDAAoD;IACpD,IAAI,UAAU,EAAE,CAAC;QACb,QAAQ,GAAG,IAAA,gDAAqB,EAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;IAC3D,CAAC;IAED,wCAAwC;IAExC,OAAO,CAAC,GAAG,CAAC,mDAAmD,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAClF,OAAO,QAAQ,CAAC;AACpB,CAAC;AAED,SAAS,qBAAqB,CAAC,QAAgB;IAC3C,MAAM,aAAa,GAAG,IAAA,sDAA2B,EAAC,QAAQ,CAAC,CAAC;IAC5D,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,SAAS,eAAe,CAAC,QAAgB;IACrC,IAAI,CAAC,QAAQ,IAAI,QAAQ,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;QACtC,OAAO,KAAK,CAAC,CAAC,0BAA0B;IAC5C,CAAC;IACD,0EAA0E;IAC1E,MAAM,OAAO,GAAG,IAAA,mCAAgB,EAAC,QAAQ,CAAC,CAAC;IAC3C,OAAO,OAAO,CAAC;AACnB,CAAC","sourcesContent":["import { extractCodeFromResponse } from \"./extractCodeFromResponse\";\r\nimport { removeCSharpOnlyLines } from \"./removeSingleLines\";\r\nimport { repairRequiredNameSpaces, repairXapRequiredNameSpaces, repairUsingStatements } from \"./repairRequiredNameSpaces\";\r\nimport { validateTestCode } from \"./validateTestCode\";\r\n\r\nexport function postGenProcess(generatedCode: string, testFramework: string, isXapCode: boolean, sourceCode?: string): string {\r\n    if (!isValidTestCode(generatedCode)) {\r\n        return undefined;\r\n    }\r\n\r\n    let testCode = extractCodeFromResponse(generatedCode);\r\n    // testCode = removeTestCodeComments(testCode);\r\n    testCode = removeCSharpOnlyLines(testCode);\r\n    testCode = repairRequiredNameSpaces(testCode, testFramework);\r\n    if (isXapCode) {\r\n        testCode = xapCodePostGenProcess(testCode);\r\n    }\r\n    \r\n    // Repair using aliases from source code if provided\r\n    if (sourceCode) {\r\n        testCode = repairUsingStatements(testCode, sourceCode);\r\n    }\r\n\r\n    // testCode = addAiAnnotation(testCode);\r\n\r\n    console.log(`Post generation process completed, code length: ${testCode.length}`);\r\n    return testCode;\r\n}\r\n\r\nfunction xapCodePostGenProcess(testCode: string): string {\r\n    const processedCode = repairXapRequiredNameSpaces(testCode);\r\n    return processedCode;\r\n}\r\n\r\nfunction isValidTestCode(testCode: string): boolean {\r\n    if (!testCode || testCode.trim() === '') {\r\n        return false; // Empty code is not valid\r\n    }\r\n    // Check if the test code contains any test attributes or using statements\r\n    const isValid = validateTestCode(testCode);\r\n    return isValid;\r\n}\r\n"]}

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

@@ -1,2 +1,19 @@
+export declare const XAP_REQUIRED_USINGS: string[];
 export declare function repairRequiredNameSpaces(code: string, testFramework: string): string;
 export declare function repairXapRequiredNameSpaces(code: string): string;
+/**
+ * Detect the using statement insertion position and indentation in the code.
+ * Returns the insert index and the indentation string.
+ */
+export declare function detectUsingInsertPosition(lines: string[]): {
+    insertIndex: number;
+    indent: string;
+};
+/**
+ * Extract using Statements from source code and add them to test code if the alias is used.
+ * Handles patterns like: using Partner = Partnerhub.PDP.Partner;
+ * @param testCode The generated test code
+ * @param sourceCode The original source code
+ * @returns The test code with missing using aliases added
+ */
+export declare function repairUsingStatements(testCode: string, sourceCode: string): string;

package/out/gen/postGen/repairRequiredNameSpaces.js

@@ -1,9 +1,19 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.XAP_REQUIRED_USINGS = void 0;
 exports.repairRequiredNameSpaces = repairRequiredNameSpaces;
 exports.repairXapRequiredNameSpaces = repairXapRequiredNameSpaces;
+exports.detectUsingInsertPosition = detectUsingInsertPosition;
+exports.repairUsingStatements = repairUsingStatements;
 const constants_1 = require("../../types/constants");
 const constants_2 = require("../../types/constants");
+exports.XAP_REQUIRED_USINGS = [
+    "using System;",
+    "using System.Threading.Tasks;",
+    "using System.Collections.Generic;",
+    "using Xap.ExecutionFramework;",
+    "using Xap.PluginFramework;",
+];
 function repairRequiredNameSpaces(code, testFramework) {
     const testFrameworkUsing = getTestFrameworkNameSpaces(testFramework);
     const requiredUsings = [
@@ -13,39 +23,50 @@ function repairRequiredNameSpaces(code, testFramework) {
     return code;
 }
 function repairXapRequiredNameSpaces(code) {
-    const xapUsings = [
-        "using System;",
-        "using System.Threading.Tasks;",
-        "using System.Collections.Generic;",
-        "using Xap.ExecutionFramework;",
-        "using Xap.PluginFramework;",
-        "using EMPTY = Microsoft.Bing.Xap.Plugins.Void;",
-    ];
-    code = repairMissingUsings(code, xapUsings);
+    const requiredUsings = [...exports.XAP_REQUIRED_USINGS];
+    // Only add XapBondRelatedNamespace if code contains XapBlob (case-sensitive)
+    if (code.includes('XapBlob')) {
+        requiredUsings.push(constants_1.XapBondRelatedNamespace);
+    }
+    code = repairMissingUsings(code, requiredUsings);
     return code;
 }
+/**
+ * Detect the using statement insertion position and indentation in the code.
+ * Returns the insert index and the indentation string.
+ */
+function detectUsingInsertPosition(lines) {
+    const firstUsingIndex = lines.findIndex(line => line.trim().startsWith('using '));
+    if (firstUsingIndex !== -1) {
+        // There are existing using statements - insert before the first one and use its indentation
+        const match = lines[firstUsingIndex].match(/^(\s*)/);
+        const indent = match ? match[1] : '';
+        return { insertIndex: firstUsingIndex, indent };
+    }
+    else {
+        // No using statements found - insert at the beginning (before namespace or at top)
+        const namespaceIndex = lines.findIndex(line => line.trim().startsWith('namespace '));
+        const insertIndex = namespaceIndex !== -1 ? namespaceIndex : 0;
+        return { insertIndex, indent: '' };
+    }
+}
 function repairMissingUsings(code, requiredUsings) {
-    // get all using lines
+    // get all using lines (trimmed to compare)
     const usingRegex = /^\s*using\s+[^\r\n;]+;/gm;
     const existingUsings = Array.from(code.matchAll(usingRegex)).map(match => match[0].trim());
     const missingUsings = requiredUsings.filter(req => !existingUsings.includes(req));
     if (missingUsings.length === 0) {
         return code;
     }
-    // find the end of the using block: the first non-using line, or the end of the file
     const lines = code.split(constants_2.LINESEPARATOR);
-    let insertIndex = 0;
-    for (let i = 0; i < lines.length; i++) {
-        if (!lines[i].trim().startsWith("using") && lines[i].trim() !== "") {
-            insertIndex = i;
-            break;
-        }
-        insertIndex = i + 1;
-    }
+    // Use the common function to detect insert position and indentation
+    const { insertIndex, indent } = detectUsingInsertPosition(lines);
+    // Apply indentation to missing usings
+    const indentedMissingUsings = missingUsings.map(u => indent + u);
     // build new code
     const newLines = [
         ...lines.slice(0, insertIndex),
-        ...missingUsings,
+        ...indentedMissingUsings,
         ...lines.slice(insertIndex)
     ];
     console.log("repaired namespaces:", missingUsings);
@@ -63,6 +84,52 @@ function getTestFrameworkNameSpaces(testFramework) {
     };
     return FRAMEWORK_USINGS[testFramework] || msTestUsing;
 }
+/**
+ * Extract using Statements from source code and add them to test code if the alias is used.
+ * Handles patterns like: using Partner = Partnerhub.PDP.Partner;
+ * @param testCode The generated test code
+ * @param sourceCode The original source code
+ * @returns The test code with missing using aliases added
+ */
+function repairUsingStatements(testCode, sourceCode) {
+    if (!sourceCode || !testCode) {
+        return testCode;
+    }
+    // Extract using aliases from source code
+    // Pattern: using AliasName = Namespace.Type;
+    const usingAliasRegex = /^\s*using\s+(\w+)\s*=\s*([^;]+);/gm;
+    const sourceAliases = new Map();
+    let match;
+    while ((match = usingAliasRegex.exec(sourceCode)) !== null) {
+        const aliasName = match[1];
+        const fullStatement = match[0].trim();
+        sourceAliases.set(aliasName, fullStatement);
+    }
+    if (sourceAliases.size === 0) {
+        return testCode;
+    }
+    // Check which aliases are used in test code but not defined
+    const missingAliases = [];
+    const existingUsings = testCode.match(/^\s*using\s+.+;/gm) || [];
+    const existingUsingsTrimmed = existingUsings.map(u => u.trim());
+    for (const [aliasName, fullStatement] of sourceAliases) {
+        // Check if alias is used in test code (as a type name, using word boundary)
+        const aliasUsagePattern = new RegExp(`\\b${aliasName}\\b`);
+        if (aliasUsagePattern.test(testCode)) {
+            // Check if the using alias is already in test code
+            if (!existingUsingsTrimmed.includes(fullStatement)) {
+                missingAliases.push(fullStatement);
+            }
+        }
+    }
+    if (missingAliases.length === 0) {
+        return testCode;
+    }
+    // Add missing aliases
+    const result = repairMissingUsings(testCode, missingAliases);
+    console.log("repaired using aliases:", missingAliases);
+    return result;
+}
 // const code = `
 // using Microsoft.VisualStudio.TestTools.UnitTesting;
 // using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;

package/out/gen/postGen/repairRequiredNameSpaces.js.map

@@ -1 +1 @@
-{"version":3,"file":"repairRequiredNameSpaces.js","sourceRoot":"","sources":["../../../src/gen/postGen/repairRequiredNameSpaces.ts"],"names":[],"mappings":";;AAKA,4DAQC;AAED,kEAYC;AA3BD,qDAAgG;AAGhG,qDAAsD;AAEtD,SAAgB,wBAAwB,CAAC,IAAY,EAAE,aAAqB;IACxE,MAAM,kBAAkB,GAAG,0BAA0B,CAAC,aAAa,CAAC,CAAC;IACrE,MAAM,cAAc,GAAG;QACnB,kBAAkB;KACrB,CAAC;IAEF,IAAI,GAAG,mBAAmB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACjD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAgB,2BAA2B,CAAC,IAAY;IACpD,MAAM,SAAS,GAAG;QACd,eAAe;QACf,+BAA+B;QAC/B,mCAAmC;QACnC,+BAA+B;QAC/B,4BAA4B;QAC5B,gDAAgD;KACnD,CAAC;IAEF,IAAI,GAAG,mBAAmB,CAAC,IAAI,EAAE,SAAS,CAAC,CAAC;IAC5C,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY,EAAE,cAAwB;IAC/D,sBAAsB;IACtB,MAAM,UAAU,GAAG,0BAA0B,CAAC;IAC9C,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAE3F,MAAM,aAAa,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAElF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,oFAAoF;IACpF,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,yBAAa,CAAC,CAAC;IACxC,IAAI,WAAW,GAAG,CAAC,CAAC;IAEpB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;QACpC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,KAAK,EAAE,EAAE,CAAC;YACjE,WAAW,GAAG,CAAC,CAAC;YAChB,MAAM;QACV,CAAC;QACD,WAAW,GAAG,CAAC,GAAG,CAAC,CAAC;IACxB,CAAC;IAED,iBAAiB;IACjB,MAAM,QAAQ,GAAG;QACb,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC;QAC9B,GAAG,aAAa;QAChB,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;KAC9B,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAC;IAEnD,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,0BAA0B,CAAC,aAAqB;IACrD,MAAM,WAAW,GAAG,qDAAqD,CAAC;IAE1E,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,OAAO,WAAW,CAAC;IACvB,CAAC;IAED,MAAM,gBAAgB,GAAG;QACrB,CAAC,2BAAe,CAAC,EAAE,WAAW;QAC9B,CAAC,8BAAkB,CAAC,EAAE,wBAAwB;QAC9C,CAAC,8BAAkB,CAAC,EAAE,cAAc;KAC9B,CAAC;IAEX,OAAO,gBAAgB,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC;AAC1D,CAAC;AAID,iBAAiB;AACjB,sDAAsD;AACtD,2EAA2E;AAC3E,gBAAgB;AAChB,oCAAoC;AACpC,qBAAqB;AACrB,aAAa;AAEb,oFAAoF;AACpF,IAAI;AACJ,kBAAkB;AAClB,oCAAoC;AACpC,QAAQ;AACR,uBAAuB;AACvB,uFAAuF;AACvF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mHAAmH;AACnH,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,wDAAwD;AACxD,YAAY;AAEZ,uBAAuB;AACvB,iFAAiF;AACjF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,qDAAqD;AACrD,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qCAAqC;AACrC,YAAY;AAEZ,uBAAuB;AACvB,2EAA2E;AAC3E,YAAY;AACZ,yBAAyB;AACzB,8DAA8D;AAE9D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,0EAA0E;AAC1E,YAAY;AACZ,yBAAyB;AACzB,0DAA0D;AAE1D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,mFAAmF;AACnF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mCAAmC;AACnC,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AACZ,QAAQ;AACR,IAAI;AACJ,KAAK;AAEL,sDAAsD;AACtD,4BAA4B","sourcesContent":["import { MSTESTFRAMEWORK, NUNITTESTFRAMEWORK, XUNITTESTFRAMEWORK } from '../../types/constants';\r\nimport fs from 'fs';\r\nimport path from 'path';\r\nimport { LINESEPARATOR } from '../../types/constants';\r\n\r\nexport function repairRequiredNameSpaces(code: string, testFramework: string): string {\r\n    const testFrameworkUsing = getTestFrameworkNameSpaces(testFramework);\r\n    const requiredUsings = [\r\n        testFrameworkUsing,\r\n    ];\r\n\r\n    code = repairMissingUsings(code, requiredUsings);\r\n    return code;\r\n}\r\n\r\nexport function repairXapRequiredNameSpaces(code: string): string {\r\n    const xapUsings = [\r\n        \"using System;\",\r\n        \"using System.Threading.Tasks;\",\r\n        \"using System.Collections.Generic;\",\r\n        \"using Xap.ExecutionFramework;\",\r\n        \"using Xap.PluginFramework;\",\r\n        \"using EMPTY = Microsoft.Bing.Xap.Plugins.Void;\",\r\n    ];\r\n\r\n    code = repairMissingUsings(code, xapUsings);\r\n    return code;\r\n}\r\n\r\nfunction repairMissingUsings(code: string, requiredUsings: string[]): string {\r\n    // get all using lines\r\n    const usingRegex = /^\\s*using\\s+[^\\r\\n;]+;/gm;\r\n    const existingUsings = Array.from(code.matchAll(usingRegex)).map(match => match[0].trim());\r\n\r\n    const missingUsings = requiredUsings.filter(req => !existingUsings.includes(req));\r\n\r\n    if (missingUsings.length === 0) {\r\n        return code;\r\n    }\r\n\r\n    // find the end of the using block: the first non-using line, or the end of the file\r\n    const lines = code.split(LINESEPARATOR);\r\n    let insertIndex = 0;\r\n\r\n    for (let i = 0; i < lines.length; i++) {\r\n        if (!lines[i].trim().startsWith(\"using\") && lines[i].trim() !== \"\") {\r\n            insertIndex = i;\r\n            break;\r\n        }\r\n        insertIndex = i + 1;\r\n    }\r\n\r\n    // build new code\r\n    const newLines = [\r\n        ...lines.slice(0, insertIndex),\r\n        ...missingUsings,\r\n        ...lines.slice(insertIndex)\r\n    ];\r\n\r\n    console.log(\"repaired namespaces:\", missingUsings);\r\n\r\n    return newLines.join('\\n');\r\n}\r\n\r\nfunction getTestFrameworkNameSpaces(testFramework: string): string {\r\n    const msTestUsing = \"using Microsoft.VisualStudio.TestTools.UnitTesting;\";\r\n\r\n    if (!testFramework) {\r\n        return msTestUsing;\r\n    }\r\n\r\n    const FRAMEWORK_USINGS = {\r\n        [MSTESTFRAMEWORK]: msTestUsing,\r\n        [NUNITTESTFRAMEWORK]: \"using NUnit.Framework;\",\r\n        [XUNITTESTFRAMEWORK]: \"using Xunit;\"\r\n    } as const;\r\n\r\n    return FRAMEWORK_USINGS[testFramework] || msTestUsing;\r\n}\r\n\r\n\r\n\r\n// const code = `\r\n// using Microsoft.VisualStudio.TestTools.UnitTesting;\r\n// using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\r\n// using System;\r\n// using System.Collections.Generic;\r\n// using System.Text;\r\n// using Moq;\r\n\r\n// namespace SharedSegments.Plugins.Tests.Workflows.Overview.HttpRequestParser.Utils\r\n// {\r\n//     [TestClass]\r\n//     public class SqmidHelperTests\r\n//     {\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithBlisId_ReturnsDecodedBlisId()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"BID=\" + Convert.ToBase64String(Encoding.UTF8.GetBytes(\"test_decodedBlisId_test\")) }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(\"decodedBlisId\", result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithoutBlisId_ReturnsNull()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"SomeOtherValue=123\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.IsNull(result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_EmptyCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>();\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             IDictionary<string, string> cookies = null;\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_UsrLocWithEmptyValue_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n//     }\r\n// }\r\n// `;\r\n\r\n// const cleanedCode = repairRequiredNameSpaces(code);\r\n// console.log(cleanedCode);"]}
+{"version":3,"file":"repairRequiredNameSpaces.js","sourceRoot":"","sources":["../../../src/gen/postGen/repairRequiredNameSpaces.ts"],"names":[],"mappings":";;;AAaA,4DAQC;AAED,kEAQC;AAMD,8DAcC;AAwDD,sDA6CC;AAxJD,qDAAyH;AAGzH,qDAAsD;AAEzC,QAAA,mBAAmB,GAAG;IAC/B,eAAe;IACf,+BAA+B;IAC/B,mCAAmC;IACnC,+BAA+B;IAC/B,4BAA4B;CAC/B,CAAC;AAEF,SAAgB,wBAAwB,CAAC,IAAY,EAAE,aAAqB;IACxE,MAAM,kBAAkB,GAAG,0BAA0B,CAAC,aAAa,CAAC,CAAC;IACrE,MAAM,cAAc,GAAG;QACnB,kBAAkB;KACrB,CAAC;IAEF,IAAI,GAAG,mBAAmB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACjD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAgB,2BAA2B,CAAC,IAAY;IACpD,MAAM,cAAc,GAAG,CAAC,GAAG,2BAAmB,CAAC,CAAC;IAChD,6EAA6E;IAC7E,IAAI,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;QAC3B,cAAc,CAAC,IAAI,CAAC,mCAAuB,CAAC,CAAC;IACjD,CAAC;IACD,IAAI,GAAG,mBAAmB,CAAC,IAAI,EAAE,cAAc,CAAC,CAAC;IACjD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED;;;GAGG;AACH,SAAgB,yBAAyB,CAAC,KAAe;IACrD,MAAM,eAAe,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC,CAAC;IAElF,IAAI,eAAe,KAAK,CAAC,CAAC,EAAE,CAAC;QACzB,4FAA4F;QAC5F,MAAM,KAAK,GAAG,KAAK,CAAC,eAAe,CAAC,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACrD,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;QACrC,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,MAAM,EAAE,CAAC;IACpD,CAAC;SAAM,CAAC;QACJ,mFAAmF;QACnF,MAAM,cAAc,GAAG,KAAK,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,UAAU,CAAC,YAAY,CAAC,CAAC,CAAC;QACrF,MAAM,WAAW,GAAG,cAAc,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,CAAC;QAC/D,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,EAAE,EAAE,CAAC;IACvC,CAAC;AACL,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY,EAAE,cAAwB;IAC/D,2CAA2C;IAC3C,MAAM,UAAU,GAAG,0BAA0B,CAAC;IAC9C,MAAM,cAAc,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAE3F,MAAM,aAAa,GAAG,cAAc,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,cAAc,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC,CAAC;IAElF,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IAChB,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,yBAAa,CAAC,CAAC;IAExC,oEAAoE;IACpE,MAAM,EAAE,WAAW,EAAE,MAAM,EAAE,GAAG,yBAAyB,CAAC,KAAK,CAAC,CAAC;IAEjE,sCAAsC;IACtC,MAAM,qBAAqB,GAAG,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAEjE,iBAAiB;IACjB,MAAM,QAAQ,GAAG;QACb,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,EAAE,WAAW,CAAC;QAC9B,GAAG,qBAAqB;QACxB,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC;KAC9B,CAAC;IAEF,OAAO,CAAC,GAAG,CAAC,sBAAsB,EAAE,aAAa,CAAC,CAAC;IAEnD,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,0BAA0B,CAAC,aAAqB;IACrD,MAAM,WAAW,GAAG,qDAAqD,CAAC;IAE1E,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,OAAO,WAAW,CAAC;IACvB,CAAC;IAED,MAAM,gBAAgB,GAAG;QACrB,CAAC,2BAAe,CAAC,EAAE,WAAW;QAC9B,CAAC,8BAAkB,CAAC,EAAE,wBAAwB;QAC9C,CAAC,8BAAkB,CAAC,EAAE,cAAc;KAC9B,CAAC;IAEX,OAAO,gBAAgB,CAAC,aAAa,CAAC,IAAI,WAAW,CAAC;AAC1D,CAAC;AAED;;;;;;GAMG;AACH,SAAgB,qBAAqB,CAAC,QAAgB,EAAE,UAAkB;IACtE,IAAI,CAAC,UAAU,IAAI,CAAC,QAAQ,EAAE,CAAC;QAC3B,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,yCAAyC;IACzC,6CAA6C;IAC7C,MAAM,eAAe,GAAG,oCAAoC,CAAC;IAC7D,MAAM,aAAa,GAAwB,IAAI,GAAG,EAAE,CAAC;IAErD,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACzD,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACtC,aAAa,CAAC,GAAG,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IAChD,CAAC;IAED,IAAI,aAAa,CAAC,IAAI,KAAK,CAAC,EAAE,CAAC;QAC3B,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,4DAA4D;IAC5D,MAAM,cAAc,GAAa,EAAE,CAAC;IACpC,MAAM,cAAc,GAAG,QAAQ,CAAC,KAAK,CAAC,mBAAmB,CAAC,IAAI,EAAE,CAAC;IACjE,MAAM,qBAAqB,GAAG,cAAc,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IAEhE,KAAK,MAAM,CAAC,SAAS,EAAE,aAAa,CAAC,IAAI,aAAa,EAAE,CAAC;QACrD,4EAA4E;QAC5E,MAAM,iBAAiB,GAAG,IAAI,MAAM,CAAC,MAAM,SAAS,KAAK,CAAC,CAAC;QAC3D,IAAI,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnC,mDAAmD;YACnD,IAAI,CAAC,qBAAqB,CAAC,QAAQ,CAAC,aAAa,CAAC,EAAE,CAAC;gBACjD,cAAc,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;YACvC,CAAC;QACL,CAAC;IACL,CAAC;IAED,IAAI,cAAc,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;QAC9B,OAAO,QAAQ,CAAC;IACpB,CAAC;IAED,sBAAsB;IACtB,MAAM,MAAM,GAAG,mBAAmB,CAAC,QAAQ,EAAE,cAAc,CAAC,CAAC;IAC7D,OAAO,CAAC,GAAG,CAAC,yBAAyB,EAAE,cAAc,CAAC,CAAC;IACvD,OAAO,MAAM,CAAC;AAClB,CAAC;AAID,iBAAiB;AACjB,sDAAsD;AACtD,2EAA2E;AAC3E,gBAAgB;AAChB,oCAAoC;AACpC,qBAAqB;AACrB,aAAa;AAEb,oFAAoF;AACpF,IAAI;AACJ,kBAAkB;AAClB,oCAAoC;AACpC,QAAQ;AACR,uBAAuB;AACvB,uFAAuF;AACvF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mHAAmH;AACnH,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,wDAAwD;AACxD,YAAY;AAEZ,uBAAuB;AACvB,iFAAiF;AACjF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,qDAAqD;AACrD,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qCAAqC;AACrC,YAAY;AAEZ,uBAAuB;AACvB,2EAA2E;AAC3E,YAAY;AACZ,yBAAyB;AACzB,8DAA8D;AAE9D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,0EAA0E;AAC1E,YAAY;AACZ,yBAAyB;AACzB,0DAA0D;AAE1D,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AAEZ,uBAAuB;AACvB,mFAAmF;AACnF,YAAY;AACZ,yBAAyB;AACzB,2DAA2D;AAC3D,gBAAgB;AAChB,mCAAmC;AACnC,iBAAiB;AAEjB,qBAAqB;AACrB,oEAAoE;AAEpE,wBAAwB;AACxB,qDAAqD;AACrD,YAAY;AACZ,QAAQ;AACR,IAAI;AACJ,KAAK;AAEL,sDAAsD;AACtD,4BAA4B","sourcesContent":["import { MSTESTFRAMEWORK, NUNITTESTFRAMEWORK, XUNITTESTFRAMEWORK, XapBondRelatedNamespace } from '../../types/constants';\r\nimport fs from 'fs';\r\nimport path from 'path';\r\nimport { LINESEPARATOR } from '../../types/constants';\r\n\r\nexport const XAP_REQUIRED_USINGS = [\r\n    \"using System;\",\r\n    \"using System.Threading.Tasks;\",\r\n    \"using System.Collections.Generic;\",\r\n    \"using Xap.ExecutionFramework;\",\r\n    \"using Xap.PluginFramework;\",\r\n];\r\n\r\nexport function repairRequiredNameSpaces(code: string, testFramework: string): string {\r\n    const testFrameworkUsing = getTestFrameworkNameSpaces(testFramework);\r\n    const requiredUsings = [\r\n        testFrameworkUsing,\r\n    ];\r\n\r\n    code = repairMissingUsings(code, requiredUsings);\r\n    return code;\r\n}\r\n\r\nexport function repairXapRequiredNameSpaces(code: string): string {\r\n    const requiredUsings = [...XAP_REQUIRED_USINGS];\r\n    // Only add XapBondRelatedNamespace if code contains XapBlob (case-sensitive)\r\n    if (code.includes('XapBlob')) {\r\n        requiredUsings.push(XapBondRelatedNamespace);\r\n    }\r\n    code = repairMissingUsings(code, requiredUsings);\r\n    return code;\r\n}\r\n\r\n/**\r\n * Detect the using statement insertion position and indentation in the code.\r\n * Returns the insert index and the indentation string.\r\n */\r\nexport function detectUsingInsertPosition(lines: string[]): { insertIndex: number, indent: string } {\r\n    const firstUsingIndex = lines.findIndex(line => line.trim().startsWith('using '));\r\n    \r\n    if (firstUsingIndex !== -1) {\r\n        // There are existing using statements - insert before the first one and use its indentation\r\n        const match = lines[firstUsingIndex].match(/^(\\s*)/);\r\n        const indent = match ? match[1] : '';\r\n        return { insertIndex: firstUsingIndex, indent };\r\n    } else {\r\n        // No using statements found - insert at the beginning (before namespace or at top)\r\n        const namespaceIndex = lines.findIndex(line => line.trim().startsWith('namespace '));\r\n        const insertIndex = namespaceIndex !== -1 ? namespaceIndex : 0;\r\n        return { insertIndex, indent: '' };\r\n    }\r\n}\r\n\r\nfunction repairMissingUsings(code: string, requiredUsings: string[]): string {\r\n    // get all using lines (trimmed to compare)\r\n    const usingRegex = /^\\s*using\\s+[^\\r\\n;]+;/gm;\r\n    const existingUsings = Array.from(code.matchAll(usingRegex)).map(match => match[0].trim());\r\n\r\n    const missingUsings = requiredUsings.filter(req => !existingUsings.includes(req));\r\n\r\n    if (missingUsings.length === 0) {\r\n        return code;\r\n    }\r\n\r\n    const lines = code.split(LINESEPARATOR);\r\n    \r\n    // Use the common function to detect insert position and indentation\r\n    const { insertIndex, indent } = detectUsingInsertPosition(lines);\r\n    \r\n    // Apply indentation to missing usings\r\n    const indentedMissingUsings = missingUsings.map(u => indent + u);\r\n\r\n    // build new code\r\n    const newLines = [\r\n        ...lines.slice(0, insertIndex),\r\n        ...indentedMissingUsings,\r\n        ...lines.slice(insertIndex)\r\n    ];\r\n\r\n    console.log(\"repaired namespaces:\", missingUsings);\r\n\r\n    return newLines.join('\\n');\r\n}\r\n\r\nfunction getTestFrameworkNameSpaces(testFramework: string): string {\r\n    const msTestUsing = \"using Microsoft.VisualStudio.TestTools.UnitTesting;\";\r\n\r\n    if (!testFramework) {\r\n        return msTestUsing;\r\n    }\r\n\r\n    const FRAMEWORK_USINGS = {\r\n        [MSTESTFRAMEWORK]: msTestUsing,\r\n        [NUNITTESTFRAMEWORK]: \"using NUnit.Framework;\",\r\n        [XUNITTESTFRAMEWORK]: \"using Xunit;\"\r\n    } as const;\r\n\r\n    return FRAMEWORK_USINGS[testFramework] || msTestUsing;\r\n}\r\n\r\n/**\r\n * Extract using Statements from source code and add them to test code if the alias is used.\r\n * Handles patterns like: using Partner = Partnerhub.PDP.Partner;\r\n * @param testCode The generated test code\r\n * @param sourceCode The original source code\r\n * @returns The test code with missing using aliases added\r\n */\r\nexport function repairUsingStatements(testCode: string, sourceCode: string): string {\r\n    if (!sourceCode || !testCode) {\r\n        return testCode;\r\n    }\r\n\r\n    // Extract using aliases from source code\r\n    // Pattern: using AliasName = Namespace.Type;\r\n    const usingAliasRegex = /^\\s*using\\s+(\\w+)\\s*=\\s*([^;]+);/gm;\r\n    const sourceAliases: Map<string, string> = new Map();\r\n    \r\n    let match;\r\n    while ((match = usingAliasRegex.exec(sourceCode)) !== null) {\r\n        const aliasName = match[1];\r\n        const fullStatement = match[0].trim();\r\n        sourceAliases.set(aliasName, fullStatement);\r\n    }\r\n\r\n    if (sourceAliases.size === 0) {\r\n        return testCode;\r\n    }\r\n\r\n    // Check which aliases are used in test code but not defined\r\n    const missingAliases: string[] = [];\r\n    const existingUsings = testCode.match(/^\\s*using\\s+.+;/gm) || [];\r\n    const existingUsingsTrimmed = existingUsings.map(u => u.trim());\r\n\r\n    for (const [aliasName, fullStatement] of sourceAliases) {\r\n        // Check if alias is used in test code (as a type name, using word boundary)\r\n        const aliasUsagePattern = new RegExp(`\\\\b${aliasName}\\\\b`);\r\n        if (aliasUsagePattern.test(testCode)) {\r\n            // Check if the using alias is already in test code\r\n            if (!existingUsingsTrimmed.includes(fullStatement)) {\r\n                missingAliases.push(fullStatement);\r\n            }\r\n        }\r\n    }\r\n\r\n    if (missingAliases.length === 0) {\r\n        return testCode;\r\n    }\r\n\r\n    // Add missing aliases\r\n    const result = repairMissingUsings(testCode, missingAliases);\r\n    console.log(\"repaired using aliases:\", missingAliases);\r\n    return result;\r\n}\r\n\r\n\r\n\r\n// const code = `\r\n// using Microsoft.VisualStudio.TestTools.UnitTesting;\r\n// using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\r\n// using System;\r\n// using System.Collections.Generic;\r\n// using System.Text;\r\n// using Moq;\r\n\r\n// namespace SharedSegments.Plugins.Tests.Workflows.Overview.HttpRequestParser.Utils\r\n// {\r\n//     [TestClass]\r\n//     public class SqmidHelperTests\r\n//     {\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithBlisId_ReturnsDecodedBlisId()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"BID=\" + Convert.ToBase64String(Encoding.UTF8.GetBytes(\"test_decodedBlisId_test\")) }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(\"decodedBlisId\", result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_ValidCookiesWithoutBlisId_ReturnsNull()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"SomeOtherValue=123\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.IsNull(result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_EmptyCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>();\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             IDictionary<string, string> cookies = null;\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n\r\n//         [TestMethod]\r\n//         public void GetSqmidFromCookie_UsrLocWithEmptyValue_ReturnsEmptyString()\r\n//         {\r\n//             // Arrange\r\n//             var cookies = new Dictionary<string, string>\r\n//             {\r\n//                 { \"USRLOC\", \"\" }\r\n//             };\r\n\r\n//             // Act\r\n//             var result = SqmidHelper.GetSqmidFromCookie(cookies);\r\n\r\n//             // Assert\r\n//             Assert.AreEqual(string.Empty, result);\r\n//         }\r\n//     }\r\n// }\r\n// `;\r\n\r\n// const cleanedCode = repairRequiredNameSpaces(code);\r\n// console.log(cleanedCode);"]}

package/out/llm/preparePrompt.d.ts

@@ -1,2 +1,3 @@
-export declare function prepareUtGenPrompt(sourceCodePath: string, sourceCode: string, dependency: string, isXapTest: boolean, testFramework: string): any[];
+export declare function prepareUtGenPrompt(sourceCodePath: string, sourceCode: string, dependency: string, isXapTest: boolean, testFramework: string, testFileExist: boolean, testFilePath?: string): any[];
 export declare function prepareUtAutoFixPrompt(prompts: any[], testCode: string, verifyError: string, isBuildFailure: boolean): any[];
+export declare function prepareMoreUtAutoFixPrompt(sourceCode: string, dependency: string, testCode: string, verifyError: string, isBuildFailure: boolean, isXapTest: boolean): any[];

package/out/llm/preparePrompt.js

@@ -35,6 +35,7 @@ var __importStar = (this && this.__importStar) || (function () {
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.prepareUtGenPrompt = prepareUtGenPrompt;
 exports.prepareUtAutoFixPrompt = prepareUtAutoFixPrompt;
+exports.prepareMoreUtAutoFixPrompt = prepareMoreUtAutoFixPrompt;
 const fs = __importStar(require("fs"));
 const path = __importStar(require("path"));
 const util = __importStar(require("util"));
@@ -44,10 +45,10 @@ const liquid = new liquidjs_1.Liquid({
     root: path.resolve(__dirname, 'prompt'), // Set root to current directory
     extname: '.liquid' // Treat included files as Markdown
 });
-function prepareUtGenPrompt(sourceCodePath, sourceCode, dependency, isXapTest, testFramework) {
+function prepareUtGenPrompt(sourceCodePath, sourceCode, dependency, isXapTest, testFramework, testFileExist, testFilePath) {
     const prompts = [];
     // prepare system prompt
-    const systemPrompt = prepareUtGenSystemPrompt(isXapTest, testFramework);
+    const systemPrompt = prepareUtGenSystemPrompt(isXapTest, testFramework, testFileExist);
     prompts.push(systemPrompt);
     // prepare user prompt
     const userPrompt = prepareUserInputPrompt(sourceCode, sourceCodePath, dependency);
@@ -57,6 +58,11 @@ function prepareUtGenPrompt(sourceCodePath, sourceCode, dependency, isXapTest, t
         const codeDependencyPrompt = prepareCodeDenpendencyPrompt(sourceCode, dependency);
         prompts.push({ role: 'user', content: codeDependencyPrompt });
     }
+    // prepare test code prompt when test file exists
+    if (testFileExist && testFilePath) {
+        const testCodePrompt = prepareOriginalTestCodePrompt(testFilePath);
+        prompts.push({ role: 'user', content: testCodePrompt });
+    }
     return prompts;
 }
 function prepareUserInputPrompt(sourceCode, sourceCodePath, dependency) {
@@ -77,6 +83,13 @@ function prepareCodeDenpendencyPrompt(sourceCode, dependency) {
     const dependencyPrompt = liquid.parseAndRenderSync(dependencyPromptTemplate, { dependency });
     return dependencyPrompt;
 }
+function prepareOriginalTestCodePrompt(testFilePath) {
+    const testCode = fs.readFileSync(testFilePath, 'utf8');
+    const testCodePromptPath = path.join(__dirname, 'prompt', 'moreUT', 'generateMoreUtTestCode.liquid');
+    const testCodePromptTemplate = fs.readFileSync(testCodePromptPath, 'utf8');
+    const testCodePrompt = liquid.parseAndRenderSync(testCodePromptTemplate, { testCode });
+    return testCodePrompt;
+}
 function prepareUtAutoFixPrompt(prompts, testCode, verifyError, isBuildFailure) {
     // const prompts = [];
     // prepare system prompt
@@ -91,8 +104,29 @@ function prepareUtAutoFixPrompt(prompts, testCode, verifyError, isBuildFailure)
     autofixPrompts.push({ role: 'user', content: autofixPrompt });
     return autofixPrompts;
 }
-function prepareUtGenSystemPrompt(isXapTest, testFramework) {
-    const systemPrompt = composeSystemPromptTemplate('generateUtTemplate.liquid', isXapTest, testFramework);
+function prepareMoreUtAutoFixPrompt(sourceCode, dependency, testCode, verifyError, isBuildFailure, isXapTest) {
+    const prompts = [];
+    // prepare autofix prompt as system prompt
+    const autofixPromptPath = path.join(__dirname, 'prompt', 'moreUT', 'generateMoreUtAutoFix.liquid');
+    const systemPromptTemplate = fs.readFileSync(autofixPromptPath, 'utf8');
+    const renderedSystemPrompt = liquid.parseAndRenderSync(systemPromptTemplate, { isBuildFailure, isXapCode: isXapTest });
+    const systemPrompt = util.format(renderedSystemPrompt, testCode, verifyError);
+    prompts.push({ role: 'system', content: systemPrompt });
+    // prepare user input prompt (source code)
+    const sourceCodePromptPath = path.join(__dirname, 'prompt', 'moreUT', 'generateMoreUTSourceCode.liquid');
+    const sourceCodePromptTemplate = fs.readFileSync(sourceCodePromptPath, 'utf8');
+    const userPrompt = liquid.parseAndRenderSync(sourceCodePromptTemplate, { sourceCode });
+    prompts.push({ role: 'user', content: userPrompt });
+    // prepare code dependency prompt
+    if (!!dependency) {
+        const codeDependencyPrompt = prepareCodeDenpendencyPrompt(sourceCode, dependency);
+        prompts.push({ role: 'user', content: codeDependencyPrompt });
+    }
+    return prompts;
+}
+function prepareUtGenSystemPrompt(isXapTest, testFramework, testFileExist) {
+    const template = testFileExist ? 'moreUT/generateMoreUtTemplate.liquid' : 'generateUtTemplate.liquid';
+    const systemPrompt = composeSystemPromptTemplate(template, isXapTest, testFramework);
     return { role: 'system', content: systemPrompt };
 }
 function composeSystemPromptTemplate(template, isXapTest, testFramework) {

package/out/llm/preparePrompt.js.map

@@ -1 +1 @@
-{"version":3,"file":"preparePrompt.js","sourceRoot":"","sources":["../../src/llm/preparePrompt.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAaA,gDAiBC;AAwBD,wDAeC;AArED,uCAAyB;AACzB,2CAA6B;AAC7B,2CAA6B;AAE7B,uCAAkC;AAGlC,2BAA2B;AAC3B,MAAM,MAAM,GAAG,IAAI,iBAAM,CAAC;IACzB,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,EAAE,gCAAgC;IACzE,OAAO,EAAE,SAAS,CAAC,mCAAmC;CACtD,CAAC,CAAC;AAEH,SAAgB,kBAAkB,CAAC,cAAsB,EAAE,UAAkB,EAAE,UAAkB,EAAE,SAAkB,EAAE,aAAqB;IAC3I,MAAM,OAAO,GAAG,EAAE,CAAC;IACnB,wBAAwB;IACxB,MAAM,YAAY,GAAG,wBAAwB,CAAC,SAAS,EAAE,aAAa,CAAC,CAAC;IACxE,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAE3B,sBAAsB;IACtB,MAAM,UAAU,GAAG,sBAAsB,CAAC,UAAU,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;IAClF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;IAEpD,iCAAiC;IACjC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,oBAAoB,GAAG,4BAA4B,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAClF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,SAAS,sBAAsB,CAAC,UAAkB,EAAE,cAAsB,EAAE,UAAkB;IAC7F,sBAAsB;IACtB,uFAAuF;IACvF,sEAAsE;IACtE,0DAA0D;IAC1D,2GAA2G;IAC3G,qBAAqB;IAErB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,4BAA4B,CAAC,CAAC;IACpF,MAAM,kBAAkB,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IACnE,MAAM,UAAU,GAAG,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IAEjF,OAAO,UAAU,CAAC;AACnB,CAAC;AAED,SAAS,4BAA4B,CAAC,UAAkB,EAAE,UAAkB;IAC3E,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,6BAA6B,CAAC,CAAC;IAC3F,MAAM,wBAAwB,GAAG,EAAE,CAAC,YAAY,CAAC,oBAAoB,EAAE,MAAM,CAAC,CAAC;IAC/E,MAAM,gBAAgB,GAAG,MAAM,CAAC,kBAAkB,CAAC,wBAAwB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IAC7F,OAAO,gBAAgB,CAAC;AACzB,CAAC;AAED,SAAgB,sBAAsB,CAAC,OAAc,EAAE,QAAgB,EAAE,WAAmB,EAAE,cAAuB;IACpH,sBAAsB;IACtB,wBAAwB;IACxB,4DAA4D;IAC5D,8BAA8B;IAC9B,MAAM,cAAc,GAAG,OAAO,IAAI,EAAE,CAAC;IAErC,yBAAyB;IACzB,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,0BAA0B,CAAC,CAAC;IACrF,MAAM,kBAAkB,GAAG,EAAE,CAAC,YAAY,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;IACtE,MAAM,gBAAgB,GAAG,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;IAC3F,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC3E,cAAc,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;IAE9D,OAAO,cAAc,CAAC;AACvB,CAAC;AAED,SAAS,wBAAwB,CAAC,SAAkB,EAAE,aAAqB;IAC1E,MAAM,YAAY,GAAG,2BAA2B,CAAC,2BAA2B,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IACxG,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;AAClD,CAAC;AAED,SAAS,2BAA2B,CAAC,QAAgB,EAAE,SAAkB,EAAE,aAAqB;IAC/F,sBAAsB;IACtB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAChE,MAAM,iBAAiB,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAEnE,mDAAmD;IACnD,MAAM,eAAe,GAAG,MAAM,CAAC,kBAAkB,CAAC,iBAAiB,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,CAAC;IAEnG,OAAO,eAAe,CAAC;AACxB,CAAC;AAED,6EAA6E;AAC7E,2EAA2E;AAE3E,gGAAgG;AAChG,+FAA+F;AAG/F,uEAAuE;AACvE,4DAA4D;AAC5D,mFAAmF;AACnF,qEAAqE;AACrE,wBAAwB;AAExB,eAAe;AACf,mCAAmC;AACnC,+TAA+T;AAC/T,gBAAgB;AAChB,YAAY;AACZ,8BAA8B;AAC9B,47GAA47G;AAC57G,uCAAuC;AACvC,aAAa;AACb,YAAY;AACZ,8BAA8B;AAC9B,kGAAkG;AAClG,aAAa;AACb,eAAe;AACf,iCAAiC;AACjC,yGAAyG;AACzG,eAAe;AACf,SAAS;AACT,sBAAsB;AACtB,IAAI","sourcesContent":["import * as fs from 'fs';\r\nimport * as path from 'path';\r\nimport * as util from 'util';\r\n\r\nimport { Liquid } from 'liquidjs';\r\nimport { getTestCodeNamespace } from '../utils/getTestCodeInfo';\r\n\r\n// Initialize Liquid engine\r\nconst liquid = new Liquid({\r\n\troot: path.resolve(__dirname, 'prompt'), // Set root to current directory\r\n\textname: '.liquid' // Treat included files as Markdown\r\n});\r\n\r\nexport function prepareUtGenPrompt(sourceCodePath: string, sourceCode: string, dependency: string, isXapTest: boolean, testFramework: string): any[] {\r\n\tconst prompts = [];\r\n\t// prepare system prompt\r\n\tconst systemPrompt = prepareUtGenSystemPrompt(isXapTest, testFramework);\r\n\tprompts.push(systemPrompt);\r\n\r\n\t// prepare user prompt\r\n\tconst userPrompt = prepareUserInputPrompt(sourceCode, sourceCodePath, dependency);\r\n\tprompts.push({ role: 'user', content: userPrompt });\r\n\r\n\t// prepare code dependency prompt\r\n\tif (!!dependency) {\r\n\t\tconst codeDependencyPrompt = prepareCodeDenpendencyPrompt(sourceCode, dependency);\r\n\t\tprompts.push({ role: 'user', content: codeDependencyPrompt });\r\n\t}\r\n\r\n\treturn prompts;\r\n}\r\n\r\nfunction prepareUserInputPrompt(sourceCode: string, sourceCodePath: string, dependency: string): string {\r\n\t// prepare user prompt\r\n\t// const userPromptPath = path.join(__dirname, 'prompt', 'generateUtUserInput.liquid');\r\n\t// const userPromptTemplate = fs.readFileSync(userPromptPath, 'utf8');\r\n\t// const namespace = getTestCodeNamespace(sourceCodePath);\r\n\t// const userPrompt = liquid.parseAndRenderSync(userPromptTemplate, { namespace, sourceCode, dependency });\r\n\t// return userPrompt;\r\n\r\n\tconst userPromptPath = path.join(__dirname, 'prompt', 'generateUtUserInput.liquid');\r\n\tconst userPromptTemplate = fs.readFileSync(userPromptPath, 'utf8');\r\n\tconst userPrompt = liquid.parseAndRenderSync(userPromptTemplate, { sourceCode });\r\n\r\n\treturn userPrompt;\r\n}\r\n\r\nfunction prepareCodeDenpendencyPrompt(sourceCode: string, dependency: string): string {\r\n\tconst dependencyPromptPath = path.join(__dirname, 'prompt', 'generateUtDependency.liquid');\r\n\tconst dependencyPromptTemplate = fs.readFileSync(dependencyPromptPath, 'utf8');\r\n\tconst dependencyPrompt = liquid.parseAndRenderSync(dependencyPromptTemplate, { dependency });\r\n\treturn dependencyPrompt;\r\n}\r\n\r\nexport function prepareUtAutoFixPrompt(prompts: any[], testCode: string, verifyError: string, isBuildFailure: boolean) {\r\n\t// const prompts = [];\r\n\t// prepare system prompt\r\n\t// const systemPrompt = prepareUtGenSystemPrompt(isXapTest);\r\n\t// prompts.push(systemPrompt);\r\n\tconst autofixPrompts = prompts || [];\r\n\r\n\t// prepare autofix prompt\r\n\tconst autofixPromptPath = path.join(__dirname, 'prompt', 'generateUtAutoFix.liquid');\r\n\tconst userPromptTemplate = fs.readFileSync(autofixPromptPath, 'utf8');\r\n\tconst renderedTemplate = liquid.parseAndRenderSync(userPromptTemplate, { isBuildFailure });\r\n\tconst autofixPrompt = util.format(renderedTemplate, testCode, verifyError);\r\n\tautofixPrompts.push({ role: 'user', content: autofixPrompt });\r\n\r\n\treturn autofixPrompts;\r\n}\r\n\r\nfunction prepareUtGenSystemPrompt(isXapTest: boolean, testFramework: string) {\r\n\tconst systemPrompt = composeSystemPromptTemplate('generateUtTemplate.liquid', isXapTest, testFramework);\r\n\treturn { role: 'system', content: systemPrompt };\r\n}\r\n\r\nfunction composeSystemPromptTemplate(template: string, isXapTest: boolean, testFramework: string) {\r\n\t// get prompt template\r\n\tconst promptTemplate = path.join(__dirname, 'prompt', template);\r\n\tconst mainPromptContent = fs.readFileSync(promptTemplate, 'utf-8');\r\n\r\n\t// render Liquid template（handle {% include %} tag）\r\n\tconst renderedContent = liquid.parseAndRenderSync(mainPromptContent, { isXapTest, testFramework });\r\n\r\n\treturn renderedContent;\r\n}\r\n\r\n// prepareUtGenSystemPrompt(false, \"MSTest\"); // Initialize the system prompt\r\n// prepareUtGenSystemPrompt(true, \"XUnit\"); // Initialize the system prompt\r\n\r\n// prepareUtAutoFixPrompt(\"\", \"\", \"\", \"MSBuild Errors\", true); // Initialize the auto-fix prompt\r\n// prepareUtAutoFixPrompt(\"\", \"\", \"\", \"VSTest result\", true); // Initialize the auto-fix prompt\r\n\r\n\r\n// function getPrompts(sourceCodePath, testFramework, codeDependency) {\r\n//     const code = fs.readFileSync(sourceCodePath, 'utf8');\r\n//     let guidancePromt = fs.readFileSync(\"D:\\\\atmp\\\\promptiteration.md\", 'utf8');\r\n//     guidancePromt = guidancePromt.replace(\"xUnit\", testFramework);\r\n//     const prompts = [\r\n\r\n//         // {\r\n//         //     \"role\": \"system\",\r\n//         //     \"content\": \"Keep your answers short and impersonal.\\nUse Markdown formatting in your answers.\\nMake sure to include the programming language name at the start of the Markdown code blocks.\\nAvoid wrapping the whole response in triple backticks.\\nYou can only give one reply for each conversation turn.\"\r\n//         // },\r\n//         {\r\n//             \"role\": \"user\",\r\n//             // \"content\": \"You are Roo, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.\\n\\n====\\n\\nRULES\\n\\n- MUST use xUnit test framework to generate test code.\\n- **MUST NOT** contain any comments or explanations in test code.\\n- Ensure the test code MUST has no syntax errors.\\n- Test public methods of class in source code only.\\n- **MUST NOT** access private members (fields or methods) of the source class in test code.\\n- Test method names must follow the pattern: [MethodName]_[Scenario]_[ExpectedResult].\\n- Return the test code in the format: ```C#\\n{Test code}\\n```.\\n- Mock all dependent methods, classes, and values not included in the user input using the Moq library.\\n- MUST include Test code in a namespace declaration and place all `using` statements above the namespace declaration.\\n- MUST import source code in the test code.\\n- MUST import all required .Net namespaces if any .Net class or interface are used in the test code. Like:\\n    - .NET built-in types (e.g., `List<T>`, `IEnumerable<T>`, `Task`, `Assert`, `Moq.Mock`, etc.).\\n    - Any attribute classes (e.g., `[TestMethod]`, `[TestClass]`, etc.).\\n    - Any LINQ or threading related classes (e.g., `System.Linq`, `System.Threading.Tasks`).\\n- All assignments and method arguments in test code MUST strictly match the declared types.\\n- When generating code, always consider the context in which the code is being used. Ensure that your changes are compatible with the existing codebase and that they follow the project's coding standards and best practices.\\n- Do not ask for more information than necessary. Use your knowledge to accomplish the user's request efficiently and effectively.\\n- The user will provide the source code directly in their message.\\n- Your goal is to try to accomplish the user's task, NOT engage in a back and forth conversation.\\n\\n====\\n\\nOBJECTIVE\\n\\nYou accomplish a given task iteratively, breaking it down into clear steps and working through them methodically.\\n\\n- **MUST** analyze the provided code and verify its logic by following these steps:\\n    1. Describe the functionality of each public method, including its inputs, outputs, and key logic.\\n    2. Identify all possible execution paths for public methods, including normal cases, edge cases, and error conditions.\\n    3. Explain any conditional branches, loops, or exception handling for public methods in the code.\\n    4. Based on this analysis, generate unit test cases that cover all identified execution paths, ensuring that mock data and assertions align with the code's logic.\\n- **MUST** remove all code analysis information in responsed test code.\\n- Absolutely forbidden to contains any other information in response except the test code.\\n- **MUST NOT** generate tests for the methods that are attributed with **[ExcludeFromCodeCoverage]**.\\n- **MUST NOT** generate tests for **private** methods.\\n- Analyze the user's task and set clear, achievable goals to accomplish it. Prioritize these goals in a logical order.\\n- Work through these goals sequentially, each goal should correspond to a distinct step in your problem-solving process.\\n\\n====\\n\\nUSER'S CUSTOM INSTRUCTIONS\\n\\nThe following additional instructions are provided by the user, and should be followed to the best of your ability without interfering with the TOOL USE guidelines.\\n\\nLanguage Preference:\\nYou should always speak and think in the English language.\"\r\n//             \"content\": guidancePromt\r\n//         },\r\n//         {\r\n//             \"role\": \"user\",\r\n//             \"content\": `generate unit test for the following csharp code:\\n\\`\\`\\`${code}\\`\\`\\``\r\n//         },\r\n//         // {\r\n//         //     \"role\": \"user\",\r\n//         //     \"content\": `following are the code dependencies of the source code:\\n${codeDependency}`\r\n//         // }\r\n//     ];\r\n//     return prompts;\r\n// }\r\n"]}
+{"version":3,"file":"preparePrompt.js","sourceRoot":"","sources":["../../src/llm/preparePrompt.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAaA,gDAuBC;AAgCD,wDAeC;AAED,gEA8BC;AAnHD,uCAAyB;AACzB,2CAA6B;AAC7B,2CAA6B;AAE7B,uCAAkC;AAGlC,2BAA2B;AAC3B,MAAM,MAAM,GAAG,IAAI,iBAAM,CAAC;IACzB,IAAI,EAAE,IAAI,CAAC,OAAO,CAAC,SAAS,EAAE,QAAQ,CAAC,EAAE,gCAAgC;IACzE,OAAO,EAAE,SAAS,CAAC,mCAAmC;CACtD,CAAC,CAAC;AAEH,SAAgB,kBAAkB,CAAC,cAAsB,EAAE,UAAkB,EAAE,UAAkB,EAAE,SAAkB,EAAE,aAAqB,EAAE,aAAsB,EAAE,YAAqB;IAC1L,MAAM,OAAO,GAAG,EAAE,CAAC;IACnB,wBAAwB;IACxB,MAAM,YAAY,GAAG,wBAAwB,CAAC,SAAS,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC;IACvF,OAAO,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IAE3B,sBAAsB;IACtB,MAAM,UAAU,GAAG,sBAAsB,CAAC,UAAU,EAAE,cAAc,EAAE,UAAU,CAAC,CAAC;IAClF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;IAEpD,iCAAiC;IACjC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,oBAAoB,GAAG,4BAA4B,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAClF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,iDAAiD;IACjD,IAAI,aAAa,IAAI,YAAY,EAAE,CAAC;QACnC,MAAM,cAAc,GAAG,6BAA6B,CAAC,YAAY,CAAC,CAAC;QACnE,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,CAAC,CAAC;IACzD,CAAC;IAED,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,SAAS,sBAAsB,CAAC,UAAkB,EAAE,cAAsB,EAAE,UAAkB;IAC7F,sBAAsB;IACtB,uFAAuF;IACvF,sEAAsE;IACtE,0DAA0D;IAC1D,2GAA2G;IAC3G,qBAAqB;IAErB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,4BAA4B,CAAC,CAAC;IACpF,MAAM,kBAAkB,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IACnE,MAAM,UAAU,GAAG,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IAEjF,OAAO,UAAU,CAAC;AACnB,CAAC;AAED,SAAS,4BAA4B,CAAC,UAAkB,EAAE,UAAkB;IAC3E,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,6BAA6B,CAAC,CAAC;IAC3F,MAAM,wBAAwB,GAAG,EAAE,CAAC,YAAY,CAAC,oBAAoB,EAAE,MAAM,CAAC,CAAC;IAC/E,MAAM,gBAAgB,GAAG,MAAM,CAAC,kBAAkB,CAAC,wBAAwB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IAC7F,OAAO,gBAAgB,CAAC;AACzB,CAAC;AAED,SAAS,6BAA6B,CAAC,YAAoB;IAC1D,MAAM,QAAQ,GAAG,EAAE,CAAC,YAAY,CAAC,YAAY,EAAE,MAAM,CAAC,CAAC;IACvD,MAAM,kBAAkB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,+BAA+B,CAAC,CAAC;IACrG,MAAM,sBAAsB,GAAG,EAAE,CAAC,YAAY,CAAC,kBAAkB,EAAE,MAAM,CAAC,CAAC;IAC3E,MAAM,cAAc,GAAG,MAAM,CAAC,kBAAkB,CAAC,sBAAsB,EAAE,EAAE,QAAQ,EAAE,CAAC,CAAC;IACvF,OAAO,cAAc,CAAC;AACvB,CAAC;AAED,SAAgB,sBAAsB,CAAC,OAAc,EAAE,QAAgB,EAAE,WAAmB,EAAE,cAAuB;IACpH,sBAAsB;IACtB,wBAAwB;IACxB,4DAA4D;IAC5D,8BAA8B;IAC9B,MAAM,cAAc,GAAG,OAAO,IAAI,EAAE,CAAC;IAErC,yBAAyB;IACzB,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,0BAA0B,CAAC,CAAC;IACrF,MAAM,kBAAkB,GAAG,EAAE,CAAC,YAAY,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;IACtE,MAAM,gBAAgB,GAAG,MAAM,CAAC,kBAAkB,CAAC,kBAAkB,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC;IAC3F,MAAM,aAAa,GAAG,IAAI,CAAC,MAAM,CAAC,gBAAgB,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC3E,cAAc,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,CAAC,CAAC;IAE9D,OAAO,cAAc,CAAC;AACvB,CAAC;AAED,SAAgB,0BAA0B,CACzC,UAAkB,EAClB,UAAkB,EAClB,QAAgB,EAChB,WAAmB,EACnB,cAAuB,EACvB,SAAkB;IAElB,MAAM,OAAO,GAAU,EAAE,CAAC;IAE1B,0CAA0C;IAC1C,MAAM,iBAAiB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,8BAA8B,CAAC,CAAC;IACnG,MAAM,oBAAoB,GAAG,EAAE,CAAC,YAAY,CAAC,iBAAiB,EAAE,MAAM,CAAC,CAAC;IACxE,MAAM,oBAAoB,GAAG,MAAM,CAAC,kBAAkB,CAAC,oBAAoB,EAAE,EAAE,cAAc,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IACvH,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,QAAQ,EAAE,WAAW,CAAC,CAAC;IAC9E,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC,CAAC;IAExD,0CAA0C;IAC1C,MAAM,oBAAoB,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,EAAE,iCAAiC,CAAC,CAAC;IACzG,MAAM,wBAAwB,GAAG,EAAE,CAAC,YAAY,CAAC,oBAAoB,EAAE,MAAM,CAAC,CAAC;IAC/E,MAAM,UAAU,GAAG,MAAM,CAAC,kBAAkB,CAAC,wBAAwB,EAAE,EAAE,UAAU,EAAE,CAAC,CAAC;IACvF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,UAAU,EAAE,CAAC,CAAC;IAEpD,iCAAiC;IACjC,IAAI,CAAC,CAAC,UAAU,EAAE,CAAC;QAClB,MAAM,oBAAoB,GAAG,4BAA4B,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;QAClF,OAAO,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC,CAAC;IAC/D,CAAC;IAED,OAAO,OAAO,CAAC;AAChB,CAAC;AAED,SAAS,wBAAwB,CAAC,SAAkB,EAAE,aAAqB,EAAE,aAAsB;IAClG,MAAM,QAAQ,GAAG,aAAa,CAAC,CAAC,CAAC,sCAAsC,CAAC,CAAC,CAAC,2BAA2B,CAAC;IACtG,MAAM,YAAY,GAAG,2BAA2B,CAAC,QAAQ,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IACrF,OAAO,EAAE,IAAI,EAAE,QAAQ,EAAE,OAAO,EAAE,YAAY,EAAE,CAAC;AAClD,CAAC;AAED,SAAS,2BAA2B,CAAC,QAAgB,EAAE,SAAkB,EAAE,aAAqB;IAC/F,sBAAsB;IACtB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,EAAE,QAAQ,CAAC,CAAC;IAChE,MAAM,iBAAiB,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAEnE,mDAAmD;IACnD,MAAM,eAAe,GAAG,MAAM,CAAC,kBAAkB,CAAC,iBAAiB,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,CAAC,CAAC;IAEnG,OAAO,eAAe,CAAC;AACxB,CAAC;AAED,6EAA6E;AAC7E,2EAA2E;AAE3E,gGAAgG;AAChG,+FAA+F;AAG/F,uEAAuE;AACvE,4DAA4D;AAC5D,mFAAmF;AACnF,qEAAqE;AACrE,wBAAwB;AAExB,eAAe;AACf,mCAAmC;AACnC,+TAA+T;AAC/T,gBAAgB;AAChB,YAAY;AACZ,8BAA8B;AAC9B,47GAA47G;AAC57G,uCAAuC;AACvC,aAAa;AACb,YAAY;AACZ,8BAA8B;AAC9B,kGAAkG;AAClG,aAAa;AACb,eAAe;AACf,iCAAiC;AACjC,yGAAyG;AACzG,eAAe;AACf,SAAS;AACT,sBAAsB;AACtB,IAAI","sourcesContent":["import * as fs from 'fs';\r\nimport * as path from 'path';\r\nimport * as util from 'util';\r\n\r\nimport { Liquid } from 'liquidjs';\r\nimport { getTestCodeNamespace } from '../utils/getTestCodeInfo';\r\n\r\n// Initialize Liquid engine\r\nconst liquid = new Liquid({\r\n\troot: path.resolve(__dirname, 'prompt'), // Set root to current directory\r\n\textname: '.liquid' // Treat included files as Markdown\r\n});\r\n\r\nexport function prepareUtGenPrompt(sourceCodePath: string, sourceCode: string, dependency: string, isXapTest: boolean, testFramework: string, testFileExist: boolean, testFilePath?: string): any[] {\r\n\tconst prompts = [];\r\n\t// prepare system prompt\r\n\tconst systemPrompt = prepareUtGenSystemPrompt(isXapTest, testFramework, testFileExist);\r\n\tprompts.push(systemPrompt);\r\n\r\n\t// prepare user prompt\r\n\tconst userPrompt = prepareUserInputPrompt(sourceCode, sourceCodePath, dependency);\r\n\tprompts.push({ role: 'user', content: userPrompt });\r\n\r\n\t// prepare code dependency prompt\r\n\tif (!!dependency) {\r\n\t\tconst codeDependencyPrompt = prepareCodeDenpendencyPrompt(sourceCode, dependency);\r\n\t\tprompts.push({ role: 'user', content: codeDependencyPrompt });\r\n\t}\r\n\r\n\t// prepare test code prompt when test file exists\r\n\tif (testFileExist && testFilePath) {\r\n\t\tconst testCodePrompt = prepareOriginalTestCodePrompt(testFilePath);\r\n\t\tprompts.push({ role: 'user', content: testCodePrompt });\r\n\t}\r\n\r\n\treturn prompts;\r\n}\r\n\r\nfunction prepareUserInputPrompt(sourceCode: string, sourceCodePath: string, dependency: string): string {\r\n\t// prepare user prompt\r\n\t// const userPromptPath = path.join(__dirname, 'prompt', 'generateUtUserInput.liquid');\r\n\t// const userPromptTemplate = fs.readFileSync(userPromptPath, 'utf8');\r\n\t// const namespace = getTestCodeNamespace(sourceCodePath);\r\n\t// const userPrompt = liquid.parseAndRenderSync(userPromptTemplate, { namespace, sourceCode, dependency });\r\n\t// return userPrompt;\r\n\r\n\tconst userPromptPath = path.join(__dirname, 'prompt', 'generateUtUserInput.liquid');\r\n\tconst userPromptTemplate = fs.readFileSync(userPromptPath, 'utf8');\r\n\tconst userPrompt = liquid.parseAndRenderSync(userPromptTemplate, { sourceCode });\r\n\r\n\treturn userPrompt;\r\n}\r\n\r\nfunction prepareCodeDenpendencyPrompt(sourceCode: string, dependency: string): string {\r\n\tconst dependencyPromptPath = path.join(__dirname, 'prompt', 'generateUtDependency.liquid');\r\n\tconst dependencyPromptTemplate = fs.readFileSync(dependencyPromptPath, 'utf8');\r\n\tconst dependencyPrompt = liquid.parseAndRenderSync(dependencyPromptTemplate, { dependency });\r\n\treturn dependencyPrompt;\r\n}\r\n\r\nfunction prepareOriginalTestCodePrompt(testFilePath: string): string {\r\n\tconst testCode = fs.readFileSync(testFilePath, 'utf8');\r\n\tconst testCodePromptPath = path.join(__dirname, 'prompt', 'moreUT', 'generateMoreUtTestCode.liquid');\r\n\tconst testCodePromptTemplate = fs.readFileSync(testCodePromptPath, 'utf8');\r\n\tconst testCodePrompt = liquid.parseAndRenderSync(testCodePromptTemplate, { testCode });\r\n\treturn testCodePrompt;\r\n}\r\n\r\nexport function prepareUtAutoFixPrompt(prompts: any[], testCode: string, verifyError: string, isBuildFailure: boolean) {\r\n\t// const prompts = [];\r\n\t// prepare system prompt\r\n\t// const systemPrompt = prepareUtGenSystemPrompt(isXapTest);\r\n\t// prompts.push(systemPrompt);\r\n\tconst autofixPrompts = prompts || [];\r\n\r\n\t// prepare autofix prompt\r\n\tconst autofixPromptPath = path.join(__dirname, 'prompt', 'generateUtAutoFix.liquid');\r\n\tconst userPromptTemplate = fs.readFileSync(autofixPromptPath, 'utf8');\r\n\tconst renderedTemplate = liquid.parseAndRenderSync(userPromptTemplate, { isBuildFailure });\r\n\tconst autofixPrompt = util.format(renderedTemplate, testCode, verifyError);\r\n\tautofixPrompts.push({ role: 'user', content: autofixPrompt });\r\n\r\n\treturn autofixPrompts;\r\n}\r\n\r\nexport function prepareMoreUtAutoFixPrompt(\r\n\tsourceCode: string,\r\n\tdependency: string,\r\n\ttestCode: string,\r\n\tverifyError: string,\r\n\tisBuildFailure: boolean,\r\n\tisXapTest: boolean,\r\n): any[] {\r\n\tconst prompts: any[] = [];\r\n\r\n\t// prepare autofix prompt as system prompt\r\n\tconst autofixPromptPath = path.join(__dirname, 'prompt', 'moreUT', 'generateMoreUtAutoFix.liquid');\r\n\tconst systemPromptTemplate = fs.readFileSync(autofixPromptPath, 'utf8');\r\n\tconst renderedSystemPrompt = liquid.parseAndRenderSync(systemPromptTemplate, { isBuildFailure, isXapCode: isXapTest });\r\n\tconst systemPrompt = util.format(renderedSystemPrompt, testCode, verifyError);\r\n\tprompts.push({ role: 'system', content: systemPrompt });\r\n\r\n\t// prepare user input prompt (source code)\r\n\tconst sourceCodePromptPath = path.join(__dirname, 'prompt', 'moreUT', 'generateMoreUTSourceCode.liquid');\r\n\tconst sourceCodePromptTemplate = fs.readFileSync(sourceCodePromptPath, 'utf8');\r\n\tconst userPrompt = liquid.parseAndRenderSync(sourceCodePromptTemplate, { sourceCode });\r\n\tprompts.push({ role: 'user', content: userPrompt });\r\n\r\n\t// prepare code dependency prompt\r\n\tif (!!dependency) {\r\n\t\tconst codeDependencyPrompt = prepareCodeDenpendencyPrompt(sourceCode, dependency);\r\n\t\tprompts.push({ role: 'user', content: codeDependencyPrompt });\r\n\t}\r\n\r\n\treturn prompts;\r\n}\r\n\r\nfunction prepareUtGenSystemPrompt(isXapTest: boolean, testFramework: string, testFileExist: boolean) {\r\n\tconst template = testFileExist ? 'moreUT/generateMoreUtTemplate.liquid' : 'generateUtTemplate.liquid';\r\n\tconst systemPrompt = composeSystemPromptTemplate(template, isXapTest, testFramework);\r\n\treturn { role: 'system', content: systemPrompt };\r\n}\r\n\r\nfunction composeSystemPromptTemplate(template: string, isXapTest: boolean, testFramework: string) {\r\n\t// get prompt template\r\n\tconst promptTemplate = path.join(__dirname, 'prompt', template);\r\n\tconst mainPromptContent = fs.readFileSync(promptTemplate, 'utf-8');\r\n\r\n\t// render Liquid template（handle {% include %} tag）\r\n\tconst renderedContent = liquid.parseAndRenderSync(mainPromptContent, { isXapTest, testFramework });\r\n\r\n\treturn renderedContent;\r\n}\r\n\r\n// prepareUtGenSystemPrompt(false, \"MSTest\"); // Initialize the system prompt\r\n// prepareUtGenSystemPrompt(true, \"XUnit\"); // Initialize the system prompt\r\n\r\n// prepareUtAutoFixPrompt(\"\", \"\", \"\", \"MSBuild Errors\", true); // Initialize the auto-fix prompt\r\n// prepareUtAutoFixPrompt(\"\", \"\", \"\", \"VSTest result\", true); // Initialize the auto-fix prompt\r\n\r\n\r\n// function getPrompts(sourceCodePath, testFramework, codeDependency) {\r\n//     const code = fs.readFileSync(sourceCodePath, 'utf8');\r\n//     let guidancePromt = fs.readFileSync(\"D:\\\\atmp\\\\promptiteration.md\", 'utf8');\r\n//     guidancePromt = guidancePromt.replace(\"xUnit\", testFramework);\r\n//     const prompts = [\r\n\r\n//         // {\r\n//         //     \"role\": \"system\",\r\n//         //     \"content\": \"Keep your answers short and impersonal.\\nUse Markdown formatting in your answers.\\nMake sure to include the programming language name at the start of the Markdown code blocks.\\nAvoid wrapping the whole response in triple backticks.\\nYou can only give one reply for each conversation turn.\"\r\n//         // },\r\n//         {\r\n//             \"role\": \"user\",\r\n//             // \"content\": \"You are Roo, a highly skilled software engineer with extensive knowledge in many programming languages, frameworks, design patterns, and best practices.\\n\\n====\\n\\nRULES\\n\\n- MUST use xUnit test framework to generate test code.\\n- **MUST NOT** contain any comments or explanations in test code.\\n- Ensure the test code MUST has no syntax errors.\\n- Test public methods of class in source code only.\\n- **MUST NOT** access private members (fields or methods) of the source class in test code.\\n- Test method names must follow the pattern: [MethodName]_[Scenario]_[ExpectedResult].\\n- Return the test code in the format: ```C#\\n{Test code}\\n```.\\n- Mock all dependent methods, classes, and values not included in the user input using the Moq library.\\n- MUST include Test code in a namespace declaration and place all `using` statements above the namespace declaration.\\n- MUST import source code in the test code.\\n- MUST import all required .Net namespaces if any .Net class or interface are used in the test code. Like:\\n    - .NET built-in types (e.g., `List<T>`, `IEnumerable<T>`, `Task`, `Assert`, `Moq.Mock`, etc.).\\n    - Any attribute classes (e.g., `[TestMethod]`, `[TestClass]`, etc.).\\n    - Any LINQ or threading related classes (e.g., `System.Linq`, `System.Threading.Tasks`).\\n- All assignments and method arguments in test code MUST strictly match the declared types.\\n- When generating code, always consider the context in which the code is being used. Ensure that your changes are compatible with the existing codebase and that they follow the project's coding standards and best practices.\\n- Do not ask for more information than necessary. Use your knowledge to accomplish the user's request efficiently and effectively.\\n- The user will provide the source code directly in their message.\\n- Your goal is to try to accomplish the user's task, NOT engage in a back and forth conversation.\\n\\n====\\n\\nOBJECTIVE\\n\\nYou accomplish a given task iteratively, breaking it down into clear steps and working through them methodically.\\n\\n- **MUST** analyze the provided code and verify its logic by following these steps:\\n    1. Describe the functionality of each public method, including its inputs, outputs, and key logic.\\n    2. Identify all possible execution paths for public methods, including normal cases, edge cases, and error conditions.\\n    3. Explain any conditional branches, loops, or exception handling for public methods in the code.\\n    4. Based on this analysis, generate unit test cases that cover all identified execution paths, ensuring that mock data and assertions align with the code's logic.\\n- **MUST** remove all code analysis information in responsed test code.\\n- Absolutely forbidden to contains any other information in response except the test code.\\n- **MUST NOT** generate tests for the methods that are attributed with **[ExcludeFromCodeCoverage]**.\\n- **MUST NOT** generate tests for **private** methods.\\n- Analyze the user's task and set clear, achievable goals to accomplish it. Prioritize these goals in a logical order.\\n- Work through these goals sequentially, each goal should correspond to a distinct step in your problem-solving process.\\n\\n====\\n\\nUSER'S CUSTOM INSTRUCTIONS\\n\\nThe following additional instructions are provided by the user, and should be followed to the best of your ability without interfering with the TOOL USE guidelines.\\n\\nLanguage Preference:\\nYou should always speak and think in the English language.\"\r\n//             \"content\": guidancePromt\r\n//         },\r\n//         {\r\n//             \"role\": \"user\",\r\n//             \"content\": `generate unit test for the following csharp code:\\n\\`\\`\\`${code}\\`\\`\\``\r\n//         },\r\n//         // {\r\n//         //     \"role\": \"user\",\r\n//         //     \"content\": `following are the code dependencies of the source code:\\n${codeDependency}`\r\n//         // }\r\n//     ];\r\n//     return prompts;\r\n// }\r\n"]}

package/out/llm/prompt/buildAfGuidelines.liquid

@@ -1,15 +1,15 @@
 # Build Failure Fix Guideline:
 - Case 1: The type or namespace name '{Class/Type}' does not exist in the namespace '{namespace}' (are you missing an assembly reference?)
   Solution: Check the source code, code dependencies. Add the required `using` statements or assembly references.
-- Case 2: No overload for method 'Execute' takes {number} arguments
-  Solution: Refer to section [The method parameters of Execute in Xap instance are different from the source code.] of Xap Unit Test Guidelines,  Adjust the method parameters to match the correct overload as per the source code.
-- Case 3: [CS0144] Cannot create an instance of the abstract type or interface '{class / bondstucture}'
+- Case 2: [CS0144] Cannot create an instance of the abstract type or interface '{class / bondstucture}'
   Solution: use ExecutionServices.CreateInstance<I{ClassName}>() to create Xap class instance or use ExecutionServices.CreateInstance<{StructName}>() to create Bond struct instance.
-- Case 4: [CS1503] cannot convert from '{Class / Bond type}' to '{Class / Bond type}'
+- Case 3: [CS1503] cannot convert from '{Class / Bond type}' to '{Class / Bond type}'
   Solution: Do explicitly type convert.
-- Case 5: [CS0037] Cannot convert null to '{Class / Bond type}' because it is a non-nullable value type
+- Case 4: [CS0037] Cannot convert null to '{Class / Bond type}' because it is a non-nullable value type
   Solution: MUST assign non-null and valid value.
-- Case 6: [CS0246] The type or namespace name '{class / interface}' could not be found (are you missing a using directive or an assembly reference?)
+- Case 5: [CS0246] The type or namespace name '{class / interface}' could not be found (are you missing a using directive or an assembly reference?)
   Solution: MUST import the **namespace** of {class / interface}
-- Case 7: [CS1501] No overload for method 'Execute' takes 2 arguments
-  Solution: - The parameters of **Execute** method  with types: **PluginServices**, **PluginOutput<T>**, or decorated with **[ConfigFile("{configFileName}")]** MUST be deleted when using Xap class instance to call it.
+- Case 6: [CS1501] No overload for method 'Execute' takes {number} arguments
+  Solution: The parameters of **Execute** method with types: **PluginServices**, **PluginOutput<T>**, **CollectionPluginOutput<IEnumerable<T>>**, or decorated with **[ConfigFile("{configFileName}")]** MUST be deleted when using Xap class instance to call it.
+- Case 7: [CA2000] Call System.IDisposable.Dispose on object created before all references to it are out of scope
+  Solution: Add "using" keyword before the variable declaration on that line, keep everything else unchanged.

package/out/llm/prompt/general/generalUtGuidelines.liquid

@@ -2,11 +2,14 @@
 
 - MUST use {{testFramework}} test framework to generate tests.
 - **MUST NOT** contain any comments or explanations in test code.
+- **MUST** preserve any copyright or file header information from the given source code.
 - Ensure the test code MUST has no syntax errors.
 - Test method names must follow the pattern: [MethodName]_[Scenario]_[ExpectedResult].
 - Return the test code in the format: ```C#\n{Test code}\n```.
 - MUST include Test code in a namespace declaration and place all `using` statements above the namespace declaration.
 - MUST import source code namespace in test code.
+- MUST import all namespaces from source code that are used in test code (e.g., types used in method parameters, return values, or mock setups).
+- MUST import all using aliases from source code that are used in test code. If source code contains `using Alias = SomeNamespace.SomeType;` and test code uses `Alias`, MUST include this using alias statement.
 - MUST import all required .Net namespaces if any .Net class or interface are used in the test code. Like:
     - .NET built-in types (e.g., `List<T>`, `IEnumerable<T>`, `Task`, `Assert`, `Moq.Mock`, etc.).
     - Any attribute classes (e.g., `[TestMethod]`, `[TestClass]`, etc.).

package/out/llm/prompt/moreUT/generateMoreUTSourceCode.liquid

@@ -0,0 +1,4 @@
+The source code to be tested:
+```csharp
+{{ sourceCode }}
+```

package/out/llm/prompt/moreUT/generateMoreUtAutoFix.liquid

@@ -0,0 +1,45 @@
+{%- if isBuildFailure -%}
+  {%- assign failedStage = "MSBuild" -%}
+  {%- assign failedTitle = "MSBuild failure errors:" -%}
+{%- else -%}
+  {%- assign failedStage = "vstest" -%}
+  {%- assign failedTitle = "vstest failure test errors:" -%}
+{%- endif -%}
+
+You are a highly skilled software engineer specializing in C# unit testing with extensive knowledge in test frameworks, test patterns, and best practices.
+
+## TASK
+Fix the unit test code that failed to {{failedStage}}. The user will provide the source code and its dependencies.
+
+## RULES
+- Return the **complete** fixed test file, not just the changed parts.
+- **ONLY fix the test method(s) that contain the error line(s)**. Do NOT modify, refactor, or change any other test methods.
+- Test methods that do NOT have errors must remain exactly unchanged - do not rename, restructure, or "improve" them.
+- Only fix the errors mentioned in the error messages.
+- **Keep existing comments unchanged**. Do NOT add new comments or explanations in the code.
+- Ensure all `using` statements are at the top of the file.
+- Ensure the code has no syntax errors.
+
+The following are the guidelines used when generating the test code. You may refer to them when fixing the test code:
+
+{%- include 'moreUT/utGenerationGuidelines' -%}
+
+## OUTPUT FORMAT
+Return the fixed test code in the format:
+```C#
+{Complete test code}
+```
+
+====
+
+Current test code (needs fixing):
+%s
+
+====
+
+{{failedTitle}}:
+%s
+
+{% if isBuildFailure %}
+{%- include 'buildAfGuidelines' -%}
+{%- endif -%}