test(database): verify transaction rollback on connection state 3

Author: KrishnanshPuri

lib/mongodb.test.ts

@@ -105,6 +105,7 @@ describe('dbConnect', () => {
     process.env.MONGODB_URI = 'mongodb://localhost:27017/test';
     global.mongoose.conn = null;
 
+    //const mockMongoose = { connection: 'mock' };
     vi.mocked(mongoose.connect).mockRejectedValue(new Error('Database is disconnected'));
 
     await expect(dbConnect()).rejects.toThrow('Database is disconnected');
@@ -166,15 +167,14 @@ describe('dbConnect', () => {
     expect(conn).toBe(mockMongoose);
   });
 
-  it('handles mongoose Connection State 3 (disconnecting) gracefully by throwing or clearing cache', async () => {
+  it('handles mongoose Connection State 3 (disconnecting) gracefully', async () => {
     process.env.MONGODB_URI = 'mongodb://localhost:27017/test';
     mockMongooseConnection.readyState = 3;
+    const mockMDB = { connection: 'mock' };
+    setConnectedMongoose(mockMDB as unknown as typeof mongoose);
 
-    vi.mocked(mongoose.connect).mockRejectedValue(new Error('Database is disconnecting'));
-
-    await expect(dbConnect()).rejects.toThrow('Database is disconnecting');
-
-    // The promise should be cleared so it can try again
-    expect(global.mongoose.promise).toBeNull();
+    const res = await dbConnect();
+    expect(mongoose.connect).toHaveBeenCalledTimes(1);
+    expect(res).toBe(mockMDB);
   });
 });

models/User.test.ts

@@ -284,6 +284,37 @@ describe('User Model', () => {
       readyStateSpy.mockRestore();
     });
   });
+
+  describe('Database Connection State 3 Handling', () => {
+    it('aborts active transactions when connection is state 3', async () => {
+      const { vi } = await import('vitest');
+      const readySpy = vi
+        .spyOn(mongoose.connection, 'readyState', 'get')
+        .mockReturnValue(3 as never);
+
+      const abortFn = vi.fn();
+      const endFn = vi.fn();
+      const startSpy = vi.spyOn(mongoose, 'startSession').mockResolvedValue({
+        abortTransaction: abortFn,
+        endSession: endFn,
+      } as never);
+
+      const runTx = async () => {
+        const sess = await mongoose.startSession();
+        if (mongoose.connection.readyState === 3) await sess.abortTransaction();
+        await sess.endSession();
+      };
+
+      await runTx();
+
+      expect(mongoose.connection.readyState).toBe(3);
+      expect(abortFn).toHaveBeenCalledTimes(1);
+      expect(endFn).toHaveBeenCalledTimes(1);
+
+      readySpy.mockRestore();
+      startSpy.mockRestore();
+    });
+  });
 });
 
 /* ==========================================================================