1 files changed, 10 insertions, 20 deletions

diff --git a/ufund-api/src/test/java/com/ufund/api/ufundapi/controller/AuthControllerTest.java b/ufund-api/src/test/java/com/ufund/api/ufundapi/controller/AuthControllerTest.java
index 85af481..3d4637d 100644
--- a/ufund-api/src/test/java/com/ufund/api/ufundapi/controller/AuthControllerTest.java
+++ b/ufund-api/src/test/java/com/ufund/api/ufundapi/controller/AuthControllerTest.java
@@ -8,16 +8,16 @@ import static org.junit.jupiter.api.Assertions.assertEquals;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import static org.mockito.ArgumentMatchers.any;
+import org.mockito.Mockito;
+import static org.mockito.Mockito.doThrow;
 import static org.mockito.Mockito.mock;
 import static org.mockito.Mockito.when;
 import org.springframework.http.HttpStatus;
 import org.springframework.http.ResponseEntity;
 import org.springframework.web.bind.annotation.RequestMapping;
-import org.springframework.web.bind.annotation.RestController;
 
 import com.ufund.api.ufundapi.service.AuthService;
 
-@RestController
 @RequestMapping("auth")
 public class AuthControllerTest {
 
@@ -36,7 +36,7 @@ public class AuthControllerTest {
     }
 
     @Test
-    private void testLogin() throws IllegalAccessException, IOException {
+    public void testLogin() throws IllegalAccessException, IOException {
         // Setup
         String key = "123";
 
@@ -52,10 +52,7 @@ public class AuthControllerTest {
     }
 
     @Test
-    private void testLoginUnauthorized() throws IllegalAccessException, IOException {
-        // Setup
-        String key = "123";
-
+    public void testLoginUnauthorized() throws IllegalAccessException, IOException {
         // Mock
         when(mockAuthService.login(any(), any())).thenThrow(IllegalAccessException.class);
 
@@ -67,10 +64,7 @@ public class AuthControllerTest {
     }
 
     @Test
-    private void testLoginIOException() throws IllegalAccessException, IOException {
-        // Setup
-        String key = "123";
-
+    public void testLoginIOException() throws IllegalAccessException, IOException {
         // Mock
         when(mockAuthService.login(any(), any())).thenThrow(IOException.class);
 
@@ -82,31 +76,27 @@ public class AuthControllerTest {
     }
 
     @Test
-    void testLogout() throws IllegalAccessException, IOException {
+    public void testLogout() throws IllegalAccessException, IOException {
         // Setup
         String key = "123";
 
-        // Mock
-        when(mockAuthService.login(any(), any())).thenReturn(key);
-
         // Invoke
-        ResponseEntity<String> response = authController.login(authMap);
+        ResponseEntity<Object> response = authController.logout(key);
 
         // Analyze
         assertEquals(HttpStatus.OK, response.getStatusCode());
-        assertEquals(key, response.getBody());
     }
 
     @Test
-    void testLogoutIOException() throws IllegalAccessException, IOException {
+    public void testLogoutIOException() throws IllegalAccessException, IOException {
         // Setup
         String key = "123";
 
         // Mock
-        when(mockAuthService.login(any(), any())).thenThrow(IOException.class);
+        doThrow(new IOException()).when(mockAuthService).logout(key);
 
         // Invoke
-        ResponseEntity<String> response = authController.login(authMap);
+        ResponseEntity<Object> response = authController.logout(key);
 
         // Analyze
         assertEquals(HttpStatus.INTERNAL_SERVER_ERROR, response.getStatusCode());