test(mongodb): verify User schema behaviors under connection state 0

Author: TanmayNelge

lib/mongodb.test.ts

@@ -105,7 +105,6 @@ 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');

models/User.test.ts

@@ -93,4 +93,37 @@ describe('User Model', () => {
       readyStateSpy.mockRestore();
     });
   });
+
+  describe('Database Connection State 0 Handling', () => {
+    it('fails queries gracefully with a ConnectionError when disconnected', async () => {
+      // Import vi locally to match the pattern used in the State 2 test
+      const { vi } = await import('vitest');
+
+      // 1. Force the mongoose connection state to 0 (Disconnected)
+      const readyStateSpy = vi
+        .spyOn(mongoose.connection, 'readyState', 'get')
+        .mockReturnValue(0 as unknown as typeof mongoose.connection.readyState);
+
+      expect(mongoose.connection.readyState).toBe(0);
+
+      // 2. Mock a database operation to simulate a connection failure
+      // Mongoose throws specific errors when bufferCommands is false and state is 0
+      const mockConnectionError = new Error('Database connection lost');
+      mockConnectionError.name = 'ConnectionError';
+
+      const findOneSpy = vi.spyOn(User, 'findOne').mockRejectedValue(mockConnectionError);
+
+      // 3. Verify that attempting to query throws the expected ConnectionError
+      await expect(User.findOne({ username: 'testuser' })).rejects.toThrow(
+        'Database connection lost'
+      );
+      await expect(User.findOne({ username: 'testuser' })).rejects.toMatchObject({
+        name: 'ConnectionError',
+      });
+
+      // 4. Clean up mocks to prevent side effects in other tests
+      readyStateSpy.mockRestore();
+      findOneSpy.mockRestore();
+    });
+  });
 });

package-lock.json

@@ -32,6 +32,7 @@
     "gsap": "^3.15.0",
     "html-to-image": "^1.11.13",
     "lucide-react": "^1.17.0",
+    "mongodb": "^7.2.0",
     "mongoose": "^9.6.2",
     "next": "^16.2.3",
     "react": "19.2.4",