Add test for shutDownClientIfNotAuthenticated in main class

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

@@ -1,8 +1,6 @@
 package org.fungover.haze;
-
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
-
 import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStreamReader;
@@ -12,22 +10,17 @@
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.List;
-import java.util.Optional;
-
 public class Main {
     static boolean serverOpen = true;
     static Logger logger = LogManager.getLogger(Main.class);
 
     public static void main(String[] args) {
-
         Initialize initialize = Initialize.getInitialize(args);
-
         HazeDatabase hazeDatabase = new HazeDatabase();
         HazeList hazeList = new HazeList(hazeDatabase);
         Auth auth = new Auth();
         initializeServer(args, initialize, auth);
         final boolean isPasswordSet = auth.isPasswordSet();
-
         addHook(hazeDatabase);
         try (ServerSocket serverSocket = new ServerSocket()) {
             initSocket(initialize, serverSocket);

src/test/java/org/fungover/haze/MainTest.java

@@ -1,4 +1,5 @@
 package org.fungover.haze;
+import org.junit.jupiter.api.AfterEach;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.DisplayName;
 import org.junit.jupiter.api.Test;
@@ -9,14 +10,30 @@
 import org.mockito.MockitoAnnotations;
 
 import java.io.BufferedReader;
+import java.io.IOException;
 import java.io.StringReader;
+import java.net.ServerSocket;
+import java.net.Socket;
 import java.util.List;
 import static org.assertj.core.api.Assertions.assertThat;
 import static org.mockito.Mockito.*;
 
 class MainTest {
 
+    private ServerSocket serverSocket;
+    private Socket clientSocket;
 
+    @BeforeEach
+    void setUp() throws IOException {
+        serverSocket = new ServerSocket(0);
+        clientSocket = new Socket("localhost", serverSocket.getLocalPort());
+    }
+
+    @AfterEach
+    void tearDown() throws IOException {
+        serverSocket.close();
+        clientSocket.close();
+    }
 
     HazeDatabase database = new HazeDatabase();
     HazeList hazeList = new HazeList(database);
@@ -151,6 +168,16 @@ void callAuthCommandReceivedWithInvalidInputShouldReturnFalse() {
         assertThat(result).isFalse();
     }
 
+    @Test
+    @DisplayName("Call AuthCommandReceived With Invalid Password Should Return False")
+    void callAuthCommandReceivedWithInvalidPasswordShouldReturnFalse() {
+        boolean isPasswordSet = false;
+        boolean clientAuthenticated = false;
+        List<String> inputList = List.of("AUTH", "password");
+        boolean result = Main.authCommandReceived(isPasswordSet, inputList, clientAuthenticated);
+        assertThat(result).isFalse();
+    }
+
     @Test
     void callToInitializeServerShouldSetPasswordToAuth(){
         String[] testArgs = {"arg1", "arg2"};
@@ -160,6 +187,26 @@ void callToInitializeServerShouldSetPasswordToAuth(){
         assertThat(auth.isPasswordSet()).isTrue();
     }
 
+    @Test
+    @DisplayName("Call to shutdownClientIfNotAuthenticated should shut down output for non authenticated user")
+    void callToShutDownClientIfNotAuthenticatedShouldShutDownOutputForNonAuthenticatedUser() throws IOException {
+        boolean clientAuthenticated = false;
+        boolean isPasswordSet = true;
+        Main.shutdownClientIfNotAuthenticated(clientSocket, clientAuthenticated, isPasswordSet);
+        assertThat(clientSocket.isOutputShutdown()).isEqualTo(true);
+    }
+
+    @Test
+    @DisplayName("Call to shutdownClientIfNotAuthenticated should not shut down output for authenticated user")
+    void callToShutDownClientIfNotAuthenticatedShouldNotShutDownOutputForAuthenticatedUser() throws IOException {
+        boolean clientAuthenticated = true;
+        boolean isPasswordSet = true;
+        Main.shutdownClientIfNotAuthenticated(clientSocket, clientAuthenticated, isPasswordSet);
+        assertThat(clientSocket.isOutputShutdown()).isEqualTo(false);
+    }
+
+
+