Test: 테스트 코드 작성

Author: jueunk617

src/main/java/com/back/domain/chat/controller/ChatApiController.java

@@ -12,7 +12,6 @@
 import org.springframework.web.bind.annotation.*;
 
 import java.time.LocalDateTime;
-import java.util.Map;
 
 @RestController
 @RequiredArgsConstructor
@@ -26,15 +25,10 @@ public class ChatApiController {
     public ResponseEntity<RsData<ChatPageResponse>> getRoomChatMessages(
             @PathVariable Long roomId,
             @RequestParam(defaultValue = "0") int page,
-            @RequestParam(defaultValue = "50") int size,
+            @RequestParam(defaultValue = "20") int size,
             @RequestParam(required = false) @DateTimeFormat(iso = DateTimeFormat.ISO.DATE_TIME) LocalDateTime before,
             @AuthenticationPrincipal CustomUserDetails userDetails) {
 
-        // size 최대값 제한 (임시: max 100)
-        if (size > 100) {
-            size = 100;
-        }
-
         ChatPageResponse chatHistory = chatService.getRoomChatHistory(roomId, page, size, before);
 
         return ResponseEntity

src/main/java/com/back/domain/chat/dto/ChatPageResponse.java

@@ -15,6 +15,7 @@ public class ChatPageResponse {
 
     private List<ChatMessageDto> content;
     private PageableDto pageable;
+    private long totalElements;
 
     // 페이징 정보 DTO
     @Data
@@ -36,6 +37,7 @@ public static ChatPageResponse from(org.springframework.data.domain.Page<ChatMes
                         .size(page.getSize())
                         .hasNext(page.hasNext())
                         .build())
+                .totalElements(page.getTotalElements())
                 .build();
     }
 }

src/main/java/com/back/domain/chat/service/ChatService.java

@@ -10,6 +10,7 @@
 import com.back.domain.chat.dto.ChatPageResponse;
 import com.back.global.exception.CustomException;
 import com.back.global.exception.ErrorCode;
+import com.back.global.security.CurrentUser;
 import lombok.RequiredArgsConstructor;
 import org.springframework.data.domain.Page;
 import org.springframework.data.domain.PageRequest;
@@ -27,6 +28,11 @@ public class ChatService {
     private final RoomChatMessageRepository roomChatMessageRepository;
     private final RoomRepository roomRepository;
     private final UserRepository userRepository;
+    private final CurrentUser currentUser;
+
+    // 페이징 설정 상수
+    private static final int DEFAULT_PAGE_SIZE = 20;
+    private static final int MAX_PAGE_SIZE = 100;
 
     // 방 채팅 메시지 저장
     @Transactional
@@ -54,7 +60,10 @@ public ChatPageResponse getRoomChatHistory(Long roomId, int page, int size, Loca
         roomRepository.findById(roomId)
                 .orElseThrow(() -> new CustomException(ErrorCode.ROOM_NOT_FOUND));
 
-        Pageable pageable = PageRequest.of(page, size);
+        // size 값 검증 및 제한
+        int validatedSize = validateAndLimitPageSize(size);
+
+        Pageable pageable = PageRequest.of(page, validatedSize);
 
         // before 파라미터가 있으면 해당 시점 이전 메시지만 조회
         Page<RoomChatMessage> messagesPage;
@@ -69,6 +78,14 @@ public ChatPageResponse getRoomChatHistory(Long roomId, int page, int size, Loca
         return ChatPageResponse.from(dtoPage);
     }
 
+    // size 값 검증 및 최대값 제한
+    private int validateAndLimitPageSize(int size) {
+        if (size <= 0) {
+            return DEFAULT_PAGE_SIZE; // 0 이하면 기본값 사용
+        }
+        return Math.min(size, MAX_PAGE_SIZE); // 최대값 제한
+    }
+
     // 메시지 엔티티를 DTO로 변환
     private ChatMessageDto convertToDto(RoomChatMessage message) {
         return ChatMessageDto.builder()
@@ -84,4 +101,4 @@ private ChatMessageDto convertToDto(RoomChatMessage message) {
                 .build();
     }
 
-}
+}

src/main/java/com/back/domain/studyroom/entity/Room.java

@@ -4,15 +4,19 @@
 import com.back.domain.user.entity.User;
 import com.back.global.entity.BaseEntity;
 import jakarta.persistence.*;
+import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
 
 import java.util.ArrayList;
 import java.util.List;
 
 @Entity
-@NoArgsConstructor
 @Getter
+@SuperBuilder
+@NoArgsConstructor
+@AllArgsConstructor
 public class Room extends BaseEntity {
     private String title;
 

src/main/java/com/back/domain/studyroom/entity/RoomChatMessage.java

@@ -6,12 +6,17 @@
 import jakarta.persistence.FetchType;
 import jakarta.persistence.JoinColumn;
 import jakarta.persistence.ManyToOne;
+import lombok.AllArgsConstructor;
+import lombok.Builder;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
 
 @Entity
 @Getter
+@SuperBuilder
 @NoArgsConstructor
+@AllArgsConstructor
 public class RoomChatMessage extends BaseEntity {
     @ManyToOne(fetch = FetchType.LAZY)
     @JoinColumn(name = "room_id")
@@ -23,10 +28,4 @@ public class RoomChatMessage extends BaseEntity {
 
     private String content;
 
-    // 채팅 메세지 생성자
-    public RoomChatMessage(Room room, User user, String content) {
-        this.room = room;
-        this.user = user;
-        this.content = content;
-    }
 }

src/main/java/com/back/domain/user/entity/User.java

@@ -9,16 +9,20 @@
 import com.back.domain.studyroom.entity.RoomParticipantHistory;
 import com.back.global.entity.BaseEntity;
 import jakarta.persistence.*;
+import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
 
 import java.util.ArrayList;
 import java.util.List;
 
 @Entity
 @Getter
-@Table(name = "USERS")
+@SuperBuilder
 @NoArgsConstructor
+@AllArgsConstructor
+@Table(name = "USERS")
 public class User extends BaseEntity {
     private String username;
 

src/main/java/com/back/global/entity/BaseEntity.java

@@ -2,15 +2,19 @@
 
 import jakarta.persistence.*;
 import lombok.Getter;
+import lombok.NoArgsConstructor;
+import lombok.experimental.SuperBuilder;
 import org.springframework.data.annotation.CreatedDate;
 import org.springframework.data.annotation.LastModifiedDate;
 import org.springframework.data.jpa.domain.support.AuditingEntityListener;
 
 import java.time.LocalDateTime;
 
+@Getter
+@SuperBuilder
 @MappedSuperclass
+@NoArgsConstructor
 @EntityListeners(AuditingEntityListener.class)
-@Getter
 public abstract class BaseEntity {
     @Id @GeneratedValue(strategy = GenerationType.IDENTITY)
     private Long id;

src/main/java/com/back/global/security/CustomUserDetails.java

@@ -1,6 +1,7 @@
 package com.back.global.security;
 
 import lombok.AllArgsConstructor;
+import lombok.Builder;
 import lombok.Getter;
 import org.springframework.security.core.authority.SimpleGrantedAuthority;
 import org.springframework.security.core.userdetails.UserDetails;
@@ -13,6 +14,7 @@
  * - JWT에서 파싱한 사용자 정보를 담고 있음
  */
 @Getter
+@Builder
 @AllArgsConstructor
 public class CustomUserDetails implements UserDetails {
     private Long userId;

src/test/java/com/back/domain/chat/controller/ChatApiControllerTest.java

@@ -0,0 +1,73 @@
+package com.back.domain.chat.controller;
+
+import com.back.domain.chat.dto.ChatPageResponse;
+import com.back.domain.chat.service.ChatService;
+import com.back.global.security.JwtTokenProvider;
+import org.junit.jupiter.api.DisplayName;
+import org.junit.jupiter.api.Test;
+
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.test.autoconfigure.web.servlet.AutoConfigureMockMvc;
+import org.springframework.boot.test.context.SpringBootTest;
+
+import org.springframework.http.MediaType;
+import org.springframework.security.authentication.UsernamePasswordAuthenticationToken;
+import org.springframework.test.context.bean.override.mockito.MockitoBean;
+import org.springframework.test.web.servlet.MockMvc;
+
+import static org.mockito.ArgumentMatchers.*;
+import static org.mockito.BDDMockito.given;
+
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+import java.util.List;
+
+@SpringBootTest
+@AutoConfigureMockMvc
+class ChatApiControllerTest {
+
+    @Autowired
+    private MockMvc mockMvc;
+
+    @MockitoBean
+    private JwtTokenProvider jwtTokenProvider;
+
+    @MockitoBean
+    private ChatService chatService;
+
+    @Test
+    @DisplayName("채팅 기록 조회 성공")
+    void t1() throws Exception {
+        ChatPageResponse mockResponse = ChatPageResponse.builder()
+                .content(List.of())
+                .pageable(ChatPageResponse.PageableDto.builder()
+                        .page(0)
+                        .size(20)
+                        .hasNext(false)
+                        .build())
+                .totalElements(0)
+                .build();
+
+        given(chatService.getRoomChatHistory(anyLong(), anyInt(), anyInt(), any()))
+                .willReturn(mockResponse);
+
+        // JWT 관련 스텁
+        given(jwtTokenProvider.validateToken(anyString())).willReturn(true);
+        given(jwtTokenProvider.getAuthentication(anyString()))
+                .willReturn(new UsernamePasswordAuthenticationToken(
+                        "mockUser", null, List.of())
+                );
+
+        mockMvc.perform(get("/api/rooms/1/messages")
+                        .param("page", "0")
+                        .param("size", "20")
+                        .header("Authorization", "Bearer faketoken") // 가짜 토큰 넣기
+                        .accept(MediaType.APPLICATION_JSON))
+                .andExpect(status().isOk())
+                .andExpect(jsonPath("$.data.content").isArray())
+                .andExpect(jsonPath("$.data.totalElements").value(0));
+    }
+}
+