JUnit test for GameEndService

source/core/src/main/com/csse3200/game/services/GameEndService.java

@@ -4,6 +4,8 @@ public class GameEndService {
 
     private int engineerCount;
 
+    private boolean gameOver = false;
+
     public GameEndService() {
         this.engineerCount = 5;
     }
@@ -16,7 +18,11 @@ public void updateEngineerCount() {
         engineerCount -= 1;
         if (engineerCount == 0) {
             // loss screen
-
+            gameOver = true;
         }
     }
+
+    public boolean hasGameEnded() {
+        return gameOver;
+    }
 }

source/core/src/test/com/csse3200/game/services/GameEndServiceTest.java

@@ -0,0 +1,40 @@
+package com.csse3200.game.services;
+
+import com.csse3200.game.extensions.GameExtension;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+@ExtendWith(GameExtension.class)
+class GameEndServiceTest {
+
+    GameEndService endService;
+
+    @BeforeEach
+    void setUp() {
+        endService = new GameEndService();
+        ServiceLocator.registerGameEndService(endService);
+    }
+
+    @Test
+    void shouldReturnCount() {
+        assertEquals(5, ServiceLocator.getGameEndService().getEngineerCount());
+    }
+
+    @Test
+    void shouldDecrementCount() {
+        ServiceLocator.getGameEndService().updateEngineerCount();
+        assertEquals(4, ServiceLocator.getGameEndService().getEngineerCount());
+    }
+
+    @Test
+    void shouldEndGame() {
+        for (int i = 0; i < 5; i++) {
+            ServiceLocator.getGameEndService().updateEngineerCount();
+        }
+        assertTrue(ServiceLocator.getGameEndService().hasGameEnded());
+    }
+}