@devness/coverit 0.1.0

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

@@ -0,0 +1,79 @@
+import type { CodeScanResult, GeneratorContext, GeneratorResult, TestFramework, TestType } from "../types/index.js";
+import type { AIProvider } from "../ai/types.js";
+/**
+ * Abstract base class for all test generators.
+ * Provides shared helpers for building test file content strings,
+ * deduplicating against existing tests, and producing consistent output.
+ *
+ * When an AIProvider is attached, generators attempt LLM-powered test
+ * generation first, falling back to AST-based templates on failure or
+ * when no provider is available.
+ */
+export declare abstract class BaseGenerator {
+    protected aiProvider: AIProvider | null;
+    abstract generate(context: GeneratorContext): Promise<GeneratorResult>;
+    setAIProvider(provider: AIProvider): void;
+    /**
+     * Attempts LLM-powered test generation using the attached AI provider.
+     * Returns the generated test file content as a string, or null if:
+     * - No AI provider is configured
+     * - The provider call fails for any reason
+     *
+     * Callers should fall back to template-based generation when this returns null.
+     */
+    protected generateWithAI(params: {
+        sourceCode: string;
+        scanResult: CodeScanResult;
+        testType: TestType;
+        framework: TestFramework;
+        existingTests?: string[];
+    }): Promise<string | null>;
+    /**
+     * Extracts raw code from an LLM response that may be wrapped in
+     * markdown code fences (```typescript ... ``` or ``` ... ```).
+     */
+    private extractCodeFromResponse;
+    /**
+     * Counts the number of test cases in generated code by looking for
+     * `it(`, `test(`, and `test.only(` invocations.
+     */
+    protected countTestCases(code: string): number;
+    protected fileHeader(): string;
+    protected buildImports(framework: TestFramework): string;
+    protected buildDescribeBlock(name: string, tests: string[]): string;
+    protected buildTestCase(name: string, body: string): string;
+    protected buildAsyncTestCase(name: string, body: string): string;
+    protected generateTestFileName(sourceFile: string, type: TestType): string;
+    /**
+     * Reads all test file contents from a directory so generators
+     * can skip creating tests that already exist.
+     */
+    protected readExistingTests(dir: string): Promise<string[]>;
+    /**
+     * Returns true if an existing test already covers the given target name
+     * (function, class, endpoint, etc.) by scanning existing test contents.
+     */
+    protected isAlreadyTested(targetName: string, existingTests: string[]): boolean;
+    protected indent(text: string, spaces: number): string;
+    protected escapeString(str: string): string;
+    private escapeRegExp;
+    /**
+     * Generates a mock variable name from a module path.
+     * e.g. "../services/user.service" -> "mockUserService"
+     */
+    protected mockNameFromModule(modulePath: string): string;
+    /**
+     * Generates a sample value for a given TypeScript type string.
+     * Used to create meaningful test data.
+     */
+    protected sampleValueForType(type: string | null): string;
+    /**
+     * Builds a full test file string from its parts.
+     */
+    protected assembleTestFile(parts: {
+        header?: string;
+        imports: string[];
+        body: string[];
+    }): string;
+}
+//# sourceMappingURL=base-generator.d.ts.map

package/dist/generators/base-generator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"base-generator.d.ts","sourceRoot":"","sources":["../../src/generators/base-generator.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EACV,cAAc,EACd,gBAAgB,EAChB,eAAe,EACf,aAAa,EACb,QAAQ,EACT,MAAM,mBAAmB,CAAC;AAC3B,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,gBAAgB,CAAC;AAGjD;;;;;;;;GAQG;AACH,8BAAsB,aAAa;IACjC,SAAS,CAAC,UAAU,EAAE,UAAU,GAAG,IAAI,CAAQ;IAE/C,QAAQ,CAAC,QAAQ,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,eAAe,CAAC;IAEtE,aAAa,CAAC,QAAQ,EAAE,UAAU,GAAG,IAAI;IAIzC;;;;;;;OAOG;cACa,cAAc,CAAC,MAAM,EAAE;QACrC,UAAU,EAAE,MAAM,CAAC;QACnB,UAAU,EAAE,cAAc,CAAC;QAC3B,QAAQ,EAAE,QAAQ,CAAC;QACnB,SAAS,EAAE,aAAa,CAAC;QACzB,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC;KAC1B,GAAG,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC;IA6B1B;;;OAGG;IACH,OAAO,CAAC,uBAAuB;IAO/B;;;OAGG;IACH,SAAS,CAAC,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,MAAM;IAQ9C,SAAS,CAAC,UAAU,IAAI,MAAM;IAM9B,SAAS,CAAC,YAAY,CAAC,SAAS,EAAE,aAAa,GAAG,MAAM;IAqBxD,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM;IAKnE,SAAS,CAAC,aAAa,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM;IAK3D,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,MAAM,EAAE,IAAI,EAAE,MAAM,GAAG,MAAM;IAOhE,SAAS,CAAC,oBAAoB,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,GAAG,MAAM;IAsB1E;;;OAGG;cACa,iBAAiB,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC;IA2BjE;;;OAGG;IACH,SAAS,CAAC,eAAe,CACvB,UAAU,EAAE,MAAM,EAClB,aAAa,EAAE,MAAM,EAAE,GACtB,OAAO;IAUV,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,GAAG,MAAM;IAQtD,SAAS,CAAC,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,MAAM;IAI3C,OAAO,CAAC,YAAY;IAIpB;;;OAGG;IACH,SAAS,CAAC,kBAAkB,CAAC,UAAU,EAAE,MAAM,GAAG,MAAM;IAWxD;;;OAGG;IACH,SAAS,CAAC,kBAAkB,CAAC,IAAI,EAAE,MAAM,GAAG,IAAI,GAAG,MAAM;IAuBzD;;OAEG;IACH,SAAS,CAAC,gBAAgB,CAAC,KAAK,EAAE;QAChC,MAAM,CAAC,EAAE,MAAM,CAAC;QAChB,OAAO,EAAE,MAAM,EAAE,CAAC;QAClB,IAAI,EAAE,MAAM,EAAE,CAAC;KAChB,GAAG,MAAM;CAWX"}

package/dist/generators/base-generator.js

@@ -0,0 +1,234 @@
+// Generated by coverit — https://coverit.dev
+import { readdir, readFile } from "node:fs/promises";
+import { basename, dirname, extname, join } from "node:path";
+import { buildTestGenerationPrompt } from "../ai/prompts.js";
+/**
+ * Abstract base class for all test generators.
+ * Provides shared helpers for building test file content strings,
+ * deduplicating against existing tests, and producing consistent output.
+ *
+ * When an AIProvider is attached, generators attempt LLM-powered test
+ * generation first, falling back to AST-based templates on failure or
+ * when no provider is available.
+ */
+export class BaseGenerator {
+    aiProvider = null;
+    setAIProvider(provider) {
+        this.aiProvider = provider;
+    }
+    /**
+     * Attempts LLM-powered test generation using the attached AI provider.
+     * Returns the generated test file content as a string, or null if:
+     * - No AI provider is configured
+     * - The provider call fails for any reason
+     *
+     * Callers should fall back to template-based generation when this returns null.
+     */
+    async generateWithAI(params) {
+        if (!this.aiProvider)
+            return null;
+        try {
+            const messages = buildTestGenerationPrompt({
+                sourceCode: params.sourceCode,
+                scanResult: params.scanResult,
+                testType: params.testType,
+                framework: params.framework,
+                existingTests: params.existingTests,
+            });
+            const response = await this.aiProvider.generate(messages, {
+                temperature: 0.2,
+                maxTokens: 4096,
+            });
+            const content = response.content.trim();
+            if (!content)
+                return null;
+            // Strip markdown fences if the model wrapped its output in them
+            return this.extractCodeFromResponse(content);
+        }
+        catch (err) {
+            const msg = err instanceof Error ? err.message : String(err);
+            console.error(`[coverit] AI generation failed: ${msg}`);
+            return null;
+        }
+    }
+    /**
+     * Extracts raw code from an LLM response that may be wrapped in
+     * markdown code fences (```typescript ... ``` or ``` ... ```).
+     */
+    extractCodeFromResponse(raw) {
+        const fenceMatch = raw.match(/^```(?:typescript|ts|javascript|js|tsx|jsx)?\s*\n([\s\S]*?)\n```\s*$/);
+        return fenceMatch?.[1] ? fenceMatch[1].trim() : raw;
+    }
+    /**
+     * Counts the number of test cases in generated code by looking for
+     * `it(`, `test(`, and `test.only(` invocations.
+     */
+    countTestCases(code) {
+        const itCount = (code.match(/\bit\s*\(/g) || []).length;
+        const testCount = (code.match(/\btest\s*\(/g) || []).length;
+        return Math.max(itCount + testCount, 1);
+    }
+    // ── File header ────────────────────────────────────────────
+    fileHeader() {
+        return "// Generated by coverit — https://coverit.dev";
+    }
+    // ── Import builders ────────────────────────────────────────
+    buildImports(framework) {
+        switch (framework) {
+            case "vitest":
+                return "import { describe, it, expect, vi, beforeEach, afterEach } from 'vitest';";
+            case "jest":
+                // Jest globals are available without explicit import in most configs,
+                // but we import jest for mock utilities.
+                return "import { jest } from '@jest/globals';";
+            case "playwright":
+                return "import { test, expect } from '@playwright/test';";
+            case "detox":
+                return [
+                    "import { device, element, by, expect as detoxExpect } from 'detox';",
+                ].join("\n");
+            default:
+                return "// No framework-specific imports";
+        }
+    }
+    // ── Block builders ─────────────────────────────────────────
+    buildDescribeBlock(name, tests) {
+        const body = tests.map((t) => this.indent(t, 2)).join("\n\n");
+        return `describe('${this.escapeString(name)}', () => {\n${body}\n});`;
+    }
+    buildTestCase(name, body) {
+        const indentedBody = this.indent(body, 2);
+        return `it('${this.escapeString(name)}', () => {\n${indentedBody}\n});`;
+    }
+    buildAsyncTestCase(name, body) {
+        const indentedBody = this.indent(body, 2);
+        return `it('${this.escapeString(name)}', async () => {\n${indentedBody}\n});`;
+    }
+    // ── File naming ────────────────────────────────────────────
+    generateTestFileName(sourceFile, type) {
+        const dir = dirname(sourceFile);
+        const ext = extname(sourceFile);
+        const name = basename(sourceFile, ext);
+        const suffixMap = {
+            unit: "test",
+            integration: "integration.test",
+            api: "api.test",
+            "e2e-browser": "e2e.test",
+            "e2e-mobile": "mobile.test",
+            "e2e-desktop": "desktop.test",
+            snapshot: "snap.test",
+            performance: "perf.test",
+        };
+        const suffix = suffixMap[type] ?? "test";
+        return join(dir, `${name}.${suffix}${ext}`);
+    }
+    // ── Existing test deduplication ────────────────────────────
+    /**
+     * Reads all test file contents from a directory so generators
+     * can skip creating tests that already exist.
+     */
+    async readExistingTests(dir) {
+        try {
+            const entries = await readdir(dir, { recursive: true });
+            const testFiles = entries.filter((e) => typeof e === "string" &&
+                (e.endsWith(".test.ts") ||
+                    e.endsWith(".test.tsx") ||
+                    e.endsWith(".spec.ts") ||
+                    e.endsWith(".spec.tsx")));
+            const contents = [];
+            for (const file of testFiles) {
+                try {
+                    const content = await readFile(join(dir, file), "utf-8");
+                    contents.push(content);
+                }
+                catch {
+                    // Silently skip unreadable files
+                }
+            }
+            return contents;
+        }
+        catch {
+            return [];
+        }
+    }
+    /**
+     * Returns true if an existing test already covers the given target name
+     * (function, class, endpoint, etc.) by scanning existing test contents.
+     */
+    isAlreadyTested(targetName, existingTests) {
+        const pattern = new RegExp(`(?:describe|it|test)\\s*\\(\\s*['"\`].*${this.escapeRegExp(targetName)}`, "i");
+        return existingTests.some((content) => pattern.test(content));
+    }
+    // ── Utility helpers ────────────────────────────────────────
+    indent(text, spaces) {
+        const pad = " ".repeat(spaces);
+        return text
+            .split("\n")
+            .map((line) => (line.trim() ? `${pad}${line}` : line))
+            .join("\n");
+    }
+    escapeString(str) {
+        return str.replace(/\\/g, "\\\\").replace(/'/g, "\\'");
+    }
+    escapeRegExp(str) {
+        return str.replace(/[.*+?^${}()|[\]\\]/g, "\\$&");
+    }
+    /**
+     * Generates a mock variable name from a module path.
+     * e.g. "../services/user.service" -> "mockUserService"
+     */
+    mockNameFromModule(modulePath) {
+        const name = basename(modulePath, extname(modulePath));
+        const camel = name
+            .split(/[-.]/)
+            .map((part, i) => i === 0 ? part : part.charAt(0).toUpperCase() + part.slice(1))
+            .join("");
+        return `mock${camel.charAt(0).toUpperCase()}${camel.slice(1)}`;
+    }
+    /**
+     * Generates a sample value for a given TypeScript type string.
+     * Used to create meaningful test data.
+     */
+    sampleValueForType(type) {
+        if (!type)
+            return "'test-value'";
+        const normalized = type.trim().toLowerCase();
+        if (normalized === "string")
+            return "'test-string'";
+        if (normalized === "number")
+            return "42";
+        if (normalized === "boolean")
+            return "true";
+        if (normalized === "void" || normalized === "undefined")
+            return "undefined";
+        if (normalized === "null")
+            return "null";
+        if (normalized.endsWith("[]"))
+            return "[]";
+        if (normalized.startsWith("promise<")) {
+            const inner = type.slice(8, -1);
+            return this.sampleValueForType(inner);
+        }
+        if (normalized === "date")
+            return "new Date('2025-01-01')";
+        if (normalized === "record<string, any>" || normalized === "object")
+            return "{}";
+        // Default to an empty object for complex types
+        return `{} as ${type}`;
+    }
+    /**
+     * Builds a full test file string from its parts.
+     */
+    assembleTestFile(parts) {
+        const sections = [
+            parts.header ?? this.fileHeader(),
+            "",
+            ...parts.imports,
+            "",
+            ...parts.body,
+            "", // trailing newline
+        ];
+        return sections.join("\n");
+    }
+}
+//# sourceMappingURL=base-generator.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"base-generator.js","sourceRoot":"","sources":["../../src/generators/base-generator.ts"],"names":[],"mappings":"AAAA,6CAA6C;AAE7C,OAAO,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,kBAAkB,CAAC;AACrD,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAS7D,OAAO,EAAE,yBAAyB,EAAE,MAAM,kBAAkB,CAAC;AAE7D;;;;;;;;GAQG;AACH,MAAM,OAAgB,aAAa;IACvB,UAAU,GAAsB,IAAI,CAAC;IAI/C,aAAa,CAAC,QAAoB;QAChC,IAAI,CAAC,UAAU,GAAG,QAAQ,CAAC;IAC7B,CAAC;IAED;;;;;;;OAOG;IACO,KAAK,CAAC,cAAc,CAAC,MAM9B;QACC,IAAI,CAAC,IAAI,CAAC,UAAU;YAAE,OAAO,IAAI,CAAC;QAElC,IAAI,CAAC;YACH,MAAM,QAAQ,GAAG,yBAAyB,CAAC;gBACzC,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,UAAU,EAAE,MAAM,CAAC,UAAU;gBAC7B,QAAQ,EAAE,MAAM,CAAC,QAAQ;gBACzB,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,aAAa,EAAE,MAAM,CAAC,aAAa;aACpC,CAAC,CAAC;YAEH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,QAAQ,CAAC,QAAQ,EAAE;gBACxD,WAAW,EAAE,GAAG;gBAChB,SAAS,EAAE,IAAI;aAChB,CAAC,CAAC;YAEH,MAAM,OAAO,GAAG,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC;YACxC,IAAI,CAAC,OAAO;gBAAE,OAAO,IAAI,CAAC;YAE1B,gEAAgE;YAChE,OAAO,IAAI,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,GAAG,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC7D,OAAO,CAAC,KAAK,CAAC,mCAAmC,GAAG,EAAE,CAAC,CAAC;YACxD,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;;OAGG;IACK,uBAAuB,CAAC,GAAW;QACzC,MAAM,UAAU,GAAG,GAAG,CAAC,KAAK,CAC1B,sEAAsE,CACvE,CAAC;QACF,OAAO,UAAU,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC;IACtD,CAAC;IAED;;;OAGG;IACO,cAAc,CAAC,IAAY;QACnC,MAAM,OAAO,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,YAAY,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QACxD,MAAM,SAAS,GAAG,CAAC,IAAI,CAAC,KAAK,CAAC,cAAc,CAAC,IAAI,EAAE,CAAC,CAAC,MAAM,CAAC;QAC5D,OAAO,IAAI,CAAC,GAAG,CAAC,OAAO,GAAG,SAAS,EAAE,CAAC,CAAC,CAAC;IAC1C,CAAC;IAED,8DAA8D;IAEpD,UAAU;QAClB,OAAO,+CAA+C,CAAC;IACzD,CAAC;IAED,8DAA8D;IAEpD,YAAY,CAAC,SAAwB;QAC7C,QAAQ,SAAS,EAAE,CAAC;YAClB,KAAK,QAAQ;gBACX,OAAO,2EAA2E,CAAC;YACrF,KAAK,MAAM;gBACT,sEAAsE;gBACtE,yCAAyC;gBACzC,OAAO,uCAAuC,CAAC;YACjD,KAAK,YAAY;gBACf,OAAO,kDAAkD,CAAC;YAC5D,KAAK,OAAO;gBACV,OAAO;oBACL,qEAAqE;iBACtE,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;YACf;gBACE,OAAO,kCAAkC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,8DAA8D;IAEpD,kBAAkB,CAAC,IAAY,EAAE,KAAe;QACxD,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC9D,OAAO,aAAa,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,eAAe,IAAI,OAAO,CAAC;IACxE,CAAC;IAES,aAAa,CAAC,IAAY,EAAE,IAAY;QAChD,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAC1C,OAAO,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,eAAe,YAAY,OAAO,CAAC;IAC1E,CAAC;IAES,kBAAkB,CAAC,IAAY,EAAE,IAAY;QACrD,MAAM,YAAY,GAAG,IAAI,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;QAC1C,OAAO,OAAO,IAAI,CAAC,YAAY,CAAC,IAAI,CAAC,qBAAqB,YAAY,OAAO,CAAC;IAChF,CAAC;IAED,8DAA8D;IAEpD,oBAAoB,CAAC,UAAkB,EAAE,IAAc;QAC/D,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,GAAG,GAAG,OAAO,CAAC,UAAU,CAAC,CAAC;QAChC,MAAM,IAAI,GAAG,QAAQ,CAAC,UAAU,EAAE,GAAG,CAAC,CAAC;QAEvC,MAAM,SAAS,GAA6B;YAC1C,IAAI,EAAE,MAAM;YACZ,WAAW,EAAE,kBAAkB;YAC/B,GAAG,EAAE,UAAU;YACf,aAAa,EAAE,UAAU;YACzB,YAAY,EAAE,aAAa;YAC3B,aAAa,EAAE,cAAc;YAC7B,QAAQ,EAAE,WAAW;YACrB,WAAW,EAAE,WAAW;SACzB,CAAC;QAEF,MAAM,MAAM,GAAG,SAAS,CAAC,IAAI,CAAC,IAAI,MAAM,CAAC;QACzC,OAAO,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,IAAI,MAAM,GAAG,GAAG,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED,8DAA8D;IAE9D;;;OAGG;IACO,KAAK,CAAC,iBAAiB,CAAC,GAAW;QAC3C,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,EAAE,EAAE,SAAS,EAAE,IAAI,EAAE,CAAC,CAAC;YACxD,MAAM,SAAS,GAAG,OAAO,CAAC,MAAM,CAC9B,CAAC,CAAC,EAAE,EAAE,CACJ,OAAO,CAAC,KAAK,QAAQ;gBACrB,CAAC,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC;oBACrB,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC;oBACvB,CAAC,CAAC,QAAQ,CAAC,UAAU,CAAC;oBACtB,CAAC,CAAC,QAAQ,CAAC,WAAW,CAAC,CAAC,CAC7B,CAAC;YAEF,MAAM,QAAQ,GAAa,EAAE,CAAC;YAC9B,KAAK,MAAM,IAAI,IAAI,SAAS,EAAE,CAAC;gBAC7B,IAAI,CAAC;oBACH,MAAM,OAAO,GAAG,MAAM,QAAQ,CAAC,IAAI,CAAC,GAAG,EAAE,IAAI,CAAC,EAAE,OAAO,CAAC,CAAC;oBACzD,QAAQ,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;gBACzB,CAAC;gBAAC,MAAM,CAAC;oBACP,iCAAiC;gBACnC,CAAC;YACH,CAAC;YACD,OAAO,QAAQ,CAAC;QAClB,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,EAAE,CAAC;QACZ,CAAC;IACH,CAAC;IAED;;;OAGG;IACO,eAAe,CACvB,UAAkB,EAClB,aAAuB;QAEvB,MAAM,OAAO,GAAG,IAAI,MAAM,CACxB,0CAA0C,IAAI,CAAC,YAAY,CAAC,UAAU,CAAC,EAAE,EACzE,GAAG,CACJ,CAAC;QACF,OAAO,aAAa,CAAC,IAAI,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC,CAAC;IAChE,CAAC;IAED,8DAA8D;IAEpD,MAAM,CAAC,IAAY,EAAE,MAAc;QAC3C,MAAM,GAAG,GAAG,GAAG,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QAC/B,OAAO,IAAI;aACR,KAAK,CAAC,IAAI,CAAC;aACX,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,GAAG,GAAG,GAAG,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC;aACrD,IAAI,CAAC,IAAI,CAAC,CAAC;IAChB,CAAC;IAES,YAAY,CAAC,GAAW;QAChC,OAAO,GAAG,CAAC,OAAO,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACzD,CAAC;IAEO,YAAY,CAAC,GAAW;QAC9B,OAAO,GAAG,CAAC,OAAO,CAAC,qBAAqB,EAAE,MAAM,CAAC,CAAC;IACpD,CAAC;IAED;;;OAGG;IACO,kBAAkB,CAAC,UAAkB;QAC7C,MAAM,IAAI,GAAG,QAAQ,CAAC,UAAU,EAAE,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;QACvD,MAAM,KAAK,GAAG,IAAI;aACf,KAAK,CAAC,MAAM,CAAC;aACb,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CACf,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,CAC9D;aACA,IAAI,CAAC,EAAE,CAAC,CAAC;QACZ,OAAO,OAAO,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,WAAW,EAAE,GAAG,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC,EAAE,CAAC;IACjE,CAAC;IAED;;;OAGG;IACO,kBAAkB,CAAC,IAAmB;QAC9C,IAAI,CAAC,IAAI;YAAE,OAAO,cAAc,CAAC;QAEjC,MAAM,UAAU,GAAG,IAAI,CAAC,IAAI,EAAE,CAAC,WAAW,EAAE,CAAC;QAE7C,IAAI,UAAU,KAAK,QAAQ;YAAE,OAAO,eAAe,CAAC;QACpD,IAAI,UAAU,KAAK,QAAQ;YAAE,OAAO,IAAI,CAAC;QACzC,IAAI,UAAU,KAAK,SAAS;YAAE,OAAO,MAAM,CAAC;QAC5C,IAAI,UAAU,KAAK,MAAM,IAAI,UAAU,KAAK,WAAW;YAAE,OAAO,WAAW,CAAC;QAC5E,IAAI,UAAU,KAAK,MAAM;YAAE,OAAO,MAAM,CAAC;QACzC,IAAI,UAAU,CAAC,QAAQ,CAAC,IAAI,CAAC;YAAE,OAAO,IAAI,CAAC;QAC3C,IAAI,UAAU,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,CAAC;YACtC,MAAM,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,CAAC;YAChC,OAAO,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC;QACD,IAAI,UAAU,KAAK,MAAM;YAAE,OAAO,wBAAwB,CAAC;QAC3D,IAAI,UAAU,KAAK,qBAAqB,IAAI,UAAU,KAAK,QAAQ;YACjE,OAAO,IAAI,CAAC;QAEd,+CAA+C;QAC/C,OAAO,SAAS,IAAI,EAAE,CAAC;IACzB,CAAC;IAED;;OAEG;IACO,gBAAgB,CAAC,KAI1B;QACC,MAAM,QAAQ,GAAa;YACzB,KAAK,CAAC,MAAM,IAAI,IAAI,CAAC,UAAU,EAAE;YACjC,EAAE;YACF,GAAG,KAAK,CAAC,OAAO;YAChB,EAAE;YACF,GAAG,KAAK,CAAC,IAAI;YACb,EAAE,EAAE,mBAAmB;SACxB,CAAC;QACF,OAAO,QAAQ,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;CACF"}

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

@@ -0,0 +1,22 @@
+import type { GeneratorContext, GeneratorResult } from "../types/index.js";
+import { BaseGenerator } from "./base-generator.js";
+/**
+ * Generates tests for Tauri desktop applications:
+ * - IPC command invoke tests (mocking @tauri-apps/api/core)
+ * - React component tests for Tauri frontends
+ */
+export declare class DesktopTestGenerator extends BaseGenerator {
+    generate(context: GeneratorContext): Promise<GeneratorResult>;
+    private buildDesktopImports;
+    private generateCommandTests;
+    private generateTauriComponentTests;
+    private generateStandardComponentTests;
+    private detectTauriCommands;
+    private getTauriCommandFunctions;
+    private isTauriIntegrated;
+    private buildCommandPayload;
+    private buildDefaultProps;
+    private toSnakeCase;
+    private relativeImportPath;
+}
+//# sourceMappingURL=desktop-test.d.ts.map

package/dist/generators/desktop-test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"desktop-test.d.ts","sourceRoot":"","sources":["../../src/generators/desktop-test.ts"],"names":[],"mappings":"AAIA,OAAO,KAAK,EAKV,gBAAgB,EAChB,eAAe,EAIhB,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD;;;;GAIG;AACH,qBAAa,oBAAqB,SAAQ,aAAa;IAC/C,QAAQ,CAAC,OAAO,EAAE,gBAAgB,GAAG,OAAO,CAAC,eAAe,CAAC;IAuInE,OAAO,CAAC,mBAAmB;IA2D3B,OAAO,CAAC,oBAAoB;IA0D5B,OAAO,CAAC,2BAA2B;IAqDnC,OAAO,CAAC,8BAA8B;IA+DtC,OAAO,CAAC,mBAAmB;IAS3B,OAAO,CAAC,wBAAwB;IAMhC,OAAO,CAAC,iBAAiB;IASzB,OAAO,CAAC,mBAAmB;IAU3B,OAAO,CAAC,iBAAiB;IAezB,OAAO,CAAC,WAAW;IAOnB,OAAO,CAAC,kBAAkB;CAK3B"}

package/dist/generators/desktop-test.js

@@ -0,0 +1,290 @@
+// Generated by coverit — https://coverit.dev
+import { readFile } from "node:fs/promises";
+import { dirname, join, relative } from "node:path";
+import { BaseGenerator } from "./base-generator.js";
+/**
+ * Generates tests for Tauri desktop applications:
+ * - IPC command invoke tests (mocking @tauri-apps/api/core)
+ * - React component tests for Tauri frontends
+ */
+export class DesktopTestGenerator extends BaseGenerator {
+    async generate(context) {
+        const { plan, project, scanResults, existingTests } = context;
+        const tests = [];
+        const warnings = [];
+        const skipped = [];
+        for (const scan of scanResults) {
+            if (!plan.target.files.includes(scan.file))
+                continue;
+            // ── AI-powered generation (preferred path) ──────────────
+            if (this.aiProvider) {
+                try {
+                    const sourceCode = await readFile(join(project.root, scan.file), "utf-8");
+                    const aiResult = await this.generateWithAI({
+                        sourceCode,
+                        scanResult: scan,
+                        testType: "e2e-desktop",
+                        framework: project.testFramework,
+                        existingTests,
+                    });
+                    if (aiResult) {
+                        const testFileName = this.generateTestFileName(scan.file, "e2e-desktop");
+                        tests.push({
+                            planId: plan.id,
+                            filePath: testFileName,
+                            content: aiResult,
+                            testType: "e2e-desktop",
+                            testCount: this.countTestCases(aiResult),
+                            framework: project.testFramework === "playwright" ||
+                                project.testFramework === "detox"
+                                ? "vitest"
+                                : project.testFramework,
+                        });
+                        continue; // Skip template fallback for this file
+                    }
+                }
+                catch {
+                    // Fall through to template-based generation
+                }
+            }
+            // ── Template-based generation (fallback) ────────────────
+            const hasComponents = scan.components.length > 0;
+            const hasTauriCommands = this.detectTauriCommands(scan);
+            if (!hasComponents && !hasTauriCommands) {
+                skipped.push({
+                    target: scan.file,
+                    reason: "No Tauri components or commands detected",
+                });
+                continue;
+            }
+            const testFileName = this.generateTestFileName(scan.file, "e2e-desktop");
+            const fw = project.testFramework;
+            const imports = this.buildDesktopImports(scan, fw, testFileName, hasTauriCommands);
+            const bodyBlocks = [];
+            let testCount = 0;
+            // Tauri IPC command tests
+            if (hasTauriCommands) {
+                const commandFns = this.getTauriCommandFunctions(scan);
+                for (const fn of commandFns) {
+                    if (this.isAlreadyTested(`invoke:${fn.name}`, existingTests)) {
+                        skipped.push({
+                            target: `invoke:${fn.name}`,
+                            reason: "Already has command tests",
+                        });
+                        continue;
+                    }
+                    const cases = this.generateCommandTests(fn, fw);
+                    bodyBlocks.push(this.buildDescribeBlock(`Tauri command: ${fn.name}`, cases));
+                    testCount += cases.length;
+                }
+            }
+            // React component tests
+            for (const component of scan.components) {
+                if (this.isAlreadyTested(component.name, existingTests)) {
+                    skipped.push({
+                        target: component.name,
+                        reason: "Already has desktop component tests",
+                    });
+                    continue;
+                }
+                const isTauriComponent = this.isTauriIntegrated(component, scan);
+                const cases = isTauriComponent
+                    ? this.generateTauriComponentTests(component, scan, fw)
+                    : this.generateStandardComponentTests(component, fw);
+                bodyBlocks.push(this.buildDescribeBlock(component.name, cases));
+                testCount += cases.length;
+            }
+            if (testCount === 0)
+                continue;
+            const content = this.assembleTestFile({
+                imports,
+                body: bodyBlocks,
+            });
+            tests.push({
+                planId: plan.id,
+                filePath: testFileName,
+                content,
+                testType: "e2e-desktop",
+                testCount,
+                framework: fw === "playwright" || fw === "detox" ? "vitest" : fw,
+            });
+        }
+        return { tests, warnings, skipped };
+    }
+    // ── Import generation ──────────────────────────────────────
+    buildDesktopImports(scan, fw, testFilePath, hasTauriCommands) {
+        const lines = [this.buildImports(fw)];
+        const mockFn = fw === "vitest" ? "vi" : "jest";
+        // Mock Tauri core if IPC commands are used
+        if (hasTauriCommands) {
+            lines.push(`${mockFn === "vi" ? "vi.mock" : "jest.mock"}('@tauri-apps/api/core', () => ({`, `  invoke: ${mockFn}.fn(),`, `}));`, "", `import { invoke } from '@tauri-apps/api/core';`);
+        }
+        // Import testing library for React component tests
+        if (scan.components.length > 0) {
+            lines.push("import { render, screen, fireEvent, waitFor } from '@testing-library/react';");
+        }
+        // Import source components/functions
+        const sourcePath = this.relativeImportPath(testFilePath, scan.file);
+        const exportedNames = scan.exports
+            .filter((e) => e.kind === "function" || e.kind === "class")
+            .map((e) => e.name);
+        const componentNames = scan.components.map((c) => c.name);
+        const allNames = [...new Set([...exportedNames, ...componentNames])];
+        if (allNames.length > 0) {
+            const defaultExport = scan.exports.find((e) => e.isDefault);
+            const namedExports = allNames.filter((n) => n !== defaultExport?.name);
+            if (defaultExport && namedExports.length > 0) {
+                lines.push(`import ${defaultExport.name}, { ${namedExports.join(", ")} } from '${sourcePath}';`);
+            }
+            else if (defaultExport) {
+                lines.push(`import ${defaultExport.name} from '${sourcePath}';`);
+            }
+            else if (namedExports.length > 0) {
+                lines.push(`import { ${namedExports.join(", ")} } from '${sourcePath}';`);
+            }
+        }
+        return lines;
+    }
+    // ── Tauri command tests ────────────────────────────────────
+    generateCommandTests(fn, fw) {
+        const cases = [];
+        const mockFn = fw === "vitest" ? "vi" : "jest";
+        const mockedInvoke = `${mockFn === "vi" ? "vi.mocked" : "jest.mocked"}(invoke)`;
+        // Build sample payload from function params
+        const payload = this.buildCommandPayload(fn.params);
+        const payloadStr = payload ? `, ${payload}` : "";
+        const returnSample = this.sampleValueForType(fn.returnType);
+        // Success case
+        cases.push(this.buildAsyncTestCase("should invoke command successfully", [
+            `${mockedInvoke}.mockResolvedValueOnce(${returnSample});`,
+            `const result = await invoke('${this.toSnakeCase(fn.name)}'${payloadStr});`,
+            `expect(invoke).toHaveBeenCalledWith('${this.toSnakeCase(fn.name)}'${payloadStr});`,
+            `expect(result).toEqual(${returnSample});`,
+        ].join("\n")));
+        // Error case
+        cases.push(this.buildAsyncTestCase("should handle command error", [
+            `${mockedInvoke}.mockRejectedValueOnce(new Error('Command failed'));`,
+            `await expect(invoke('${this.toSnakeCase(fn.name)}'${payloadStr})).rejects.toThrow('Command failed');`,
+        ].join("\n")));
+        // Payload validation (if params exist)
+        if (fn.params.length > 0) {
+            cases.push(this.buildAsyncTestCase("should send correct payload", [
+                `${mockedInvoke}.mockResolvedValueOnce(undefined);`,
+                `await invoke('${this.toSnakeCase(fn.name)}'${payloadStr});`,
+                `expect(invoke).toHaveBeenCalledTimes(1);`,
+                `const calledPayload = ${mockedInvoke}.mock.calls[0][1];`,
+                `expect(calledPayload).toBeDefined();`,
+            ].join("\n")));
+        }
+        return cases;
+    }
+    // ── Tauri-integrated component tests ───────────────────────
+    generateTauriComponentTests(component, _scan, fw) {
+        const cases = [];
+        const mockFn = fw === "vitest" ? "vi" : "jest";
+        const mockedInvoke = `${mockFn === "vi" ? "vi.mocked" : "jest.mocked"}(invoke)`;
+        const propStr = this.buildDefaultProps(component.props, fw);
+        // Rendering test
+        cases.push(this.buildTestCase("should render without crashing", [
+            `const { container } = render(<${component.name}${propStr} />);`,
+            `expect(container).toBeTruthy();`,
+        ].join("\n")));
+        // Test that Tauri commands are invoked on mount/interaction
+        cases.push(this.buildAsyncTestCase("should invoke Tauri commands", [
+            `${mockedInvoke}.mockResolvedValue({});`,
+            `render(<${component.name}${propStr} />);`,
+            `await waitFor(() => {`,
+            `  expect(invoke).toHaveBeenCalled();`,
+            `});`,
+        ].join("\n")));
+        // Reset mocks
+        cases.push(this.buildTestCase("should clean up on unmount", [
+            `${mockedInvoke}.mockResolvedValue({});`,
+            `const { unmount } = render(<${component.name}${propStr} />);`,
+            `unmount();`,
+            `// Component should unmount without errors`,
+        ].join("\n")));
+        return cases;
+    }
+    // ── Standard React component tests ─────────────────────────
+    generateStandardComponentTests(component, fw) {
+        const cases = [];
+        const propStr = this.buildDefaultProps(component.props, fw);
+        // Basic render
+        cases.push(this.buildTestCase("should render without crashing", [
+            `const { container } = render(<${component.name}${propStr} />);`,
+            `expect(container).toBeTruthy();`,
+        ].join("\n")));
+        // Event handlers
+        const eventProps = component.props.filter((p) => /^on[A-Z]/.test(p.name));
+        for (const prop of eventProps) {
+            const mockName = `mock${prop.name.charAt(0).toUpperCase()}${prop.name.slice(1)}`;
+            const mockInit = fw === "vitest" ? "vi.fn()" : "jest.fn()";
+            cases.push(this.buildTestCase(`should handle ${prop.name}`, [
+                `const ${mockName} = ${mockInit};`,
+                `render(<${component.name}${propStr} ${prop.name}={${mockName}} />);`,
+                `const target = screen.getByRole('button');`,
+                `fireEvent.click(target);`,
+                `expect(${mockName}).toHaveBeenCalled();`,
+            ].join("\n")));
+        }
+        // Conditional rendering for boolean props
+        const boolProps = component.props.filter((p) => p.type?.toLowerCase() === "boolean");
+        for (const prop of boolProps) {
+            cases.push(this.buildTestCase(`should render correctly when ${prop.name} is false`, [
+                `const { container } = render(<${component.name}${propStr} ${prop.name}={false} />);`,
+                `expect(container).toBeTruthy();`,
+            ].join("\n")));
+        }
+        return cases;
+    }
+    // ── Detection helpers ──────────────────────────────────────
+    detectTauriCommands(scan) {
+        return scan.imports.some((m) => m.source === "@tauri-apps/api/core" ||
+            m.source === "@tauri-apps/api" ||
+            m.source.startsWith("@tauri-apps/"));
+    }
+    getTauriCommandFunctions(scan) {
+        // Functions that likely invoke Tauri commands are exported async functions
+        // in files that import from @tauri-apps
+        return scan.functions.filter((f) => f.isExported && f.isAsync);
+    }
+    isTauriIntegrated(_component, scan) {
+        return this.detectTauriCommands(scan);
+    }
+    // ── Utility helpers ────────────────────────────────────────
+    buildCommandPayload(params) {
+        if (params.length === 0)
+            return null;
+        const fields = params
+            .map((p) => `${p.name}: ${this.sampleValueForType(p.type)}`)
+            .join(", ");
+        return `{ ${fields} }`;
+    }
+    buildDefaultProps(props, fw) {
+        const required = props.filter((p) => !p.isOptional);
+        if (required.length === 0)
+            return "";
+        const mockInit = fw === "vitest" ? "vi.fn()" : "jest.fn()";
+        const assignments = required
+            .map((p) => {
+            if (/^on[A-Z]/.test(p.name))
+                return `${p.name}={${mockInit}}`;
+            return `${p.name}={${this.sampleValueForType(p.type)}}`;
+        })
+            .join(" ");
+        return ` ${assignments}`;
+    }
+    toSnakeCase(name) {
+        return name
+            .replace(/([a-z])([A-Z])/g, "$1_$2")
+            .replace(/([A-Z])([A-Z][a-z])/g, "$1_$2")
+            .toLowerCase();
+    }
+    relativeImportPath(from, to) {
+        const rel = relative(dirname(from), to);
+        const withoutExt = rel.replace(/\.(ts|tsx|js|jsx)$/, "");
+        return rel.startsWith(".") ? withoutExt : `./${withoutExt}`;
+    }
+}
+//# sourceMappingURL=desktop-test.js.map