1515import com .ject .vs .user .domain .ImageColor ;
1616import com .ject .vs .user .domain .User ;
1717import com .ject .vs .user .port .in .UserQueryUseCase ;
18- import com .ject .vs .vote .domain .VoteOption ;
1918import com .ject .vs .vote .domain .VoteOptionCode ;
2019import com .ject .vs .vote .port .in .VoteParticipationQueryUseCase ;
2120import com .ject .vs .vote .port .in .VoteQueryUseCase ;
@@ -61,9 +60,6 @@ class ChatServiceTest {
6160 @ Mock
6261 private UserQueryUseCase userQueryUseCase ;
6362
64- @ Mock
65- private VoteOption selectedOption ;
66-
6763 @ Nested
6864 class sendMessage {
6965
@@ -98,8 +94,7 @@ class sendMessage {
9894 given (sender .getNickname ()).willReturn ("테스트유저" );
9995 given (sender .getImageColor ()).willReturn (ImageColor .GREEN );
10096 given (userQueryUseCase .getUser (2L )).willReturn (sender );
101- given (voteQueryUseCase .getSelectedOption (1L , 2L )).willReturn (selectedOption );
102- given (selectedOption .getCode ()).willReturn (VoteOptionCode .A );
97+ given (voteQueryUseCase .findSelectedOptionCode (1L , 2L )).willReturn (Optional .of (VoteOptionCode .A ));
10398
10499 // when
105100 MessageResult result = chatService .sendMessage (new SendMessageCommand (1L , 2L , "hello" ));
@@ -206,14 +201,14 @@ class getMessages {
206201 given (sender .getNickname ()).willReturn ("" );
207202 given (sender .getImageColor ()).willReturn (ImageColor .GREEN );
208203 given (userQueryUseCase .getUser (2L )).willReturn (sender );
209- given (voteQueryUseCase .getSelectedOption (1L , 2L )).willReturn (selectedOption );
210- given (selectedOption .getCode ()).willReturn (null );
204+ given (voteQueryUseCase .findSelectedOptionCode (1L , 2L )).willReturn (Optional .empty ());
211205
212206 // when
213207 MessagePageResult result = chatService .getMessages (1L , 2L , null , 30 );
214208
215209 // then
216210 assertThat (result .messages ()).hasSize (1 );
211+ assertThat (result .messages ().getFirst ().senderVoteOption ()).isNull ();
217212 assertThat (result .hasNext ()).isFalse ();
218213 verify (chatMessageRepository ).findAllByVoteIdOrderByIdDesc (eq (1L ), any (PageRequest .class ));
219214 }
@@ -228,8 +223,7 @@ class getMessages {
228223 given (sender .getNickname ()).willReturn ("" );
229224 given (sender .getImageColor ()).willReturn (ImageColor .GREEN );
230225 given (userQueryUseCase .getUser (2L )).willReturn (sender );
231- given (voteQueryUseCase .getSelectedOption (1L , 2L )).willReturn (selectedOption );
232- given (selectedOption .getCode ()).willReturn (null );
226+ given (voteQueryUseCase .findSelectedOptionCode (1L , 2L )).willReturn (Optional .of (VoteOptionCode .B ));
233227
234228 // when
235229 MessagePageResult result = chatService .getMessages (1L , 2L , 100L , 30 );
0 commit comments