improved error handling

Author: Aad1tya27

setup.ts

@@ -922,11 +922,17 @@ export async function dataImportWithoutDocker(
   importData: () => Promise<void>,
 ): Promise<void> {
   if (!process.env.MONGO_DB_URL) {
-    console.log("Couldn't find mongodb url");
-    return;
+    throw new Error("MongoDB URL is not configured. Please run setup first.");
   }
 
-  const isDbEmpty = await checkDb(process.env.MONGO_DB_URL);
+  let isDbEmpty: boolean;
+  try {
+    isDbEmpty = await checkDb(process.env.MONGO_DB_URL);
+  } catch (error) {
+    throw new Error(
+      `Failed to check database: ${error instanceof Error ? error.message : String(error)}`,
+    );
+  }
   if (!isDbEmpty) {
     const { shouldOverwriteData } = await inquirer.prompt({
       type: "confirm",

tests/setup/dataImportFlow.spec.ts

@@ -198,6 +198,44 @@ describe("Data Importation Without Docker", () => {
     expect(importDefaultDataMock).not.toBeCalled();
     expect(importDataMock).not.toBeCalled();
   });
+
+  it("should handle database connection failure gracefully", async () => {
+    const checkDbMock = vi
+      .fn()
+      .mockImplementation(async (): Promise<boolean> => {
+        return false;
+      });
+    const wipeExistingDataMock = vi
+      .fn()
+      .mockImplementation(async (): Promise<void> => {
+        return Promise.resolve();
+      });
+    const importDataMock = vi
+      .fn()
+      .mockImplementation(async (): Promise<void> => {
+        return Promise.resolve();
+      });
+    const importDefaultDataMock = vi
+      .fn()
+      .mockImplementation(async (): Promise<void> => {
+        return Promise.resolve();
+      });
+    const errorMessage = "Database connection failed";
+    checkDbMock.mockRejectedValueOnce(new Error(errorMessage));
+
+    await expect(
+      dataImportWithoutDocker(
+        checkDbMock,
+        wipeExistingDataMock,
+        importDefaultDataMock,
+        importDataMock,
+      ),
+    ).rejects.toThrow(errorMessage);
+
+    expect(wipeExistingDataMock).not.toBeCalled();
+    expect(importDefaultDataMock).not.toBeCalled();
+    expect(importDataMock).not.toBeCalled();
+  });
 });
 
 describe("Data Importation With Docker", () => {