test: FCM, File, Friend, FriendRequest, Like, Member 관련 테스트 코드 작성

kakao-tech-campus-2nd-step3 · Nov 6, 2024 · c859e7a · c859e7a
1 parent bfd58ca
commit c859e7a
Show file tree

Hide file tree

Showing 28 changed files with 3,394 additions and 127 deletions.
diff --git a/src/main/java/com/potatocake/everymoment/dto/request/FcmTokenRequest.java b/src/main/java/com/potatocake/everymoment/dto/request/FcmTokenRequest.java
@@ -2,8 +2,12 @@
 
 import jakarta.validation.constraints.NotBlank;
 import jakarta.validation.constraints.Size;
+import lombok.AllArgsConstructor;
 import lombok.Getter;
+import lombok.NoArgsConstructor;
 
+@NoArgsConstructor
+@AllArgsConstructor
 @Getter
 public class FcmTokenRequest {
 

diff --git a/src/main/java/com/potatocake/everymoment/entity/Like.java b/src/main/java/com/potatocake/everymoment/entity/Like.java
@@ -9,10 +9,12 @@
 import jakarta.persistence.Table;
 import lombok.AccessLevel;
 import lombok.Builder;
+import lombok.Getter;
 import lombok.NoArgsConstructor;
 
 @Table(name = "likes")
 @NoArgsConstructor(access = AccessLevel.PROTECTED)
+@Getter
 @Entity
 public class Like {
 

diff --git a/src/test/java/com/potatocake/everymoment/controller/FcmControllerTest.java b/src/test/java/com/potatocake/everymoment/controller/FcmControllerTest.java
@@ -0,0 +1,126 @@
+package com.potatocake.everymoment.controller;
+
+import static org.mockito.BDDMockito.then;
+import static org.mockito.BDDMockito.willDoNothing;
+import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.csrf;
+import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.user;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.delete;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.post;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+import com.fasterxml.jackson.databind.ObjectMapper;
+import com.potatocake.everymoment.dto.request.FcmTokenRequest;
+import com.potatocake.everymoment.entity.Member;
+import com.potatocake.everymoment.security.MemberDetails;
+import com.potatocake.everymoment.service.FcmService;
+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.WebMvcTest;
+import org.springframework.boot.test.mock.mockito.MockBean;
+import org.springframework.http.MediaType;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.ResultActions;
+
+@WebMvcTest(FcmController.class)
+class FcmControllerTest {
+
+    @Autowired
+    private MockMvc mockMvc;
+
+    @Autowired
+    private ObjectMapper objectMapper;
+
+    @MockBean
+    private FcmService fcmService;
+
+    @Test
+    @DisplayName("FCM 토큰이 성공적으로 등록된다.")
+    void should_RegisterToken_When_ValidInput() throws Exception {
+        // given
+        Long memberId = 1L;
+        Member member = Member.builder()
+                .id(memberId)
+                .number(1234L)
+                .nickname("testUser")
+                .build();
+        MemberDetails memberDetails = new MemberDetails(member);
+
+        FcmTokenRequest request = new FcmTokenRequest("fcm-token-123", "device123");
+
+        willDoNothing().given(fcmService).registerToken(memberId, "device123", "fcm-token-123");
+
+        // when
+        ResultActions result = mockMvc.perform(post("/api/fcm/token")
+                .with(user(memberDetails))
+                .with(csrf())
+                .contentType(MediaType.APPLICATION_JSON)
+                .content(objectMapper.writeValueAsString(request)));
+
+        // then
+        result.andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value(200))
+                .andExpect(jsonPath("$.message").value("success"));
+
+        then(fcmService).should().registerToken(memberId, "device123", "fcm-token-123");
+    }
+
+    @Test
+    @DisplayName("FCM 토큰이 성공적으로 삭제된다.")
+    void should_RemoveToken_When_ValidInput() throws Exception {
+        // given
+        Long memberId = 1L;
+        Member member = Member.builder()
+                .id(memberId)
+                .number(1234L)
+                .nickname("testUser")
+                .build();
+        MemberDetails memberDetails = new MemberDetails(member);
+        String deviceId = "device123";
+
+        willDoNothing().given(fcmService).removeToken(memberId, deviceId);
+
+        // when
+        ResultActions result = mockMvc.perform(delete("/api/fcm/token")
+                .with(user(memberDetails))
+                .with(csrf())
+                .param("deviceId", deviceId));
+
+        // then
+        result.andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value(200))
+                .andExpect(jsonPath("$.message").value("success"));
+
+        then(fcmService).should().removeToken(memberId, deviceId);
+    }
+
+    @Test
+    @DisplayName("FCM 토큰 요청 데이터가 누락되면 등록에 실패한다.")
+    void should_FailToRegister_When_InvalidInput() throws Exception {
+        // given
+        Long memberId = 1L;
+        Member member = Member.builder()
+                .id(memberId)
+                .number(1234L)
+                .nickname("testUser")
+                .build();
+        MemberDetails memberDetails = new MemberDetails(member);
+
+        FcmTokenRequest request = new FcmTokenRequest();  // deviceId와 fcmToken 누락
+
+        // when
+        ResultActions result = mockMvc.perform(post("/api/fcm/token")
+                .with(user(memberDetails))
+                .with(csrf())
+                .contentType(MediaType.APPLICATION_JSON)
+                .content(objectMapper.writeValueAsString(request)));
+
+        // then
+        result.andExpect(status().isBadRequest())
+                .andExpect(jsonPath("$.code").value(400));
+
+        then(fcmService).shouldHaveNoInteractions();
+    }
+
+}
diff --git a/src/test/java/com/potatocake/everymoment/controller/FileControllerTest.java b/src/test/java/com/potatocake/everymoment/controller/FileControllerTest.java
@@ -0,0 +1,156 @@
+package com.potatocake.everymoment.controller;
+
+import static org.mockito.BDDMockito.given;
+import static org.mockito.BDDMockito.then;
+import static org.mockito.BDDMockito.willDoNothing;
+import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.csrf;
+import static org.springframework.security.test.web.servlet.request.SecurityMockMvcRequestPostProcessors.user;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.get;
+import static org.springframework.test.web.servlet.request.MockMvcRequestBuilders.multipart;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.jsonPath;
+import static org.springframework.test.web.servlet.result.MockMvcResultMatchers.status;
+
+import com.potatocake.everymoment.dto.response.FileResponse;
+import com.potatocake.everymoment.entity.Member;
+import com.potatocake.everymoment.security.MemberDetails;
+import com.potatocake.everymoment.service.FileService;
+import java.util.List;
+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.WebMvcTest;
+import org.springframework.boot.test.mock.mockito.MockBean;
+import org.springframework.http.MediaType;
+import org.springframework.mock.web.MockMultipartFile;
+import org.springframework.security.test.context.support.WithMockUser;
+import org.springframework.test.web.servlet.MockMvc;
+import org.springframework.test.web.servlet.ResultActions;
+
+@WithMockUser
+@WebMvcTest(FileController.class)
+class FileControllerTest {
+
+    @Autowired
+    private MockMvc mockMvc;
+
+    @MockBean
+    private FileService fileService;
+
+    @Test
+    @DisplayName("파일 목록이 성공적으로 조회된다.")
+    void should_ReturnFiles_When_ValidDiaryId() throws Exception {
+        // given
+        Long diaryId = 1L;
+        List<FileResponse> responses = List.of(
+                FileResponse.builder()
+                        .id(1L)
+                        .imageUrl("https://example.com/image1.jpg")
+                        .order(1)
+                        .build(),
+                FileResponse.builder()
+                        .id(2L)
+                        .imageUrl("https://example.com/image2.jpg")
+                        .order(2)
+                        .build()
+        );
+
+        given(fileService.getFiles(diaryId)).willReturn(responses);
+
+        // when
+        ResultActions result = mockMvc.perform(get("/api/diaries/{diaryId}/files", diaryId));
+
+        // then
+        result.andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value(200))
+                .andExpect(jsonPath("$.message").value("success"))
+                .andExpect(jsonPath("$.info").isArray())
+                .andExpect(jsonPath("$.info[0].imageUrl").value("https://example.com/image1.jpg"))
+                .andExpect(jsonPath("$.info[1].imageUrl").value("https://example.com/image2.jpg"));
+
+        then(fileService).should().getFiles(diaryId);
+    }
+
+    @Test
+    @DisplayName("파일이 성공적으로 업로드된다.")
+    void should_UploadFiles_When_ValidInput() throws Exception {
+        // given
+        Long diaryId = 1L;
+        Long memberId = 1L;
+        Member member = Member.builder()
+                .id(memberId)
+                .number(1234L)
+                .nickname("testUser")
+                .build();
+        MemberDetails memberDetails = new MemberDetails(member);
+
+        MockMultipartFile file1 = new MockMultipartFile(
+                "files",
+                "test1.jpg",
+                MediaType.IMAGE_JPEG_VALUE,
+                "test image 1".getBytes()
+        );
+        MockMultipartFile file2 = new MockMultipartFile(
+                "files",
+                "test2.jpg",
+                MediaType.IMAGE_JPEG_VALUE,
+                "test image 2".getBytes()
+        );
+
+        willDoNothing().given(fileService).uploadFiles(diaryId, memberId, List.of(file1, file2));
+
+        // when
+        ResultActions result = mockMvc.perform(multipart("/api/diaries/{diaryId}/files", diaryId)
+                .file(file1)
+                .file(file2)
+                .with(user(memberDetails))
+                .with(csrf()));
+
+        // then
+        result.andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value(200))
+                .andExpect(jsonPath("$.message").value("success"));
+
+        then(fileService).should().uploadFiles(diaryId, memberId, List.of(file1, file2));
+    }
+
+    @Test
+    @DisplayName("파일이 성공적으로 수정된다.")
+    void should_UpdateFiles_When_ValidInput() throws Exception {
+        // given
+        Long diaryId = 1L;
+        Long memberId = 1L;
+        Member member = Member.builder()
+                .id(memberId)
+                .number(1234L)
+                .nickname("testUser")
+                .build();
+        MemberDetails memberDetails = new MemberDetails(member);
+
+        MockMultipartFile file = new MockMultipartFile(
+                "files",
+                "test.jpg",
+                MediaType.IMAGE_JPEG_VALUE,
+                "test image".getBytes()
+        );
+
+        willDoNothing().given(fileService).updateFiles(diaryId, memberId, List.of(file));
+
+        // when
+        ResultActions result = mockMvc.perform(multipart("/api/diaries/{diaryId}/files", diaryId)
+                .file(file)
+                .with(request -> {
+                    request.setMethod("PUT");
+                    return request;
+                })
+                .with(user(memberDetails))
+                .with(csrf()));
+
+        // then
+        result.andExpect(status().isOk())
+                .andExpect(jsonPath("$.code").value(200))
+                .andExpect(jsonPath("$.message").value("success"));
+
+        then(fileService).should().updateFiles(diaryId, memberId, List.of(file));
+    }
+
+}