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

package/out/core/config/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/config/index.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,UAAU;IACvB,MAAM,CAAC,EAAE;QACL,eAAe,CAAC,EAAE,OAAO,CAAC;QAC1B,iBAAiB,CAAC,EAAE,OAAO,CAAC;QAC5B,WAAW,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;QAC1B,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,UAAU,CAAC,EAAE,OAAO,CAAC;QACrB,gBAAgB,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC,CAAC;QAC1D,UAAU,CAAC,EAAE,MAAM,CAAC;QACpB,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,aAAa,CAAC,EAAE,MAAM,CAAC;QACvB,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,qBAAqB,CAAC,EAAE,OAAO,CAAC;QAChC,wBAAwB,CAAC,EAAE,MAAM,CAAC;QAClC,iBAAiB,CAAC,EAAE,MAAM,CAAC;QAC3B,iBAAiB,CAAC,EAAE,MAAM,CAAC;KAC9B,CAAC;IACF,GAAG,CAAC,EAAE;QACF,GAAG,EAAE,MAAM,CAAC;KACf,CAAA;CACJ;AAED,MAAM,MAAM,YAAY,GAAG,MAAM,UAAU,CAAC;AAM5C,wBAAgB,eAAe,CAAC,EAAE,EAAE,MAAM,UAAU,QAEnD;AAED,wBAAgB,SAAS,IAAI,UAAU,CAEtC"}

package/out/core/config/index.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.setConfigGetter = setConfigGetter;
+exports.getConfig = getConfig;
+let _getConfig = () => ({
+    common: {}
+});
+function setConfigGetter(cb) {
+    _getConfig = cb;
+}
+function getConfig() {
+    return _getConfig();
+}
+//# sourceMappingURL=index.js.map

package/out/core/config/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/config/index.ts"],"names":[],"mappings":";;AA4BA,0CAEC;AAED,8BAEC;AAVD,IAAI,UAAU,GAAqB,GAAG,EAAE,CAAC,CAAC;IACtC,MAAM,EAAE,EAAE;CACb,CAAC,CAAC;AAEH,SAAgB,eAAe,CAAC,EAAoB;IAChD,UAAU,GAAG,EAAE,CAAC;AACpB,CAAC;AAED,SAAgB,SAAS;IACrB,OAAO,UAAU,EAAE,CAAC;AACxB,CAAC"}

package/out/core/constants.d.ts

@@ -0,0 +1,9 @@
+export declare const TEST_COPILOT_CONFIG_DIR = ".test_copilot";
+export declare const TEST_COPILOT_CONFIG_PROMPTS_UT_TS = ".test_copilot/prompts/ts";
+export declare const DISABLE_ESLINT_RULE_TEMPLATE = "/* eslint-disable %s */";
+export declare const JEST_PARSE_ERROR = "JestParseError";
+export declare const REPO_DIR_PLACEHOLDER = "{repoDir}";
+export declare const PACKAGE_DIR_PLACEHOLDER = "{packageDir}";
+export declare const TEST_FILE_PATH_PLACEHOLDER = "{testFilePath}";
+export declare const DEFAULT_RETRY_TIMES = 3;
+//# sourceMappingURL=constants.d.ts.map

package/out/core/constants.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.d.ts","sourceRoot":"","sources":["../../src/core/constants.ts"],"names":[],"mappings":"AAAA,eAAO,MAAM,uBAAuB,kBAAkB,CAAC;AACvD,eAAO,MAAM,iCAAiC,6BAA0C,CAAC;AACzF,eAAO,MAAM,4BAA4B,4BAA4B,CAAC;AACtE,eAAO,MAAM,gBAAgB,mBAAmB,CAAC;AACjD,eAAO,MAAM,oBAAoB,cAAc,CAAC;AAChD,eAAO,MAAM,uBAAuB,iBAAiB,CAAC;AACtD,eAAO,MAAM,0BAA0B,mBAAmB,CAAC;AAC3D,eAAO,MAAM,mBAAmB,IAAI,CAAC"}

package/out/core/constants.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DEFAULT_RETRY_TIMES = exports.TEST_FILE_PATH_PLACEHOLDER = exports.PACKAGE_DIR_PLACEHOLDER = exports.REPO_DIR_PLACEHOLDER = exports.JEST_PARSE_ERROR = exports.DISABLE_ESLINT_RULE_TEMPLATE = exports.TEST_COPILOT_CONFIG_PROMPTS_UT_TS = exports.TEST_COPILOT_CONFIG_DIR = void 0;
+exports.TEST_COPILOT_CONFIG_DIR = '.test_copilot';
+exports.TEST_COPILOT_CONFIG_PROMPTS_UT_TS = `${exports.TEST_COPILOT_CONFIG_DIR}/prompts/ts`;
+exports.DISABLE_ESLINT_RULE_TEMPLATE = '/* eslint-disable %s */';
+exports.JEST_PARSE_ERROR = 'JestParseError';
+exports.REPO_DIR_PLACEHOLDER = '{repoDir}';
+exports.PACKAGE_DIR_PLACEHOLDER = '{packageDir}';
+exports.TEST_FILE_PATH_PLACEHOLDER = '{testFilePath}';
+exports.DEFAULT_RETRY_TIMES = 3;
+//# sourceMappingURL=constants.js.map

package/out/core/constants.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"constants.js","sourceRoot":"","sources":["../../src/core/constants.ts"],"names":[],"mappings":";;;AAAa,QAAA,uBAAuB,GAAG,eAAe,CAAC;AAC1C,QAAA,iCAAiC,GAAG,GAAG,+BAAuB,aAAa,CAAC;AAC5E,QAAA,4BAA4B,GAAG,yBAAyB,CAAC;AACzD,QAAA,gBAAgB,GAAG,gBAAgB,CAAC;AACpC,QAAA,oBAAoB,GAAG,WAAW,CAAC;AACnC,QAAA,uBAAuB,GAAG,cAAc,CAAC;AACzC,QAAA,0BAA0B,GAAG,gBAAgB,CAAC;AAC9C,QAAA,mBAAmB,GAAG,CAAC,CAAC"}

package/out/core/features/common/base.d.ts

@@ -0,0 +1,20 @@
+import { OnProgress } from "../../../types";
+import { LLMMessage, PromptKey } from "../../llm";
+import type { TaskOptions } from "./types";
+export declare abstract class FeatureBase<TParams, TResult> {
+    #private;
+    protected params: TParams;
+    get stopped(): boolean;
+    get taskOptions(): TaskOptions;
+    private onAbort;
+    get testFramework(): string;
+    constructor(options?: TaskOptions);
+    run(params: TParams): Promise<TResult>;
+    protected abstract task(): Promise<any>;
+    protected sleep(ms: number): Promise<void>;
+    protected progress: OnProgress;
+    protected getPrompt(promptKey: PromptKey, values: Record<string, string>): LLMMessage[];
+    protected getCodeFromLLM(messages: LLMMessage[], emptyFlag?: string): Promise<string>;
+    protected runSubTask<TFeature extends FeatureBase<any, any>>(feat: new (options: any) => TFeature, params: TFeature extends FeatureBase<infer P, any> ? P : any): Promise<TFeature extends FeatureBase<any, infer R> ? R : any>;
+}
+//# sourceMappingURL=base.d.ts.map

package/out/core/features/common/base.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.d.ts","sourceRoot":"","sources":["../../../../src/core/features/common/base.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAC5C,OAAO,EAA6B,UAAU,EAAE,SAAS,EAAqB,MAAM,WAAW,CAAC;AAEhG,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,SAAS,CAAC;AAE3C,8BAAsB,WAAW,CAAC,OAAO,EAAE,OAAO;;IAI9C,SAAS,CAAC,MAAM,EAAE,OAAO,CAAC;IAE1B,IAAW,OAAO,IAAI,OAAO,CAE5B;IAED,IAAW,WAAW,IAAI,WAAW,CAEpC;IAED,OAAO,CAAC,OAAO,CAAmC;IAElD,IAAW,aAAa,IAAI,MAAM,CAEjC;gBAEW,OAAO,CAAC,EAAE,WAAW;IAOpB,GAAG,CAAC,MAAM,EAAE,OAAO,GAAG,OAAO,CAAC,OAAO,CAAC;IAcnD,SAAS,CAAC,QAAQ,CAAC,IAAI,IAAI,OAAO,CAAC,GAAG,CAAC;IAGvC,SAAS,CAAC,KAAK,CAAC,EAAE,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAI1C,SAAS,CAAC,QAAQ,EAAE,UAAU,CAK5B;IAEF,SAAS,CAAC,SAAS,CAAC,SAAS,EAAE,SAAS,EAAE,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,CAAC;cAWxD,cAAc,CAAC,QAAQ,EAAE,UAAU,EAAE,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;cAgB3E,UAAU,CAAC,QAAQ,SAAS,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,EAC7D,IAAI,EAAE,KAAK,OAAO,KAAA,KAAK,QAAQ,EAC/B,MAAM,EAAE,QAAQ,SAAS,WAAW,CAAC,MAAM,CAAC,EAAE,GAAG,CAAC,GAAG,CAAC,GAAG,GAAG,GAC7D,OAAO,CAAC,QAAQ,SAAS,WAAW,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC,GAAG,CAAC,GAAG,GAAG,CAAC;CAInE"}

package/out/core/features/common/base.js

@@ -0,0 +1,77 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FeatureBase = void 0;
+const llm_1 = require("../../llm");
+const common_1 = require("../../utils/common");
+class FeatureBase {
+    #stopped = false;
+    #taskOptions;
+    params;
+    get stopped() {
+        return this.#stopped;
+    }
+    get taskOptions() {
+        return this.#taskOptions;
+    }
+    onAbort = () => { this.#stopped = true; };
+    get testFramework() {
+        return this.taskOptions?.testFramework || 'jest';
+    }
+    constructor(options) {
+        this.#taskOptions = options;
+        if (options?.abortSignal) {
+            options.abortSignal.addEventListener('abort', this.onAbort, { once: true });
+        }
+    }
+    async run(params) {
+        try {
+            if (this.stopped) {
+                return;
+            }
+            this.params = params;
+            return await this.task();
+        }
+        finally {
+            if (this.#taskOptions?.abortSignal) {
+                this.#taskOptions?.abortSignal.removeEventListener('abort', this.onAbort);
+            }
+        }
+    }
+    // use this method to allow the long synchronous task to be interrupted by the user
+    sleep(ms) {
+        return new Promise(resolve => setTimeout(resolve, ms));
+    }
+    progress = (options) => {
+        if (!this.taskOptions?.progress) {
+            return;
+        }
+        this.taskOptions.progress(options);
+    };
+    getPrompt(promptKey, values) {
+        return (0, llm_1.getPrompt)(promptKey, {
+            ...values,
+            ["<<<TestFramework>>>" /* PromptPlaceholder.TestFramework */]: this.testFramework
+        }, this.taskOptions.promptConfigs?.[promptKey]);
+    }
+    async getCodeFromLLM(messages, emptyFlag) {
+        const result = await (0, llm_1.getLLMResponse)(messages);
+        if (!result?.success) {
+            throw new Error('Failed to get response from LLM');
+        }
+        if (emptyFlag && result.content.includes(emptyFlag)) {
+            return '';
+        }
+        const resultCode = (0, common_1.extractCodeFromResponse)(result.content);
+        if (!resultCode) {
+            throw new Error('Failed to extract code from response');
+        }
+        this.progress({ message: 'Code generated' });
+        return resultCode;
+    }
+    async runSubTask(feat, params) {
+        const task = new feat(this.taskOptions);
+        return await task.run(params);
+    }
+}
+exports.FeatureBase = FeatureBase;
+//# sourceMappingURL=base.js.map

package/out/core/features/common/base.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"base.js","sourceRoot":"","sources":["../../../../src/core/features/common/base.ts"],"names":[],"mappings":";;;AACA,mCAAgG;AAChG,+CAA6D;AAG7D,MAAsB,WAAW;IAC7B,QAAQ,GAAG,KAAK,CAAC;IACjB,YAAY,CAAe;IAEjB,MAAM,CAAU;IAE1B,IAAW,OAAO;QACd,OAAO,IAAI,CAAC,QAAQ,CAAC;IACzB,CAAC;IAED,IAAW,WAAW;QAClB,OAAO,IAAI,CAAC,YAAY,CAAC;IAC7B,CAAC;IAEO,OAAO,GAAG,GAAG,EAAE,GAAG,IAAI,CAAC,QAAQ,GAAG,IAAI,CAAC,CAAC,CAAC,CAAC;IAElD,IAAW,aAAa;QACpB,OAAO,IAAI,CAAC,WAAW,EAAE,aAAa,IAAI,MAAM,CAAC;IACrD,CAAC;IAED,YAAY,OAAqB;QAC7B,IAAI,CAAC,YAAY,GAAG,OAAO,CAAC;QAC5B,IAAI,OAAO,EAAE,WAAW,EAAE,CAAC;YACvB,OAAO,CAAC,WAAW,CAAC,gBAAgB,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;QAChF,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,GAAG,CAAC,MAAe;QAC5B,IAAI,CAAC;YACD,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,OAAO;YACX,CAAC;YACD,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;YACrB,OAAO,MAAM,IAAI,CAAC,IAAI,EAAE,CAAC;QAC7B,CAAC;gBAAS,CAAC;YACP,IAAI,IAAI,CAAC,YAAY,EAAE,WAAW,EAAE,CAAC;gBACjC,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC,mBAAmB,CAAC,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,CAAC;YAC9E,CAAC;QACL,CAAC;IACL,CAAC;IAID,mFAAmF;IACzE,KAAK,CAAC,EAAU;QACtB,OAAO,IAAI,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC,CAAC;IAC3D,CAAC;IAES,QAAQ,GAAe,CAAC,OAAO,EAAE,EAAE;QACzC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,EAAE,CAAC;YAC9B,OAAO;QACX,CAAC;QACD,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;IACvC,CAAC,CAAC;IAEQ,SAAS,CAAC,SAAoB,EAAE,MAA8B;QACpE,OAAO,IAAA,eAAS,EACZ,SAAS,EACT;YACI,GAAG,MAAM;YACT,6DAAiC,EAAE,IAAI,CAAC,aAAa;SACxD,EACD,IAAI,CAAC,WAAW,CAAC,aAAa,EAAE,CAAC,SAAS,CAAC,CAC9C,CAAC;IACN,CAAC;IAES,KAAK,CAAC,cAAc,CAAC,QAAsB,EAAE,SAAkB;QACrE,MAAM,MAAM,GAAG,MAAM,IAAA,oBAAc,EAAC,QAAQ,CAAC,CAAC;QAC9C,IAAI,CAAC,MAAM,EAAE,OAAO,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;QACvD,CAAC;QACD,IAAI,SAAS,IAAI,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC,EAAE,CAAC;YAClD,OAAO,EAAE,CAAC;QACd,CAAC;QACD,MAAM,UAAU,GAAG,IAAA,gCAAuB,EAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC3D,IAAI,CAAC,UAAU,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;QAC5D,CAAC;QACD,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,gBAAgB,EAAE,CAAC,CAAC;QAC7C,OAAO,UAAU,CAAC;IACtB,CAAC;IAES,KAAK,CAAC,UAAU,CACtB,IAA+B,EAC/B,MAA4D;QAE5D,MAAM,IAAI,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QACxC,OAAO,MAAM,IAAI,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;IAClC,CAAC;CACJ;AAzFD,kCAyFC"}

package/out/core/features/common/types.d.ts

@@ -0,0 +1,16 @@
+import type { OnProgress } from "../../../types";
+import type { PromptConfigs } from "../../llm";
+export declare enum UTGenerateResult {
+    Success = "success",
+    Error = "error",
+    Cancelled = "cancelled",
+    NotFound = "notFound",
+    Skipped = "skipped"
+}
+export interface TaskOptions {
+    abortSignal?: AbortSignal;
+    progress?: OnProgress;
+    promptConfigs?: Partial<PromptConfigs>;
+    testFramework?: string;
+}
+//# sourceMappingURL=types.d.ts.map

package/out/core/features/common/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../../src/core/features/common/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AACjD,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,WAAW,CAAC;AAE/C,oBAAY,gBAAgB;IACxB,OAAO,YAAY;IACnB,KAAK,UAAU;IACf,SAAS,cAAc;IACvB,QAAQ,aAAa;IACrB,OAAO,YAAY;CACtB;AAED,MAAM,WAAW,WAAW;IACxB,WAAW,CAAC,EAAE,WAAW,CAAC;IAC1B,QAAQ,CAAC,EAAE,UAAU,CAAC;IACtB,aAAa,CAAC,EAAE,OAAO,CAAC,aAAa,CAAC,CAAC;IACvC,aAAa,CAAC,EAAE,MAAM,CAAC;CAC1B"}

package/out/core/features/common/types.js

@@ -0,0 +1,12 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.UTGenerateResult = void 0;
+var UTGenerateResult;
+(function (UTGenerateResult) {
+    UTGenerateResult["Success"] = "success";
+    UTGenerateResult["Error"] = "error";
+    UTGenerateResult["Cancelled"] = "cancelled";
+    UTGenerateResult["NotFound"] = "notFound";
+    UTGenerateResult["Skipped"] = "skipped";
+})(UTGenerateResult || (exports.UTGenerateResult = UTGenerateResult = {}));
+//# sourceMappingURL=types.js.map

package/out/core/features/common/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../../src/core/features/common/types.ts"],"names":[],"mappings":";;;AAGA,IAAY,gBAMX;AAND,WAAY,gBAAgB;IACxB,uCAAmB,CAAA;IACnB,mCAAe,CAAA;IACf,2CAAuB,CAAA;IACvB,yCAAqB,CAAA;IACrB,uCAAmB,CAAA;AACvB,CAAC,EANW,gBAAgB,gCAAhB,gBAAgB,QAM3B"}

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

@@ -0,0 +1,31 @@
+import { FeatureBase } from "./common/base";
+import { CodeDetailData } from "../analyze";
+import { UTGenerateResult } from "./common/types";
+export interface GenerateFileParams {
+    filePath: string;
+    selectedLine?: number;
+    lineRanges?: number[][];
+    methods?: Set<string>;
+    updateExisting?: boolean;
+}
+export interface GenerateFileResultItem {
+    targetName: string;
+    expectedPath: string;
+    code: string;
+    status: UTGenerateResult;
+    isNew: boolean;
+    sourceCodeData: CodeDetailData;
+    sourceCodeMockCode?: string;
+}
+export interface GenerateFileResult {
+    results: GenerateFileResultItem[];
+    projectFolder?: string;
+}
+export declare class GenerateFile extends FeatureBase<GenerateFileParams, GenerateFileResult> {
+    #private;
+    protected task(): Promise<GenerateFileResult>;
+    private isMatchTarget;
+    private isSpecExists;
+    private getSpecName;
+}
+//# sourceMappingURL=generateFile.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generateFile.d.ts","sourceRoot":"","sources":["../../../src/core/features/generateFile.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,cAAc,EAAkB,MAAM,YAAY,CAAC;AAE5D,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAGlD,MAAM,WAAW,kBAAkB;IAC/B,QAAQ,EAAE,MAAM,CAAC;IACjB,YAAY,CAAC,EAAE,MAAM,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;IACxB,OAAO,CAAC,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;IACtB,cAAc,CAAC,EAAE,OAAO,CAAC;CAC5B;AAED,MAAM,WAAW,sBAAsB;IACnC,UAAU,EAAE,MAAM,CAAC;IAEnB,YAAY,EAAE,MAAM,CAAC;IAErB,IAAI,EAAE,MAAM,CAAC;IACb,MAAM,EAAE,gBAAgB,CAAC;IACzB,KAAK,EAAE,OAAO,CAAC;IACf,cAAc,EAAE,cAAc,CAAC;IAC/B,kBAAkB,CAAC,EAAE,MAAM,CAAC;CAC/B;AAED,MAAM,WAAW,kBAAkB;IAC/B,OAAO,EAAE,sBAAsB,EAAE,CAAC;IAClC,aAAa,CAAC,EAAE,MAAM,CAAC;CAC1B;AAED,qBAAa,YAAa,SAAQ,WAAW,CAAC,kBAAkB,EAAE,kBAAkB,CAAC;;cAMjE,IAAI;IAqGpB,OAAO,CAAC,aAAa;IAerB,OAAO,CAAC,YAAY;IAgBpB,OAAO,CAAC,WAAW;CAGtB"}

package/out/core/features/generateFile.js

@@ -0,0 +1,148 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GenerateFile = void 0;
+const fs_1 = __importDefault(require("fs"));
+const path_1 = __importDefault(require("path"));
+const tsMorph_1 = require("../utils/tsMorph");
+const base_1 = require("./common/base");
+const analyze_1 = require("../analyze");
+const generateSingle_1 = require("./generateSingle");
+const types_1 = require("./common/types");
+const updateSpec_1 = require("./updateSpec");
+class GenerateFile extends base_1.FeatureBase {
+    #fileDir;
+    #specDir;
+    #pureFileName;
+    #fileSuffix;
+    async task() {
+        const { filePath } = this.params;
+        this.#fileSuffix = path_1.default.extname(filePath);
+        this.#pureFileName = path_1.default.basename(filePath, this.#fileSuffix);
+        this.#fileDir = path_1.default.dirname(filePath);
+        this.#specDir = path_1.default.join(this.#fileDir, `${this.#pureFileName}.spec`);
+        this.progress({ message: `Analyzing ${filePath}...` });
+        const { project, projectRoot } = (0, tsMorph_1.getProject)(filePath);
+        const taskResult = {
+            results: [],
+            projectFolder: projectRoot
+        };
+        await this.sleep(0);
+        if (this.stopped) {
+            return taskResult;
+        }
+        const exporteds = await (0, analyze_1.read)(filePath, {
+            deepTypesCheck: false,
+            cachedProject: project
+        });
+        await this.sleep(0);
+        if (this.stopped) {
+            return taskResult;
+        }
+        if (!exporteds.length) {
+            this.progress({ message: `No exported function/class found in ${filePath}`, failed: true });
+            return taskResult;
+        }
+        const targets = [];
+        this.progress({ message: `Found ${exporteds.length} exported function/class` });
+        for (const codeData of exporteds) {
+            if (this.stopped) {
+                return taskResult;
+            }
+            const { type, linesRange } = codeData;
+            if (type === analyze_1.CodeType.Function) {
+                if (this.isMatchTarget(linesRange, codeData.name)) {
+                    targets.push({ codeData, filePath });
+                }
+            }
+            else if (type === analyze_1.CodeType.Class) {
+                const classFunctions = codeData.classFunctions || [];
+                for (const classFunction of classFunctions) {
+                    if (this.isMatchTarget(classFunction.linesRange, classFunction.name)) {
+                        targets.push({ codeData, filePath, targetMember: classFunction });
+                    }
+                }
+            }
+        }
+        if (!targets.length) {
+            this.progress({ message: `No target found in ${filePath}`, failed: true });
+            return taskResult;
+        }
+        for (const target of targets) {
+            if (this.stopped) {
+                return taskResult;
+            }
+            const existedSpecFile = this.isSpecExists(target.codeData.name);
+            if (!existedSpecFile) {
+                const genResult = await this.runSubTask(generateSingle_1.GenerateSingle, {
+                    ...target,
+                });
+                taskResult.results.push({
+                    ...genResult,
+                    expectedPath: path_1.default.resolve(this.#specDir, this.getSpecName(target.codeData.name)),
+                    targetName: target.codeData.name,
+                    isNew: true,
+                    sourceCodeData: target.codeData,
+                    sourceCodeMockCode: genResult.mockCode
+                });
+                continue;
+            }
+            if (!this.params.updateExisting) {
+                this.progress({ message: `Spec file already exists for ${target.codeData.name}, please use "Update test"`, failed: true });
+                continue;
+            }
+            const updateResult = await this.runSubTask(updateSpec_1.UpdateSpec, {
+                sourceCodeFilePath: filePath,
+                sourceCode: target.codeData.code,
+                specFilePath: existedSpecFile
+            });
+            if (updateResult.status === types_1.UTGenerateResult.Success) {
+                taskResult.results.push({
+                    code: updateResult.code,
+                    status: updateResult.status,
+                    expectedPath: existedSpecFile,
+                    targetName: target.codeData.name,
+                    isNew: false,
+                    sourceCodeData: target.codeData,
+                });
+            }
+        }
+        return taskResult;
+    }
+    isMatchTarget(pos, functionName) {
+        if (this.params.selectedLine != null) {
+            return pos[0] <= this.params.selectedLine && pos[1] >= this.params.selectedLine;
+        }
+        if (this.params.lineRanges?.length) {
+            return this.params.lineRanges.some(([start, end]) => {
+                return !(pos[1] < start || pos[0] > end);
+            });
+        }
+        if (this.params.methods?.size) {
+            return this.params.methods.has(functionName);
+        }
+        return true;
+    }
+    isSpecExists(testName) {
+        const filename = this.getSpecName(testName);
+        const oldSpecName = `${this.#pureFileName}.${testName}.spec${this.#fileSuffix}`;
+        const oldWebSpecName = `${this.#pureFileName}.${testName}.web.spec${this.#fileSuffix}`;
+        const rawSpecName = `${this.#pureFileName}.spec${this.#fileSuffix}`;
+        const checkPaths = [
+            path_1.default.resolve(this.#fileDir, filename),
+            path_1.default.resolve(this.#fileDir, rawSpecName),
+            path_1.default.resolve(this.#specDir, filename),
+            path_1.default.resolve(this.#specDir, rawSpecName),
+            path_1.default.resolve(this.#specDir, oldSpecName),
+            path_1.default.resolve(this.#specDir, oldWebSpecName),
+        ];
+        return checkPaths.find(fs_1.default.existsSync);
+    }
+    getSpecName(testName) {
+        return `${testName}.web.spec${this.#fileSuffix}`;
+    }
+}
+exports.GenerateFile = GenerateFile;
+//# sourceMappingURL=generateFile.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generateFile.js","sourceRoot":"","sources":["../../../src/core/features/generateFile.ts"],"names":[],"mappings":";;;;;;AAAA,4CAAoB;AACpB,gDAAwB;AACxB,8CAA8C;AAC9C,wCAA4C;AAC5C,wCAA4D;AAC5D,qDAAwE;AACxE,0CAAkD;AAClD,6CAA0C;AA2B1C,MAAa,YAAa,SAAQ,kBAAmD;IACjF,QAAQ,CAAS;IACjB,QAAQ,CAAS;IACjB,aAAa,CAAS;IACtB,WAAW,CAAS;IAEV,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAEjC,IAAI,CAAC,WAAW,GAAG,cAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAC1C,IAAI,CAAC,aAAa,GAAG,cAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,CAAC;QAC/D,IAAI,CAAC,QAAQ,GAAG,cAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,CAAC,QAAQ,GAAG,cAAI,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,GAAG,IAAI,CAAC,aAAa,OAAO,CAAC,CAAC;QAEvE,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,aAAa,QAAQ,KAAK,EAAE,CAAC,CAAC;QACvD,MAAM,EAAE,OAAO,EAAE,WAAW,EAAE,GAAG,IAAA,oBAAU,EAAC,QAAQ,CAAC,CAAC;QAEtD,MAAM,UAAU,GAAuB;YACnC,OAAO,EAAE,EAAE;YACX,aAAa,EAAE,WAAW;SAC7B,CAAC;QAEF,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,OAAO,UAAU,CAAC;QACtB,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,IAAA,cAAI,EAAC,QAAQ,EAAE;YACnC,cAAc,EAAE,KAAK;YACrB,aAAa,EAAE,OAAO;SACzB,CAAC,CAAC;QACH,MAAM,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QACpB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;YACf,OAAO,UAAU,CAAC;QACtB,CAAC;QACD,IAAI,CAAC,SAAS,CAAC,MAAM,EAAE,CAAC;YACpB,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,uCAAuC,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAC5F,OAAO,UAAU,CAAC;QACtB,CAAC;QAED,MAAM,OAAO,GAA2B,EAAE,CAAC;QAC3C,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,SAAS,SAAS,CAAC,MAAM,0BAA0B,EAAE,CAAC,CAAC;QAChF,KAAK,MAAM,QAAQ,IAAI,SAAS,EAAE,CAAC;YAC/B,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,OAAO,UAAU,CAAC;YACtB,CAAC;YACD,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,GAAG,QAAQ,CAAC;YACtC,IAAI,IAAI,KAAK,kBAAQ,CAAC,QAAQ,EAAE,CAAC;gBAC7B,IAAI,IAAI,CAAC,aAAa,CAAC,UAAU,EAAE,QAAQ,CAAC,IAAI,CAAC,EAAE,CAAC;oBAChD,OAAO,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;gBACzC,CAAC;YACL,CAAC;iBAAM,IAAI,IAAI,KAAK,kBAAQ,CAAC,KAAK,EAAE,CAAC;gBACjC,MAAM,cAAc,GAAG,QAAQ,CAAC,cAAc,IAAI,EAAE,CAAC;gBACrD,KAAK,MAAM,aAAa,IAAI,cAAc,EAAE,CAAC;oBACzC,IAAI,IAAI,CAAC,aAAa,CAAC,aAAa,CAAC,UAAU,EAAE,aAAa,CAAC,IAAI,CAAC,EAAE,CAAC;wBACnE,OAAO,CAAC,IAAI,CAAC,EAAE,QAAQ,EAAE,QAAQ,EAAE,YAAY,EAAE,aAAa,EAAE,CAAC,CAAC;oBACtE,CAAC;gBACL,CAAC;YACL,CAAC;QACL,CAAC;QACD,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC;YAClB,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,sBAAsB,QAAQ,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3E,OAAO,UAAU,CAAC;QACtB,CAAC;QAED,KAAK,MAAM,MAAM,IAAI,OAAO,EAAE,CAAC;YAC3B,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;gBACf,OAAO,UAAU,CAAC;YACtB,CAAC;YACD,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;YAChE,IAAI,CAAC,eAAe,EAAE,CAAC;gBACnB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,+BAAc,EAAE;oBACpD,GAAG,MAAM;iBACZ,CAAC,CAAC;gBACH,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC;oBACpB,GAAG,SAAS;oBACZ,YAAY,EAAE,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC;oBACjF,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;oBAChC,KAAK,EAAE,IAAI;oBACX,cAAc,EAAE,MAAM,CAAC,QAAQ;oBAC/B,kBAAkB,EAAE,SAAS,CAAC,QAAQ;iBACzC,CAAC,CAAC;gBACH,SAAS;YACb,CAAC;YACD,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;gBAC9B,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,gCAAgC,MAAM,CAAC,QAAQ,CAAC,IAAI,4BAA4B,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;gBAC3H,SAAS;YACb,CAAC;YACD,MAAM,YAAY,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,uBAAU,EAAE;gBACnD,kBAAkB,EAAE,QAAQ;gBAC5B,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;gBAChC,YAAY,EAAE,eAAe;aAChC,CAAC,CAAC;YACH,IAAI,YAAY,CAAC,MAAM,KAAK,wBAAgB,CAAC,OAAO,EAAE,CAAC;gBACnD,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC;oBACpB,IAAI,EAAE,YAAY,CAAC,IAAI;oBACvB,MAAM,EAAE,YAAY,CAAC,MAAM;oBAC3B,YAAY,EAAE,eAAe;oBAC7B,UAAU,EAAE,MAAM,CAAC,QAAQ,CAAC,IAAI;oBAChC,KAAK,EAAE,KAAK;oBACZ,cAAc,EAAE,MAAM,CAAC,QAAQ;iBAClC,CAAC,CAAC;YACP,CAAC;QACL,CAAC;QACD,OAAO,UAAU,CAAC;IACtB,CAAC;IAEO,aAAa,CAAC,GAAiC,EAAE,YAAoB;QACzE,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,IAAI,EAAE,CAAC;YACnC,OAAO,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,IAAI,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,MAAM,CAAC,YAAY,CAAC;QACpF,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,EAAE,CAAC;YACjC,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,KAAK,EAAE,GAAG,CAAC,EAAE,EAAE;gBAChD,OAAO,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,KAAK,IAAI,GAAG,CAAC,CAAC,CAAC,GAAG,GAAG,CAAC,CAAC;YAC7C,CAAC,CAAC,CAAC;QACP,CAAC;QACD,IAAI,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC;YAC5B,OAAO,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;QACjD,CAAC;QACD,OAAO,IAAI,CAAC;IAChB,CAAC;IAEO,YAAY,CAAC,QAAgB;QACjC,MAAM,QAAQ,GAAG,IAAI,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;QAC5C,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,aAAa,IAAI,QAAQ,QAAQ,IAAI,CAAC,WAAW,EAAE,CAAC;QAChF,MAAM,cAAc,GAAG,GAAG,IAAI,CAAC,aAAa,IAAI,QAAQ,YAAY,IAAI,CAAC,WAAW,EAAE,CAAC;QACvF,MAAM,WAAW,GAAG,GAAG,IAAI,CAAC,aAAa,QAAQ,IAAI,CAAC,WAAW,EAAE,CAAC;QACpE,MAAM,UAAU,GAAG;YACf,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC;YACrC,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC;YACxC,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,QAAQ,CAAC;YACrC,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC;YACxC,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,CAAC;YACxC,cAAI,CAAC,OAAO,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,CAAC;SAC9C,CAAC;QACF,OAAO,UAAU,CAAC,IAAI,CAAC,YAAE,CAAC,UAAU,CAAC,CAAC;IAC1C,CAAC;IAEO,WAAW,CAAC,QAAgB;QAChC,OAAO,GAAG,QAAQ,YAAY,IAAI,CAAC,WAAW,EAAE,CAAC;IACrD,CAAC;CACJ;AA7ID,oCA6IC"}

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

@@ -0,0 +1,28 @@
+import { ClassFunction, CodeDetailData } from "../analyze";
+import { FeatureBase } from "./common/base";
+import { UTGenerateResult } from "./common/types";
+export interface GenerateSingleParams {
+    codeData: CodeDetailData;
+    filePath: string;
+    targetMember?: ClassFunction;
+}
+interface GenerateSingleResult {
+    code: string;
+    mockCode: string;
+    status: UTGenerateResult;
+}
+/**
+ * Generate new test for a single function/method.
+ * !! Assume there is no existing test for the function/method !!
+ */
+export declare class GenerateSingle extends FeatureBase<GenerateSingleParams, GenerateSingleResult> {
+    private specName;
+    private pureFileName;
+    private sourceDir;
+    protected task(): Promise<GenerateSingleResult>;
+    private genFunctionTestCode;
+    private genClassMemberTestCode;
+    private formatNewCode;
+}
+export {};
+//# sourceMappingURL=generateSingle.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generateSingle.d.ts","sourceRoot":"","sources":["../../../src/core/features/generateSingle.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,cAAc,EAA8B,MAAM,YAAY,CAAC;AAGvF,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,OAAO,EAAE,gBAAgB,EAAE,MAAM,gBAAgB,CAAC;AAMlD,MAAM,WAAW,oBAAoB;IACjC,QAAQ,EAAE,cAAc,CAAC;IACzB,QAAQ,EAAE,MAAM,CAAC;IAEjB,YAAY,CAAC,EAAE,aAAa,CAAC;CAChC;AAED,UAAU,oBAAoB;IAC1B,IAAI,EAAE,MAAM,CAAC;IACb,QAAQ,EAAE,MAAM,CAAC;IACjB,MAAM,EAAE,gBAAgB,CAAC;CAC5B;AAED;;;GAGG;AACH,qBAAa,cAAe,SAAQ,WAAW,CAAC,oBAAoB,EAAE,oBAAoB,CAAC;IACvF,OAAO,CAAC,QAAQ,CAAS;IACzB,OAAO,CAAC,YAAY,CAAS;IAC7B,OAAO,CAAC,SAAS,CAAS;cAEV,IAAI,IAAI,OAAO,CAAC,oBAAoB,CAAC;YAgCvC,mBAAmB;YA8BnB,sBAAsB;IA2EpC,OAAO,CAAC,aAAa;CAKxB"}

package/out/core/features/generateSingle.js

@@ -0,0 +1,149 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GenerateSingle = void 0;
+const path_1 = __importDefault(require("path"));
+const analyze_1 = require("../analyze");
+const llm_1 = require("../llm");
+const deps_1 = require("../utils/deps");
+const base_1 = require("./common/base");
+const types_1 = require("./common/types");
+const tag_1 = require("../valid/tag");
+const mock_1 = require("../valid/mock");
+const common_1 = require("../utils/common");
+/**
+ * Generate new test for a single function/method.
+ * !! Assume there is no existing test for the function/method !!
+ */
+class GenerateSingle extends base_1.FeatureBase {
+    specName;
+    pureFileName;
+    sourceDir;
+    async task() {
+        const { codeData: codeData, filePath } = this.params;
+        const ext = path_1.default.extname(filePath);
+        const baseFileName = path_1.default.basename(filePath, ext);
+        this.sourceDir = path_1.default.dirname(filePath);
+        this.specName = `./${baseFileName}.spec${ext}`;
+        this.pureFileName = `${baseFileName}${ext}`;
+        try {
+            if (codeData.type === analyze_1.CodeType.Function) {
+                return this.genFunctionTestCode();
+            }
+            else if (codeData.type === analyze_1.CodeType.Class) {
+                return this.genClassMemberTestCode();
+            }
+            return {
+                code: '',
+                mockCode: '',
+                status: types_1.UTGenerateResult.Skipped,
+            };
+        }
+        catch (e) {
+            this.progress({ message: `[Error] Failed to generate test code: ${e.message}`, failed: true });
+            return {
+                code: '',
+                mockCode: '',
+                status: types_1.UTGenerateResult.Error,
+            };
+        }
+    }
+    async genFunctionTestCode() {
+        const { codeData: codeData } = this.params;
+        this.progress({ message: `Generating test code for function ${codeData.name}` });
+        const targetSourceCode = (0, analyze_1.getCode)(codeData, {
+            beforeImports: '// external import declarations:\n',
+            beforeCode: '// test target code:\n'
+        });
+        let mockString = (0, deps_1.getPackageMocks)(codeData.importDeclares);
+        const messages = this.getPrompt("GenFuncUT" /* PromptKey.GenFuncUT */, {
+            ["<<<SourceFileName>>>" /* PromptPlaceholder.SourceFileName */]: this.pureFileName,
+            ["<<<FunctionName>>>" /* PromptPlaceholder.FunctionName */]: (0, common_1.formatImportName)(codeData.name, codeData.isDefaultExport),
+            ["<<<SpecFileName>>>" /* PromptPlaceholder.SpecFileName */]: this.specName,
+            ["<<<SourceCode>>>" /* PromptPlaceholder.SourceCode */]: targetSourceCode,
+            ["<<<MockCode>>>" /* PromptPlaceholder.MockCode */]: mockString,
+            ["<<<Language>>>" /* PromptPlaceholder.Language */]: (0, common_1.getLanguage)(this.specName),
+            ["<<<ModuleType>>>" /* PromptPlaceholder.ModuleType */]: codeData.isCommonJS ? "cjs" /* ModuleType.CJS */ : "esm" /* ModuleType.ESM */,
+        });
+        const resultCode = await this.getCodeFromLLM(messages);
+        return {
+            code: this.formatNewCode(resultCode),
+            mockCode: mockString,
+            status: types_1.UTGenerateResult.Success,
+        };
+    }
+    async genClassMemberTestCode() {
+        const { codeData, targetMember } = this.params;
+        if (!targetMember) {
+            throw new Error('targetMember is required for class member test generation');
+        }
+        this.progress({ message: `Generating test code for class member ${targetMember.name} in ${codeData.name}` });
+        const isDataConnector = this.pureFileName.includes('.connector.') && codeData.name.endsWith('Connector');
+        const classDeclare = (0, analyze_1.getCode)(codeData, {
+            beforeImports: '// external import declarations:\n',
+            beforeCode: '// test target code:\n'
+        });
+        const declareMap = {};
+        // for wc
+        const commonDeclares = {};
+        for (const module in codeData.importDeclares) {
+            const declares = codeData.importDeclares[module];
+            for (const declare of declares) {
+                if ((module === '@msnews/experiences-web-component' && declare.name === 'WebComponentExperience')
+                    || (module === '@cs-core/foundation' && ['attr', 'observable', 'volatile', 'FASTElement'].includes(declare.name))) {
+                    if (!commonDeclares[module]) {
+                        commonDeclares[module] = [];
+                    }
+                    commonDeclares[module].push(declare);
+                    // skip the declares of web component
+                    continue;
+                }
+                declareMap[declare.name] = module;
+            }
+        }
+        // get mocks
+        const declares = {
+            ...commonDeclares
+        };
+        for (const [id] of targetMember.externalIdentifiers || []) {
+            const module = declareMap[id];
+            if (!module) {
+                continue;
+            }
+            if (!declares[module]) {
+                declares[module] = [];
+            }
+            declares[module].push({ name: id, declare: '' });
+        }
+        let mockString = (0, deps_1.getPackageMocks)(declares);
+        if (isDataConnector) {
+            mockString += `\n${llm_1.Templates.dataConnector(codeData.name, targetMember.name)}`;
+        }
+        const messages = this.getPrompt("GenClassMemberUT" /* PromptKey.GenClassMemberUT */, {
+            ["<<<SourceFileName>>>" /* PromptPlaceholder.SourceFileName */]: this.pureFileName,
+            ["<<<ClassName>>>" /* PromptPlaceholder.ClassName */]: (0, common_1.formatImportName)(codeData.name, codeData.isDefaultExport),
+            ["<<<FunctionName>>>" /* PromptPlaceholder.FunctionName */]: targetMember.name,
+            ["<<<SpecFileName>>>" /* PromptPlaceholder.SpecFileName */]: this.specName,
+            ["<<<ClassDeclare>>>" /* PromptPlaceholder.ClassDeclare */]: classDeclare,
+            ["<<<SourceCode>>>" /* PromptPlaceholder.SourceCode */]: targetMember.body,
+            ["<<<MockCode>>>" /* PromptPlaceholder.MockCode */]: mockString,
+            ["<<<Language>>>" /* PromptPlaceholder.Language */]: (0, common_1.getLanguage)(this.specName),
+            ["<<<ModuleType>>>" /* PromptPlaceholder.ModuleType */]: codeData.isCommonJS ? "cjs" /* ModuleType.CJS */ : "esm" /* ModuleType.ESM */,
+        });
+        const resultCode = await this.getCodeFromLLM(messages);
+        return {
+            code: this.formatNewCode(resultCode),
+            mockCode: mockString,
+            status: types_1.UTGenerateResult.Success,
+        };
+    }
+    formatNewCode(code) {
+        let newCode = code;
+        newCode = (0, mock_1.checkMock)(newCode);
+        return (0, tag_1.addTag)(newCode);
+    }
+}
+exports.GenerateSingle = GenerateSingle;
+//# sourceMappingURL=generateSingle.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"generateSingle.js","sourceRoot":"","sources":["../../../src/core/features/generateSingle.ts"],"names":[],"mappings":";;;;;;AAAA,gDAAwB;AACxB,wCAAuF;AACvF,gCAAiE;AACjE,wCAAgD;AAChD,wCAA4C;AAC5C,0CAAkD;AAClD,sCAAsC;AACtC,wCAA0C;AAC1C,4CAAgE;AAgBhE;;;GAGG;AACH,MAAa,cAAe,SAAQ,kBAAuD;IAC/E,QAAQ,CAAS;IACjB,YAAY,CAAS;IACrB,SAAS,CAAS;IAEhB,KAAK,CAAC,IAAI;QAChB,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAErD,MAAM,GAAG,GAAG,cAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QAEnC,MAAM,YAAY,GAAG,cAAI,CAAC,QAAQ,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC;QAClD,IAAI,CAAC,SAAS,GAAG,cAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;QACxC,IAAI,CAAC,QAAQ,GAAG,KAAK,YAAY,QAAQ,GAAG,EAAE,CAAC;QAC/C,IAAI,CAAC,YAAY,GAAG,GAAG,YAAY,GAAG,GAAG,EAAE,CAAC;QAE5C,IAAI,CAAC;YACD,IAAI,QAAQ,CAAC,IAAI,KAAK,kBAAQ,CAAC,QAAQ,EAAE,CAAC;gBACtC,OAAO,IAAI,CAAC,mBAAmB,EAAE,CAAC;YACtC,CAAC;iBAAM,IAAI,QAAQ,CAAC,IAAI,KAAK,kBAAQ,CAAC,KAAK,EAAE,CAAC;gBAC1C,OAAO,IAAI,CAAC,sBAAsB,EAAE,CAAC;YACzC,CAAC;YACD,OAAO;gBACH,IAAI,EAAE,EAAE;gBACR,QAAQ,EAAE,EAAE;gBACZ,MAAM,EAAE,wBAAgB,CAAC,OAAO;aACnC,CAAA;QACL,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,yCAAyC,CAAC,CAAC,OAAO,EAAE,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;YAC/F,OAAO;gBACH,IAAI,EAAE,EAAE;gBACR,QAAQ,EAAE,EAAE;gBACZ,MAAM,EAAE,wBAAgB,CAAC,KAAK;aAEjC,CAAA;QACL,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,mBAAmB;QAC7B,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAE3C,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,qCAAqC,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QACjF,MAAM,gBAAgB,GAAG,IAAA,iBAAO,EAAC,QAAQ,EAAE;YACvC,aAAa,EAAE,oCAAoC;YACnD,UAAU,EAAE,wBAAwB;SACvC,CAAC,CAAC;QAEH,IAAI,UAAU,GAAG,IAAA,sBAAe,EAAC,QAAQ,CAAC,cAAc,CAAC,CAAC;QAC1D,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,wCAE3B;YACI,+DAAkC,EAAE,IAAI,CAAC,YAAY;YACrD,2DAAgC,EAAE,IAAA,yBAAgB,EAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,eAAe,CAAC;YAC3F,2DAAgC,EAAE,IAAI,CAAC,QAAQ;YAC/C,uDAA8B,EAAE,gBAAgB;YAChD,mDAA4B,EAAE,UAAU;YACxC,mDAA4B,EAAE,IAAA,oBAAW,EAAC,IAAI,CAAC,QAAQ,CAAC;YACxD,uDAA8B,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC,4BAAgB,CAAC,2BAAe;SACxF,CACJ,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACvD,OAAO;YACH,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC;YACpC,QAAQ,EAAE,UAAU;YACpB,MAAM,EAAE,wBAAgB,CAAC,OAAO;SACnC,CAAC;IACN,CAAC;IAEO,KAAK,CAAC,sBAAsB;QAChC,MAAM,EAAE,QAAQ,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,MAAM,CAAC;QAC/C,IAAI,CAAC,YAAY,EAAE,CAAC;YAChB,MAAM,IAAI,KAAK,CAAC,2DAA2D,CAAC,CAAC;QACjF,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,EAAE,OAAO,EAAE,yCAAyC,YAAY,CAAC,IAAI,OAAO,QAAQ,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;QAC7G,MAAM,eAAe,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,QAAQ,CAAC,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC;QACzG,MAAM,YAAY,GAAG,IAAA,iBAAO,EAAC,QAAQ,EAAE;YACnC,aAAa,EAAE,oCAAoC;YACnD,UAAU,EAAE,wBAAwB;SACvC,CAAC,CAAC;QAEH,MAAM,UAAU,GAA2B,EAAE,CAAC;QAC9C,SAAS;QACT,MAAM,cAAc,GAA8B,EAAE,CAAC;QAErD,KAAK,MAAM,MAAM,IAAI,QAAQ,CAAC,cAAc,EAAE,CAAC;YAC3C,MAAM,QAAQ,GAAG,QAAQ,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;YACjD,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;gBAC7B,IAAI,CAAC,MAAM,KAAK,mCAAmC,IAAI,OAAO,CAAC,IAAI,KAAK,wBAAwB,CAAC;uBAC1F,CAAC,MAAM,KAAK,qBAAqB,IAAI,CAAC,MAAM,EAAE,YAAY,EAAE,UAAU,EAAE,aAAa,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,EACnH,CAAC;oBACC,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,EAAE,CAAC;wBAC1B,cAAc,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;oBAChC,CAAC;oBACD,cAAc,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;oBACrC,qCAAqC;oBACrC,SAAS;gBACb,CAAC;gBACD,UAAU,CAAC,OAAO,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC;YACtC,CAAC;QACL,CAAC;QAED,YAAY;QACZ,MAAM,QAAQ,GAA8B;YACxC,GAAG,cAAc;SACpB,CAAC;QACF,KAAK,MAAM,CAAC,EAAE,CAAC,IAAI,YAAY,CAAC,mBAAmB,IAAI,EAAE,EAAE,CAAC;YACxD,MAAM,MAAM,GAAG,UAAU,CAAC,EAAE,CAAC,CAAC;YAC9B,IAAI,CAAC,MAAM,EAAE,CAAC;gBACV,SAAS;YACb,CAAC;YACD,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBACpB,QAAQ,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC;YAC1B,CAAC;YACD,QAAQ,CAAC,MAAM,CAAC,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC;QACrD,CAAC;QACD,IAAI,UAAU,GAAG,IAAA,sBAAe,EAAC,QAAQ,CAAC,CAAC;QAC3C,IAAI,eAAe,EAAE,CAAC;YAClB,UAAU,IAAI,KAAK,eAAS,CAAC,aAAa,CAAC,QAAQ,CAAC,IAAI,EAAE,YAAY,CAAC,IAAI,CAAC,EAAE,CAAC;QACnF,CAAC;QAED,MAAM,QAAQ,GAAG,IAAI,CAAC,SAAS,sDAE3B;YACI,+DAAkC,EAAE,IAAI,CAAC,YAAY;YACrD,qDAA6B,EAAE,IAAA,yBAAgB,EAAC,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC,eAAe,CAAC;YACxF,2DAAgC,EAAE,YAAY,CAAC,IAAI;YACnD,2DAAgC,EAAE,IAAI,CAAC,QAAQ;YAC/C,2DAAgC,EAAE,YAAY;YAC9C,uDAA8B,EAAE,YAAY,CAAC,IAAI;YACjD,mDAA4B,EAAE,UAAU;YACxC,mDAA4B,EAAE,IAAA,oBAAW,EAAC,IAAI,CAAC,QAAQ,CAAC;YACxD,uDAA8B,EAAE,QAAQ,CAAC,UAAU,CAAC,CAAC,4BAAgB,CAAC,2BAAe;SACxF,CACJ,CAAC;QACF,MAAM,UAAU,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,CAAC;QACvD,OAAO;YACH,IAAI,EAAE,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC;YACpC,QAAQ,EAAE,UAAU;YACpB,MAAM,EAAE,wBAAgB,CAAC,OAAO;SACnC,CAAC;IACN,CAAC;IAEO,aAAa,CAAC,IAAY;QAC9B,IAAI,OAAO,GAAG,IAAI,CAAC;QACnB,OAAO,GAAG,IAAA,gBAAS,EAAC,OAAO,CAAC,CAAC;QAC7B,OAAO,IAAA,YAAM,EAAC,OAAO,CAAC,CAAA;IAC1B,CAAC;CACJ;AAnJD,wCAmJC"}

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

@@ -0,0 +1,9 @@
+import { FeatureBase } from './common/base';
+export type IFeature = FeatureBase<any, any>;
+export { UTGenerateResult, TaskOptions } from './common/types';
+export { GenerateFile, GenerateFileParams, GenerateFileResult } from './generateFile';
+export { UpdateSpec, UpdateSpecParams, UpdateSpecResult } from './updateSpec';
+export { ValidFix, ValidFixParams, ValidFixResult } from './validFix';
+export { MergeUTFile, MergeUTFileParams, MergeUTFileResult } from './mergeUTFile';
+export { RemoveFailedCases, RemoveFailedCasesParams, RemoveFailedCasesResult } from './removeFailedCases';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../../src/core/features/index.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,WAAW,EAAE,MAAM,eAAe,CAAC;AAC5C,MAAM,MAAM,QAAQ,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAE7C,OAAO,EAAE,gBAAgB,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE/D,OAAO,EAAE,YAAY,EAAE,kBAAkB,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AACtF,OAAO,EAAE,UAAU,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,MAAM,cAAc,CAAC;AAC9E,OAAO,EAAE,QAAQ,EAAE,cAAc,EAAE,cAAc,EAAE,MAAM,YAAY,CAAC;AACtE,OAAO,EAAE,WAAW,EAAE,iBAAiB,EAAE,iBAAiB,EAAE,MAAM,eAAe,CAAC;AAClF,OAAO,EAAE,iBAAiB,EAAE,uBAAuB,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC"}

package/out/core/features/index.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RemoveFailedCases = exports.MergeUTFile = exports.ValidFix = exports.UpdateSpec = exports.GenerateFile = exports.UTGenerateResult = void 0;
+var types_1 = require("./common/types");
+Object.defineProperty(exports, "UTGenerateResult", { enumerable: true, get: function () { return types_1.UTGenerateResult; } });
+var generateFile_1 = require("./generateFile");
+Object.defineProperty(exports, "GenerateFile", { enumerable: true, get: function () { return generateFile_1.GenerateFile; } });
+var updateSpec_1 = require("./updateSpec");
+Object.defineProperty(exports, "UpdateSpec", { enumerable: true, get: function () { return updateSpec_1.UpdateSpec; } });
+var validFix_1 = require("./validFix");
+Object.defineProperty(exports, "ValidFix", { enumerable: true, get: function () { return validFix_1.ValidFix; } });
+var mergeUTFile_1 = require("./mergeUTFile");
+Object.defineProperty(exports, "MergeUTFile", { enumerable: true, get: function () { return mergeUTFile_1.MergeUTFile; } });
+var removeFailedCases_1 = require("./removeFailedCases");
+Object.defineProperty(exports, "RemoveFailedCases", { enumerable: true, get: function () { return removeFailedCases_1.RemoveFailedCases; } });
+//# sourceMappingURL=index.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../src/core/features/index.ts"],"names":[],"mappings":";;;AAGA,wCAA+D;AAAtD,yGAAA,gBAAgB,OAAA;AAEzB,+CAAsF;AAA7E,4GAAA,YAAY,OAAA;AACrB,2CAA8E;AAArE,wGAAA,UAAU,OAAA;AACnB,uCAAsE;AAA7D,oGAAA,QAAQ,OAAA;AACjB,6CAAkF;AAAzE,0GAAA,WAAW,OAAA;AACpB,yDAA0G;AAAjG,sHAAA,iBAAiB,OAAA"}

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

@@ -0,0 +1,2 @@
+export declare function fixLintErrors(filePath: string, lintErrors: string, testFramework: string): Promise<string>;
+//# sourceMappingURL=lintFix.d.ts.map