Update SubLobbyTest.java

utastudents · May 2, 2024 · 85523e1 · 85523e1
1 parent bdda5b2
commit 85523e1
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/test/java/uta/cse3310/SubLobbyTest.java b/src/test/java/uta/cse3310/SubLobbyTest.java
@@ -81,4 +81,20 @@ public void testAllPlayersReady() {
         player1.setReady(true);
         assertTrue(subLobby.allPlayersReady());
     }
+
+    // Test if players in a lobby are assigned a color
+    public void testReassignColors() {
+
+        subLobby.addPlayer(player2);
+        subLobby.addPlayer(player3);
+        subLobby.addPlayer(player4);
+
+        subLobby.reassignColors(subLobby);
+
+        // Verify that each player has been assigned a color
+        assertEquals("blue", player1.getColor());
+        assertEquals("pink", player2.getColor());
+        assertEquals("green", player3.getColor());
+        assertEquals("orange", player4.getColor());
+    }
 }