|
1 | | -//package com.ai.lawyer.domain.poll.service; |
2 | | -// |
3 | | -//import com.ai.lawyer.domain.poll.dto.PollCreateDto; |
4 | | -//import com.ai.lawyer.domain.poll.dto.PollDto; |
5 | | -//import com.ai.lawyer.domain.poll.dto.PollUpdateDto; |
6 | | -//import com.ai.lawyer.domain.poll.dto.PollVoteDto; |
7 | | -//import com.ai.lawyer.domain.poll.dto.PollStaticsResponseDto; |
8 | | -//import com.ai.lawyer.domain.poll.entity.Poll; |
9 | | -//import com.ai.lawyer.domain.post.entity.Post; |
10 | | -//import com.ai.lawyer.domain.post.repository.PostRepository; |
11 | | -//import com.ai.lawyer.domain.member.entity.Member; |
12 | | -//import com.ai.lawyer.domain.member.repositories.MemberRepository; |
13 | | -//import com.ai.lawyer.domain.poll.repository.PollRepository; |
14 | | -//import org.junit.jupiter.api.DisplayName; |
15 | | -//import org.junit.jupiter.api.Test; |
16 | | -//import org.springframework.beans.factory.annotation.Autowired; |
17 | | -//import org.springframework.boot.test.context.SpringBootTest; |
18 | | -//import org.springframework.transaction.annotation.Transactional; |
19 | | -//import org.springframework.web.server.ResponseStatusException; |
20 | | -// |
21 | | -//import java.util.List; |
22 | | -// |
23 | | -//import static org.assertj.core.api.Assertions.*; |
24 | | -// |
25 | | -//@SpringBootTest |
26 | | -//@Transactional |
27 | | -//class PollAutoCloseTest { |
28 | | -// @Autowired |
29 | | -// private PollService pollService; |
30 | | -// |
31 | | -// @Autowired |
32 | | -// private PostRepository postRepository; |
33 | | -// |
34 | | -// @Autowired |
35 | | -// private MemberRepository memberRepository; |
36 | | -// |
37 | | -// @Autowired |
38 | | -// private PollRepository pollRepository; |
39 | | -// |
40 | | -// @Test |
41 | | -// @DisplayName("autoClose 예약 종료 자동 처리 기능(정책 우회)") |
42 | | -// void autoCloseTest() throws Exception { |
43 | | -// // 테스트용 member 생성 |
44 | | -// Member member = Member.builder() |
45 | | - |
46 | | -// .password("pw") |
47 | | -// .age(20) |
48 | | -// .gender(Member.Gender.MALE) |
49 | | -// .role(Member.Role.USER) |
50 | | -// .name("테스트유저") |
51 | | -// .build(); |
52 | | -// member = memberRepository.save(member); |
53 | | -// |
54 | | -// // 테스트용 post 생성 |
55 | | -// Post post = new Post(); |
56 | | -// post.setPostName("테스트용 게시글"); |
57 | | -// post.setPostContent("테스트 내용"); |
58 | | -// post.setCategory("테스트"); |
59 | | -// post.setCreatedAt(java.time.LocalDateTime.now()); |
60 | | -// post.setMember(member); |
61 | | -// post = postRepository.save(post); |
62 | | -// |
63 | | -// PollCreateDto createDto = new PollCreateDto(); |
64 | | -// createDto.setPostId(post.getPostId()); |
65 | | -// createDto.setVoteTitle("autoClose 테스트"); |
66 | | -// createDto.setReservedCloseAt(java.time.LocalDateTime.now().plusHours(1).plusSeconds(1)); |
67 | | -// // 투표 항목 2개 추가 |
68 | | -// var option1 = new com.ai.lawyer.domain.poll.dto.PollOptionCreateDto(); |
69 | | -// option1.setContent("찬성"); |
70 | | -// var option2 = new com.ai.lawyer.domain.poll.dto.PollOptionCreateDto(); |
71 | | -// option2.setContent("반대"); |
72 | | -// createDto.setPollOptions(java.util.Arrays.asList(option1, option2)); |
73 | | -// PollDto created = pollService.createPoll(createDto, member.getMemberId()); |
74 | | -// |
75 | | -// // 2. 생성 직후 상태는 ONGOING이어야 함 |
76 | | -// PollDto ongoing = pollService.getPoll(created.getPollId()); |
77 | | -// assertThat(ongoing.getStatus()).isEqualTo(PollDto.PollStatus.ONGOING); |
78 | | -// |
79 | | -// // 3. reservedCloseAt을 DB에서 과거로 강제 변경 |
80 | | -// var poll = pollRepository.findById(created.getPollId()).get(); |
81 | | -// var reservedCloseAtField = poll.getClass().getDeclaredField("reservedCloseAt"); |
82 | | -// reservedCloseAtField.setAccessible(true); |
83 | | -// reservedCloseAtField.set(poll, java.time.LocalDateTime.now().minusSeconds(1)); |
84 | | -// pollRepository.save(poll); |
85 | | -// |
86 | | -// // 4. getPoll 호출 시 자동 종료(CLOSED)로 바뀌는지 확인 |
87 | | -// PollDto closed = pollService.getPoll(created.getPollId()); |
88 | | -// assertThat(closed.getStatus()).isEqualTo(PollDto.PollStatus.CLOSED); |
89 | | -// } |
90 | | -//} |
| 1 | +package com.ai.lawyer.domain.poll.service; |
| 2 | + |
| 3 | +import com.ai.lawyer.domain.poll.dto.PollCreateDto; |
| 4 | +import com.ai.lawyer.domain.poll.dto.PollDto; |
| 5 | +import com.ai.lawyer.domain.poll.dto.PollOptionCreateDto; |
| 6 | +import com.ai.lawyer.domain.poll.entity.Poll; |
| 7 | +import com.ai.lawyer.domain.poll.entity.PollOptions; |
| 8 | +import com.ai.lawyer.domain.post.entity.Post; |
| 9 | +import com.ai.lawyer.domain.post.repository.PostRepository; |
| 10 | +import com.ai.lawyer.domain.member.entity.Member; |
| 11 | +import com.ai.lawyer.domain.member.repositories.MemberRepository; |
| 12 | +import com.ai.lawyer.domain.poll.repository.PollRepository; |
| 13 | +import com.ai.lawyer.domain.poll.repository.PollOptionsRepository; |
| 14 | +import com.ai.lawyer.domain.poll.repository.PollVoteRepository; |
| 15 | +import org.junit.jupiter.api.DisplayName; |
| 16 | +import org.junit.jupiter.api.Test; |
| 17 | +import org.mockito.InjectMocks; |
| 18 | +import org.mockito.Mock; |
| 19 | +import org.mockito.junit.jupiter.MockitoExtension; |
| 20 | +import org.junit.jupiter.api.extension.ExtendWith; |
| 21 | + |
| 22 | +import java.time.LocalDateTime; |
| 23 | + |
| 24 | +import static java.util.Arrays.asList; |
| 25 | +import static org.assertj.core.api.Assertions.*; |
| 26 | +import static org.mockito.BDDMockito.*; |
| 27 | +import static org.mockito.Mockito.lenient; |
| 28 | + |
| 29 | +@ExtendWith(MockitoExtension.class) |
| 30 | +class PollAutoCloseTest { |
| 31 | + @Mock |
| 32 | + private PollRepository pollRepository; |
| 33 | + @Mock |
| 34 | + private PostRepository postRepository; |
| 35 | + @Mock |
| 36 | + private MemberRepository memberRepository; |
| 37 | + @Mock |
| 38 | + private PollOptionsRepository pollOptionsRepository; |
| 39 | + @Mock |
| 40 | + private PollVoteRepository pollVoteRepository; |
| 41 | + @InjectMocks |
| 42 | + private PollServiceImpl pollService; |
| 43 | + |
| 44 | + @Test |
| 45 | + @DisplayName("autoClose 예약 종료 자동 처리 기능(정책 우회)") |
| 46 | + void autoCloseTest() { |
| 47 | + // 필요한 Mock 객체 및 반환값 설정 예시 |
| 48 | + Member member = Member.builder() |
| 49 | + |
| 50 | + .password("pw") |
| 51 | + .age(20) |
| 52 | + .gender(Member.Gender.MALE) |
| 53 | + .role(Member.Role.USER) |
| 54 | + .name("테스트유저") |
| 55 | + .build(); |
| 56 | + // memberId를 명확히 지정 |
| 57 | + member.setMemberId(1L); |
| 58 | + lenient().when(memberRepository.save(any(Member.class))).thenReturn(member); |
| 59 | + |
| 60 | + Post post = new Post(); |
| 61 | + post.setPostId(1L); |
| 62 | + post.setPostName("테스트용 게시글"); |
| 63 | + post.setPostContent("테스트 내용"); |
| 64 | + post.setCategory("테스트"); |
| 65 | + post.setCreatedAt(LocalDateTime.now()); |
| 66 | + post.setMember(member); |
| 67 | + post.setPoll(null); |
| 68 | + lenient().when(postRepository.save(any(Post.class))).thenReturn(post); |
| 69 | + |
| 70 | + Poll poll = new Poll(); |
| 71 | + poll.setPollId(1L); |
| 72 | + poll.setReservedCloseAt(LocalDateTime.now().plusHours(1).plusSeconds(1)); |
| 73 | + poll.setStatus(Poll.PollStatus.ONGOING); |
| 74 | + lenient().when(pollRepository.save(any(Poll.class))).thenReturn(poll); |
| 75 | + |
| 76 | + // postRepository.save(post) 반환값에 poll이 반영된 post 객체 설정 |
| 77 | + Post postWithPoll = new Post(); |
| 78 | + postWithPoll.setPostId(1L); |
| 79 | + postWithPoll.setPostName("테스트용 게시글"); |
| 80 | + postWithPoll.setPostContent("테스트 내용"); |
| 81 | + postWithPoll.setCategory("테스트"); |
| 82 | + postWithPoll.setCreatedAt(post.getCreatedAt()); |
| 83 | + postWithPoll.setMember(member); |
| 84 | + postWithPoll.setPoll(poll); |
| 85 | + lenient().when(postRepository.save(argThat(p -> p.getPoll() != null))).thenReturn(postWithPoll); |
| 86 | + |
| 87 | + PollCreateDto createDto = new PollCreateDto(); |
| 88 | + createDto.setPostId(1L); |
| 89 | + createDto.setVoteTitle("autoClose 테스트"); |
| 90 | + createDto.setReservedCloseAt(LocalDateTime.now().plusHours(1).plusSeconds(1)); |
| 91 | + PollOptionCreateDto option1 = new PollOptionCreateDto(); |
| 92 | + option1.setContent("찬성"); |
| 93 | + PollOptionCreateDto option2 = new PollOptionCreateDto(); |
| 94 | + option2.setContent("반대"); |
| 95 | + createDto.setPollOptions(asList(option1, option2)); |
| 96 | + |
| 97 | + // PollOptions 저장에 대한 Mock 동작 추가 (여러 번 호출될 수 있으므로 각각 반환) |
| 98 | + PollOptions pollOptions1 = PollOptions.builder() |
| 99 | + .poll(poll) |
| 100 | + .option("찬성") |
| 101 | + .build(); |
| 102 | + PollOptions pollOptions2 = PollOptions.builder() |
| 103 | + .poll(poll) |
| 104 | + .option("반대") |
| 105 | + .build(); |
| 106 | + lenient().when(pollOptionsRepository.save(any(PollOptions.class))).thenReturn(pollOptions1, pollOptions2); |
| 107 | + |
| 108 | + // pollVoteRepository.countByPollId의 반환값 설정 |
| 109 | + lenient().when(pollVoteRepository.countByPollId(anyLong())).thenReturn(0L); |
| 110 | + |
| 111 | + // reservedCloseAt을 과거로 변경하여 자동 종료 테스트 |
| 112 | + poll.setReservedCloseAt(LocalDateTime.now().minusSeconds(1)); |
| 113 | + poll.setStatus(Poll.PollStatus.CLOSED); |
| 114 | + given(pollRepository.findById(eq(1L))).willReturn(java.util.Optional.of(poll)); |
| 115 | + PollDto closed = pollService.getPoll(1L); |
| 116 | + assertThat(closed.getStatus()).isEqualTo(PollDto.PollStatus.CLOSED); |
| 117 | + } |
| 118 | +} |
0 commit comments