diff --git a/src/main/java/ch/unibas/dmi/dbis/cs108/casono/server/app/commands/logout/LogoutHandler.java b/src/main/java/ch/unibas/dmi/dbis/cs108/casono/server/app/commands/logout/LogoutHandler.java index af08e0c..c8e00b6 100644 --- a/src/main/java/ch/unibas/dmi/dbis/cs108/casono/server/app/commands/logout/LogoutHandler.java +++ b/src/main/java/ch/unibas/dmi/dbis/cs108/casono/server/app/commands/logout/LogoutHandler.java @@ -16,7 +16,15 @@ public class LogoutHandler implements CommandHandler { @Override public void execute(LogoutRequest request) { - userRegistry.removeBySessionId(request.getSessionId()); - responseDispatcher.dispatch(new OkResponse(request.getContext())); + boolean was_removed = userRegistry.removeBySessionId(request.getSessionId()); + if (was_removed) { + responseDispatcher.dispatch(new OkResponse(request.getContext())); + } else { + responseDispatcher.dispatch( + new ErrorResponse( + request.getContext(), + "NO_USER_ASSOCIATED", + "No user is associated with your session. Did you login before?")); + } } }