Add tests for askQuestion function and refactor main function for better error handling

Author: mayank1513

lib/src/index.test.ts

@@ -0,0 +1,46 @@
+import { describe, it, expect, vi } from "vitest";
+import readline from "readline";
+import { execSync } from "child_process";
+import { askQuestion } from ".";
+
+vi.mock("child_process");
+
+describe("askQuestion", () => {
+  it("resolves with user input", async () => {
+    const mockInterface = {
+      question: vi.fn((_q: string, cb: (ans: string) => void) => cb("my-app")),
+      close: vi.fn(),
+    } as unknown as readline.Interface;
+
+    vi.spyOn(readline, "createInterface").mockReturnValue(mockInterface);
+
+    const answer = await askQuestion("Project name?");
+    expect(answer).toBe("my-app");
+  });
+
+  it("resolves with empty string if input is empty", async () => {
+    const mockInterface = {
+      question: vi.fn((_q: string, cb: (ans: string) => void) => cb("")),
+      close: vi.fn(),
+    } as unknown as readline.Interface;
+
+    vi.spyOn(readline, "createInterface").mockReturnValue(mockInterface);
+
+    const answer = await askQuestion("Project name?");
+    expect(answer).toBe("");
+  });
+});
+
+describe("error handling", () => {
+  it("handles execSync errors gracefully", () => {
+    (execSync as unknown as ReturnType<typeof vi.fn>).mockImplementation(() => {
+      throw new Error("mock failure");
+    });
+
+    try {
+      execSync("pnpm dlx create-turbo");
+    } catch (err: any) {
+      expect(err.message).toContain("mock failure");
+    }
+  });
+});

lib/src/index.ts

@@ -20,7 +20,7 @@ import readline from "readline";
 
 const DEFAULT_NAME = "react18-tools-turborepo-template";
 
-function askQuestion(query: string): Promise<string> {
+export const askQuestion = (query: string): Promise<string> => {
   const rl = readline.createInterface({
     input: process.stdin,
     output: process.stdout,
@@ -31,9 +31,9 @@ function askQuestion(query: string): Promise<string> {
       resolve(ans);
     }),
   );
-}
+};
 
-async function main(): Promise<void> {
+const main = async (): Promise<void> => {
   let projectName = process.argv[2];
 
   if (!projectName) {
@@ -51,13 +51,13 @@ async function main(): Promise<void> {
     execSync(cmd, { stdio: "inherit" });
   } catch (err: any) {
     console.error("❌ Error while running create-turbo:", err.message);
-    process.exit(1);
+    return;
   }
 
   // Step 2: Verify directory exists
   if (!existsSync(projectDir)) {
     console.error(`❌ Expected directory not found: ${projectDir}`);
-    process.exit(1);
+    return;
   }
 
   // Step 3: Run rebrand
@@ -69,10 +69,10 @@ async function main(): Promise<void> {
     });
   } catch (err: any) {
     console.error("❌ Rebrand step failed:", err.message);
-    process.exit(1);
+    return;
   }
 
   console.log(`\n✅ Done! Your project is ready in: ${projectDir}\n`);
-}
+};
 
 main();