mcp-test-generator 1.0.0

package/dist/utils/file-utils.js

@@ -0,0 +1,157 @@
+import * as fs from "fs";
+import * as path from "path";
+import { glob } from "glob";
+import { getSourceExtensions } from "./language-detector.js";
+const DEFAULT_EXCLUDE_DIRS = [
+    "node_modules",
+    ".git",
+    "dist",
+    "build",
+    "coverage",
+    ".next",
+    "__pycache__",
+    ".pytest_cache",
+    "venv",
+    "env",
+    ".venv",
+    "target",
+    ".gradle",
+    ".idea",
+    ".vscode",
+];
+const DEFAULT_EXCLUDE_FILES = [
+    "*.d.ts",
+    "*.config.*",
+    "*.spec.*",
+    "*.test.*",
+    "*_test.*",
+    "*Test.*",
+    "index.ts",
+    "index.js",
+    "main.ts",
+    "main.js",
+    "setup.*",
+    "jest.setup.*",
+];
+export async function findSourceFiles(projectPath, language, includePatterns, excludePatterns) {
+    const extensions = getSourceExtensions(language);
+    let patterns;
+    if (includePatterns && includePatterns.length > 0) {
+        patterns = includePatterns;
+    }
+    else {
+        patterns = extensions.map((ext) => `**/*${ext}`);
+    }
+    const excludeDirs = DEFAULT_EXCLUDE_DIRS.map((d) => `**/${d}/**`);
+    const defaultExcludes = [
+        ...excludeDirs,
+        ...DEFAULT_EXCLUDE_FILES.map((f) => `**/${f}`),
+    ];
+    const allExcludes = [
+        ...defaultExcludes,
+        ...(excludePatterns || []),
+    ];
+    const sourceFiles = [];
+    for (const pattern of patterns) {
+        const files = await glob(pattern, {
+            cwd: projectPath,
+            ignore: allExcludes,
+            absolute: false,
+            nodir: true,
+        });
+        for (const file of files) {
+            const absolutePath = path.join(projectPath, file);
+            const stat = fs.statSync(absolutePath);
+            // Skip test files
+            if (isTestFile(file))
+                continue;
+            // Skip very small files (likely just exports)
+            if (stat.size < 50)
+                continue;
+            const existingTestPath = findExistingTest(projectPath, file, language);
+            sourceFiles.push({
+                filePath: absolutePath,
+                relativePath: file,
+                language,
+                size: stat.size,
+                hasExistingTests: existingTestPath !== null,
+                existingTestPath,
+            });
+        }
+    }
+    return sourceFiles;
+}
+export function isTestFile(filePath) {
+    const basename = path.basename(filePath);
+    return (basename.includes(".test.") ||
+        basename.includes(".spec.") ||
+        basename.includes("_test.") ||
+        basename.endsWith("Test.java") ||
+        basename.endsWith("Tests.java") ||
+        filePath.includes("__tests__") ||
+        filePath.includes("/tests/") ||
+        filePath.includes("/test/"));
+}
+export function findExistingTest(projectPath, sourceRelativePath, language) {
+    const parsed = path.parse(sourceRelativePath);
+    const nameWithoutExt = parsed.name;
+    const dir = parsed.dir;
+    const possibleTestPaths = [
+        // Co-located tests
+        path.join(dir, `${nameWithoutExt}.test${parsed.ext}`),
+        path.join(dir, `${nameWithoutExt}.spec${parsed.ext}`),
+        // __tests__ directory
+        path.join(dir, "__tests__", `${nameWithoutExt}.test${parsed.ext}`),
+        path.join(dir, "__tests__", `${nameWithoutExt}.spec${parsed.ext}`),
+        // tests directory at root
+        path.join("tests", dir, `${nameWithoutExt}.test${parsed.ext}`),
+        path.join("test", dir, `${nameWithoutExt}.test${parsed.ext}`),
+    ];
+    // Python-specific
+    if (language === "python") {
+        possibleTestPaths.push(path.join(dir, `test_${nameWithoutExt}.py`), path.join(dir, `${nameWithoutExt}_test.py`), path.join("tests", `test_${nameWithoutExt}.py`));
+    }
+    // Java-specific
+    if (language === "java") {
+        possibleTestPaths.push(path.join(dir.replace("main", "test"), `${nameWithoutExt}Test.java`));
+    }
+    for (const testPath of possibleTestPaths) {
+        const absoluteTestPath = path.join(projectPath, testPath);
+        if (fs.existsSync(absoluteTestPath)) {
+            return testPath;
+        }
+    }
+    return null;
+}
+export function generateTestFilePath(projectPath, sourceRelativePath, language, testFramework, outputDir) {
+    const parsed = path.parse(sourceRelativePath);
+    const nameWithoutExt = parsed.name;
+    let testFileName;
+    let testDir;
+    switch (language) {
+        case "python":
+            testFileName = `test_${nameWithoutExt}.py`;
+            testDir = outputDir || "tests";
+            break;
+        case "java":
+            testFileName = `${nameWithoutExt}Test.java`;
+            testDir = outputDir || parsed.dir.replace("src/main", "src/test");
+            break;
+        default:
+            testFileName = `${nameWithoutExt}.test${parsed.ext}`;
+            testDir = outputDir || path.join(parsed.dir, "__tests__");
+            break;
+    }
+    return path.join(projectPath, testDir, testFileName);
+}
+export function ensureDirectoryExists(filePath) {
+    const dir = path.dirname(filePath);
+    if (!fs.existsSync(dir)) {
+        fs.mkdirSync(dir, { recursive: true });
+    }
+}
+export function writeTestFile(filePath, content) {
+    ensureDirectoryExists(filePath);
+    fs.writeFileSync(filePath, content, "utf-8");
+}
+//# sourceMappingURL=file-utils.js.map

package/dist/utils/file-utils.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"file-utils.js","sourceRoot":"","sources":["../../src/utils/file-utils.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAC7B,OAAO,EAAE,IAAI,EAAE,MAAM,MAAM,CAAC;AAK5B,OAAO,EAAE,mBAAmB,EAAwB,MAAM,wBAAwB,CAAC;AAEnF,MAAM,oBAAoB,GAAG;IAC3B,cAAc;IACd,MAAM;IACN,MAAM;IACN,OAAO;IACP,UAAU;IACV,OAAO;IACP,aAAa;IACb,eAAe;IACf,MAAM;IACN,KAAK;IACL,OAAO;IACP,QAAQ;IACR,SAAS;IACT,OAAO;IACP,SAAS;CACV,CAAC;AAEF,MAAM,qBAAqB,GAAG;IAC5B,QAAQ;IACR,YAAY;IACZ,UAAU;IACV,UAAU;IACV,UAAU;IACV,SAAS;IACT,UAAU;IACV,UAAU;IACV,SAAS;IACT,SAAS;IACT,SAAS;IACT,cAAc;CACf,CAAC;AAEF,MAAM,CAAC,KAAK,UAAU,eAAe,CACnC,WAAmB,EACnB,QAA6B,EAC7B,eAA0B,EAC1B,eAA0B;IAE1B,MAAM,UAAU,GAAG,mBAAmB,CAAC,QAAQ,CAAC,CAAC;IAEjD,IAAI,QAAkB,CAAC;IAEvB,IAAI,eAAe,IAAI,eAAe,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAClD,QAAQ,GAAG,eAAe,CAAC;IAC7B,CAAC;SAAM,CAAC;QACN,QAAQ,GAAG,UAAU,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,GAAG,EAAE,CAAC,CAAC;IACnD,CAAC;IAED,MAAM,WAAW,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;IAClE,MAAM,eAAe,GAAG;QACtB,GAAG,WAAW;QACd,GAAG,qBAAqB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,MAAM,CAAC,EAAE,CAAC;KAC/C,CAAC;IAEF,MAAM,WAAW,GAAG;QAClB,GAAG,eAAe;QAClB,GAAG,CAAC,eAAe,IAAI,EAAE,CAAC;KAC3B,CAAC;IAEF,MAAM,WAAW,GAAqB,EAAE,CAAC;IAEzC,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,OAAO,EAAE;YAChC,GAAG,EAAE,WAAW;YAChB,MAAM,EAAE,WAAW;YACnB,QAAQ,EAAE,KAAK;YACf,KAAK,EAAE,IAAI;SACZ,CAAC,CAAC;QAEH,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,MAAM,YAAY,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,CAAC;YAClD,MAAM,IAAI,GAAG,EAAE,CAAC,QAAQ,CAAC,YAAY,CAAC,CAAC;YAEvC,kBAAkB;YAClB,IAAI,UAAU,CAAC,IAAI,CAAC;gBAAE,SAAS;YAE/B,8CAA8C;YAC9C,IAAI,IAAI,CAAC,IAAI,GAAG,EAAE;gBAAE,SAAS;YAE7B,MAAM,gBAAgB,GAAG,gBAAgB,CAAC,WAAW,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC;YAEvE,WAAW,CAAC,IAAI,CAAC;gBACf,QAAQ,EAAE,YAAY;gBACtB,YAAY,EAAE,IAAI;gBAClB,QAAQ;gBACR,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,gBAAgB,EAAE,gBAAgB,KAAK,IAAI;gBAC3C,gBAAgB;aACjB,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,WAAW,CAAC;AACrB,CAAC;AAED,MAAM,UAAU,UAAU,CAAC,QAAgB;IACzC,MAAM,QAAQ,GAAG,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;IACzC,OAAO,CACL,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC3B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC3B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC;QAC3B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC9B,QAAQ,CAAC,QAAQ,CAAC,YAAY,CAAC;QAC/B,QAAQ,CAAC,QAAQ,CAAC,WAAW,CAAC;QAC9B,QAAQ,CAAC,QAAQ,CAAC,SAAS,CAAC;QAC5B,QAAQ,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAC5B,CAAC;AACJ,CAAC;AAED,MAAM,UAAU,gBAAgB,CAC9B,WAAmB,EACnB,kBAA0B,EAC1B,QAA6B;IAE7B,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAC9C,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC;IACnC,MAAM,GAAG,GAAG,MAAM,CAAC,GAAG,CAAC;IAEvB,MAAM,iBAAiB,GAAG;QACxB,mBAAmB;QACnB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,QAAQ,MAAM,CAAC,GAAG,EAAE,CAAC;QACrD,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,QAAQ,MAAM,CAAC,GAAG,EAAE,CAAC;QACrD,sBAAsB;QACtB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,WAAW,EAAE,GAAG,cAAc,QAAQ,MAAM,CAAC,GAAG,EAAE,CAAC;QAClE,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,WAAW,EAAE,GAAG,cAAc,QAAQ,MAAM,CAAC,GAAG,EAAE,CAAC;QAClE,0BAA0B;QAC1B,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,GAAG,cAAc,QAAQ,MAAM,CAAC,GAAG,EAAE,CAAC;QAC9D,IAAI,CAAC,IAAI,CAAC,MAAM,EAAE,GAAG,EAAE,GAAG,cAAc,QAAQ,MAAM,CAAC,GAAG,EAAE,CAAC;KAC9D,CAAC;IAEF,kBAAkB;IAClB,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,iBAAiB,CAAC,IAAI,CACpB,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,QAAQ,cAAc,KAAK,CAAC,EAC3C,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,cAAc,UAAU,CAAC,EAC3C,IAAI,CAAC,IAAI,CAAC,OAAO,EAAE,QAAQ,cAAc,KAAK,CAAC,CAChD,CAAC;IACJ,CAAC;IAED,gBAAgB;IAChB,IAAI,QAAQ,KAAK,MAAM,EAAE,CAAC;QACxB,iBAAiB,CAAC,IAAI,CACpB,IAAI,CAAC,IAAI,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,EAAE,MAAM,CAAC,EAAE,GAAG,cAAc,WAAW,CAAC,CACrE,CAAC;IACJ,CAAC;IAED,KAAK,MAAM,QAAQ,IAAI,iBAAiB,EAAE,CAAC;QACzC,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,QAAQ,CAAC,CAAC;QAC1D,IAAI,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,EAAE,CAAC;YACpC,OAAO,QAAQ,CAAC;QAClB,CAAC;IACH,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,WAAmB,EACnB,kBAA0B,EAC1B,QAA6B,EAC7B,aAAqB,EACrB,SAAkB;IAElB,MAAM,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,kBAAkB,CAAC,CAAC;IAC9C,MAAM,cAAc,GAAG,MAAM,CAAC,IAAI,CAAC;IAEnC,IAAI,YAAoB,CAAC;IACzB,IAAI,OAAe,CAAC;IAEpB,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,QAAQ;YACX,YAAY,GAAG,QAAQ,cAAc,KAAK,CAAC;YAC3C,OAAO,GAAG,SAAS,IAAI,OAAO,CAAC;YAC/B,MAAM;QACR,KAAK,MAAM;YACT,YAAY,GAAG,GAAG,cAAc,WAAW,CAAC;YAC5C,OAAO,GAAG,SAAS,IAAI,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;YAClE,MAAM;QACR;YACE,YAAY,GAAG,GAAG,cAAc,QAAQ,MAAM,CAAC,GAAG,EAAE,CAAC;YACrD,OAAO,GAAG,SAAS,IAAI,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YAC1D,MAAM;IACV,CAAC;IAED,OAAO,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,OAAO,EAAE,YAAY,CAAC,CAAC;AACvD,CAAC;AAED,MAAM,UAAU,qBAAqB,CAAC,QAAgB;IACpD,MAAM,GAAG,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;IACnC,IAAI,CAAC,EAAE,CAAC,UAAU,CAAC,GAAG,CAAC,EAAE,CAAC;QACxB,EAAE,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;IACzC,CAAC;AACH,CAAC;AAED,MAAM,UAAU,aAAa,CAAC,QAAgB,EAAE,OAAe;IAC7D,qBAAqB,CAAC,QAAQ,CAAC,CAAC;IAChC,EAAE,CAAC,aAAa,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;AAC/C,CAAC"}

package/dist/utils/language-detector.d.ts

@@ -0,0 +1,6 @@
+import { ProgrammingLanguage, ProjectFramework, TestFramework } from "../models/project-info.js";
+export declare function detectLanguage(projectPath: string): ProgrammingLanguage;
+export declare function detectFramework(projectPath: string): ProjectFramework;
+export declare function detectTestFramework(projectPath: string, language: ProgrammingLanguage): TestFramework;
+export declare function getSourceExtensions(language: ProgrammingLanguage): string[];
+export declare function getTestFileExtension(language: ProgrammingLanguage, framework: TestFramework): string;

package/dist/utils/language-detector.js

@@ -0,0 +1,148 @@
+import * as fs from "fs";
+import * as path from "path";
+export function detectLanguage(projectPath) {
+    const files = fs.readdirSync(projectPath);
+    // Check for TypeScript
+    if (files.includes("tsconfig.json") ||
+        files.some((f) => f.endsWith(".ts") || f.endsWith(".tsx"))) {
+        return "typescript";
+    }
+    // Check for Python
+    if (files.includes("setup.py") ||
+        files.includes("pyproject.toml") ||
+        files.includes("requirements.txt") ||
+        files.some((f) => f.endsWith(".py"))) {
+        return "python";
+    }
+    // Check for Java
+    if (files.includes("pom.xml") ||
+        files.includes("build.gradle") ||
+        files.some((f) => f.endsWith(".java"))) {
+        return "java";
+    }
+    // Check for JavaScript
+    if (files.includes("package.json") ||
+        files.some((f) => f.endsWith(".js") || f.endsWith(".jsx"))) {
+        return "javascript";
+    }
+    return "unknown";
+}
+export function detectFramework(projectPath) {
+    const packageJsonPath = path.join(projectPath, "package.json");
+    if (fs.existsSync(packageJsonPath)) {
+        try {
+            const pkg = JSON.parse(fs.readFileSync(packageJsonPath, "utf-8"));
+            const allDeps = {
+                ...pkg.dependencies,
+                ...pkg.devDependencies,
+            };
+            if (allDeps["next"])
+                return "nextjs";
+            if (allDeps["@nestjs/core"])
+                return "nestjs";
+            if (allDeps["react"])
+                return "react";
+            if (allDeps["vue"])
+                return "vue";
+            if (allDeps["@angular/core"])
+                return "angular";
+            if (allDeps["express"])
+                return "express";
+        }
+        catch {
+            // Ignore parse errors
+        }
+    }
+    // Check for Python frameworks
+    const requirementsPath = path.join(projectPath, "requirements.txt");
+    if (fs.existsSync(requirementsPath)) {
+        const content = fs.readFileSync(requirementsPath, "utf-8");
+        if (content.includes("fastapi"))
+            return "fastapi";
+        if (content.includes("flask"))
+            return "flask";
+        if (content.includes("django"))
+            return "django";
+    }
+    // Check for pyproject.toml
+    const pyprojectPath = path.join(projectPath, "pyproject.toml");
+    if (fs.existsSync(pyprojectPath)) {
+        const content = fs.readFileSync(pyprojectPath, "utf-8");
+        if (content.includes("fastapi"))
+            return "fastapi";
+        if (content.includes("flask"))
+            return "flask";
+        if (content.includes("django"))
+            return "django";
+    }
+    // Check for Java frameworks
+    const pomPath = path.join(projectPath, "pom.xml");
+    if (fs.existsSync(pomPath)) {
+        const content = fs.readFileSync(pomPath, "utf-8");
+        if (content.includes("spring"))
+            return "spring";
+    }
+    return null;
+}
+export function detectTestFramework(projectPath, language) {
+    const packageJsonPath = path.join(projectPath, "package.json");
+    if (fs.existsSync(packageJsonPath)) {
+        try {
+            const pkg = JSON.parse(fs.readFileSync(packageJsonPath, "utf-8"));
+            const allDeps = {
+                ...pkg.dependencies,
+                ...pkg.devDependencies,
+            };
+            if (allDeps["vitest"])
+                return "vitest";
+            if (allDeps["jest"] || allDeps["@jest/core"])
+                return "jest";
+            if (allDeps["mocha"])
+                return "mocha";
+        }
+        catch {
+            // Ignore
+        }
+    }
+    // Defaults by language
+    switch (language) {
+        case "typescript":
+        case "javascript":
+            return "jest";
+        case "python":
+            return "pytest";
+        case "java":
+            return "junit";
+        default:
+            return "jest";
+    }
+}
+export function getSourceExtensions(language) {
+    switch (language) {
+        case "typescript":
+            return [".ts", ".tsx"];
+        case "javascript":
+            return [".js", ".jsx", ".mjs"];
+        case "python":
+            return [".py"];
+        case "java":
+            return [".java"];
+        default:
+            return [".ts", ".js"];
+    }
+}
+export function getTestFileExtension(language, framework) {
+    switch (language) {
+        case "typescript":
+            return framework === "vitest" ? ".test.ts" : ".test.ts";
+        case "javascript":
+            return ".test.js";
+        case "python":
+            return "_test.py";
+        case "java":
+            return "Test.java";
+        default:
+            return ".test.ts";
+    }
+}
+//# sourceMappingURL=language-detector.js.map

package/dist/utils/language-detector.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"language-detector.js","sourceRoot":"","sources":["../../src/utils/language-detector.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,MAAM,IAAI,CAAC;AACzB,OAAO,KAAK,IAAI,MAAM,MAAM,CAAC;AAO7B,MAAM,UAAU,cAAc,CAAC,WAAmB;IAChD,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,CAAC,WAAW,CAAC,CAAC;IAE1C,uBAAuB;IACvB,IACE,KAAK,CAAC,QAAQ,CAAC,eAAe,CAAC;QAC/B,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAC1D,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,mBAAmB;IACnB,IACE,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC;QAC1B,KAAK,CAAC,QAAQ,CAAC,gBAAgB,CAAC;QAChC,KAAK,CAAC,QAAQ,CAAC,kBAAkB,CAAC;QAClC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC,EACpC,CAAC;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAED,iBAAiB;IACjB,IACE,KAAK,CAAC,QAAQ,CAAC,SAAS,CAAC;QACzB,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,EACtC,CAAC;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAED,uBAAuB;IACvB,IACE,KAAK,CAAC,QAAQ,CAAC,cAAc,CAAC;QAC9B,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,QAAQ,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAC1D,CAAC;QACD,OAAO,YAAY,CAAC;IACtB,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,MAAM,UAAU,eAAe,CAAC,WAAmB;IACjD,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IAE/D,IAAI,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC;YAClE,MAAM,OAAO,GAAG;gBACd,GAAG,GAAG,CAAC,YAAY;gBACnB,GAAG,GAAG,CAAC,eAAe;aACvB,CAAC;YAEF,IAAI,OAAO,CAAC,MAAM,CAAC;gBAAE,OAAO,QAAQ,CAAC;YACrC,IAAI,OAAO,CAAC,cAAc,CAAC;gBAAE,OAAO,QAAQ,CAAC;YAC7C,IAAI,OAAO,CAAC,OAAO,CAAC;gBAAE,OAAO,OAAO,CAAC;YACrC,IAAI,OAAO,CAAC,KAAK,CAAC;gBAAE,OAAO,KAAK,CAAC;YACjC,IAAI,OAAO,CAAC,eAAe,CAAC;gBAAE,OAAO,SAAS,CAAC;YAC/C,IAAI,OAAO,CAAC,SAAS,CAAC;gBAAE,OAAO,SAAS,CAAC;QAC3C,CAAC;QAAC,MAAM,CAAC;YACP,sBAAsB;QACxB,CAAC;IACH,CAAC;IAED,8BAA8B;IAC9B,MAAM,gBAAgB,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,kBAAkB,CAAC,CAAC;IACpE,IAAI,EAAE,CAAC,UAAU,CAAC,gBAAgB,CAAC,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,gBAAgB,EAAE,OAAO,CAAC,CAAC;QAC3D,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,SAAS,CAAC;QAClD,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,OAAO,CAAC;QAC9C,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;IAClD,CAAC;IAED,2BAA2B;IAC3B,MAAM,aAAa,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,gBAAgB,CAAC,CAAC;IAC/D,IAAI,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,EAAE,CAAC;QACjC,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,aAAa,EAAE,OAAO,CAAC,CAAC;QACxD,IAAI,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;YAAE,OAAO,SAAS,CAAC;QAClD,IAAI,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,OAAO,CAAC;QAC9C,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;IAClD,CAAC;IAED,4BAA4B;IAC5B,MAAM,OAAO,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;IAClD,IAAI,EAAE,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;QAC3B,MAAM,OAAO,GAAG,EAAE,CAAC,YAAY,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QAClD,IAAI,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,QAAQ,CAAC;IAClD,CAAC;IAED,OAAO,IAAI,CAAC;AACd,CAAC;AAED,MAAM,UAAU,mBAAmB,CACjC,WAAmB,EACnB,QAA6B;IAE7B,MAAM,eAAe,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,cAAc,CAAC,CAAC;IAE/D,IAAI,EAAE,CAAC,UAAU,CAAC,eAAe,CAAC,EAAE,CAAC;QACnC,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,YAAY,CAAC,eAAe,EAAE,OAAO,CAAC,CAAC,CAAC;YAClE,MAAM,OAAO,GAAG;gBACd,GAAG,GAAG,CAAC,YAAY;gBACnB,GAAG,GAAG,CAAC,eAAe;aACvB,CAAC;YAEF,IAAI,OAAO,CAAC,QAAQ,CAAC;gBAAE,OAAO,QAAQ,CAAC;YACvC,IAAI,OAAO,CAAC,MAAM,CAAC,IAAI,OAAO,CAAC,YAAY,CAAC;gBAAE,OAAO,MAAM,CAAC;YAC5D,IAAI,OAAO,CAAC,OAAO,CAAC;gBAAE,OAAO,OAAO,CAAC;QACvC,CAAC;QAAC,MAAM,CAAC;YACP,SAAS;QACX,CAAC;IACH,CAAC;IAED,uBAAuB;IACvB,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,YAAY,CAAC;QAClB,KAAK,YAAY;YACf,OAAO,MAAM,CAAC;QAChB,KAAK,QAAQ;YACX,OAAO,QAAQ,CAAC;QAClB,KAAK,MAAM;YACT,OAAO,OAAO,CAAC;QACjB;YACE,OAAO,MAAM,CAAC;IAClB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,mBAAmB,CAAC,QAA6B;IAC/D,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,YAAY;YACf,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC;QACzB,KAAK,YAAY;YACf,OAAO,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC,CAAC;QACjC,KAAK,QAAQ;YACX,OAAO,CAAC,KAAK,CAAC,CAAC;QACjB,KAAK,MAAM;YACT,OAAO,CAAC,OAAO,CAAC,CAAC;QACnB;YACE,OAAO,CAAC,KAAK,EAAE,KAAK,CAAC,CAAC;IAC1B,CAAC;AACH,CAAC;AAED,MAAM,UAAU,oBAAoB,CAClC,QAA6B,EAC7B,SAAwB;IAExB,QAAQ,QAAQ,EAAE,CAAC;QACjB,KAAK,YAAY;YACf,OAAO,SAAS,KAAK,QAAQ,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC;QAC1D,KAAK,YAAY;YACf,OAAO,UAAU,CAAC;QACpB,KAAK,QAAQ;YACX,OAAO,UAAU,CAAC;QACpB,KAAK,MAAM;YACT,OAAO,WAAW,CAAC;QACrB;YACE,OAAO,UAAU,CAAC;IACtB,CAAC;AACH,CAAC"}

package/dist/utils/test-suite-renderer.d.ts

@@ -0,0 +1,3 @@
+import { TestSuite } from "../models/test-case.js";
+import { ProgrammingLanguage } from "../models/project-info.js";
+export declare function renderTestSuiteToCode(suite: TestSuite, _testFramework: string, language: ProgrammingLanguage): string;

package/dist/utils/test-suite-renderer.js

@@ -0,0 +1,62 @@
+export function renderTestSuiteToCode(suite, _testFramework, language) {
+    if (language === "python") {
+        return renderPython(suite);
+    }
+    return renderJestOrVitest(suite);
+}
+function renderJestOrVitest(suite) {
+    const lines = [...suite.imports, ""];
+    for (const block of suite.describes) {
+        lines.push(...renderDescribeBlock(block, 0));
+    }
+    return lines.join("\n");
+}
+function renderDescribeBlock(block, indent) {
+    const lines = [];
+    const pad = "  ".repeat(indent);
+    lines.push(`${pad}describe("${escapeStr(block.name)}", () => {`);
+    if (block.beforeEach) {
+        lines.push(`${pad}  beforeEach(() => {`);
+        block.beforeEach.split("\n").forEach((l) => lines.push(`${pad}    ${l.trim()}`));
+        lines.push(`${pad}  });`);
+    }
+    for (const tc of block.testCases) {
+        const testFn = tc.isAsync ? "async () =>" : "() =>";
+        lines.push(`${pad}  it("${escapeStr(tc.name)}", ${testFn} {`);
+        if (tc.arrangement) {
+            tc.arrangement.split("\n").forEach((l) => lines.push(`${pad}    ${l.trim()}`));
+        }
+        tc.action.split("\n").forEach((l) => lines.push(`${pad}    ${l.trim()}`));
+        tc.assertion.split("\n").forEach((l) => lines.push(`${pad}    ${l.trim()}`));
+        lines.push(`${pad}  });`);
+    }
+    for (const nested of block.nestedDescribes) {
+        lines.push(...renderDescribeBlock(nested, indent + 1));
+    }
+    lines.push(`${pad}});`);
+    return lines;
+}
+function renderPython(suite) {
+    const lines = [...suite.imports, ""];
+    for (const block of suite.describes) {
+        lines.push(`class Test${block.name.replace(/\W/g, "_")}:`);
+        if (block.beforeEach) {
+            lines.push("    def setUp(self):");
+            block.beforeEach.split("\n").forEach((l) => lines.push(`        ${l.trim()}`));
+        }
+        for (const tc of block.testCases) {
+            const methodName = tc.name.replace(/\s+/g, "_").replace(/[^a-zA-Z0-9_]/g, "");
+            lines.push(`    def test_${methodName}(self):`);
+            [tc.arrangement, tc.action, tc.assertion].forEach((s) => {
+                if (s)
+                    s.split("\n").forEach((l) => lines.push(`        ${l.trim()}`));
+            });
+            lines.push("");
+        }
+    }
+    return lines.join("\n");
+}
+function escapeStr(s) {
+    return s.replace(/\\/g, "\\\\").replace(/"/g, '\\"');
+}
+//# sourceMappingURL=test-suite-renderer.js.map

package/dist/utils/test-suite-renderer.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"test-suite-renderer.js","sourceRoot":"","sources":["../../src/utils/test-suite-renderer.ts"],"names":[],"mappings":"AAGA,MAAM,UAAU,qBAAqB,CACnC,KAAgB,EAChB,cAAsB,EACtB,QAA6B;IAE7B,IAAI,QAAQ,KAAK,QAAQ,EAAE,CAAC;QAC1B,OAAO,YAAY,CAAC,KAAK,CAAC,CAAC;IAC7B,CAAC;IACD,OAAO,kBAAkB,CAAC,KAAK,CAAC,CAAC;AACnC,CAAC;AAED,SAAS,kBAAkB,CAAC,KAAgB;IAC1C,MAAM,KAAK,GAAa,CAAC,GAAG,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC/C,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QACpC,KAAK,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC;IAC/C,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAoB,EAAE,MAAc;IAC/D,MAAM,KAAK,GAAa,EAAE,CAAC;IAC3B,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;IAChC,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,aAAa,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;IACjE,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;QACrB,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,sBAAsB,CAAC,CAAC;QACzC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QACjF,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC;IAC5B,CAAC;IACD,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QACjC,MAAM,MAAM,GAAG,EAAE,CAAC,OAAO,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,OAAO,CAAC;QACpD,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,SAAS,SAAS,CAAC,EAAE,CAAC,IAAI,CAAC,MAAM,MAAM,IAAI,CAAC,CAAC;QAC9D,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;YACnB,EAAE,CAAC,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,EAAE,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QAC1E,EAAE,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QAC7E,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,OAAO,CAAC,CAAC;IAC5B,CAAC;IACD,KAAK,MAAM,MAAM,IAAI,KAAK,CAAC,eAAe,EAAE,CAAC;QAC3C,KAAK,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,MAAM,EAAE,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC;IACzD,CAAC;IACD,KAAK,CAAC,IAAI,CAAC,GAAG,GAAG,KAAK,CAAC,CAAC;IACxB,OAAO,KAAK,CAAC;AACf,CAAC;AAED,SAAS,YAAY,CAAC,KAAgB;IACpC,MAAM,KAAK,GAAa,CAAC,GAAG,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;IAC/C,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;QACpC,KAAK,CAAC,IAAI,CAAC,aAAa,KAAK,CAAC,IAAI,CAAC,OAAO,CAAC,KAAK,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC;QAC3D,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;YACrB,KAAK,CAAC,IAAI,CAAC,sBAAsB,CAAC,CAAC;YACnC,KAAK,CAAC,UAAU,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;QACjF,CAAC;QACD,KAAK,MAAM,EAAE,IAAI,KAAK,CAAC,SAAS,EAAE,CAAC;YACjC,MAAM,UAAU,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,CAAC,MAAM,EAAE,GAAG,CAAC,CAAC,OAAO,CAAC,gBAAgB,EAAE,EAAE,CAAC,CAAC;YAC9E,KAAK,CAAC,IAAI,CAAC,gBAAgB,UAAU,SAAS,CAAC,CAAC;YAChD,CAAC,EAAE,CAAC,WAAW,EAAE,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE;gBACtD,IAAI,CAAC;oBAAE,CAAC,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;YACzE,CAAC,CAAC,CAAC;YACH,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IACD,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AAC1B,CAAC;AAED,SAAS,SAAS,CAAC,CAAS;IAC1B,OAAO,CAAC,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;AACvD,CAAC"}

package/package.json

@@ -0,0 +1,49 @@
+{
+  "name": "mcp-test-generator",
+  "version": "1.0.0",
+  "description": "MCP server that generates test cases for TypeScript, JavaScript, Python, and Java projects. Use with Cursor, Claude Desktop, or any MCP client.",
+  "main": "dist/index.js",
+  "type": "module",
+  "bin": {
+    "mcp-test-generator": "dist/index.js"
+  },
+  "files": [
+    "dist",
+    "README.md"
+  ],
+  "scripts": {
+    "build": "tsc",
+    "start": "node dist/index.js",
+    "dev": "tsx src/index.ts",
+    "watch": "tsc --watch",
+    "prepublishOnly": "npm run build"
+  },
+  "keywords": [
+    "mcp",
+    "model-context-protocol",
+    "test-generator",
+    "jest",
+    "pytest",
+    "vitest",
+    "testing"
+  ],
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/ranjats/mcp-test-generator.git"
+  },
+  "license": "MIT",
+  "engines": {
+    "node": ">=18"
+  },
+  "dependencies": {
+    "@modelcontextprotocol/sdk": "^1.0.0",
+    "glob": "^10.3.10",
+    "typescript": "^5.3.3",
+    "ts-morph": "^21.0.1",
+    "zod": "^3.22.4"
+  },
+  "devDependencies": {
+    "@types/node": "^20.10.0",
+    "tsx": "^4.7.0"
+  }
+}