From 0b4bd8a04a35e79b59114cbb6609bf8918ac60fc Mon Sep 17 00:00:00 2001
From: Bruno Roesner <bruno.roesner.1@ens.etsmtl.ca>
Date: Tue, 3 Dec 2024 14:51:23 -0500
Subject: [PATCH] modified insert room test

---
 server/__tests__/room.test.js | 28 ++++++++++++++++++++++------
 1 file changed, 22 insertions(+), 6 deletions(-)

diff --git a/server/__tests__/room.test.js b/server/__tests__/room.test.js
index a83ba00..801b092 100644
--- a/server/__tests__/room.test.js
+++ b/server/__tests__/room.test.js
@@ -50,9 +50,20 @@ describe('Rooms', () => {
         const roomID = '123456';
         const roomName = 'room123456';
         const hostname = 'localhost';
+        const mongoObjectID = new ObjectId();
 
-        collection.findOne.mockResolvedValue(null);
-        collection.insertOne.mockResolvedValue({ insertedId: new ObjectId() });
+        // Mock first findOne to simulate no entry
+        collection.findOne.mockResolvedValueOnce(null);
+
+        //Mock second findOne to simulate new room
+        collection.findOne.mockResolvedValueOnce({
+            _id: mongoObjectID,
+            id: roomID,
+            name: roomName,
+            host: hostname,
+        });
+
+        collection.insertOne.mockResolvedValue({ insertedId: mongoObjectID });
 
         const room = new Room(roomID, roomName, hostname);
 
@@ -60,16 +71,21 @@ describe('Rooms', () => {
 
         expect(db.connect).toHaveBeenCalled();
         expect(db.getConnection).toHaveBeenCalled();
-        expect(collection.findOne).toHaveBeenCalled();
-        expect(result).toBeDefined();
+        expect(collection.findOne).toHaveBeenCalledWith({id: '123456'});
     });
 
     it('should throw an error if a room with same ID exists', async () => {
         const roomID = '123456';
-        const roomName = 'room123456';
+        const roomName = 'otherRoom';
         const hostname = 'localhost';
+        const mongoObjectID = new ObjectId();
 
-        collection.findOne.mockResolvedValue({ id: roomID });
+        collection.findOne.mockResolvedValueOnce({
+            _id: mongoObjectID,
+            id: roomID,
+            name: roomName,
+            host: hostname,
+        });
 
         const room = new Room(roomID, roomName, hostname);