@ai-dev-tools/test-copilot-core 1.0.19

package/out/core/features/lintFix.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"lintFix.d.ts","sourceRoot":"","sources":["../../../src/core/features/lintFix.ts"],"names":[],"mappings":"AASA,wBAAsB,aAAa,CAAC,QAAQ,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM,mBA0B9F"}

package/out/core/features/lintFix.js

@@ -0,0 +1,33 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.fixLintErrors = fixLintErrors;
+const common_1 = require("../utils/common");
+const llm_1 = require("../llm");
+const fs_1 = __importDefault(require("fs"));
+const console_1 = require("../utils/console");
+const path_1 = __importDefault(require("path"));
+async function fixLintErrors(filePath, lintErrors, testFramework) {
+    const fileExt = path_1.default.extname(filePath);
+    const testCode = fs_1.default.readFileSync(filePath, 'utf-8');
+    const lintFixPrompt = (0, llm_1.getPrompt)("FixLintErrors" /* PromptKey.FixLintErrors */, {
+        ["<<<TestFramework>>>" /* PromptPlaceholder.TestFramework */]: testFramework,
+        ["<<<Spec>>>" /* PromptPlaceholder.SpecCode */]: testCode,
+        ["<<<LintErrors>>>" /* PromptPlaceholder.LintErrors */]: lintErrors,
+        ["<<<Language>>>" /* PromptPlaceholder.Language */]: (0, common_1.getLanguage)(fileExt)
+    });
+    const res = await (0, llm_1.getLLMResponse)(lintFixPrompt);
+    if (!res.success) {
+        (0, console_1.getConsole)().error(` lint fix - Failed to get code from AI, res: ${JSON.stringify(res)}`);
+        return "";
+    }
+    let fixedTestCode = (0, common_1.extractCodeFromResponse)(res.content);
+    if (!fixedTestCode) {
+        (0, console_1.getConsole)().error(` lint fix - Failed to extract code from AI response, res.content: ${res.content}`);
+        return "";
+    }
+    return fixedTestCode;
+}
+//# sourceMappingURL=lintFix.js.map

package/out/core/features/lintFix.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"lintFix.js","sourceRoot":"","sources":["../../../src/core/features/lintFix.ts"],"names":[],"mappings":";;;;;AASA,sCA0BC;AAlCD,4CAAuE;AACvE,gCAAmD;AAEnD,4CAAoB;AACpB,8CAA8C;AAE9C,gDAAwB;AAEjB,KAAK,UAAU,aAAa,CAAC,QAAgB,EAAE,UAAkB,EAAE,aAAqB;IAC3F,MAAM,OAAO,GAAG,cAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACvC,MAAM,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACpD,MAAM,aAAa,GAAG,IAAA,eAAS,iDAE3B;QACI,6DAAiC,EAAE,aAAa;QAChD,+CAA4B,EAAE,QAAQ;QACtC,uDAA8B,EAAE,UAAU;QAC1C,mDAA4B,EAAE,IAAA,oBAAW,EAAC,OAAO,CAAC;KACrD,CACJ,CAAC;IAEF,MAAM,GAAG,GAAG,MAAM,IAAA,oBAAc,EAAC,aAAa,CAAC,CAAC;IAChD,IAAI,CAAC,GAAG,CAAC,OAAO,EAAE,CAAC;QACf,IAAA,oBAAU,GAAE,CAAC,KAAK,CAAC,gDAAgD,IAAI,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QAC1F,OAAO,EAAE,CAAC;IACd,CAAC;IAED,IAAI,aAAa,GAAG,IAAA,gCAAuB,EAAC,GAAG,CAAC,OAAO,CAAC,CAAC;IACzD,IAAI,CAAC,aAAa,EAAE,CAAC;QACjB,IAAA,oBAAU,GAAE,CAAC,KAAK,CAAC,qEAAqE,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QACvG,OAAO,EAAE,CAAC;IACd,CAAC;IAED,OAAO,aAAa,CAAC;AACzB,CAAC"}

package/out/core/features/mergeUTFile.d.ts

@@ -0,0 +1,19 @@
+import { FeatureBase } from "./common/base";
+export interface MergeUTFileParams {
+    sourceCodePath: string;
+    sourceSpecPath: string;
+    targetSpecPath: string;
+}
+export interface MergeUTFileResult {
+    success: boolean;
+}
+/**
+ * Generate new test for a single function/method.
+ * !! Assume there is no existing test for the function/method !!
+ */
+export declare class MergeUTFile extends FeatureBase<MergeUTFileParams, MergeUTFileResult> {
+    protected task(): Promise<{
+        success: boolean;
+    }>;
+}
+//# sourceMappingURL=mergeUTFile.d.ts.map

package/out/core/features/mergeUTFile.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"mergeUTFile.d.ts","sourceRoot":"","sources":["../../../src/core/features/mergeUTFile.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAO5C,MAAM,WAAW,iBAAiB;IAE9B,cAAc,EAAE,MAAM,CAAC;IAEvB,cAAc,EAAE,MAAM,CAAC;IAEvB,cAAc,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,iBAAiB;IAC9B,OAAO,EAAE,OAAO,CAAC;CACpB;AAED;;;GAGG;AACH,qBAAa,WAAY,SAAQ,WAAW,CAAC,iBAAiB,EAAE,iBAAiB,CAAC;cAC9D,IAAI;;;CAiDvB"}

package/out/core/features/mergeUTFile.js

@@ -0,0 +1,63 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MergeUTFile = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const base_1 = require("./common/base");
+const file_1 = require("../utils/file");
+const llm_1 = require("../llm");
+const config_1 = require("../config");
+const validFix_1 = require("./validFix");
+const system_1 = require("../utils/system");
+/**
+ * Generate new test for a single function/method.
+ * !! Assume there is no existing test for the function/method !!
+ */
+class MergeUTFile extends base_1.FeatureBase {
+    async task() {
+        const { sourceCodePath, sourceSpecPath, targetSpecPath } = this.params;
+        const sourceCodeFilename = './' + path_1.default.basename(sourceCodePath);
+        const specRelativePath = (0, file_1.getNormalizedRelativePath)(sourceCodePath, sourceSpecPath);
+        const targetFileName = path_1.default.basename(targetSpecPath);
+        const targetFileRelativePath = (0, file_1.getNormalizedRelativePath)(sourceCodePath, targetSpecPath);
+        const separatedFileContent = fs_1.default.readFileSync(sourceSpecPath, 'utf-8');
+        let targetFileContent = '// Empty file';
+        if (fs_1.default.existsSync(targetSpecPath)) {
+            targetFileContent = fs_1.default.readFileSync(targetSpecPath, 'utf-8');
+        }
+        const prompt = (0, llm_1.getPrompt)("MergeUTFile" /* PromptKey.MergeUTFile */, {
+            ["<<<SourceFileName>>>" /* PromptPlaceholder.SourceFileName */]: sourceCodeFilename,
+            ["<<<SpecFileName>>>" /* PromptPlaceholder.SpecFileName */]: specRelativePath,
+            ["<<<TargetFilePath>>>" /* PromptPlaceholder.TargetFilePath */]: targetFileRelativePath,
+            ["<<<TestFramework>>>" /* PromptPlaceholder.TestFramework */]: (0, config_1.getConfig)().common.testFramework,
+            ["<<<Spec>>>" /* PromptPlaceholder.SpecCode */]: separatedFileContent,
+            ["<<<SpecTemplate>>>" /* PromptPlaceholder.SpecTemplate */]: targetFileContent,
+        });
+        this.progress({ message: `Start merging test file \`${specRelativePath}\` into \`${targetFileRelativePath}\`` });
+        const llmRes = await this.getCodeFromLLM(prompt);
+        const testRes = await this.runSubTask(validFix_1.ValidFix, {
+            sourceFilePath: sourceCodePath,
+            inputSpecCode: llmRes,
+            specFilePath: targetSpecPath.replace(targetFileName, `tmp.${targetFileName}`),
+        });
+        if (testRes.status === 'valid') {
+            this.progress({ message: 'Merging test file successful' });
+            fs_1.default.writeFileSync(targetSpecPath, llmRes, 'utf-8');
+            (0, system_1.removeFile)(sourceSpecPath);
+            return { success: true };
+        }
+        this.progress({
+            message: `Merging \`${specRelativePath} into ${targetFileRelativePath} failed`,
+            extra: {
+                title: 'Test error',
+                content: JSON.stringify(testRes.failedCases, null, 2),
+            }
+        });
+        return { success: false };
+    }
+}
+exports.MergeUTFile = MergeUTFile;
+//# sourceMappingURL=mergeUTFile.js.map

package/out/core/features/mergeUTFile.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mergeUTFile.js","sourceRoot":"","sources":["../../../src/core/features/mergeUTFile.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,gDAAwB;AACxB,wCAA4C;AAC5C,wCAAuE;AACvE,gCAAiE;AACjE,sCAAsC;AACtC,yCAAsC;AACtC,4CAA6C;AAe7C;;;GAGG;AACH,MAAa,WAAY,SAAQ,kBAAiD;IACpE,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,cAAc,EAAE,cAAc,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAEvE,MAAM,kBAAkB,GAAG,IAAI,GAAG,cAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QAEhE,MAAM,gBAAgB,GAAG,IAAA,gCAAyB,EAAC,cAAc,EAAE,cAAc,CAAC,CAAC;QACnF,MAAM,cAAc,GAAG,cAAI,CAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QACrD,MAAM,sBAAsB,GAAG,IAAA,gCAAyB,EAAC,cAAc,EAAE,cAAc,CAAC,CAAC;QAEzF,MAAM,oBAAoB,GAAG,YAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACtE,IAAI,iBAAiB,GAAG,eAAe,CAAC;QACxC,IAAI,YAAE,CAAC,UAAU,CAAC,cAAc,CAAC,EAAE,CAAC;YAChC,iBAAiB,GAAG,YAAE,CAAC,YAAY,CAAC,cAAc,EAAE,OAAO,CAAC,CAAC;QACjE,CAAC;QAED,MAAM,MAAM,GAAG,IAAA,eAAS,6CAAwB;YAC5C,+DAAkC,EAAE,kBAAkB;YACtD,2DAAgC,EAAE,gBAAgB;YAClD,+DAAkC,EAAE,sBAAsB;YAC1D,6DAAiC,EAAE,IAAA,kBAAS,GAAE,CAAC,MAAM,CAAC,aAAa;YACnE,+CAA4B,EAAE,oBAAoB;YAClD,2DAAgC,EAAE,iBAAiB;SACtD,CAAC,CAAC;QAEH,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,6BAA6B,gBAAgB,aAAa,sBAAsB,IAAI,EAAE,CAAC,CAAC;QACjH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAEjD,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,mBAAQ,EAAE;YAC5C,cAAc,EAAE,cAAc;YAC9B,aAAa,EAAE,MAAM;YACrB,YAAY,EAAE,cAAc,CAAC,OAAO,CAAC,cAAc,EAAE,OAAO,cAAc,EAAE,CAAC;SAChF,CAAC,CAAC;QAEH,IAAI,OAAO,CAAC,MAAM,KAAK,OAAO,EAAE,CAAC;YAC7B,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,8BAA8B,EAAE,CAAC,CAAC;YAC3D,YAAE,CAAC,aAAa,CAAC,cAAc,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAClD,IAAA,mBAAU,EAAC,cAAc,CAAC,CAAC;YAC3B,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,CAAC;QAC7B,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC;YACV,OAAO,EAAE,aAAa,gBAAgB,SAAS,sBAAsB,SAAS;YAC9E,KAAK,EAAE;gBACH,KAAK,EAAE,YAAY;gBACnB,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC;aACxD;SACJ,CAAC,CAAC;QACH,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;IAC9B,CAAC;CACJ;AAlDD,kCAkDC"}

package/out/core/features/removeFailedCases.d.ts

@@ -0,0 +1,23 @@
+import { FeatureBase } from "./common/base";
+import { FailedCaseDetails, TestOutput } from "../valid/type";
+export interface RemoveFailedCasesParams {
+    inputSpecCode: string;
+    targetName: string;
+    inputFailedCases: FailedCaseDetails[];
+    sourceFilePath: string;
+    specFilePath?: string;
+    projectFolder?: string;
+}
+export interface RemoveFailedCasesResult {
+    code: string;
+    status: 'success' | 'failed' | 'error';
+    message?: string;
+    testResult?: TestOutput;
+    failedCases?: FailedCaseDetails[];
+}
+export declare class RemoveFailedCases extends FeatureBase<RemoveFailedCasesParams, RemoveFailedCasesResult> {
+    #private;
+    protected task(): Promise<RemoveFailedCasesResult>;
+    private validSpecCode;
+}
+//# sourceMappingURL=removeFailedCases.d.ts.map

package/out/core/features/removeFailedCases.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"removeFailedCases.d.ts","sourceRoot":"","sources":["../../../src/core/features/removeFailedCases.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAE5C,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAK9D,MAAM,WAAW,uBAAuB;IACpC,aAAa,EAAE,MAAM,CAAC;IACtB,UAAU,EAAE,MAAM,CAAC;IACnB,gBAAgB,EAAE,iBAAiB,EAAE,CAAC;IACtC,cAAc,EAAE,MAAM,CAAC;IACvB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,aAAa,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,MAAM,WAAW,uBAAuB;IACpC,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,SAAS,GAAG,QAAQ,GAAG,OAAO,CAAC;IACvC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,WAAW,CAAC,EAAE,iBAAiB,EAAE,CAAC;CACrC;AAED,qBAAa,iBAAkB,SAAQ,WAAW,CAAC,uBAAuB,EAAE,uBAAuB,CAAC;;cAKhF,IAAI,IAAI,OAAO,CAAC,uBAAuB,CAAC;YA4C1C,aAAa;CA6B9B"}

package/out/core/features/removeFailedCases.js

@@ -0,0 +1,81 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RemoveFailedCases = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const base_1 = require("./common/base");
+const valid_1 = require("../valid");
+const system_1 = require("../utils/system");
+const remove_1 = require("../valid/remove");
+const config_1 = require("../config");
+class RemoveFailedCases extends base_1.FeatureBase {
+    #sourceDir;
+    #pureFileName;
+    #fileSuffix;
+    async task() {
+        const { inputSpecCode, targetName, inputFailedCases, sourceFilePath, specFilePath } = this.params;
+        this.#sourceDir = path_1.default.dirname(sourceFilePath);
+        this.#fileSuffix = path_1.default.extname(sourceFilePath);
+        this.#pureFileName = path_1.default.basename(sourceFilePath, this.#fileSuffix);
+        let testCode = inputSpecCode;
+        let failedCases = inputFailedCases;
+        let retry = 0;
+        let success = false;
+        const maxRetry = 5;
+        this.progress({ message: `Starting to remove failed cases for ${targetName}` });
+        while (failedCases.length && retry < maxRetry) {
+            ++retry;
+            this.progress({ message: `Remove attempt ${retry}/${maxRetry}` });
+            const targets = new Set(failedCases.map(({ casePath }) => casePath));
+            if (failedCases.length) {
+                testCode = (0, remove_1.removeTestCases)(testCode, targets, {
+                    deleteFailed: (0, config_1.getConfig)().common.removeFailedCases
+                });
+            }
+            const testResult = await this.validSpecCode(testCode, specFilePath, targetName);
+            if (testResult.success) {
+                success = true;
+                this.progress({ message: `Successfully removed failed cases` });
+                break;
+            }
+            const failedData = (0, valid_1.getFailedCase)(testResult, specFilePath ? path_1.default.basename(specFilePath) : `${targetName}.spec${this.#fileSuffix}`);
+            failedCases = failedData.cases;
+        }
+        return {
+            code: testCode,
+            status: success ? 'success' : 'failed',
+            failedCases: success ? [] : failedCases
+        };
+    }
+    async validSpecCode(testCode, expectedSpecFilePath, targetName) {
+        let isTmpFile = false;
+        let specPath = expectedSpecFilePath;
+        try {
+            // for new spec file and not saved yet, use tmp file
+            if (!specPath || !fs_1.default.existsSync(specPath)) {
+                isTmpFile = true;
+                const dir = path_1.default.dirname(expectedSpecFilePath || this.#sourceDir);
+                if (!fs_1.default.existsSync(dir)) {
+                    fs_1.default.mkdirSync(dir, { recursive: true });
+                }
+                specPath = path_1.default.join(dir, `${this.#pureFileName}.${targetName}.tmp.spec${this.#fileSuffix}`);
+                fs_1.default.writeFileSync(specPath, testCode, 'utf-8');
+            }
+            const testResult = await (0, valid_1.runTest)(this.testFramework, specPath, {
+                projectFolder: this.params.projectFolder,
+                abortSignal: this.taskOptions.abortSignal,
+            });
+            return testResult;
+        }
+        finally {
+            if (isTmpFile) {
+                (0, system_1.removeFile)(specPath);
+            }
+        }
+    }
+}
+exports.RemoveFailedCases = RemoveFailedCases;
+//# sourceMappingURL=removeFailedCases.js.map

package/out/core/features/removeFailedCases.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"removeFailedCases.js","sourceRoot":"","sources":["../../../src/core/features/removeFailedCases.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,gDAAwB;AACxB,wCAA4C;AAC5C,oCAAkD;AAElD,4CAA6C;AAC7C,4CAAkD;AAClD,sCAAsC;AAmBtC,MAAa,iBAAkB,SAAQ,kBAA6D;IAChG,UAAU,CAAS;IACnB,aAAa,CAAS;IACtB,WAAW,CAAS;IAEV,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,aAAa,EAAE,UAAU,EAAE,gBAAgB,EAAE,cAAc,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAClG,IAAI,CAAC,UAAU,GAAG,cAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAC/C,IAAI,CAAC,WAAW,GAAG,cAAI,CAAC,OAAO,CAAC,cAAc,CAAC,CAAC;QAChD,IAAI,CAAC,aAAa,GAAG,cAAI,CAAC,QAAQ,CAAC,cAAc,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAErE,IAAI,QAAQ,GAAG,aAAa,CAAC;QAC7B,IAAI,WAAW,GAAG,gBAAgB,CAAC;QACnC,IAAI,KAAK,GAAG,CAAC,CAAC;QACd,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,MAAM,QAAQ,GAAG,CAAC,CAAC;QAEnB,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,uCAAuC,UAAU,EAAE,EAAE,CAAC,CAAC;QAEhF,OAAO,WAAW,CAAC,MAAM,IAAI,KAAK,GAAG,QAAQ,EAAE,CAAC;YAC5C,EAAE,KAAK,CAAC;YACR,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,kBAAkB,KAAK,IAAI,QAAQ,EAAE,EAAE,CAAC,CAAC;YAElE,MAAM,OAAO,GAAG,IAAI,GAAG,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC;YACrE,IAAI,WAAW,CAAC,MAAM,EAAE,CAAC;gBACrB,QAAQ,GAAG,IAAA,wBAAe,EAAC,QAAQ,EAAE,OAAO,EAAE;oBAC1C,YAAY,EAAE,IAAA,kBAAS,GAAE,CAAC,MAAM,CAAC,iBAAiB;iBACrD,CAAC,CAAC;YACP,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;YAEhF,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;gBACrB,OAAO,GAAG,IAAI,CAAC;gBACf,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,mCAAmC,EAAE,CAAC,CAAC;gBAChE,MAAM;YACV,CAAC;YAED,MAAM,UAAU,GAAG,IAAA,qBAAa,EAAC,UAAU,EAAE,YAAY,CAAC,CAAC,CAAC,cAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,GAAG,UAAU,QAAQ,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;YACnI,WAAW,GAAG,UAAU,CAAC,KAAK,CAAC;QACnC,CAAC;QAED,OAAO;YACH,IAAI,EAAE,QAAQ;YACd,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ;YACtC,WAAW,EAAE,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,WAAW;SAC1C,CAAC;IACN,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,QAAgB,EAAE,oBAA4B,EAAE,UAAkB;QAC1F,IAAI,SAAS,GAAG,KAAK,CAAC;QACtB,IAAI,QAAQ,GAAG,oBAAoB,CAAC;QACpC,IAAI,CAAC;YACD,oDAAoD;YACpD,IAAI,CAAC,QAAQ,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;gBACxC,SAAS,GAAG,IAAI,CAAC;gBACjB,MAAM,GAAG,GAAG,cAAI,CAAC,OAAO,CAAC,oBAAoB,IAAI,IAAI,CAAC,UAAU,CAAC,CAAC;gBAClE,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;oBACtB,YAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC3C,CAAC;gBACD,QAAQ,GAAG,cAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC,aAAa,IAAI,UAAU,YAAY,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;gBAC7F,YAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,QAAQ,EAAE,OAAO,CAAC,CAAC;YAClD,CAAC;YACD,MAAM,UAAU,GAAG,MAAM,IAAA,eAAO,EAC5B,IAAI,CAAC,aAAa,EAClB,QAAQ,EACR;gBACI,aAAa,EAAE,IAAI,CAAC,MAAM,CAAC,aAAa;gBACxC,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,WAAW;aAC5C,CACJ,CAAC;YACF,OAAO,UAAU,CAAC;QACtB,CAAC;gBAAS,CAAC;YACP,IAAI,SAAS,EAAE,CAAC;gBACZ,IAAA,mBAAU,EAAC,QAAQ,CAAC,CAAC;YACzB,CAAC;QACL,CAAC;IACL,CAAC;CACJ;AA9ED,8CA8EC"}

package/out/core/features/updateSpec.d.ts

@@ -0,0 +1,20 @@
+import { FeatureBase } from "./common/base";
+import { UTGenerateResult } from "./common/types";
+export interface UpdateSpecParams {
+    sourceCodeFilePath: string;
+    sourceCode: string;
+    specFilePath: string;
+    maxSpecCodeLength?: number;
+    testBlockStartLines?: number[];
+}
+export interface UpdateSpecResult {
+    code: string;
+    status: UTGenerateResult;
+}
+export declare class UpdateSpec extends FeatureBase<UpdateSpecParams, UpdateSpecResult> {
+    #private;
+    protected task(): Promise<UpdateSpecResult>;
+    private fixSpecFile;
+    private fixSpecBlock;
+}
+//# sourceMappingURL=updateSpec.d.ts.map

package/out/core/features/updateSpec.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"updateSpec.d.ts","sourceRoot":"","sources":["../../../src/core/features/updateSpec.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAKlD,MAAM,WAAW,gBAAgB;IAC7B,kBAAkB,EAAE,MAAM,CAAC;IAC3B,UAAU,EAAE,MAAM,CAAC;IACnB,YAAY,EAAE,MAAM,CAAC;IAErB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,mBAAmB,CAAC,EAAE,MAAM,EAAE,CAAC;CAClC;AAED,MAAM,WAAW,gBAAgB;IAC7B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,gBAAgB,CAAC;CAC5B;AAID,qBAAa,UAAW,SAAQ,WAAW,CAAC,gBAAgB,EAAE,gBAAgB,CAAC;;cAI3D,IAAI,IAAI,OAAO,CAAC,gBAAgB,CAAC;YAcnC,WAAW;YAmBX,YAAY;CAwD7B"}

package/out/core/features/updateSpec.js

@@ -0,0 +1,98 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UpdateSpec = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const base_1 = require("./common/base");
+const types_1 = require("./common/types");
+const specBlock_1 = require("../utils/specBlock");
+const common_1 = require("../utils/common");
+const EMPTY_FLAG = 'NO_NEED_TO_CHANGE';
+class UpdateSpec extends base_1.FeatureBase {
+    #sourceFileName;
+    #specRelativePath;
+    async task() {
+        const specCode = fs_1.default.readFileSync(this.params.specFilePath, 'utf-8');
+        this.#sourceFileName = path_1.default.basename(this.params.sourceCodeFilePath);
+        this.#specRelativePath = path_1.default.relative(this.params.sourceCodeFilePath, this.params.specFilePath);
+        const isTooLong = specCode.length > (this.params.maxSpecCodeLength || 40000);
+        let newSpecCode = await (isTooLong ? this.fixSpecBlock(this.params.specFilePath) : this.fixSpecFile(specCode));
+        return {
+            code: newSpecCode,
+            status: newSpecCode ? types_1.UTGenerateResult.Success : types_1.UTGenerateResult.Skipped,
+        };
+    }
+    async fixSpecFile(oldSpecCode) {
+        const messages = this.getPrompt("UpdateSpecFile" /* PromptKey.UpdateSpecFile */, {
+            ["<<<SourceFileName>>>" /* PromptPlaceholder.SourceFileName */]: this.#sourceFileName,
+            ["<<<SourceCode>>>" /* PromptPlaceholder.SourceCode */]: this.params.sourceCode,
+            ["<<<SpecFileName>>>" /* PromptPlaceholder.SpecFileName */]: this.#specRelativePath,
+            ["<<<Spec>>>" /* PromptPlaceholder.SpecCode */]: oldSpecCode,
+            ["<<<Language>>>" /* PromptPlaceholder.Language */]: (0, common_1.getLanguage)(this.#sourceFileName),
+        });
+        this.progress({ message: 'Generating code...' });
+        const newSpecCode = await this.getCodeFromLLM(messages, EMPTY_FLAG);
+        if (!newSpecCode) {
+            this.progress({ message: 'No need to update.', finish: 1 });
+            return;
+        }
+        return newSpecCode;
+    }
+    async fixSpecBlock(specPath) {
+        if (!this.params.testBlockStartLines?.length) {
+            this.progress({ message: 'Spec file is too large, cannot update, please provide test block start lines' });
+            return;
+        }
+        this.progress({ message: 'File is too large, splitting into blocks...' });
+        const { imports, blocks, updateBlocks } = (0, specBlock_1.getSpecBlocks)(specPath, this.params.testBlockStartLines);
+        this.progress({ message: `Found ${blocks.length} related blocks` });
+        const newContent = [];
+        let updateCount = 0;
+        const importStats = imports.join('\n');
+        for (const block of blocks) {
+            try {
+                this.progress({ message: `Generating code for ${block.type}("${block.name}")` });
+                const testDeps = block.mocks.concat(block.lifecycles).join('\n');
+                const codeExt = [importStats, testDeps].filter(el => !!el.trim()).join('\n');
+                const messages = this.getPrompt("UpdateSpecBlock" /* PromptKey.UpdateSpecBlock */, {
+                    ["<<<SourceFileName>>>" /* PromptPlaceholder.SourceFileName */]: this.#sourceFileName,
+                    ["<<<SourceCode>>>" /* PromptPlaceholder.SourceCode */]: this.params.sourceCode,
+                    ["<<<SpecFileName>>>" /* PromptPlaceholder.SpecFileName */]: this.#specRelativePath,
+                    ["<<<Spec>>>" /* PromptPlaceholder.SpecCode */]: block.body,
+                    ["<<<CodeExtInfo>>>" /* PromptPlaceholder.CodeExtInfo */]: codeExt,
+                    ["<<<Language>>>" /* PromptPlaceholder.Language */]: (0, common_1.getLanguage)(this.#sourceFileName),
+                });
+                const newSpecCode = await this.getCodeFromLLM(messages, EMPTY_FLAG);
+                if (!newSpecCode) {
+                    this.progress({ message: 'No need to update' });
+                }
+                else {
+                    updateCount++;
+                }
+                newContent.push(newSpecCode || block.body);
+            }
+            catch (error) {
+                this.progress({
+                    message: `Error in generating code: ${error.message}`,
+                    failed: true,
+                    lastFailed: true,
+                    extra: {
+                        title: 'Failed to update spec block',
+                        content: JSON.stringify(error, null, 2)
+                    }
+                });
+            }
+        }
+        if (!updateCount) {
+            this.progress({ message: `No block updated` });
+            return;
+        }
+        this.progress({ message: `Updating ${this.#specRelativePath} with ${updateCount} blocks` });
+        return updateBlocks(newContent);
+    }
+}
+exports.UpdateSpec = UpdateSpec;
+//# sourceMappingURL=updateSpec.js.map

package/out/core/features/updateSpec.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"updateSpec.js","sourceRoot":"","sources":["../../../src/core/features/updateSpec.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,gDAAwB;AACxB,wCAA4C;AAC5C,0CAAkD;AAElD,kDAAmD;AACnD,4CAA8C;AAgB9C,MAAM,UAAU,GAAG,mBAAmB,CAAC;AAEvC,MAAa,UAAW,SAAQ,kBAA+C;IAC3E,eAAe,CAAS;IACxB,iBAAiB,CAAS;IAEhB,KAAK,CAAC,IAAI;QAChB,MAAM,QAAQ,GAAG,YAAE,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAEpE,IAAI,CAAC,eAAe,GAAG,cAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,CAAC,CAAC;QACrE,IAAI,CAAC,iBAAiB,GAAG,cAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,MAAM,CAAC,kBAAkB,EAAE,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC;QAEjG,MAAM,SAAS,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,CAAC,MAAM,CAAC,iBAAiB,IAAI,KAAK,CAAC,CAAC;QAC7E,IAAI,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC,CAAC;QAC/G,OAAO;YACH,IAAI,EAAE,WAAW;YACjB,MAAM,EAAE,WAAW,CAAC,CAAC,CAAC,wBAAgB,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAgB,CAAC,OAAO;SAC5E,CAAA;IACL,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,WAAmB;QACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,kDAE3B;YACI,+DAAkC,EAAE,IAAI,CAAC,eAAe;YACxD,uDAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;YACtD,2DAAgC,EAAE,IAAI,CAAC,iBAAiB;YACxD,+CAA4B,EAAE,WAAW;YACzC,mDAA4B,EAAE,IAAA,oBAAW,EAAC,IAAI,CAAC,eAAe,CAAC;SAClE,CAAC,CAAA;QACN,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,oBAAoB,EAAE,CAAC,CAAC;QACjD,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;QACpE,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,oBAAoB,EAAE,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC;YAC5D,OAAO;QACX,CAAC;QACD,OAAO,WAAW,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,YAAY,CAAC,QAAgB;QACvC,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,mBAAmB,EAAE,MAAM,EAAE,CAAC;YAC3C,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,8EAA8E,EAAE,CAAC,CAAC;YAC3G,OAAO;QACX,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,6CAA6C,EAAE,CAAC,CAAC;QAE1E,MAAM,EAAE,OAAO,EAAE,MAAM,EAAE,YAAY,EAAE,GAAG,IAAA,yBAAa,EAAC,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,mBAAmB,CAAC,CAAC;QACnG,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,SAAS,MAAM,CAAC,MAAM,iBAAiB,EAAE,CAAC,CAAC;QACpE,MAAM,UAAU,GAAa,EAAE,CAAC;QAChC,IAAI,WAAW,GAAG,CAAC,CAAC;QACpB,MAAM,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvC,KAAK,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;YACzB,IAAI,CAAC;gBACD,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,uBAAuB,KAAK,CAAC,IAAI,KAAK,KAAK,CAAC,IAAI,IAAI,EAAE,CAAC,CAAC;gBACjF,MAAM,QAAQ,GAAG,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBACjE,MAAM,OAAO,GAAG,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC,MAAM,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;gBAE7E,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,oDAE3B;oBACI,+DAAkC,EAAE,IAAI,CAAC,eAAe;oBACxD,uDAA8B,EAAE,IAAI,CAAC,MAAM,CAAC,UAAU;oBACtD,2DAAgC,EAAE,IAAI,CAAC,iBAAiB;oBACxD,+CAA4B,EAAE,KAAK,CAAC,IAAI;oBACxC,yDAA+B,EAAE,OAAO;oBACxC,mDAA4B,EAAE,IAAA,oBAAW,EAAC,IAAI,CAAC,eAAe,CAAC;iBAClE,CAAC,CAAC;gBACP,MAAM,WAAW,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,EAAE,UAAU,CAAC,CAAC;gBAEpE,IAAI,CAAC,WAAW,EAAE,CAAC;oBACf,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,mBAAmB,EAAE,CAAC,CAAC;gBACpD,CAAC;qBAAM,CAAC;oBACJ,WAAW,EAAE,CAAC;gBAClB,CAAC;gBACD,UAAU,CAAC,IAAI,CAAC,WAAW,IAAI,KAAK,CAAC,IAAI,CAAC,CAAC;YAC/C,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACb,IAAI,CAAC,QAAQ,CAAC;oBACV,OAAO,EAAE,6BAA6B,KAAK,CAAC,OAAO,EAAE;oBACrD,MAAM,EAAE,IAAI;oBACZ,UAAU,EAAE,IAAI;oBAChB,KAAK,EAAE;wBACH,KAAK,EAAE,6BAA6B;wBACpC,OAAO,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,EAAE,IAAI,EAAE,CAAC,CAAC;qBAC1C;iBACJ,CAAC,CAAC;YACP,CAAC;QACL,CAAC;QACD,IAAI,CAAC,WAAW,EAAE,CAAC;YACf,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,kBAAkB,EAAE,CAAC,CAAC;YAC/C,OAAO;QACX,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,YAAY,IAAI,CAAC,iBAAiB,SAAS,WAAW,SAAS,EAAE,CAAC,CAAC;QAC5F,OAAO,YAAY,CAAC,UAAU,CAAC,CAAC;IACpC,CAAC;CACJ;AA7FD,gCA6FC"}

package/out/core/features/validFix.d.ts

@@ -0,0 +1,29 @@
+import { FailedCaseDetails, TestOutput } from "../valid/type";
+import { FeatureBase } from "./common/base";
+export interface ValidFixParams {
+    targetName?: string;
+    sourceFilePath: string;
+    targetSourceCode?: string;
+    sourceMockCode?: string;
+    inputSpecCode?: string;
+    specFilePath?: string;
+    projectFolder?: string;
+    testFileSpecifier?: string;
+    maxRetry?: number;
+}
+export interface ValidFixResult {
+    code: string;
+    status: 'valid' | 'failed' | 'error';
+    message?: string;
+    testResult?: TestOutput;
+    failedCases?: FailedCaseDetails[];
+    isDataErrorOnly?: boolean;
+}
+export declare class ValidFix extends FeatureBase<ValidFixParams, ValidFixResult> {
+    #private;
+    protected task(): Promise<ValidFixResult>;
+    private validSpecCode;
+    private fixSpecCodeWithRetry;
+    private fixSpecCode;
+}
+//# sourceMappingURL=validFix.d.ts.map

package/out/core/features/validFix.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"validFix.d.ts","sourceRoot":"","sources":["../../../src/core/features/validFix.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,iBAAiB,EAAE,UAAU,EAAE,MAAM,eAAe,CAAC;AAI9D,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAU5C,MAAM,WAAW,cAAc;IAE3B,UAAU,CAAC,EAAE,MAAM,CAAC;IAEpB,cAAc,EAAE,MAAM,CAAC;IAEvB,gBAAgB,CAAC,EAAE,MAAM,CAAC;IAE1B,cAAc,CAAC,EAAE,MAAM,CAAC;IAExB,aAAa,CAAC,EAAE,MAAM,CAAC;IAEvB,YAAY,CAAC,EAAE,MAAM,CAAC;IAEtB,aAAa,CAAC,EAAE,MAAM,CAAC;IAEvB,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,QAAQ,CAAC,EAAE,MAAM,CAAC;CACrB;AAED,MAAM,WAAW,cAAc;IAC3B,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,OAAO,GAAG,QAAQ,GAAG,OAAO,CAAC;IACrC,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,UAAU,CAAC,EAAE,UAAU,CAAC;IACxB,WAAW,CAAC,EAAE,iBAAiB,EAAE,CAAC;IAClC,eAAe,CAAC,EAAE,OAAO,CAAC;CAC7B;AAED,qBAAa,QAAS,SAAQ,WAAW,CAAC,cAAc,EAAE,cAAc,CAAC;;cAKrD,IAAI,IAAI,OAAO,CAAC,cAAc,CAAC;YAqDjC,aAAa;YAuDb,oBAAoB;YA4DpB,WAAW;CAmD5B"}

package/out/core/features/validFix.js

@@ -0,0 +1,219 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ValidFix = void 0;
+const valid_1 = require("../valid");
+const base_1 = require("./common/base");
+const fs_1 = __importDefault(require("fs"));
+const common_1 = require("../utils/common");
+const deps_1 = require("../utils/deps");
+const tsMorph_1 = require("../utils/tsMorph");
+const path_1 = __importDefault(require("path"));
+const analyze_1 = require("../analyze");
+const system_1 = require("../utils/system");
+const move_1 = require("../utils/move");
+class ValidFix extends base_1.FeatureBase {
+    #sourceDir;
+    #pureFileName;
+    #fileSuffix;
+    async task() {
+        const { inputSpecCode, sourceFilePath, specFilePath } = this.params;
+        this.#sourceDir = path_1.default.dirname(sourceFilePath);
+        this.#fileSuffix = path_1.default.extname(sourceFilePath);
+        this.#pureFileName = path_1.default.basename(sourceFilePath, this.#fileSuffix);
+        let specCode = inputSpecCode;
+        if (!specCode) {
+            if (!specFilePath) {
+                this.progress({ message: `No spec file found`, failed: true });
+                return {
+                    code: '',
+                    status: 'error',
+                    message: `No spec file found`,
+                };
+            }
+            if (!fs_1.default.existsSync(specFilePath)) {
+                this.progress({ message: `Spec file not found`, failed: true });
+                return {
+                    code: '',
+                    status: 'error',
+                    message: `Spec file not exists: ${specFilePath}`,
+                };
+            }
+            specCode = fs_1.default.readFileSync(specFilePath, 'utf-8');
+        }
+        const res = await this.validSpecCode(specCode, specFilePath);
+        const testResult = res.testResult;
+        let failedCases = res.failedCases;
+        let isDataErrorOnly = res.isDataErrorOnly;
+        if (testResult.success) {
+            return {
+                code: specCode,
+                status: 'valid',
+            };
+        }
+        if (!failedCases) {
+            const failedData = (0, valid_1.getFailedCase)(testResult, specFilePath ? path_1.default.basename(specFilePath) : undefined);
+            failedCases = failedData.cases;
+            isDataErrorOnly = failedData.isDataErrorOnly;
+        }
+        this.progress({ message: `Generate fix code` });
+        const result = await this.fixSpecCodeWithRetry(specCode, failedCases, isDataErrorOnly);
+        if (!result.testResult.success) {
+            this.progress({ message: `Test validation failed`, failed: true });
+        }
+        return result;
+    }
+    async validSpecCode(testCode, expectedSpecFilePath) {
+        let isTmpFile = false;
+        let specPath = expectedSpecFilePath;
+        try {
+            // for new spec file and not saved yet, use tmp file
+            if (!specPath || !fs_1.default.existsSync(specPath)) {
+                const { targetName } = this.params;
+                isTmpFile = true;
+                const dir = path_1.default.dirname(expectedSpecFilePath || this.#sourceDir);
+                if (!fs_1.default.existsSync(dir)) {
+                    fs_1.default.mkdirSync(dir, { recursive: true });
+                }
+                const specName = `tmp.${this.#pureFileName}${targetName ? '.' + targetName : ''}.${this.params.testFileSpecifier}${this.#fileSuffix}`;
+                specPath = path_1.default.join(dir, specName);
+                // if test code and source code are in different dirs, adjust the import path
+                const testFileDir = path_1.default.dirname(specPath);
+                let finalCode = testCode;
+                finalCode = await (0, move_1.updateCodeReferences)(this.#sourceDir, testFileDir, specName, finalCode);
+                fs_1.default.writeFileSync(specPath, finalCode, 'utf-8');
+            }
+            this.progress({ message: 'Running test' });
+            const testResult = await (0, valid_1.runTest)(this.testFramework, specPath, {
+                projectFolder: this.params.projectFolder,
+                abortSignal: this.taskOptions.abortSignal,
+            });
+            if (testResult?.success) {
+                this.progress({ message: `All tests passed` });
+            }
+            else {
+                this.progress({ message: `Some tests failed` });
+            }
+            const specFileName = path_1.default.basename(specPath);
+            const failedData = testResult.success
+                ? {
+                    cases: [],
+                    isDataErrorOnly: false,
+                }
+                : (0, valid_1.getFailedCase)(testResult, path_1.default.basename(specFileName));
+            return {
+                failedCases: failedData.cases,
+                isDataErrorOnly: failedData.isDataErrorOnly || false,
+                testResult,
+            };
+        }
+        finally {
+            if (isTmpFile) {
+                (0, system_1.removeFile)(specPath);
+            }
+        }
+    }
+    async fixSpecCodeWithRetry(testCode, initialFailedCases, isDataErrorOnly) {
+        const maxRetry = this.params.maxRetry || 3;
+        let currentTestCode = testCode;
+        let currentFailedCases = initialFailedCases;
+        let currentIsDataErrorOnly = isDataErrorOnly;
+        for (let i = 0; i < maxRetry; i++) {
+            this.progress({
+                message: `(${i + 1}/${maxRetry}) Generating fix code`
+            });
+            const fixedCode = await this.fixSpecCode(currentTestCode, currentFailedCases, currentIsDataErrorOnly);
+            this.progress({ message: 'Start to run test' });
+            // Test the fixed code
+            const validRes = await this.validSpecCode(fixedCode, this.params.specFilePath);
+            if (validRes.testResult.success) {
+                this.progress({ message: `Fix successful after ${i + 1} attempts` });
+                return {
+                    code: fixedCode,
+                    status: 'valid',
+                    testResult: validRes.testResult,
+                    failedCases: validRes.failedCases,
+                    isDataErrorOnly: validRes.isDataErrorOnly,
+                };
+            }
+            // If this is the last retry, return the result even if tests failed
+            if (i === maxRetry - 1) {
+                return {
+                    code: fixedCode,
+                    status: 'failed',
+                    testResult: validRes.testResult,
+                    failedCases: validRes.failedCases,
+                    isDataErrorOnly: validRes.isDataErrorOnly,
+                };
+            }
+            // Prepare for next iteration
+            currentTestCode = fixedCode;
+            if (validRes.failedCases && validRes.failedCases.length > 0) {
+                currentFailedCases = validRes.failedCases;
+                const failedData = (0, valid_1.getFailedCase)(validRes.testResult, this.params.specFilePath ? path_1.default.basename(this.params.specFilePath) : undefined);
+                currentIsDataErrorOnly = failedData.isDataErrorOnly;
+                this.progress({
+                    message: `(${i + 1}/${maxRetry}) Fix failed, retrying with ${currentFailedCases.length} failed cases`,
+                    lastFailed: true
+                });
+            }
+            else {
+                // No specific failed cases found, but test still failed
+                this.progress({
+                    message: `(${i + 1}/${maxRetry}) Fix failed with unknown errors, retrying`,
+                    lastFailed: true
+                });
+            }
+        }
+    }
+    async fixSpecCode(testCode, cases, isDataErrorOnly) {
+        let repairPrompt;
+        if (isDataErrorOnly) {
+            repairPrompt = this.getPrompt("FixTestDataErrors" /* PromptKey.FixTestDataErrors */, {
+                ["<<<Spec>>>" /* PromptPlaceholder.SpecCode */]: testCode,
+                ["<<<TestErrors>>>" /* PromptPlaceholder.TestErrors */]: JSON.stringify((0, valid_1.getDataErrors)(testCode, cases)),
+                ["<<<Language>>>" /* PromptPlaceholder.Language */]: (0, common_1.getLanguage)(this.#fileSuffix),
+                ["<<<TestFramework>>>" /* PromptPlaceholder.TestFramework */]: this.testFramework,
+            });
+        }
+        else {
+            let { sourceFilePath, targetSourceCode, sourceMockCode, specFilePath, targetName } = this.params;
+            if (!targetSourceCode && sourceFilePath) {
+                if (targetName) {
+                    const { project } = (0, tsMorph_1.getProject)(sourceFilePath);
+                    const exportedList = await (0, analyze_1.read)(sourceFilePath, {
+                        cachedProject: project
+                    });
+                    for (const data of exportedList) {
+                        if (data.name === targetName) {
+                            targetSourceCode = data.code;
+                            sourceMockCode = (0, deps_1.getPackageMocks)(data.importDeclares);
+                            break;
+                        }
+                    }
+                }
+                else {
+                    targetSourceCode = fs_1.default.readFileSync(sourceFilePath, 'utf-8');
+                }
+            }
+            const specFileName = specFilePath
+                ? path_1.default.basename(specFilePath)
+                : `${this.#pureFileName}.spec${this.#fileSuffix}`;
+            repairPrompt = this.getPrompt("FixTestErrors" /* PromptKey.FixTestErrors */, {
+                ["<<<Spec>>>" /* PromptPlaceholder.SpecCode */]: testCode,
+                ["<<<SourceFileName>>>" /* PromptPlaceholder.SourceFileName */]: sourceFilePath,
+                ["<<<SpecFileName>>>" /* PromptPlaceholder.SpecFileName */]: specFileName,
+                ["<<<SourceCode>>>" /* PromptPlaceholder.SourceCode */]: targetSourceCode || '',
+                ["<<<MockCode>>>" /* PromptPlaceholder.MockCode */]: sourceMockCode || '',
+                ["<<<TestErrors>>>" /* PromptPlaceholder.TestErrors */]: JSON.stringify(cases),
+                ["<<<Language>>>" /* PromptPlaceholder.Language */]: (0, common_1.getLanguage)(this.#fileSuffix),
+                ["<<<TestFramework>>>" /* PromptPlaceholder.TestFramework */]: this.testFramework,
+            });
+        }
+        return await this.getCodeFromLLM(repairPrompt);
+    }
+}
+exports.ValidFix = ValidFix;
+//# sourceMappingURL=validFix.js.map