relocate some test-classes

Author: guFalcon

src/main/java/info/unterrainer/websocketserver/AiComm.java renamed to src/test/java/info/unterrainer/websocketserver/AiComm.java

@@ -3,6 +3,7 @@
 import java.util.Set;
 import java.util.concurrent.ConcurrentHashMap;
 
+import io.javalin.websocket.WsCloseContext;
 import io.javalin.websocket.WsConnectContext;
 import io.javalin.websocket.WsMessageContext;
 import lombok.extern.slf4j.Slf4j;
@@ -19,6 +20,11 @@ public void onConnect(WsConnectContext ctx) throws Exception {
 		ctx.send("Welcome to our websocket-server!");
 	}
 
+	@Override
+	public void onClose(WsCloseContext ctx) {
+		connectedWsClients.removeIf(client -> client.session.equals(ctx.session));
+	}
+
 	@Override
 	public void onMessage(WsMessageContext ctx) throws Exception {
 		super.onMessage(ctx);