From 8aeb13f44fbeec2d836d1a71e5cc2d4cdb8486b9 Mon Sep 17 00:00:00 2001
From: Joffrey BION <joffrey.bion@gmail.com>
Date: Tue, 17 Jan 2017 00:05:23 +0100
Subject: Add test for LobbyRepository

---
 .../sevenwonders/repositories/LobbyRepository.java       | 16 ++++++++++++----
 1 file changed, 12 insertions(+), 4 deletions(-)

(limited to 'src/main/java/org/luxons/sevenwonders')

diff --git a/src/main/java/org/luxons/sevenwonders/repositories/LobbyRepository.java b/src/main/java/org/luxons/sevenwonders/repositories/LobbyRepository.java
index dce824ad..8f305791 100644
--- a/src/main/java/org/luxons/sevenwonders/repositories/LobbyRepository.java
+++ b/src/main/java/org/luxons/sevenwonders/repositories/LobbyRepository.java
@@ -35,17 +35,25 @@ public class LobbyRepository {
         return lobby;
     }
 
-    public Lobby find(long lobbyId) {
+    public Lobby find(long lobbyId) throws LobbyNotFoundException {
         Lobby lobby = lobbies.get(lobbyId);
         if (lobby == null) {
-            throw new LobbyNotFoundException(String.valueOf(lobbyId));
+            throw new LobbyNotFoundException(lobbyId);
+        }
+        return lobby;
+    }
+
+    public Lobby remove(long lobbyId) throws LobbyNotFoundException {
+        Lobby lobby = lobbies.remove(lobbyId);
+        if (lobby == null) {
+            throw new LobbyNotFoundException(lobbyId);
         }
         return lobby;
     }
 
     public static class LobbyNotFoundException extends RuntimeException {
-        LobbyNotFoundException(String name) {
-            super("Lobby not found for game '" + name + "'");
+        LobbyNotFoundException(long id) {
+            super("Lobby not found for id '" + id + "'");
         }
     }
 }
-- 
cgit