test: 이벤트 호출 테스트 추가

woowacourse-teams · May 1, 2024 · 0f33ab1 · 0f33ab1
1 parent 4f7a9d9
commit 0f33ab1
Showing 1 changed file with 38 additions and 0 deletions.
diff --git a/...rc/test/java/com/ddang/ddang/chat/handler/ChatWebSocketHandleTextMessageProviderTest.java b/...rc/test/java/com/ddang/ddang/chat/handler/ChatWebSocketHandleTextMessageProviderTest.java
@@ -1,5 +1,6 @@
 package com.ddang.ddang.chat.handler;
 
+import com.ddang.ddang.chat.application.event.CreateReadMessageLogEvent;
 import com.ddang.ddang.chat.application.event.MessageNotificationEvent;
 import com.ddang.ddang.chat.application.event.UpdateReadMessageLogEvent;
 import com.ddang.ddang.chat.domain.WebSocketChatSessions;
@@ -40,6 +41,9 @@
 @SuppressWarnings("NonAsciiCharacters")
 class ChatWebSocketHandleTextMessageProviderTest extends ChatWebSocketHandleTextMessageProviderTestFixture {
 
+    @Autowired
+    ApplicationEvents applicationEvents;
+
     @Autowired
     ChatWebSocketHandleTextMessageProvider provider;
 
@@ -63,6 +67,8 @@ class ChatWebSocketHandleTextMessageProviderTest extends ChatWebSocketHandleText
 
     @Autowired
     ObjectMapper objectMapper;
+    @Autowired
+    private ChatWebSocketHandleTextMessageProvider chatWebSocketHandleTextMessageProvider;
 
     @Test
     void 지원하는_웹소켓_핸들링_타입을_반환한다() {
@@ -174,6 +180,38 @@ class ChatWebSocketHandleTextMessageProviderTest extends ChatWebSocketHandleText
         assertThat(actual).hasSize(1);
     }
 
+    @Test
+    void 메시지_전송시_알림_전송_이벤트를_호출한다() throws Exception {
+        // given
+        given(writerSession.getAttributes()).willReturn(발신자_세션_attribute_정보);
+        willDoNothing().given(sessions).add(writerSession, 채팅방.getId());
+        willReturn(false).given(sessions).containsByUserId(채팅방.getId(), 수신자.getId());
+        willReturn(Set.of(writerSession)).given(sessions).getSessionsByChatRoomId(채팅방.getId());
+
+        // when
+        provider.handleCreateSendMessage(writerSession, 메시지_전송_데이터);
+        final long actual = events.stream(MessageNotificationEvent.class).count();
+
+        // then
+        assertThat(actual).isEqualTo(1);
+    }
+
+    @Test
+    void 메시지_전송시_메시지_로그_업데이트_이벤트를_호출한다() throws Exception {
+        // given
+        given(writerSession.getAttributes()).willReturn(발신자_세션_attribute_정보);
+        willDoNothing().given(sessions).add(writerSession, 채팅방.getId());
+        willReturn(false).given(sessions).containsByUserId(채팅방.getId(), 수신자.getId());
+        willReturn(Set.of(writerSession)).given(sessions).getSessionsByChatRoomId(채팅방.getId());
+
+        // when
+        provider.handleCreateSendMessage(writerSession, 메시지_전송_데이터);
+        final long actual = events.stream(UpdateReadMessageLogEvent.class).count();
+
+        // then
+        assertThat(actual).isEqualTo(1);
+    }
+
     @Test
     void 잘못된_데이터_타입_전달시_예외가_발생한다() throws JsonProcessingException {
         // given