Added test for multiple connections

src/main/java/org/fungover/thunder/Client.java

@@ -1,5 +1,9 @@
 package org.fungover.thunder;
 
+import java.io.IOException;
+import java.io.ObjectInputStream;
+import java.io.ObjectOutputStream;
+import java.net.Socket;
 import java.util.HashSet;
 import java.util.Set;
 
@@ -30,4 +34,12 @@ public void subscribeToTopic(String topic) {
     public Set<String> getSubscribedTopics() {
         return new HashSet<>(subscribedTopics);
     }
+
+    public void connectToServer(String serverAddress, int serverPort) throws IOException {
+        try (Socket socket = new Socket(serverAddress, serverPort);
+             ObjectOutputStream out = new ObjectOutputStream(socket.getOutputStream());
+             ObjectInputStream in = new ObjectInputStream(socket.getInputStream())) {
+            this.connected = true;
+        }
+    }
 }

src/main/java/org/fungover/thunder/ClientHandler.java

@@ -53,5 +53,4 @@ public void removeDisconnectedClients() {
     public List<Socket> getClients() {
         return clients;
     }
-
 }

src/main/java/org/fungover/thunder/Main.java

@@ -6,7 +6,7 @@
 
 public class Main {
 
-    private static final Logger logger = Logger.getLogger(Main.class.getName());
+    static final Logger logger = Logger.getLogger(Main.class.getName());
 
     public static void main(String[] args) {
         Server server;

src/main/java/org/fungover/thunder/Server.java

@@ -3,6 +3,10 @@
 import java.io.IOException;
 import java.net.ServerSocket;
 import java.net.Socket;
+import java.util.logging.Level;
+
+import static org.fungover.thunder.Main.logger;
+
 
 public class Server {
     private final ServerSocket serverSocket;
@@ -26,4 +30,12 @@ public void start() throws IOException {
             e.printStackTrace();
         }
     }
+    public void stop(){
+        try {
+            serverSocket.close();
+            System.out.println("Server stopped");
+        } catch (IOException e) {
+            logger.log(Level.SEVERE, e.getMessage(), e);
+        }
+    }
 }

src/test/java/org/fungover/thunder/ClientHandlerTest.java

@@ -5,13 +5,19 @@
 import org.junit.jupiter.api.Test;
 
 import java.io.*;
+import java.lang.reflect.Field;
 import java.net.InetAddress;
 import java.net.InetSocketAddress;
 import java.net.Socket;
+import java.util.ArrayList;
 import java.util.Collections;
 import java.util.List;
+import java.util.concurrent.Executors;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.ExecutorService;
 
 import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.ArgumentMatchers.any;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 

src/test/java/org/fungover/thunder/ServerTest.java

@@ -0,0 +1,63 @@
+package org.fungover.thunder;
+
+import org.junit.jupiter.api.AfterAll;
+import org.junit.jupiter.api.BeforeAll;
+import org.junit.jupiter.api.Test;
+import static org.junit.jupiter.api.Assertions.assertTrue;
+
+import java.io.IOException;
+import java.util.concurrent.ExecutorService;
+import java.util.concurrent.Executors;
+
+public class ServerTest {
+    private static Server server;
+    @BeforeAll
+    public static void setUp() {
+        try {
+            server = new Server();
+            ExecutorService executorService = Executors.newFixedThreadPool(5);
+            executorService.submit(() -> {
+                try {
+                    server.start();
+                } catch (IOException e) {
+                    e.printStackTrace();
+                }
+            });
+            Thread.sleep(1000);
+        } catch (IOException | InterruptedException e) {
+            e.printStackTrace();
+        }
+    }
+    @AfterAll
+    public static void tearDown() {
+        if (server != null) {
+            server.stop();
+        }
+    }
+    @Test
+    public void testMultipleConnections() {
+        int numClients = 5;
+        ExecutorService executorService = Executors.newFixedThreadPool(numClients);
+
+        for (int i = 0; i < numClients; i++) {
+            final int clientId = i;
+            executorService.submit(() -> {
+                try {
+                    Client client = new Client("Client" + clientId);
+                    client.connectToServer("localhost", 1883);
+                    assertTrue(client.isConnected());
+                } catch (Exception e) {
+                    e.printStackTrace();
+                }
+            });
+        }
+        executorService.shutdown();
+        while (!executorService.isTerminated()) {
+            try {
+                Thread.sleep(100);
+            } catch (InterruptedException e) {
+                e.printStackTrace();
+            }
+        }
+    }
+}