mcp-test-generator 1.0.0

package/dist/analyzers/typescript-analyzer.js

@@ -0,0 +1,386 @@
+import * as fs from "fs";
+export function analyzeTypeScriptFile(filePath) {
+    const content = fs.readFileSync(filePath, "utf-8");
+    const lines = content.split("\n");
+    return {
+        filePath,
+        imports: extractImports(content),
+        exports: extractExports(content),
+        functions: extractFunctions(content),
+        classes: extractClasses(content),
+        constants: extractConstants(content),
+        interfaces: extractInterfaces(content),
+    };
+}
+function extractImports(content) {
+    const imports = [];
+    const importRegex = /import\s+(?:(type)\s+)?(?:(\w+)(?:\s*,\s*)?)?(?:\{([^}]+)\})?\s+from\s+['"]([^'"]+)['"]/g;
+    const importStarRegex = /import\s+\*\s+as\s+(\w+)\s+from\s+['"]([^'"]+)['"]/g;
+    let match;
+    while ((match = importRegex.exec(content)) !== null) {
+        const isTypeOnly = match[1] === "type";
+        const defaultImport = match[2] || null;
+        const namedImportsStr = match[3] || "";
+        const module = match[4];
+        const namedImports = namedImportsStr
+            ? namedImportsStr
+                .split(",")
+                .map((s) => s.trim())
+                .filter((s) => s.length > 0)
+                .map((s) => {
+                // Handle "Name as Alias"
+                const parts = s.split(/\s+as\s+/);
+                return parts[parts.length - 1].trim();
+            })
+            : [];
+        imports.push({ module, namedImports, defaultImport, isTypeOnly });
+    }
+    while ((match = importStarRegex.exec(content)) !== null) {
+        imports.push({
+            module: match[2],
+            namedImports: [],
+            defaultImport: match[1],
+            isTypeOnly: false,
+        });
+    }
+    return imports;
+}
+function extractExports(content) {
+    const exports = [];
+    const exportRegex = /export\s+(?:default\s+)?(?:async\s+)?(?:function|class|const|let|var|interface|type|enum)\s+(\w+)/g;
+    const exportNamedRegex = /export\s*\{([^}]+)\}/g;
+    let match;
+    while ((match = exportRegex.exec(content)) !== null) {
+        exports.push(match[1]);
+    }
+    while ((match = exportNamedRegex.exec(content)) !== null) {
+        const names = match[1].split(",").map((s) => s.trim().split(/\s+as\s+/)[0].trim());
+        exports.push(...names);
+    }
+    return [...new Set(exports)];
+}
+function extractFunctions(content) {
+    const functions = [];
+    // Regular function declarations
+    const funcRegex = /(?:(export)\s+)?(?:(default)\s+)?(?:(async)\s+)?function\s+(\w+)\s*(?:<[^>]*>)?\s*\(([^)]*)\)(?:\s*:\s*([^\n{]+))?\s*\{/g;
+    let match;
+    while ((match = funcRegex.exec(content)) !== null) {
+        const isExported = !!match[1] || !!match[2];
+        const isAsync = !!match[3];
+        const name = match[4];
+        const paramsStr = match[5];
+        const returnType = match[6]?.trim() || null;
+        const body = extractBody(content, match.index + match[0].length - 1);
+        functions.push({
+            name,
+            type: "function",
+            isAsync,
+            isExported,
+            isStatic: false,
+            parameters: parseParameters(paramsStr),
+            returnType,
+            className: null,
+            decorators: [],
+            jsDoc: extractJsDoc(content, match.index),
+            complexity: calculateComplexity(body),
+            startLine: getLineNumber(content, match.index),
+            endLine: getLineNumber(content, match.index + match[0].length + body.length),
+            body,
+        });
+    }
+    // Arrow functions assigned to const/let/var
+    const arrowRegex = /(?:(export)\s+)?(?:const|let|var)\s+(\w+)\s*(?::\s*[^=]+)?\s*=\s*(?:(async)\s+)?(?:\([^)]*\)|(\w+))\s*(?::\s*[^=]+)?\s*=>/g;
+    while ((match = arrowRegex.exec(content)) !== null) {
+        const isExported = !!match[1];
+        const name = match[2];
+        const isAsync = !!match[3];
+        // Find the arrow and extract parameters
+        const preArrow = content.substring(match.index, content.indexOf("=>", match.index));
+        const paramsMatch = preArrow.match(/\(([^)]*)\)/);
+        const paramsStr = paramsMatch ? paramsMatch[1] : match[4] || "";
+        const arrowIndex = content.indexOf("=>", match.index);
+        const afterArrow = content.substring(arrowIndex + 2).trim();
+        let body;
+        if (afterArrow.startsWith("{")) {
+            body = extractBody(content, arrowIndex + 2 + content.substring(arrowIndex + 2).indexOf("{"));
+        }
+        else {
+            // Single expression
+            const endIndex = findExpressionEnd(content, arrowIndex + 2);
+            body = content.substring(arrowIndex + 2, endIndex).trim();
+        }
+        functions.push({
+            name,
+            type: "arrow",
+            isAsync,
+            isExported,
+            isStatic: false,
+            parameters: parseParameters(paramsStr),
+            returnType: null,
+            className: null,
+            decorators: [],
+            jsDoc: extractJsDoc(content, match.index),
+            complexity: calculateComplexity(body),
+            startLine: getLineNumber(content, match.index),
+            endLine: getLineNumber(content, match.index + body.length),
+            body,
+        });
+    }
+    return functions;
+}
+function extractClasses(content) {
+    const classes = [];
+    const classRegex = /(?:(export)\s+)?(?:(default)\s+)?class\s+(\w+)(?:\s+extends\s+(\w+))?(?:\s+implements\s+([^{]+))?\s*\{/g;
+    let match;
+    while ((match = classRegex.exec(content)) !== null) {
+        const isExported = !!match[1];
+        const name = match[3];
+        const extendsClass = match[4] || null;
+        const implementsStr = match[5]?.trim() || "";
+        const implementsInterfaces = implementsStr
+            ? implementsStr.split(",").map((s) => s.trim())
+            : [];
+        const classBody = extractBody(content, match.index + match[0].length - 1);
+        const methods = extractMethods(classBody, name);
+        const constructorInfo = methods.find((m) => m.type === "constructor") || null;
+        const properties = extractClassProperties(classBody);
+        classes.push({
+            name,
+            isExported,
+            constructorInfo,
+            methods: methods.filter((m) => m.type !== "constructor"),
+            properties,
+            decorators: extractDecorators(content, match.index),
+            extendsClass,
+            implementsInterfaces,
+        });
+    }
+    return classes;
+}
+function extractMethods(classBody, className) {
+    const methods = [];
+    const methodRegex = /(?:(static)\s+)?(?:(async)\s+)?(\w+)\s*\(([^)]*)\)(?:\s*:\s*([^\n{]+))?\s*\{/g;
+    let match;
+    while ((match = methodRegex.exec(classBody)) !== null) {
+        const isStatic = !!match[1];
+        const isAsync = !!match[2];
+        const name = match[3];
+        const paramsStr = match[4];
+        const returnType = match[5]?.trim() || null;
+        if (name === "constructor") {
+            methods.push({
+                name: "constructor",
+                type: "constructor",
+                isAsync: false,
+                isExported: false,
+                isStatic: false,
+                parameters: parseParameters(paramsStr),
+                returnType: null,
+                className,
+                decorators: [],
+                jsDoc: null,
+                complexity: 1,
+                startLine: 0,
+                endLine: 0,
+                body: "",
+            });
+        }
+        else {
+            const body = extractBody(classBody, match.index + match[0].length - 1);
+            methods.push({
+                name,
+                type: "method",
+                isAsync,
+                isExported: false,
+                isStatic,
+                parameters: parseParameters(paramsStr),
+                returnType,
+                className,
+                decorators: [],
+                jsDoc: extractJsDoc(classBody, match.index),
+                complexity: calculateComplexity(body),
+                startLine: 0,
+                endLine: 0,
+                body,
+            });
+        }
+    }
+    return methods;
+}
+function parseParameters(paramsStr) {
+    if (!paramsStr.trim())
+        return [];
+    const params = [];
+    let depth = 0;
+    let current = "";
+    for (const char of paramsStr) {
+        if (char === "(" || char === "<" || char === "{" || char === "[")
+            depth++;
+        if (char === ")" || char === ">" || char === "}" || char === "]")
+            depth--;
+        if (char === "," && depth === 0) {
+            params.push(parseOneParameter(current.trim()));
+            current = "";
+        }
+        else {
+            current += char;
+        }
+    }
+    if (current.trim()) {
+        params.push(parseOneParameter(current.trim()));
+    }
+    return params;
+}
+function parseOneParameter(param) {
+    const isRest = param.startsWith("...");
+    if (isRest)
+        param = param.substring(3);
+    let defaultValue = null;
+    const defaultMatch = param.match(/\s*=\s*(.+)$/);
+    if (defaultMatch) {
+        defaultValue = defaultMatch[1].trim();
+        param = param.substring(0, param.indexOf("=")).trim();
+    }
+    const isOptional = param.includes("?");
+    param = param.replace("?", "");
+    let type = null;
+    const typeMatch = param.match(/:\s*(.+)$/);
+    if (typeMatch) {
+        type = typeMatch[1].trim();
+        param = param.substring(0, param.indexOf(":")).trim();
+    }
+    return {
+        name: param.trim(),
+        type,
+        isOptional: isOptional || defaultValue !== null,
+        defaultValue,
+        isRest,
+    };
+}
+function extractBody(content, openBraceIndex) {
+    let depth = 0;
+    let i = openBraceIndex;
+    while (i < content.length) {
+        if (content[i] === "{")
+            depth++;
+        if (content[i] === "}") {
+            depth--;
+            if (depth === 0) {
+                return content.substring(openBraceIndex + 1, i);
+            }
+        }
+        i++;
+    }
+    return content.substring(openBraceIndex + 1);
+}
+function findExpressionEnd(content, startIndex) {
+    let i = startIndex;
+    let depth = 0;
+    while (i < content.length) {
+        const char = content[i];
+        if (char === "(" || char === "[" || char === "{")
+            depth++;
+        if (char === ")" || char === "]" || char === "}") {
+            if (depth === 0)
+                return i;
+            depth--;
+        }
+        if ((char === ";" || char === "\n") && depth === 0)
+            return i;
+        i++;
+    }
+    return content.length;
+}
+function extractJsDoc(content, functionIndex) {
+    const before = content.substring(0, functionIndex).trimEnd();
+    const jsDocMatch = before.match(/\/\*\*[\s\S]*?\*\/\s*$/);
+    return jsDocMatch ? jsDocMatch[0] : null;
+}
+function extractDecorators(content, classIndex) {
+    const before = content.substring(0, classIndex).trimEnd();
+    const decorators = [];
+    const decoratorRegex = /@(\w+)(?:\([^)]*\))?\s*$/gm;
+    let match;
+    while ((match = decoratorRegex.exec(before)) !== null) {
+        decorators.push(match[1]);
+    }
+    return decorators;
+}
+function extractClassProperties(classBody) {
+    const properties = [];
+    const propRegex = /(?:(private|protected|public)\s+)?(?:(static)\s+)?(?:(readonly)\s+)?(\w+)(?:\?)?(?::\s*([^;=\n]+))?(?:\s*=\s*([^;\n]+))?;/g;
+    let match;
+    while ((match = propRegex.exec(classBody)) !== null) {
+        const visibility = match[1] || "public";
+        const name = match[4];
+        if (["if", "else", "return", "const", "let", "var", "for", "while"].includes(name))
+            continue;
+        properties.push({
+            name,
+            type: match[5]?.trim() || null,
+            isPrivate: visibility === "private",
+            isStatic: !!match[2],
+            defaultValue: match[6]?.trim() || null,
+        });
+    }
+    return properties;
+}
+function extractConstants(content) {
+    const constants = [];
+    const constRegex = /(?:(export)\s+)?const\s+(\w+)(?::\s*([^=]+))?\s*=\s*([^;\n]+)/g;
+    let match;
+    while ((match = constRegex.exec(content)) !== null) {
+        const name = match[2];
+        // Skip if it's a function (arrow function)
+        const value = match[4].trim();
+        if (value.includes("=>") || value.startsWith("function"))
+            continue;
+        constants.push({
+            name,
+            type: match[3]?.trim() || null,
+            value,
+            isExported: !!match[1],
+        });
+    }
+    return constants;
+}
+function extractInterfaces(content) {
+    const interfaces = [];
+    const interfaceRegex = /(?:(export)\s+)?interface\s+(\w+)(?:\s+extends\s+[^{]+)?\s*\{/g;
+    let match;
+    while ((match = interfaceRegex.exec(content)) !== null) {
+        interfaces.push({
+            name: match[2],
+            properties: [],
+            isExported: !!match[1],
+        });
+    }
+    return interfaces;
+}
+function calculateComplexity(body) {
+    let complexity = 1;
+    const complexityKeywords = [
+        /\bif\s*\(/g,
+        /\belse\s+if\s*\(/g,
+        /\bfor\s*\(/g,
+        /\bwhile\s*\(/g,
+        /\bswitch\s*\(/g,
+        /\bcase\s+/g,
+        /\bcatch\s*\(/g,
+        /\?\?/g,
+        /\?\./g,
+        /&&/g,
+        /\|\|/g,
+        /\?[^.?]/g,
+    ];
+    for (const regex of complexityKeywords) {
+        const matches = body.match(regex);
+        if (matches)
+            complexity += matches.length;
+    }
+    return complexity;
+}
+function getLineNumber(content, index) {
+    return content.substring(0, index).split("\n").length;
+}
+//# sourceMappingURL=typescript-analyzer.js.map

package/dist/analyzers/typescript-analyzer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"typescript-analyzer.js","sourceRoot":"","sources":["../../src/analyzers/typescript-analyzer.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AAWzB,MAAM,UAAU,qBAAqB,CAAC,QAAgB;IACpD,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;IACnD,MAAM,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAElC,OAAO;QACL,QAAQ;QACR,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC;QAChC,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC;QAChC,SAAS,EAAE,gBAAgB,CAAC,OAAO,CAAC;QACpC,OAAO,EAAE,cAAc,CAAC,OAAO,CAAC;QAChC,SAAS,EAAE,gBAAgB,CAAC,OAAO,CAAC;QACpC,UAAU,EAAE,iBAAiB,CAAC,OAAO,CAAC;KACvC,CAAC;AACJ,CAAC;AAED,SAAS,cAAc,CAAC,OAAe;IACrC,MAAM,OAAO,GAAiB,EAAE,CAAC;IACjC,MAAM,WAAW,GACf,0FAA0F,CAAC;IAC7F,MAAM,eAAe,GACnB,qDAAqD,CAAC;IAExD,IAAI,KAAK,CAAC;IAEV,OAAO,CAAC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACpD,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,KAAK,MAAM,CAAC;QACvC,MAAM,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;QACvC,MAAM,eAAe,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACvC,MAAM,MAAM,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAExB,MAAM,YAAY,GAAG,eAAe;YAClC,CAAC,CAAC,eAAe;iBACZ,KAAK,CAAC,GAAG,CAAC;iBACV,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;iBACpB,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC;iBAC3B,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE;gBACT,yBAAyB;gBACzB,MAAM,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC;gBAClC,OAAO,KAAK,CAAC,KAAK,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YACxC,CAAC,CAAC;YACN,CAAC,CAAC,EAAE,CAAC;QAEP,OAAO,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,YAAY,EAAE,aAAa,EAAE,UAAU,EAAE,CAAC,CAAC;IACpE,CAAC;IAED,OAAO,CAAC,KAAK,GAAG,eAAe,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACxD,OAAO,CAAC,IAAI,CAAC;YACX,MAAM,EAAE,KAAK,CAAC,CAAC,CAAC;YAChB,YAAY,EAAE,EAAE;YAChB,aAAa,EAAE,KAAK,CAAC,CAAC,CAAC;YACvB,UAAU,EAAE,KAAK;SAClB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,cAAc,CAAC,OAAe;IACrC,MAAM,OAAO,GAAa,EAAE,CAAC;IAC7B,MAAM,WAAW,GACf,oGAAoG,CAAC;IACvG,MAAM,gBAAgB,GAAG,uBAAuB,CAAC;IAEjD,IAAI,KAAK,CAAC;IAEV,OAAO,CAAC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACpD,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,CAAC,KAAK,GAAG,gBAAgB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACzD,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,KAAK,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;QACnF,OAAO,CAAC,IAAI,CAAC,GAAG,KAAK,CAAC,CAAC;IACzB,CAAC;IAED,OAAO,CAAC,GAAG,IAAI,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC;AAC/B,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAe;IACvC,MAAM,SAAS,GAAmB,EAAE,CAAC;IAErC,gCAAgC;IAChC,MAAM,SAAS,GACb,0HAA0H,CAAC;IAE7H,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QAClD,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5C,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC;QAC5C,MAAM,IAAI,GAAG,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAErE,SAAS,CAAC,IAAI,CAAC;YACb,IAAI;YACJ,IAAI,EAAE,UAAU;YAChB,OAAO;YACP,UAAU;YACV,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,eAAe,CAAC,SAAS,CAAC;YACtC,UAAU;YACV,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,EAAE;YACd,KAAK,EAAE,YAAY,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC;YACzC,UAAU,EAAE,mBAAmB,CAAC,IAAI,CAAC;YACrC,SAAS,EAAE,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC;YAC9C,OAAO,EAAE,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;YAC5E,IAAI;SACL,CAAC,CAAC;IACL,CAAC;IAED,4CAA4C;IAC5C,MAAM,UAAU,GACd,4HAA4H,CAAC;IAE/H,OAAO,CAAC,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAE3B,wCAAwC;QACxC,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,EAAE,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;QACpF,MAAM,WAAW,GAAG,QAAQ,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;QAClD,MAAM,SAAS,GAAG,WAAW,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAEhE,MAAM,UAAU,GAAG,OAAO,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;QACtD,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC5D,IAAI,IAAY,CAAC;QAEjB,IAAI,UAAU,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/B,IAAI,GAAG,WAAW,CAAC,OAAO,EAAE,UAAU,GAAG,CAAC,GAAG,OAAO,CAAC,SAAS,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC;QAC/F,CAAC;aAAM,CAAC;YACN,oBAAoB;YACpB,MAAM,QAAQ,GAAG,iBAAiB,CAAC,OAAO,EAAE,UAAU,GAAG,CAAC,CAAC,CAAC;YAC5D,IAAI,GAAG,OAAO,CAAC,SAAS,CAAC,UAAU,GAAG,CAAC,EAAE,QAAQ,CAAC,CAAC,IAAI,EAAE,CAAC;QAC5D,CAAC;QAED,SAAS,CAAC,IAAI,CAAC;YACb,IAAI;YACJ,IAAI,EAAE,OAAO;YACb,OAAO;YACP,UAAU;YACV,QAAQ,EAAE,KAAK;YACf,UAAU,EAAE,eAAe,CAAC,SAAS,CAAC;YACtC,UAAU,EAAE,IAAI;YAChB,SAAS,EAAE,IAAI;YACf,UAAU,EAAE,EAAE;YACd,KAAK,EAAE,YAAY,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC;YACzC,UAAU,EAAE,mBAAmB,CAAC,IAAI,CAAC;YACrC,SAAS,EAAE,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC;YAC9C,OAAO,EAAE,aAAa,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,MAAM,CAAC;YAC1D,IAAI;SACL,CAAC,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,cAAc,CAAC,OAAe;IACrC,MAAM,OAAO,GAAgB,EAAE,CAAC;IAChC,MAAM,UAAU,GACd,yGAAyG,CAAC;IAE5G,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACnD,MAAM,UAAU,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC9B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,YAAY,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC;QACtC,MAAM,aAAa,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;QAC7C,MAAM,oBAAoB,GAAG,aAAa;YACxC,CAAC,CAAC,aAAa,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;YAC/C,CAAC,CAAC,EAAE,CAAC;QAEP,MAAM,SAAS,GAAG,WAAW,CAC3B,OAAO,EACP,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAClC,CAAC;QAEF,MAAM,OAAO,GAAG,cAAc,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAChD,MAAM,eAAe,GACnB,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC,IAAI,IAAI,CAAC;QACxD,MAAM,UAAU,GAAG,sBAAsB,CAAC,SAAS,CAAC,CAAC;QAErD,OAAO,CAAC,IAAI,CAAC;YACX,IAAI;YACJ,UAAU;YACV,eAAe;YACf,OAAO,EAAE,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,aAAa,CAAC;YACxD,UAAU;YACV,UAAU,EAAE,iBAAiB,CAAC,OAAO,EAAE,KAAK,CAAC,KAAK,CAAC;YACnD,YAAY;YACZ,oBAAoB;SACrB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,cAAc,CAAC,SAAiB,EAAE,SAAiB;IAC1D,MAAM,OAAO,GAAmB,EAAE,CAAC;IACnC,MAAM,WAAW,GACf,+EAA+E,CAAC;IAElF,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACtD,MAAM,QAAQ,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC5B,MAAM,OAAO,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,MAAM,SAAS,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QAC3B,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI,CAAC;QAE5C,IAAI,IAAI,KAAK,aAAa,EAAE,CAAC;YAC3B,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,aAAa;gBACnB,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,KAAK;gBACd,UAAU,EAAE,KAAK;gBACjB,QAAQ,EAAE,KAAK;gBACf,UAAU,EAAE,eAAe,CAAC,SAAS,CAAC;gBACtC,UAAU,EAAE,IAAI;gBAChB,SAAS;gBACT,UAAU,EAAE,EAAE;gBACd,KAAK,EAAE,IAAI;gBACX,UAAU,EAAE,CAAC;gBACb,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC;gBACV,IAAI,EAAE,EAAE;aACT,CAAC,CAAC;QACL,CAAC;aAAM,CAAC;YACN,MAAM,IAAI,GAAG,WAAW,CACtB,SAAS,EACT,KAAK,CAAC,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,MAAM,GAAG,CAAC,CAClC,CAAC;YAEF,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI;gBACJ,IAAI,EAAE,QAAQ;gBACd,OAAO;gBACP,UAAU,EAAE,KAAK;gBACjB,QAAQ;gBACR,UAAU,EAAE,eAAe,CAAC,SAAS,CAAC;gBACtC,UAAU;gBACV,SAAS;gBACT,UAAU,EAAE,EAAE;gBACd,KAAK,EAAE,YAAY,CAAC,SAAS,EAAE,KAAK,CAAC,KAAK,CAAC;gBAC3C,UAAU,EAAE,mBAAmB,CAAC,IAAI,CAAC;gBACrC,SAAS,EAAE,CAAC;gBACZ,OAAO,EAAE,CAAC;gBACV,IAAI;aACL,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,eAAe,CAAC,SAAiB;IACxC,IAAI,CAAC,SAAS,CAAC,IAAI,EAAE;QAAE,OAAO,EAAE,CAAC;IAEjC,MAAM,MAAM,GAAoB,EAAE,CAAC;IACnC,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,OAAO,GAAG,EAAE,CAAC;IAEjB,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;QAC7B,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;YAAE,KAAK,EAAE,CAAC;QAC1E,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;YAAE,KAAK,EAAE,CAAC;QAC1E,IAAI,IAAI,KAAK,GAAG,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;YAChC,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;YAC/C,OAAO,GAAG,EAAE,CAAC;QACf,CAAC;aAAM,CAAC;YACN,OAAO,IAAI,IAAI,CAAC;QAClB,CAAC;IACH,CAAC;IACD,IAAI,OAAO,CAAC,IAAI,EAAE,EAAE,CAAC;QACnB,MAAM,CAAC,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACjD,CAAC;IAED,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAa;IACtC,MAAM,MAAM,GAAG,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACvC,IAAI,MAAM;QAAE,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC;IAEvC,IAAI,YAAY,GAAkB,IAAI,CAAC;IACvC,MAAM,YAAY,GAAG,KAAK,CAAC,KAAK,CAAC,cAAc,CAAC,CAAC;IACjD,IAAI,YAAY,EAAE,CAAC;QACjB,YAAY,GAAG,YAAY,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QACtC,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,CAAC;IAED,MAAM,UAAU,GAAG,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,CAAC;IACvC,KAAK,GAAG,KAAK,CAAC,OAAO,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;IAE/B,IAAI,IAAI,GAAkB,IAAI,CAAC;IAC/B,MAAM,SAAS,GAAG,KAAK,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC;IAC3C,IAAI,SAAS,EAAE,CAAC;QACd,IAAI,GAAG,SAAS,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC3B,KAAK,GAAG,KAAK,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,CAAC;IAED,OAAO;QACL,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE;QAClB,IAAI;QACJ,UAAU,EAAE,UAAU,IAAI,YAAY,KAAK,IAAI;QAC/C,YAAY;QACZ,MAAM;KACP,CAAC;AACJ,CAAC;AAED,SAAS,WAAW,CAAC,OAAe,EAAE,cAAsB;IAC1D,IAAI,KAAK,GAAG,CAAC,CAAC;IACd,IAAI,CAAC,GAAG,cAAc,CAAC;IAEvB,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QAC1B,IAAI,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG;YAAE,KAAK,EAAE,CAAC;QAChC,IAAI,OAAO,CAAC,CAAC,CAAC,KAAK,GAAG,EAAE,CAAC;YACvB,KAAK,EAAE,CAAC;YACR,IAAI,KAAK,KAAK,CAAC,EAAE,CAAC;gBAChB,OAAO,OAAO,CAAC,SAAS,CAAC,cAAc,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC;YAClD,CAAC;QACH,CAAC;QACD,CAAC,EAAE,CAAC;IACN,CAAC;IAED,OAAO,OAAO,CAAC,SAAS,CAAC,cAAc,GAAG,CAAC,CAAC,CAAC;AAC/C,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAe,EAAE,UAAkB;IAC5D,IAAI,CAAC,GAAG,UAAU,CAAC;IACnB,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;QAC1B,MAAM,IAAI,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACxB,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG;YAAE,KAAK,EAAE,CAAC;QAC1D,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,GAAG,EAAE,CAAC;YACjD,IAAI,KAAK,KAAK,CAAC;gBAAE,OAAO,CAAC,CAAC;YAC1B,KAAK,EAAE,CAAC;QACV,CAAC;QACD,IAAI,CAAC,IAAI,KAAK,GAAG,IAAI,IAAI,KAAK,IAAI,CAAC,IAAI,KAAK,KAAK,CAAC;YAAE,OAAO,CAAC,CAAC;QAC7D,CAAC,EAAE,CAAC;IACN,CAAC;IAED,OAAO,OAAO,CAAC,MAAM,CAAC;AACxB,CAAC;AAED,SAAS,YAAY,CAAC,OAAe,EAAE,aAAqB;IAC1D,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,aAAa,CAAC,CAAC,OAAO,EAAE,CAAC;IAC7D,MAAM,UAAU,GAAG,MAAM,CAAC,KAAK,CAAC,wBAAwB,CAAC,CAAC;IAC1D,OAAO,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC;AAC3C,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAe,EAAE,UAAkB;IAC5D,MAAM,MAAM,GAAG,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,UAAU,CAAC,CAAC,OAAO,EAAE,CAAC;IAC1D,MAAM,UAAU,GAAa,EAAE,CAAC;IAChC,MAAM,cAAc,GAAG,4BAA4B,CAAC;IAEpD,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACtD,UAAU,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC;IAC5B,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,sBAAsB,CAAC,SAAiB;IAC/C,MAAM,UAAU,GAAmB,EAAE,CAAC;IACtC,MAAM,SAAS,GACb,4HAA4H,CAAC;IAE/H,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,SAAS,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACpD,MAAM,UAAU,GAAG,KAAK,CAAC,CAAC,CAAC,IAAI,QAAQ,CAAC;QACxC,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,IACE,CAAC,IAAI,EAAE,MAAM,EAAE,QAAQ,EAAE,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,EAAE,OAAO,CAAC,CAAC,QAAQ,CACtE,IAAI,CACL;YAED,SAAS;QAEX,UAAU,CAAC,IAAI,CAAC;YACd,IAAI;YACJ,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI;YAC9B,SAAS,EAAE,UAAU,KAAK,SAAS;YACnC,QAAQ,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;YACpB,YAAY,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI;SACvC,CAAC,CAAC;IACL,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,gBAAgB,CAAC,OAAe;IACvC,MAAM,SAAS,GAAU,EAAE,CAAC;IAC5B,MAAM,UAAU,GACd,gEAAgE,CAAC;IAEnE,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,UAAU,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACnD,MAAM,IAAI,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC;QACtB,2CAA2C;QAC3C,MAAM,KAAK,GAAG,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC;QAC9B,IAAI,KAAK,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,KAAK,CAAC,UAAU,CAAC,UAAU,CAAC;YAAE,SAAS;QAEnE,SAAS,CAAC,IAAI,CAAC;YACb,IAAI;YACJ,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,IAAI;YAC9B,KAAK;YACL,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;SACvB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAe;IACxC,MAAM,UAAU,GAAU,EAAE,CAAC;IAC7B,MAAM,cAAc,GAClB,gEAAgE,CAAC;IAEnE,IAAI,KAAK,CAAC;IACV,OAAO,CAAC,KAAK,GAAG,cAAc,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;QACvD,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;YACd,UAAU,EAAE,EAAE;YACd,UAAU,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC;SACvB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAY;IACvC,IAAI,UAAU,GAAG,CAAC,CAAC;IACnB,MAAM,kBAAkB,GAAG;QACzB,YAAY;QACZ,mBAAmB;QACnB,aAAa;QACb,eAAe;QACf,gBAAgB;QAChB,YAAY;QACZ,eAAe;QACf,OAAO;QACP,OAAO;QACP,KAAK;QACL,OAAO;QACP,UAAU;KACX,CAAC;IAEF,KAAK,MAAM,KAAK,IAAI,kBAAkB,EAAE,CAAC;QACvC,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QAClC,IAAI,OAAO;YAAE,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC;IAC5C,CAAC;IAED,OAAO,UAAU,CAAC;AACpB,CAAC;AAED,SAAS,aAAa,CAAC,OAAe,EAAE,KAAa;IACnD,OAAO,OAAO,CAAC,SAAS,CAAC,CAAC,EAAE,KAAK,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;AACxD,CAAC"}

package/dist/generators/test-generator.d.ts

@@ -0,0 +1,4 @@
+import { FileAnalysis } from "../models/function-info.js";
+import { TestSuite } from "../models/test-case.js";
+import { GenerationConfig, ProgrammingLanguage } from "../models/project-info.js";
+export declare function generateTestSuite(analysis: FileAnalysis, language: ProgrammingLanguage, config: GenerationConfig): TestSuite;

package/dist/generators/test-generator.js

@@ -0,0 +1,223 @@
+export function generateTestSuite(analysis, language, config) {
+    const describes = [];
+    // Generate tests for standalone functions
+    for (const func of analysis.functions) {
+        if (!func.isExported && !config.coverageTargets.functions)
+            continue;
+        const testCases = generateTestCasesForFunction(func, config);
+        describes.push({
+            name: func.name,
+            testCases,
+            nestedDescribes: [],
+            beforeEach: null,
+        });
+    }
+    // Generate tests for classes
+    if (config.coverageTargets.classes) {
+        for (const cls of analysis.classes) {
+            const classDescribe = generateTestsForClass(cls, config);
+            describes.push(classDescribe);
+        }
+    }
+    // Generate import statements
+    const imports = generateImportStatements(analysis, language, config);
+    return {
+        sourceFile: analysis.filePath,
+        testFile: "",
+        imports,
+        beforeEach: null,
+        afterEach: null,
+        describes,
+    };
+}
+function generateTestCasesForFunction(func, config) {
+    const testCases = [];
+    // Basic positive test case
+    testCases.push({
+        name: `should ${func.name} correctly with valid input`,
+        description: `Tests ${func.name} with standard valid input`,
+        type: "positive",
+        arrangement: generateArrangement(func),
+        action: generateAction(func),
+        assertion: generateAssertion(func),
+        isAsync: func.isAsync,
+        targetFunction: func.name,
+        targetClass: func.className,
+        mockDependencies: [],
+    });
+    // Parameter-based tests
+    if (func.parameters.length > 0) {
+        // Test with each parameter having edge values
+        for (const param of func.parameters) {
+            if (config.coverageTargets.edgeCases) {
+                testCases.push(...generateEdgeCaseTests(func, param));
+            }
+            if (config.coverageTargets.errorHandling) {
+                testCases.push(...generateErrorHandlingTests(func, param));
+            }
+        }
+    }
+    // Test with no arguments if all params are optional
+    if (func.parameters.every((p) => p.isOptional)) {
+        testCases.push({
+            name: `should handle ${func.name} called with no arguments`,
+            description: `Tests ${func.name} with no arguments when all params are optional`,
+            type: "edge_case",
+            arrangement: "// No arrangement needed - testing with no arguments",
+            action: func.isAsync
+                ? `const result = await ${func.name}();`
+                : `const result = ${func.name}();`,
+            assertion: "expect(result).toBeDefined();",
+            isAsync: func.isAsync,
+            targetFunction: func.name,
+            targetClass: func.className,
+            mockDependencies: [],
+        });
+    }
+    // Async-specific tests
+    if (func.isAsync) {
+        testCases.push({
+            name: `should handle ${func.name} rejection/error`,
+            description: `Tests that ${func.name} properly handles async errors`,
+            type: "error_handling",
+            arrangement: "// Setup to trigger error condition",
+            action: `const promise = ${func.name}(${generateInvalidArgs(func)});`,
+            assertion: `await expect(promise).rejects.toThrow();`,
+            isAsync: true,
+            targetFunction: func.name,
+            targetClass: func.className,
+            mockDependencies: [],
+        });
+    }
+    // Return type specific tests
+    if (func.returnType) {
+        testCases.push(...generateReturnTypeTests(func));
+    }
+    return testCases;
+}
+function generateTestsForClass(cls, config) {
+    const nestedDescribes = [];
+    const classCases = [];
+    // Constructor tests
+    if (cls.constructorInfo) {
+        classCases.push({
+            name: "should create an instance",
+            description: `Tests that ${cls.name} can be instantiated`,
+            type: "positive",
+            arrangement: generateConstructorArrangement(cls),
+            action: `const instance = new ${cls.name}(${generateConstructorArgs(cls)});`,
+            assertion: `expect(instance).toBeInstanceOf(${cls.name});`,
+            isAsync: false,
+            targetFunction: "constructor",
+            targetClass: cls.name,
+            mockDependencies: [],
+        });
+    }
+    // Method tests
+    for (const method of cls.methods) {
+        if (method.name.startsWith("_") && method.name !== "__init__")
+            continue;
+        const methodDescribe = {
+            name: method.name,
+            testCases: generateTestCasesForFunction({ ...method, className: cls.name }, config),
+            nestedDescribes: [],
+            beforeEach: `const instance = new ${cls.name}(${generateConstructorArgs(cls)});`,
+        };
+        nestedDescribes.push(methodDescribe);
+    }
+    return {
+        name: cls.name,
+        testCases: classCases,
+        nestedDescribes,
+        beforeEach: null,
+    };
+}
+function generateEdgeCaseTests(func, param) {
+    const tests = [];
+    const paramType = param.type?.toLowerCase() || "any";
+    if (paramType.includes("string") ||
+        paramType === "any" ||
+        !param.type) {
+        tests.push({
+            name: `should handle ${func.name} with empty string for ${param.name}`,
+            description: `Edge case: empty string for ${param.name}`,
+            type: "edge_case",
+            arrangement: generateArrangementWithOverride(func, param.name, '""'),
+            action: generateActionWithOverride(func, param.name, '""'),
+            assertion: "expect(result).toBeDefined();",
+            isAsync: func.isAsync,
+            targetFunction: func.name,
+            targetClass: func.className,
+            mockDependencies: [],
+        });
+    }
+    if (paramType.includes("number") ||
+        paramType === "any" ||
+        !param.type) {
+        tests.push({
+            name: `should handle ${func.name} with zero for ${param.name}`,
+            description: `Edge case: zero value for ${param.name}`,
+            type: "boundary",
+            arrangement: generateArrangementWithOverride(func, param.name, "0"),
+            action: generateActionWithOverride(func, param.name, "0"),
+            assertion: "expect(result).toBeDefined();",
+            isAsync: func.isAsync,
+            targetFunction: func.name,
+            targetClass: func.className,
+            mockDependencies: [],
+        });
+    }
+    return tests;
+}
+function generateErrorHandlingTests(func, param) {
+    const tests = [];
+    tests.push({
+        name: `should handle ${func.name} with null/undefined for ${param.name}`,
+        description: `Error handling: null or undefined for ${param.name}`,
+        type: "error_handling",
+        arrangement: generateArrangementWithOverride(func, param.name, "null"),
+        action: generateActionWithOverride(func, param.name, "null"),
+        assertion: "expect(result).toBeDefined();",
+        isAsync: func.isAsync,
+        targetFunction: func.name,
+        targetClass: func.className,
+        mockDependencies: [],
+    });
+    return tests;
+}
+function generateReturnTypeTests(_func) {
+    return [];
+}
+function generateArrangement(func) {
+    const args = func.parameters.map((p) => p.defaultValue ?? "undefined").join(", ");
+    return `// Arrange\nconst result = ${func.isAsync ? "await " : ""}${func.name}(${args});`;
+}
+function generateAction(func) {
+    const args = func.parameters.map((p) => p.name + ": " + (p.defaultValue ?? "undefined")).join(", ");
+    return func.isAsync ? `const result = await ${func.name}(${args});` : `const result = ${func.name}(${args});`;
+}
+function generateAssertion(_func) {
+    return "expect(result).toBeDefined();";
+}
+function generateInvalidArgs(func) {
+    return func.parameters.map((_p) => "undefined").join(", ");
+}
+function generateConstructorArrangement(_cls) {
+    return "// Arrange";
+}
+function generateConstructorArgs(cls) {
+    return (cls.constructorInfo?.parameters ?? []).map((p) => p.defaultValue ?? "undefined").join(", ");
+}
+function generateArrangementWithOverride(func, paramName, value) {
+    const args = func.parameters.map((p) => (p.name === paramName ? value : p.defaultValue ?? "undefined")).join(", ");
+    return `// Arrange with ${paramName}=${value}\nconst result = ${func.isAsync ? "await " : ""}${func.name}(${args});`;
+}
+function generateActionWithOverride(func, paramName, value) {
+    const args = func.parameters.map((p) => (p.name === paramName ? value : p.defaultValue ?? "undefined")).join(", ");
+    return func.isAsync ? `const result = await ${func.name}(${args});` : `const result = ${func.name}(${args});`;
+}
+function generateImportStatements(analysis, _language, _config) {
+    const base = analysis.filePath.replace(/\.(ts|tsx|js|jsx)$/, "").split("/").pop() ?? "module";
+    return ["import { } from './" + base + "';"];
+}
+//# sourceMappingURL=test-generator.js.map

package/dist/generators/test-generator.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-generator.js","sourceRoot":"","sources":["../../src/generators/test-generator.ts"],"names":[],"mappings":"AAOA,MAAM,UAAU,iBAAiB,CAC/B,QAAsB,EACtB,QAA6B,EAC7B,MAAwB;IAExB,MAAM,SAAS,GAAoB,EAAE,CAAC;IAEtC,0CAA0C;IAC1C,KAAK,MAAM,IAAI,IAAI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACtC,IAAI,CAAC,IAAI,CAAC,UAAU,IAAI,CAAC,MAAM,CAAC,eAAe,CAAC,SAAS;YAAE,SAAS;QAEpE,MAAM,SAAS,GAAG,4BAA4B,CAAC,IAAI,EAAE,MAAM,CAAC,CAAC;QAE7D,SAAS,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,SAAS;YACT,eAAe,EAAE,EAAE;YACnB,UAAU,EAAE,IAAI;SACjB,CAAC,CAAC;IACL,CAAC;IAED,6BAA6B;IAC7B,IAAI,MAAM,CAAC,eAAe,CAAC,OAAO,EAAE,CAAC;QACnC,KAAK,MAAM,GAAG,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACnC,MAAM,aAAa,GAAG,qBAAqB,CAAC,GAAG,EAAE,MAAM,CAAC,CAAC;YACzD,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;QAChC,CAAC;IACH,CAAC;IAED,6BAA6B;IAC7B,MAAM,OAAO,GAAG,wBAAwB,CAAC,QAAQ,EAAE,QAAQ,EAAE,MAAM,CAAC,CAAC;IAErE,OAAO;QACL,UAAU,EAAE,QAAQ,CAAC,QAAQ;QAC7B,QAAQ,EAAE,EAAE;QACZ,OAAO;QACP,UAAU,EAAE,IAAI;QAChB,SAAS,EAAE,IAAI;QACf,SAAS;KACV,CAAC;AACJ,CAAC;AAED,SAAS,4BAA4B,CACnC,IAAS,EACT,MAAwB;IAExB,MAAM,SAAS,GAAe,EAAE,CAAC;IAEjC,2BAA2B;IAC3B,SAAS,CAAC,IAAI,CAAC;QACb,IAAI,EAAE,UAAU,IAAI,CAAC,IAAI,6BAA6B;QACtD,WAAW,EAAE,SAAS,IAAI,CAAC,IAAI,4BAA4B;QAC3D,IAAI,EAAE,UAAU;QAChB,WAAW,EAAE,mBAAmB,CAAC,IAAI,CAAC;QACtC,MAAM,EAAE,cAAc,CAAC,IAAI,CAAC;QAC5B,SAAS,EAAE,iBAAiB,CAAC,IAAI,CAAC;QAClC,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,cAAc,EAAE,IAAI,CAAC,IAAI;QACzB,WAAW,EAAE,IAAI,CAAC,SAAS;QAC3B,gBAAgB,EAAE,EAAE;KACrB,CAAC,CAAC;IAEH,wBAAwB;IACxB,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAC/B,8CAA8C;QAC9C,KAAK,MAAM,KAAK,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;YACpC,IAAI,MAAM,CAAC,eAAe,CAAC,SAAS,EAAE,CAAC;gBACrC,SAAS,CAAC,IAAI,CACZ,GAAG,qBAAqB,CAAC,IAAI,EAAE,KAAK,CAAC,CACtC,CAAC;YACJ,CAAC;YAED,IAAI,MAAM,CAAC,eAAe,CAAC,aAAa,EAAE,CAAC;gBACzC,SAAS,CAAC,IAAI,CACZ,GAAG,0BAA0B,CAAC,IAAI,EAAE,KAAK,CAAC,CAC3C,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED,oDAAoD;IACpD,IAAI,IAAI,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,UAAU,CAAC,EAAE,CAAC;QACpD,SAAS,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,iBAAiB,IAAI,CAAC,IAAI,2BAA2B;YAC3D,WAAW,EAAE,SAAS,IAAI,CAAC,IAAI,iDAAiD;YAChF,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,sDAAsD;YACnE,MAAM,EAAE,IAAI,CAAC,OAAO;gBAClB,CAAC,CAAC,wBAAwB,IAAI,CAAC,IAAI,KAAK;gBACxC,CAAC,CAAC,kBAAkB,IAAI,CAAC,IAAI,KAAK;YACpC,SAAS,EAAE,+BAA+B;YAC1C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,WAAW,EAAE,IAAI,CAAC,SAAS;YAC3B,gBAAgB,EAAE,EAAE;SACrB,CAAC,CAAC;IACL,CAAC;IAED,uBAAuB;IACvB,IAAI,IAAI,CAAC,OAAO,EAAE,CAAC;QACjB,SAAS,CAAC,IAAI,CAAC;YACb,IAAI,EAAE,iBAAiB,IAAI,CAAC,IAAI,kBAAkB;YAClD,WAAW,EAAE,cAAc,IAAI,CAAC,IAAI,gCAAgC;YACpE,IAAI,EAAE,gBAAgB;YACtB,WAAW,EAAE,qCAAqC;YAClD,MAAM,EAAE,mBAAmB,IAAI,CAAC,IAAI,IAAI,mBAAmB,CAAC,IAAI,CAAC,IAAI;YACrE,SAAS,EAAE,0CAA0C;YACrD,OAAO,EAAE,IAAI;YACb,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,WAAW,EAAE,IAAI,CAAC,SAAS;YAC3B,gBAAgB,EAAE,EAAE;SACrB,CAAC,CAAC;IACL,CAAC;IAED,6BAA6B;IAC7B,IAAI,IAAI,CAAC,UAAU,EAAE,CAAC;QACpB,SAAS,CAAC,IAAI,CACZ,GAAG,uBAAuB,CAAC,IAAI,CAAC,CACjC,CAAC;IACJ,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,qBAAqB,CAAC,GAAQ,EAAE,MAAwB;IAC/D,MAAM,eAAe,GAAoB,EAAE,CAAC;IAC5C,MAAM,UAAU,GAAe,EAAE,CAAC;IAElC,oBAAoB;IACpB,IAAI,GAAG,CAAC,eAAe,EAAE,CAAC;QACxB,UAAU,CAAC,IAAI,CAAC;YACd,IAAI,EAAE,2BAA2B;YACjC,WAAW,EAAE,cAAc,GAAG,CAAC,IAAI,sBAAsB;YACzD,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,8BAA8B,CAAC,GAAG,CAAC;YAChD,MAAM,EAAE,wBAAwB,GAAG,CAAC,IAAI,IAAI,uBAAuB,CAAC,GAAG,CAAC,IAAI;YAC5E,SAAS,EAAE,mCAAmC,GAAG,CAAC,IAAI,IAAI;YAC1D,OAAO,EAAE,KAAK;YACd,cAAc,EAAE,aAAa;YAC7B,WAAW,EAAE,GAAG,CAAC,IAAI;YACrB,gBAAgB,EAAE,EAAE;SACrB,CAAC,CAAC;IACL,CAAC;IAED,eAAe;IACf,KAAK,MAAM,MAAM,IAAI,GAAG,CAAC,OAAO,EAAE,CAAC;QACjC,IAAI,MAAM,CAAC,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,IAAI,MAAM,CAAC,IAAI,KAAK,UAAU;YAAE,SAAS;QAExE,MAAM,cAAc,GAAkB;YACpC,IAAI,EAAE,MAAM,CAAC,IAAI;YACjB,SAAS,EAAE,4BAA4B,CACrC,EAAE,GAAG,MAAM,EAAE,SAAS,EAAE,GAAG,CAAC,IAAI,EAAE,EAClC,MAAM,CACP;YACD,eAAe,EAAE,EAAE;YACnB,UAAU,EAAE,wBAAwB,GAAG,CAAC,IAAI,IAAI,uBAAuB,CAAC,GAAG,CAAC,IAAI;SACjF,CAAC;QACF,eAAe,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;IACvC,CAAC;IAED,OAAO;QACL,IAAI,EAAE,GAAG,CAAC,IAAI;QACd,SAAS,EAAE,UAAU;QACrB,eAAe;QACf,UAAU,EAAE,IAAI;KACjB,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,IAAS,EAAE,KAAU;IAClD,MAAM,KAAK,GAAe,EAAE,CAAC;IAC7B,MAAM,SAAS,GAAG,KAAK,CAAC,IAAI,EAAE,WAAW,EAAE,IAAI,KAAK,CAAC;IAErD,IACE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC5B,SAAS,KAAK,KAAK;QACnB,CAAC,KAAK,CAAC,IAAI,EACX,CAAC;QACD,KAAK,CAAC,IAAI,CAAC;YACT,IAAI,EAAE,iBAAiB,IAAI,CAAC,IAAI,0BAA0B,KAAK,CAAC,IAAI,EAAE;YACtE,WAAW,EAAE,+BAA+B,KAAK,CAAC,IAAI,EAAE;YACxD,IAAI,EAAE,WAAW;YACjB,WAAW,EAAE,+BAA+B,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC;YACpE,MAAM,EAAE,0BAA0B,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC;YAC1D,SAAS,EAAE,+BAA+B;YAC1C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,WAAW,EAAE,IAAI,CAAC,SAAS;YAC3B,gBAAgB,EAAE,EAAE;SACrB,CAAC,CAAC;IACL,CAAC;IAED,IACE,SAAS,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC5B,SAAS,KAAK,KAAK;QACnB,CAAC,KAAK,CAAC,IAAI,EACX,CAAC;QACD,KAAK,CAAC,IAAI,CAAC;YACT,IAAI,EAAE,iBAAiB,IAAI,CAAC,IAAI,kBAAkB,KAAK,CAAC,IAAI,EAAE;YAC9D,WAAW,EAAE,6BAA6B,KAAK,CAAC,IAAI,EAAE;YACtD,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,+BAA+B,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC;YACnE,MAAM,EAAE,0BAA0B,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC;YACzD,SAAS,EAAE,+BAA+B;YAC1C,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,cAAc,EAAE,IAAI,CAAC,IAAI;YACzB,WAAW,EAAE,IAAI,CAAC,SAAS;YAC3B,gBAAgB,EAAE,EAAE;SACrB,CAAC,CAAC;IACL,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,0BAA0B,CAAC,IAAS,EAAE,KAAU;IACvD,MAAM,KAAK,GAAe,EAAE,CAAC;IAC7B,KAAK,CAAC,IAAI,CAAC;QACT,IAAI,EAAE,iBAAiB,IAAI,CAAC,IAAI,4BAA4B,KAAK,CAAC,IAAI,EAAE;QACxE,WAAW,EAAE,yCAAyC,KAAK,CAAC,IAAI,EAAE;QAClE,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE,+BAA+B,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC;QACtE,MAAM,EAAE,0BAA0B,CAAC,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,MAAM,CAAC;QAC5D,SAAS,EAAE,+BAA+B;QAC1C,OAAO,EAAE,IAAI,CAAC,OAAO;QACrB,cAAc,EAAE,IAAI,CAAC,IAAI;QACzB,WAAW,EAAE,IAAI,CAAC,SAAS;QAC3B,gBAAgB,EAAE,EAAE;KACrB,CAAC,CAAC;IACH,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,uBAAuB,CAAC,KAAU;IACzC,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAS;IACpC,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACvF,OAAO,8BAA8B,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC;AAC5F,CAAC;AAED,SAAS,cAAc,CAAC,IAAS;IAC/B,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,IAAI,GAAG,IAAI,GAAG,CAAC,CAAC,CAAC,YAAY,IAAI,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACzG,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,kBAAkB,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC;AAChH,CAAC;AAED,SAAS,iBAAiB,CAAC,KAAU;IACnC,OAAO,+BAA+B,CAAC;AACzC,CAAC;AAED,SAAS,mBAAmB,CAAC,IAAS;IACpC,OAAO,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,EAAO,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,8BAA8B,CAAC,IAAS;IAC/C,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAS,uBAAuB,CAAC,GAAQ;IACvC,OAAO,CAAC,GAAG,CAAC,eAAe,EAAE,UAAU,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,YAAY,IAAI,WAAW,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC3G,CAAC;AAED,SAAS,+BAA+B,CAAC,IAAS,EAAE,SAAiB,EAAE,KAAa;IAClF,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,IAAI,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxH,OAAO,mBAAmB,SAAS,IAAI,KAAK,oBAAoB,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,GAAG,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC;AACvH,CAAC;AAED,SAAS,0BAA0B,CAAC,IAAS,EAAE,SAAiB,EAAE,KAAa;IAC7E,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAM,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,KAAK,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,CAAC,YAAY,IAAI,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IACxH,OAAO,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC,wBAAwB,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC,CAAC,CAAC,kBAAkB,IAAI,CAAC,IAAI,IAAI,IAAI,IAAI,CAAC;AAChH,CAAC;AAED,SAAS,wBAAwB,CAAC,QAAsB,EAAE,SAA8B,EAAE,OAAyB;IACjH,MAAM,IAAI,GAAG,QAAQ,CAAC,QAAQ,CAAC,OAAO,CAAC,oBAAoB,EAAE,EAAE,CAAC,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,IAAI,QAAQ,CAAC;IAC9F,OAAO,CAAC,qBAAqB,GAAG,IAAI,GAAG,IAAI,CAAC,CAAC;AAC/C,CAAC"}

package/dist/index.d.ts

@@ -0,0 +1,2 @@
+#!/usr/bin/env node
+export {};

package/dist/index.js

@@ -0,0 +1,14 @@
+#!/usr/bin/env node
+import { createMCPServer } from "./server.js";
+async function main() {
+    const server = createMCPServer();
+    const transport = await import("@modelcontextprotocol/sdk/server/stdio.js");
+    const stdioTransport = new transport.StdioServerTransport();
+    await server.connect(stdioTransport);
+    console.error("MCP Test Generator Server running on stdio");
+}
+main().catch((error) => {
+    console.error("Fatal error:", error);
+    process.exit(1);
+});
+//# sourceMappingURL=index.js.map