@ai-dev-tools/csharp-copilot-core 0.0.19

package/out/gen/csharpUtGen.js

@@ -0,0 +1,151 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.generateCsharpUtCode = generateCsharpUtCode;
+const fs = __importStar(require("fs"));
+const path = __importStar(require("path"));
+const constants_1 = require("../types/constants");
+const fileUtils_1 = require("../utils/fileUtils");
+const autoFix_1 = require("./autoFix");
+const checkXapCode_1 = require("../utils/checkXapCode");
+const detectTestFramework_1 = require("../utils/detectTestFramework");
+const ensureValidLLMResponse_1 = require("./ensureValidLLMResponse");
+const generateCodeDependency_1 = require("../analyze/generateCodeDependency");
+const getTestFile_1 = require("../utils/getTestFile");
+const installAnalyzerTool_1 = require("../analyze/installAnalyzerTool");
+const preGenProcess_1 = require("./preGen/preGenProcess");
+const preparePrompt_1 = require("../llm/preparePrompt");
+const removeFailedTestMethods_1 = require("../utils/removeFailedTestMethods");
+const appInsightTelemetry_1 = require("../metrics/appInsightTelemetry");
+const RETRYTIMES = 3;
+async function generateCsharpUtCode(sourceCodePath, testCodePath, enableAutoFix, keepFailedTests, source = constants_1.CSHARPUTGEN, isBenchmark = false) {
+    const startTime = Date.now();
+    console.log(`*****************Generating C# unit test code for: ${sourceCodePath}*****************`);
+    console.log(`Current work dir: ${process.cwd()}`);
+    // send usage metrics
+    (0, appInsightTelemetry_1.sendUserUsageEvent)(source);
+    sourceCodePath = path.normalize(sourceCodePath);
+    if (!fs.existsSync(sourceCodePath)) {
+        console.error(`Source code file not found: ${sourceCodePath}`);
+        return;
+    }
+    // install nuget cli tool to analyze code
+    (0, installAnalyzerTool_1.installOrUpdateAnalyzerTool)();
+    let sourceCode = fs.readFileSync(sourceCodePath, 'utf-8');
+    // check if source code is xap code, if so, use xap code generation prompt
+    const isXapCode = (0, checkXapCode_1.checkXapCode)(sourceCode);
+    sourceCode = await (0, preGenProcess_1.preGenProcess)(sourceCode, sourceCodePath, isXapCode);
+    if (!sourceCode) {
+        console.error(`Skip generate test for source code: ${sourceCodePath}`);
+        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
+    }
+    // get code dependencies
+    let codeDependency = await (0, generateCodeDependency_1.generateCodeDependency)(sourceCodePath);
+    // detect existing test file or create a new one by source code path
+    const { testFilePath, testProjectPath, testFileExist } = (0, getTestFile_1.getOrCreatTestFile)(sourceCodePath, testCodePath, isXapCode);
+    if (!testFilePath || !testProjectPath) {
+        console.error(`Failed to get or create test file for source code: ${sourceCodePath}`);
+        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
+    }
+    // get test framework
+    const testFramework = (0, detectTestFramework_1.detectTestFramework)(testProjectPath);
+    // prepare prompt
+    let prompt = (0, preparePrompt_1.prepareUtGenPrompt)(sourceCodePath, sourceCode, codeDependency, isXapCode, testFramework);
+    // let prompt = getPrompts(sourceCodePath, testFramework, codeDependency);
+    // call LLM service to gen code with retry
+    const testCode = await (0, ensureValidLLMResponse_1.ensureValidLLMResponse)(prompt, isXapCode);
+    // write test code to test project
+    // To run vstest, need to write the test code to test project first, then build the test project and run the tests
+    (0, fileUtils_1.writeTestCode)(testCode, testFilePath);
+    // autofix
+    let autoFixResult = { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };
+    if (enableAutoFix) {
+        const isVsPlugin = source === constants_1.CSVSPLUGIN;
+        autoFixResult = await (0, autoFix_1.autoFixTestCode)(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);
+        // autoFixResult = await autoRetryTestCode(prompt, sourceCode, testCode, testProjectPath, testFilePath, isXapTest);
+    }
+    if (autoFixResult.success) {
+        console.log(`Gerate UT success time: ${(Date.now() - startTime) / 1000} seconds`);
+        return autoFixResult;
+    }
+    // if retry max times or test generation failed, and keepFailedTests is false, delete failed test file or remove failed test methods
+    if (!keepFailedTests) {
+        // keep succeed test code, or delete test file
+        if (!autoFixResult.testResult || autoFixResult.testResult.passed === 0) {
+            !testFileExist && (0, fileUtils_1.deleteTestFileIfExists)(testFilePath);
+            console.log(`Failed to generate csharp test code, deleted test file: ${testFilePath}`);
+        }
+        else {
+            (0, removeFailedTestMethods_1.removeFailedTestCasesInTestFile)(autoFixResult.testResult, testFilePath);
+            console.log(`write succeed test code to file: ${testFilePath}`);
+        }
+    }
+    else {
+        console.log(`keep failed test file: ${testFilePath}`);
+    }
+    console.log(`Gerate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);
+    return autoFixResult;
+}
+// generateCsharpUtCode("D:\\code\\IndexServe\\private\\indexserve\\RecommendationEngine\\RecoPS\\PSTableClient\\PSTableClient.cs", undefined, false, true, false);
+// generateCsharpUtCode("D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Overview\\HttpRequestParser\\Utils\\SqmidHelper.cs", undefined, true, true, false);
+// generateCsharpUtCode("D:\\code\\CS.Service.Fundamental\\SharedSegments\\SharedSegments\\SharedSegments.Plugins\\Workflows\\Eplant\\ResponseBuilders\\EplantMigrateResponseBuilder.cs", undefined, true, true, true);
+// generateCsharpUtCode("D:\\code\\UCMDynamics\\private\\DynamicCode\\Plugin\\UCMDynamics.AppointmentPlugin\\AppointmentOperation.cs", undefined, false, true, false);
+// generateCsharpUtCode("D:\\code\\AIUnitTestExample\\src\\ToListinator.Analyzers\\ToListCountAnalyzer.cs", undefined, true, true, false);
+// function getPrompts(sourceCodePath, testFramework, codeDependency) {
+//     const code = fs.readFileSync(sourceCodePath, 'utf8');
+//     let guidancePromt = fs.readFileSync("D:\\atmp\\promptiteration.md", 'utf8');
+//     guidancePromt = guidancePromt.replace("xUnit", testFramework);
+//     const prompts = [
+//         // {
+//         //     "role": "system",
+//         //     "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."
+//         // },
+//         {
+//             "role": "user",
+//             // "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."
+//             "content": guidancePromt
+//         },
+//         {
+//             "role": "user",
+//             "content": `generate unit test for the following csharp code:\n\`\`\`${code}\`\`\``
+//         },
+//         // {
+//         //     "role": "user",
+//         //     "content": `following are the code dependencies of the source code:\n${codeDependency}`
+//         // }
+//     ];
+//     return prompts;
+// }
+//# sourceMappingURL=csharpUtGen.js.map

package/out/gen/csharpUtGen.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"csharpUtGen.js","sourceRoot":"","sources":["../../src/gen/csharpUtGen.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;AAqBA,oDAwFC;AA7GD,uCAAyB;AACzB,2CAA6B;AAE7B,kDAA6D;AAC7D,kDAA2E;AAG3E,uCAA4C;AAC5C,wDAAqD;AACrD,sEAAmE;AACnE,qEAAkE;AAClE,8EAA2E;AAC3E,sDAA0D;AAC1D,wEAA6E;AAC7E,0DAAuD;AACvD,wDAA0D;AAC1D,8EAAmF;AACnF,wEAAoE;AAEpE,MAAM,UAAU,GAAG,CAAC,CAAC;AAEd,KAAK,UAAU,oBAAoB,CACtC,cAAsB,EACtB,YAAoB,EACpB,aAAsB,EACtB,eAAwB,EACxB,SAAiB,uBAAW,EAC5B,cAAuB,KAAK;IAE5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;IAE7B,OAAO,CAAC,GAAG,CAAC,sDAAsD,cAAc,mBAAmB,CAAC,CAAC;IACrG,OAAO,CAAC,GAAG,CAAC,qBAAqB,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAElD,qBAAqB;IACrB,IAAA,wCAAkB,EAAC,MAAM,CAAC,CAAC;IAE3B,cAAc,GAAG,IAAI,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAChD,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;QACjC,OAAO,CAAC,KAAK,CAAC,+BAA+B,cAAc,EAAE,CAAC,CAAC;QAC/D,OAAO;IACX,CAAC;IAED,yCAAyC;IACzC,IAAA,iDAA2B,GAAE,CAAC;IAE9B,IAAI,UAAU,GAAG,EAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;IAE1D,0EAA0E;IAC1E,MAAM,SAAS,GAAG,IAAA,2BAAY,EAAC,UAAU,CAAC,CAAC;IAE3C,UAAU,GAAG,MAAM,IAAA,6BAAa,EAAC,UAAU,EAAE,cAAc,EAAE,SAAS,CAAC,CAAC;IACxE,IAAI,CAAC,UAAU,EAAE,CAAC;QACd,OAAO,CAAC,KAAK,CAAC,uCAAuC,cAAc,EAAE,CAAC,CAAC;QACvE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,wBAAwB;IACxB,IAAI,cAAc,GAAG,MAAM,IAAA,+CAAsB,EAAC,cAAc,CAAC,CAAC;IAElE,oEAAoE;IACpE,MAAM,EAAE,YAAY,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,IAAA,gCAAkB,EAAC,cAAc,EAAE,YAAY,EAAE,SAAS,CAAC,CAAC;IACrH,IAAI,CAAC,YAAY,IAAI,CAAC,eAAe,EAAE,CAAC;QACpC,OAAO,CAAC,KAAK,CAAC,sDAAsD,cAAc,EAAE,CAAC,CAAC;QACtF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,YAAY,EAAE,SAAS,EAAE,eAAe,EAAE,SAAS,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACtJ,CAAC;IAED,qBAAqB;IACrB,MAAM,aAAa,GAAG,IAAA,yCAAmB,EAAC,eAAe,CAAC,CAAC;IAE3D,iBAAiB;IACjB,IAAI,MAAM,GAAG,IAAA,kCAAkB,EAAC,cAAc,EAAE,UAAU,EAAE,cAAc,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;IACtG,0EAA0E;IAE1E,0CAA0C;IAC1C,MAAM,QAAQ,GAAG,MAAM,IAAA,+CAAsB,EAAC,MAAM,EAAE,SAAS,CAAC,CAAC;IAEjE,kCAAkC;IAClC,kHAAkH;IAClH,IAAA,yBAAa,EAAC,QAAQ,EAAE,YAAY,CAAC,CAAC;IAEtC,UAAU;IACV,IAAI,aAAa,GAAc,EAAE,OAAO,EAAE,IAAI,EAAE,YAAY,EAAE,eAAe,EAAE,YAAY,EAAE,cAAc,EAAE,WAAW,EAAE,IAAI,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;IACnJ,IAAI,aAAa,EAAE,CAAC;QAChB,MAAM,UAAU,GAAG,MAAM,KAAK,sBAAU,CAAC;QACzC,aAAa,GAAG,MAAM,IAAA,yBAAe,EAAC,MAAM,EAAE,cAAc,EAAE,QAAQ,EAAE,eAAe,EAAE,YAAY,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,EAAE,WAAW,CAAC,CAAC;QACvJ,mHAAmH;IACvH,CAAC;IAED,IAAI,aAAa,CAAC,OAAO,EAAE,CAAC;QACxB,OAAO,CAAC,GAAG,CAAC,2BAA2B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;QAClF,OAAO,aAAa,CAAC;IACzB,CAAC;IAED,oIAAoI;IACpI,IAAI,CAAC,eAAe,EAAE,CAAC;QACnB,8CAA8C;QAC9C,IAAI,CAAC,aAAa,CAAC,UAAU,IAAI,aAAa,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACrE,CAAC,aAAa,IAAI,IAAA,kCAAsB,EAAC,YAAY,CAAC,CAAC;YACvD,OAAO,CAAC,GAAG,CAAC,2DAA2D,YAAY,EAAE,CAAC,CAAC;QAC3F,CAAC;aAAM,CAAC;YACJ,IAAA,yDAA+B,EAAC,aAAa,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;YACxE,OAAO,CAAC,GAAG,CAAC,oCAAoC,YAAY,EAAE,CAAC,CAAC;QACpE,CAAC;IACL,CAAC;SAAM,CAAC;QACJ,OAAO,CAAC,GAAG,CAAC,0BAA0B,YAAY,EAAE,CAAC,CAAC;IAC1D,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,0BAA0B,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,UAAU,CAAC,CAAC;IACjF,OAAO,aAAa,CAAC;AACzB,CAAC;AAED,mKAAmK;AACnK,iNAAiN;AACjN,uNAAuN;AACvN,sKAAsK;AACtK,0IAA0I;AAE1I,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\n\r\nimport { CSHARPUTGEN, CSVSPLUGIN } from '../types/constants';\r\nimport { deleteTestFileIfExists, writeTestCode } from \"../utils/fileUtils\";\r\n\r\nimport { GenResult } from '../types/genResult';\r\nimport { autoFixTestCode } from './autoFix';\r\nimport { checkXapCode } from '../utils/checkXapCode';\r\nimport { detectTestFramework } from '../utils/detectTestFramework';\r\nimport { ensureValidLLMResponse } from './ensureValidLLMResponse';\r\nimport { generateCodeDependency } from '../analyze/generateCodeDependency';\r\nimport { getOrCreatTestFile } from '../utils/getTestFile';\r\nimport { installOrUpdateAnalyzerTool } from '../analyze/installAnalyzerTool';\r\nimport { preGenProcess } from './preGen/preGenProcess';\r\nimport { prepareUtGenPrompt } from '../llm/preparePrompt';\r\nimport { removeFailedTestCasesInTestFile } from '../utils/removeFailedTestMethods';\r\nimport { sendUserUsageEvent } from '../metrics/appInsightTelemetry';\r\n\r\nconst RETRYTIMES = 3;\r\n\r\nexport async function generateCsharpUtCode(\r\n    sourceCodePath: string,\r\n    testCodePath: string,\r\n    enableAutoFix: boolean,\r\n    keepFailedTests: boolean,\r\n    source: string = CSHARPUTGEN,\r\n    isBenchmark: boolean = false\r\n): Promise<GenResult> {\r\n    const startTime = Date.now();\r\n\r\n    console.log(`*****************Generating C# unit test code for: ${sourceCodePath}*****************`);\r\n    console.log(`Current work dir: ${process.cwd()}`);\r\n\r\n    // send usage metrics\r\n    sendUserUsageEvent(source);\r\n\r\n    sourceCodePath = path.normalize(sourceCodePath);\r\n    if (!fs.existsSync(sourceCodePath)) {\r\n        console.error(`Source code file not found: ${sourceCodePath}`);\r\n        return;\r\n    }\r\n\r\n    // install nuget cli tool to analyze code\r\n    installOrUpdateAnalyzerTool();\r\n\r\n    let sourceCode = fs.readFileSync(sourceCodePath, 'utf-8');\r\n\r\n    // check if source code is xap code, if so, use xap code generation prompt\r\n    const isXapCode = checkXapCode(sourceCode);\r\n\r\n    sourceCode = await preGenProcess(sourceCode, sourceCodePath, isXapCode);\r\n    if (!sourceCode) {\r\n        console.error(`Skip generate test for source code: ${sourceCodePath}`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get code dependencies\r\n    let codeDependency = await generateCodeDependency(sourceCodePath);\r\n\r\n    // detect existing test file or create a new one by source code path\r\n    const { testFilePath, testProjectPath, testFileExist } = getOrCreatTestFile(sourceCodePath, testCodePath, isXapCode);\r\n    if (!testFilePath || !testProjectPath) {\r\n        console.error(`Failed to get or create test file for source code: ${sourceCodePath}`);\r\n        return { success: false, testFilePath: undefined, testProjectPath: undefined, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    }\r\n\r\n    // get test framework\r\n    const testFramework = detectTestFramework(testProjectPath);\r\n\r\n    // prepare prompt\r\n    let prompt = prepareUtGenPrompt(sourceCodePath, sourceCode, codeDependency, isXapCode, testFramework);\r\n    // let prompt = getPrompts(sourceCodePath, testFramework, codeDependency);\r\n\r\n    // call LLM service to gen code with retry\r\n    const testCode = await ensureValidLLMResponse(prompt, isXapCode);\r\n\r\n    // write test code to test project\r\n    // To run vstest, need to write the test code to test project first, then build the test project and run the tests\r\n    writeTestCode(testCode, testFilePath);\r\n\r\n    // autofix\r\n    let autoFixResult: GenResult = { success: true, testFilePath, testProjectPath, codeFilePath: sourceCodePath, buildResult: null, testResult: null };\r\n    if (enableAutoFix) {\r\n        const isVsPlugin = source === CSVSPLUGIN;\r\n        autoFixResult = await autoFixTestCode(prompt, sourceCodePath, testCode, testProjectPath, testFilePath, isXapCode, isVsPlugin, RETRYTIMES, isBenchmark);\r\n        // autoFixResult = await autoRetryTestCode(prompt, sourceCode, testCode, testProjectPath, testFilePath, isXapTest);\r\n    }\r\n\r\n    if (autoFixResult.success) {\r\n        console.log(`Gerate UT success time: ${(Date.now() - startTime) / 1000} seconds`);\r\n        return autoFixResult;\r\n    }\r\n\r\n    // if retry max times or test generation failed, and keepFailedTests is false, delete failed test file or remove failed test methods\r\n    if (!keepFailedTests) {\r\n        // keep succeed test code, or delete test file\r\n        if (!autoFixResult.testResult || autoFixResult.testResult.passed === 0) {\r\n            !testFileExist && deleteTestFileIfExists(testFilePath);\r\n            console.log(`Failed to generate csharp test code, deleted test file: ${testFilePath}`);\r\n        } else {\r\n            removeFailedTestCasesInTestFile(autoFixResult.testResult, testFilePath);\r\n            console.log(`write succeed test code to file: ${testFilePath}`);\r\n        }\r\n    } else {\r\n        console.log(`keep failed test file: ${testFilePath}`);\r\n    }\r\n    console.log(`Gerate UT failed time: ${(Date.now() - startTime) / 1000} seconds`);\r\n    return autoFixResult;\r\n}\r\n\r\n// generateCsharpUtCode(\"D:\\\\code\\\\IndexServe\\\\private\\\\indexserve\\\\RecommendationEngine\\\\RecoPS\\\\PSTableClient\\\\PSTableClient.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Overview\\\\HttpRequestParser\\\\Utils\\\\SqmidHelper.cs\", undefined, true, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\CS.Service.Fundamental\\\\SharedSegments\\\\SharedSegments\\\\SharedSegments.Plugins\\\\Workflows\\\\Eplant\\\\ResponseBuilders\\\\EplantMigrateResponseBuilder.cs\", undefined, true, true, true);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\UCMDynamics\\\\private\\\\DynamicCode\\\\Plugin\\\\UCMDynamics.AppointmentPlugin\\\\AppointmentOperation.cs\", undefined, false, true, false);\r\n// generateCsharpUtCode(\"D:\\\\code\\\\AIUnitTestExample\\\\src\\\\ToListinator.Analyzers\\\\ToListCountAnalyzer.cs\", undefined, true, true, false);\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\r\n\r\n"]}

package/out/gen/ensureValidLLMResponse.d.ts

@@ -0,0 +1 @@
+export declare function ensureValidLLMResponse(prompt: any[], isXapCode: boolean, retryTimes?: number): Promise<string>;

package/out/gen/ensureValidLLMResponse.js

@@ -0,0 +1,19 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ensureValidLLMResponse = ensureValidLLMResponse;
+const postGenProcess_1 = require("./postGen/postGenProcess");
+const sendRequestToLLM_1 = require("../llm/sendRequestToLLM");
+async function ensureValidLLMResponse(prompt, isXapCode, retryTimes = 3) {
+    for (let i = 0; i < retryTimes; i++) {
+        // call LLM service to gen code
+        const { content: generatedCode } = await (0, sendRequestToLLM_1.sendRequestToLLM)(prompt);
+        // extract valid code from LLM response
+        const testCode = (0, postGenProcess_1.postGenProcess)(generatedCode, isXapCode);
+        if (!!testCode) {
+            return testCode;
+        }
+        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);
+    }
+    return undefined;
+}
+//# sourceMappingURL=ensureValidLLMResponse.js.map

package/out/gen/ensureValidLLMResponse.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ensureValidLLMResponse.js","sourceRoot":"","sources":["../../src/gen/ensureValidLLMResponse.ts"],"names":[],"mappings":";;AAGA,wDAcC;AAjBD,6DAA0D;AAC1D,8DAA2D;AAEpD,KAAK,UAAU,sBAAsB,CAAC,MAAa,EAAE,SAAkB,EAAE,aAAqB,CAAC;IAClG,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,UAAU,EAAE,CAAC,EAAE,EAAE,CAAC;QAClC,+BAA+B;QAC/B,MAAM,EAAE,OAAO,EAAE,aAAa,EAAE,GAAG,MAAM,IAAA,mCAAgB,EAAC,MAAM,CAAC,CAAC;QAElE,uCAAuC;QACvC,MAAM,QAAQ,GAAG,IAAA,+BAAc,EAAC,aAAa,EAAE,SAAS,CAAC,CAAC;QAC1D,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QACpB,CAAC;QACD,OAAO,CAAC,GAAG,CAAC,oDAAoD,CAAC,GAAG,CAAC,IAAI,UAAU,GAAG,CAAC,CAAC;IAC5F,CAAC;IAED,OAAO,SAAS,CAAC;AACrB,CAAC","sourcesContent":["import { postGenProcess } from \"./postGen/postGenProcess\";\r\nimport { sendRequestToLLM } from \"../llm/sendRequestToLLM\";\r\n\r\nexport async function ensureValidLLMResponse(prompt: any[], isXapCode: boolean, retryTimes: number = 3): Promise<string> {\r\n    for (let i = 0; i < retryTimes; i++) {\r\n        // call LLM service to gen code\r\n        const { content: generatedCode } = await sendRequestToLLM(prompt);\r\n    \r\n        // extract valid code from LLM response\r\n        const testCode = postGenProcess(generatedCode, isXapCode);\r\n        if (!!testCode) {\r\n            return testCode;\r\n        }\r\n        console.log(`Failed to generate valid test code, retrying... (${i + 1}/${retryTimes})`);\r\n    }\r\n\r\n    return undefined;\r\n}"]}

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

@@ -0,0 +1 @@
+export declare function extractCodeFromResponse(response: string): string;

package/out/gen/postGen/extractCodeFromResponse.js

@@ -0,0 +1,37 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.extractCodeFromResponse = extractCodeFromResponse;
+const regexList = [
+    /```(?:C#|csharp|CSharp|json|mermaid)?([\s\S]*?)(\n?```\n*|$)/g,
+    /```([\s\S]*?)(\n?```\n*|$)/g,
+    /```C#\s*([\s\S]*?)```/gi
+];
+function extractCodeFromResponse(response) {
+    if (!response) {
+        return '';
+    }
+    const text = response.trim();
+    for (const regex of regexList) {
+        let testCode = getRegexResult(text, regex);
+        if (!!testCode) {
+            return testCode;
+        }
+    }
+    const lines = text.split(/\r?\n/);
+    const hasMarkdown = lines.some(line => line.startsWith('#'));
+    if (hasMarkdown) {
+        return '';
+    }
+    return text;
+}
+function getRegexResult(response, regex) {
+    const match = regex.exec(response);
+    if (!match || match.length < 2) {
+        return null;
+    }
+    return match[1].trim();
+}
+// const input = '```C#\nusing Microsoft.VisualStudio.TestTools.UnitTesting;\nusing Moq;\nusing System;\nusing System.Collections.Generic;\nusing SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\n\nnamespace SharedSegments.Tests\n{\n    [TestClass]\n    public class SqmidHelperTests\n    {\n        [TestMethod]\n        public void GetSqmidFromCookie_CookieWithValidBlisId_ReturnsDecodedSqmid()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"BID=QlJfU01JRA==\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(\"SMID\", result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_CookieWithoutBlisId_ReturnsNull()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"ANOTHERDATA=123\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.IsNull(result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_CookieWithEmptyValues_ReturnsEmptyString()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"BID=\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_InvalidBase64BlisId_ThrowsArgumentException_ReturnsEmptyString()\n        {\n            var cookies = new Dictionary<string, string>\n            {\n                { \"USRLOC\", \"BID=InvalidBase64\" }\n            };\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_NoLocationCookie_ReturnsEmptyString()\n        {\n            var cookies = new Dictionary<string, string>();\n\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\n        {\n            var result = SqmidHelper.GetSqmidFromCookie(null);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_ValidBase64_ReturnsExpectedValue()\n        {\n            var blisId = \"QlJfU01JRA==\";\n\n            var result = InvokeDecodeBlisId(blisId);\n\n            Assert.AreEqual(\"SMID\", result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_InvalidBase64_ReturnsEmptyString()\n        {\n            var blisId = \"InvalidBase64\";\n\n            var result = InvokeDecodeBlisId(blisId);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_EmptyBlisId_ReturnsEmptyString()\n        {\n            var blisId = string.Empty;\n\n            var result = InvokeDecodeBlisId(blisId);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        [TestMethod]\n        public void DecodeBlisId_NullBlisId_ReturnsEmptyString()\n        {\n            var result = InvokeDecodeBlisId(null);\n\n            Assert.AreEqual(string.Empty, result);\n        }\n\n        private string InvokeDecodeBlisId(string blisId)\n        {\n            var method = typeof(SqmidHelper).GetMethod(\"DecodeBlisId\", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Static);\n            return (string)method.Invoke(null, new object[] { blisId });\n        }\n    }\n}\n```';
+// const code = extractCode(input);
+// console.log(code);
+//# sourceMappingURL=extractCodeFromResponse.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"extractCodeFromResponse.js","sourceRoot":"","sources":["../../../src/gen/postGen/extractCodeFromResponse.ts"],"names":[],"mappings":";;AAOA,0DAkBC;AAxBD,MAAM,SAAS,GAAG;IACd,+DAA+D;IAC/D,6BAA6B;IAC7B,yBAAyB;CAC5B,CAAC;AAEF,SAAgB,uBAAuB,CAAC,QAAgB;IACpD,IAAI,CAAC,QAAQ,EAAE,CAAC;QACZ,OAAO,EAAE,CAAC;IACd,CAAC;IACD,MAAM,IAAI,GAAG,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC7B,KAAK,MAAM,KAAK,IAAI,SAAS,EAAE,CAAC;QAC5B,IAAI,QAAQ,GAAG,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;QAC3C,IAAI,CAAC,CAAC,QAAQ,EAAE,CAAC;YACb,OAAO,QAAQ,CAAC;QACpB,CAAC;IACL,CAAC;IAED,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAClC,MAAM,WAAW,GAAG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;IAC7D,IAAI,WAAW,EAAE,CAAC;QACd,OAAO,EAAE,CAAC;IACd,CAAC;IACD,OAAO,IAAI,CAAC;AAChB,CAAC;AAED,SAAS,cAAc,CAAC,QAAgB,EAAE,KAAa;IACnD,MAAM,KAAK,GAAG,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,KAAK,IAAI,KAAK,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC7B,OAAO,IAAI,CAAC;IAChB,CAAC;IACD,OAAO,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;AAC3B,CAAC;AAED,+qHAA+qH;AAC/qH,mCAAmC;AACnC,qBAAqB","sourcesContent":["\r\nconst regexList = [\r\n    /```(?:C#|csharp|CSharp|json|mermaid)?([\\s\\S]*?)(\\n?```\\n*|$)/g,\r\n    /```([\\s\\S]*?)(\\n?```\\n*|$)/g,\r\n    /```C#\\s*([\\s\\S]*?)```/gi\r\n];\r\n\r\nexport function extractCodeFromResponse(response: string): string {\r\n    if (!response) {\r\n        return '';\r\n    }\r\n    const text = response.trim();\r\n    for (const regex of regexList) {\r\n        let testCode = getRegexResult(text, regex);\r\n        if (!!testCode) {\r\n            return testCode;\r\n        }\r\n    }\r\n\r\n    const lines = text.split(/\\r?\\n/);\r\n    const hasMarkdown = lines.some(line => line.startsWith('#'));\r\n    if (hasMarkdown) {\r\n        return '';\r\n    }\r\n    return text;\r\n}\r\n\r\nfunction getRegexResult(response: string, regex: RegExp): string | null {\r\n    const match = regex.exec(response);\r\n    if (!match || match.length < 2) {\r\n        return null;\r\n    }\r\n    return match[1].trim();\r\n}\r\n\r\n// const input = '```C#\\nusing Microsoft.VisualStudio.TestTools.UnitTesting;\\nusing Moq;\\nusing System;\\nusing System.Collections.Generic;\\nusing SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;\\n\\nnamespace SharedSegments.Tests\\n{\\n    [TestClass]\\n    public class SqmidHelperTests\\n    {\\n        [TestMethod]\\n        public void GetSqmidFromCookie_CookieWithValidBlisId_ReturnsDecodedSqmid()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"BID=QlJfU01JRA==\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(\\\"SMID\\\", result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_CookieWithoutBlisId_ReturnsNull()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"ANOTHERDATA=123\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.IsNull(result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_CookieWithEmptyValues_ReturnsEmptyString()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"BID=\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_InvalidBase64BlisId_ThrowsArgumentException_ReturnsEmptyString()\\n        {\\n            var cookies = new Dictionary<string, string>\\n            {\\n                { \\\"USRLOC\\\", \\\"BID=InvalidBase64\\\" }\\n            };\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_NoLocationCookie_ReturnsEmptyString()\\n        {\\n            var cookies = new Dictionary<string, string>();\\n\\n            var result = SqmidHelper.GetSqmidFromCookie(cookies);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()\\n        {\\n            var result = SqmidHelper.GetSqmidFromCookie(null);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_ValidBase64_ReturnsExpectedValue()\\n        {\\n            var blisId = \\\"QlJfU01JRA==\\\";\\n\\n            var result = InvokeDecodeBlisId(blisId);\\n\\n            Assert.AreEqual(\\\"SMID\\\", result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_InvalidBase64_ReturnsEmptyString()\\n        {\\n            var blisId = \\\"InvalidBase64\\\";\\n\\n            var result = InvokeDecodeBlisId(blisId);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_EmptyBlisId_ReturnsEmptyString()\\n        {\\n            var blisId = string.Empty;\\n\\n            var result = InvokeDecodeBlisId(blisId);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        [TestMethod]\\n        public void DecodeBlisId_NullBlisId_ReturnsEmptyString()\\n        {\\n            var result = InvokeDecodeBlisId(null);\\n\\n            Assert.AreEqual(string.Empty, result);\\n        }\\n\\n        private string InvokeDecodeBlisId(string blisId)\\n        {\\n            var method = typeof(SqmidHelper).GetMethod(\\\"DecodeBlisId\\\", System.Reflection.BindingFlags.NonPublic | System.Reflection.BindingFlags.Static);\\n            return (string)method.Invoke(null, new object[] { blisId });\\n        }\\n    }\\n}\\n```';\r\n// const code = extractCode(input);\r\n// console.log(code);\r\n"]}

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

@@ -0,0 +1 @@
+export declare function postGenProcess(generatedCode: string, isXapCode: boolean): string;

package/out/gen/postGen/postGenProcess.js

@@ -0,0 +1,36 @@
+"use strict";
+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");
+function postGenProcess(generatedCode, isXapCode) {
+    if (!isValidTestCode(generatedCode)) {
+        return undefined;
+    }
+    let testCode = (0, extractCodeFromResponse_1.extractCodeFromResponse)(generatedCode);
+    testCode = (0, removeComments_1.removeTestCodeComments)(testCode);
+    testCode = (0, removeSingleLines_1.removeCSharpOnlyLines)(testCode);
+    if (isXapCode) {
+        testCode = xapCodePostGenProcess(testCode);
+    }
+    console.log(`Post generation process completed, code length: ${testCode.length}`);
+    return testCode;
+}
+function xapCodePostGenProcess(testCode) {
+    const processedCode = (0, repairRequiredNameSpaces_1.repairRequiredNameSpaces)(testCode);
+    return processedCode;
+}
+function isValidTestCode(testCode) {
+    if (!testCode || testCode.trim() === '') {
+        return false; // Empty code is not valid
+    }
+    // Check if the test code contains any test attributes or using statements
+    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

@@ -0,0 +1 @@
+{"version":3,"file":"postGenProcess.js","sourceRoot":"","sources":["../../../src/gen/postGen/postGenProcess.ts"],"names":[],"mappings":";;AAQA,wCAaC;AAnBD,uEAAoE;AACpE,2DAA4D;AAC5D,qDAA0D;AAC1D,yEAAsE;AACtE,yDAAsD;AAEtD,SAAgB,cAAc,CAAC,aAAqB,EAAE,SAAkB;IACpE,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,IAAI,SAAS,EAAE,CAAC;QACZ,QAAQ,GAAG,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,CAAC,GAAG,CAAC,mDAAmD,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC;IAClF,OAAO,QAAQ,CAAC;AACpB,CAAC;AAGD,SAAS,qBAAqB,CAAC,QAAgB;IAC3C,MAAM,aAAa,GAAG,IAAA,mDAAwB,EAAC,QAAQ,CAAC,CAAC;IACzD,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 * as fs from 'fs';\r\n\r\nimport { extractCodeFromResponse } from \"./extractCodeFromResponse\";\r\nimport { removeCSharpOnlyLines } from \"./removeSingleLines\";\r\nimport { removeTestCodeComments } from \"./removeComments\";\r\nimport { repairRequiredNameSpaces } from \"./repairRequiredNameSpaces\";\r\nimport { validateTestCode } from \"./validateTestCode\";\r\n\r\nexport function postGenProcess(generatedCode: 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    if (isXapCode) {\r\n        testCode = xapCodePostGenProcess(testCode);\r\n    }\r\n    console.log(`Post generation process completed, code length: ${testCode.length}`);\r\n    return testCode;\r\n}\r\n\r\n\r\nfunction xapCodePostGenProcess(testCode: string): string {\r\n    const processedCode = repairRequiredNameSpaces(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"]}

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

@@ -0,0 +1 @@
+export declare function removeTestCodeComments(code: string): string;

package/out/gen/postGen/removeComments.js

@@ -0,0 +1,92 @@
+"use strict";
+// todo: remove line：
+// C#(just contains C# in a line)
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.removeTestCodeComments = removeTestCodeComments;
+function removeTestCodeComments(code) {
+    // Remove multi-line comments (/* ... */)
+    let cleanedCode = code.replace(/\/\*[\s\S]*?\*\//g, '');
+    // Remove single-line comments (// ...)
+    cleanedCode = cleanedCode.replace(/\/\/.*$/gm, '');
+    // Remove extra blank lines that might result from comment removal, only left at most one blank line
+    // cleanedCode = cleanedCode.replace(/\n\s*\n/g, '\n');
+    cleanedCode = cleanedCode.replace(/(\r?\n){3,}/g, '\n\n');
+    return cleanedCode.trim();
+}
+// const code = `
+// using Microsoft.VisualStudio.TestTools.UnitTesting;
+// using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;
+// using System;
+// using System.Collections.Generic;
+// using System.Text;
+// using Moq;
+// namespace SharedSegments.Plugins.Tests.Workflows.Overview.HttpRequestParser.Utils
+// {
+//     [TestClass]
+//     public class SqmidHelperTests
+//     {
+//         [TestMethod]
+//         public void GetSqmidFromCookie_ValidCookiesWithBlisId_ReturnsDecodedBlisId()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "BID=" + Convert.ToBase64String(Encoding.UTF8.GetBytes("test_decodedBlisId_test")) }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual("decodedBlisId", result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_ValidCookiesWithoutBlisId_ReturnsNull()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "SomeOtherValue=123" }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.IsNull(result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_EmptyCookies_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>();
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             IDictionary<string, string> cookies = null;
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_UsrLocWithEmptyValue_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "" }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//     }
+// }
+// `;
+// const cleanedCode = removeTestCodeComments(code);
+// console.log(cleanedCode);
+//# sourceMappingURL=removeComments.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"removeComments.js","sourceRoot":"","sources":["../../../src/gen/postGen/removeComments.ts"],"names":[],"mappings":";AAAA,qBAAqB;AACrB,iCAAiC;;AAEjC,wDAYC;AAZD,SAAgB,sBAAsB,CAAC,IAAY;IAC/C,yCAAyC;IACzC,IAAI,WAAW,GAAG,IAAI,CAAC,OAAO,CAAC,mBAAmB,EAAE,EAAE,CAAC,CAAC;IAExD,uCAAuC;IACvC,WAAW,GAAG,WAAW,CAAC,OAAO,CAAC,WAAW,EAAE,EAAE,CAAC,CAAC;IAEnD,oGAAoG;IACpG,uDAAuD;IACvD,WAAW,GAAG,WAAW,CAAC,OAAO,CAAC,cAAc,EAAE,MAAM,CAAC,CAAC;IAE1D,OAAO,WAAW,CAAC,IAAI,EAAE,CAAC;AAC9B,CAAC;AAED,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,oDAAoD;AACpD,4BAA4B","sourcesContent":["// todo: remove line：\r\n// C#(just contains C# in a line)\r\n\r\nexport function removeTestCodeComments(code: string): string {\r\n    // Remove multi-line comments (/* ... */)\r\n    let cleanedCode = code.replace(/\\/\\*[\\s\\S]*?\\*\\//g, '');\r\n    \r\n    // Remove single-line comments (// ...)\r\n    cleanedCode = cleanedCode.replace(/\\/\\/.*$/gm, '');\r\n    \r\n    // Remove extra blank lines that might result from comment removal, only left at most one blank line\r\n    // cleanedCode = cleanedCode.replace(/\\n\\s*\\n/g, '\\n');\r\n    cleanedCode = cleanedCode.replace(/(\\r?\\n){3,}/g, '\\n\\n');\r\n\r\n    return cleanedCode.trim();\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 = removeTestCodeComments(code);\r\n// console.log(cleanedCode);"]}

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

@@ -0,0 +1 @@
+export declare function removeCSharpOnlyLines(generatedCode: string): string;

package/out/gen/postGen/removeSingleLines.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.removeCSharpOnlyLines = removeCSharpOnlyLines;
+const illegalSingleLines = [
+    'c#',
+    'csharp',
+    '```',
+    '```c#',
+    '```csharp',
+];
+function removeCSharpOnlyLines(generatedCode) {
+    const lines = generatedCode.split(/\r?\n/);
+    const filteredLines = lines.map(line => {
+        // remove trailing spaces and tabs
+        const trimmedLine = line.trim();
+        if (trimmedLine.length === 0) {
+            return trimmedLine; // keep empty lines
+        }
+        else {
+            return line;
+        }
+    }).filter(line => {
+        const lowerCasedLine = line.toLowerCase();
+        return !illegalSingleLines.includes(lowerCasedLine);
+    });
+    return filteredLines.join('\n');
+}
+//# sourceMappingURL=removeSingleLines.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"removeSingleLines.js","sourceRoot":"","sources":["../../../src/gen/postGen/removeSingleLines.ts"],"names":[],"mappings":";;AAOA,sDAiBC;AAxBD,MAAM,kBAAkB,GAAG;IACvB,IAAI;IACJ,QAAQ;IACR,KAAK;IACL,OAAO;IACP,WAAW;CACd,CAAC;AACF,SAAgB,qBAAqB,CAAC,aAAqB;IACvD,MAAM,KAAK,GAAG,aAAa,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;IAC3C,MAAM,aAAa,GAAG,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE;QAE/B,kCAAkC;QAClC,MAAM,WAAW,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC;QAChC,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC3B,OAAO,WAAW,CAAC,CAAC,mBAAmB;QAC3C,CAAC;aAAM,CAAC;YACJ,OAAO,IAAI,CAAC;QAChB,CAAC;IACL,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,EAAE;QACjB,MAAM,cAAc,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QAC1C,OAAO,CAAC,kBAAkB,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;IACxD,CAAC,CAAC,CAAC;IAEH,OAAO,aAAa,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACpC,CAAC","sourcesContent":["const illegalSingleLines = [\r\n    'c#',\r\n    'csharp',\r\n    '```',\r\n    '```c#',\r\n    '```csharp',\r\n];\r\nexport function removeCSharpOnlyLines(generatedCode: string): string {\r\n    const lines = generatedCode.split(/\\r?\\n/);\r\n    const filteredLines = lines.map(line => \r\n        {\r\n            // remove trailing spaces and tabs\r\n            const trimmedLine = line.trim();\r\n            if (trimmedLine.length === 0) {\r\n                return trimmedLine; // keep empty lines\r\n            } else {\r\n                return line;\r\n            }\r\n        }).filter(line => {\r\n        const lowerCasedLine = line.toLowerCase();\r\n        return !illegalSingleLines.includes(lowerCasedLine);\r\n    });\r\n\r\n    return filteredLines.join('\\n');\r\n}"]}

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

@@ -0,0 +1 @@
+export declare function repairRequiredNameSpaces(code: string): string;

package/out/gen/postGen/repairRequiredNameSpaces.js

@@ -0,0 +1,116 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.repairRequiredNameSpaces = repairRequiredNameSpaces;
+function repairRequiredNameSpaces(code) {
+    const requiredUsings = [
+        "using Xap.ExecutionFramework;",
+        "using Xap.PluginFramework;",
+        "using System;",
+        "using System.Threading.Tasks;",
+        "using System.Collections.Generic;",
+        "using EMPTY = Microsoft.Bing.Xap.Plugins.Void;",
+        "using Microsoft.VisualStudio.TestTools.UnitTesting;"
+    ];
+    // get all using lines
+    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(/\r?\n/);
+    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;
+    }
+    // build new code
+    const newLines = [
+        ...lines.slice(0, insertIndex),
+        ...missingUsings,
+        ...lines.slice(insertIndex)
+    ];
+    console.log("repaired namespaces:", missingUsings);
+    return newLines.join('\n');
+}
+// const code = `
+// using Microsoft.VisualStudio.TestTools.UnitTesting;
+// using SharedSegments.Plugins.Workflows.Overview.HttpRequestParser.Utils;
+// using System;
+// using System.Collections.Generic;
+// using System.Text;
+// using Moq;
+// namespace SharedSegments.Plugins.Tests.Workflows.Overview.HttpRequestParser.Utils
+// {
+//     [TestClass]
+//     public class SqmidHelperTests
+//     {
+//         [TestMethod]
+//         public void GetSqmidFromCookie_ValidCookiesWithBlisId_ReturnsDecodedBlisId()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "BID=" + Convert.ToBase64String(Encoding.UTF8.GetBytes("test_decodedBlisId_test")) }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual("decodedBlisId", result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_ValidCookiesWithoutBlisId_ReturnsNull()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "SomeOtherValue=123" }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.IsNull(result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_EmptyCookies_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>();
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_NullCookies_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             IDictionary<string, string> cookies = null;
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//         [TestMethod]
+//         public void GetSqmidFromCookie_UsrLocWithEmptyValue_ReturnsEmptyString()
+//         {
+//             // Arrange
+//             var cookies = new Dictionary<string, string>
+//             {
+//                 { "USRLOC", "" }
+//             };
+//             // Act
+//             var result = SqmidHelper.GetSqmidFromCookie(cookies);
+//             // Assert
+//             Assert.AreEqual(string.Empty, result);
+//         }
+//     }
+// }
+// `;
+// const cleanedCode = repairRequiredNameSpaces(code);
+// console.log(cleanedCode);
+//# sourceMappingURL=repairRequiredNameSpaces.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"repairRequiredNameSpaces.js","sourceRoot":"","sources":["../../../src/gen/postGen/repairRequiredNameSpaces.ts"],"names":[],"mappings":";;AAGA,4DA2CC;AA3CD,SAAgB,wBAAwB,CAAC,IAAY;IACjD,MAAM,cAAc,GAAG;QACnB,+BAA+B;QAC/B,4BAA4B;QAC5B,eAAe;QACf,+BAA+B;QAC/B,mCAAmC;QACnC,gDAAgD;QAChD,qDAAqD;KACxD,CAAC;IAEF,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,OAAO,CAAC,CAAC;IAClC,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;AAGD,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 fs from 'fs';\r\nimport path from 'path';\r\n\r\nexport function repairRequiredNameSpaces(code: string): string {\r\n    const requiredUsings = [\r\n        \"using Xap.ExecutionFramework;\",\r\n        \"using Xap.PluginFramework;\",\r\n        \"using System;\",\r\n        \"using System.Threading.Tasks;\",\r\n        \"using System.Collections.Generic;\",\r\n        \"using EMPTY = Microsoft.Bing.Xap.Plugins.Void;\",\r\n        \"using Microsoft.VisualStudio.TestTools.UnitTesting;\"\r\n    ];\r\n\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(/\\r?\\n/);\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\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);"]}