@@ -35,7 +35,7 @@ public async Task HandleAsync_WithEmptySegments_ShouldLogWarningAndReturn()
3535 await _handler . HandleAsync ( context ) ;
3636
3737 VerifyWarningLogged ( "채팅 처리 결과에 유효한 세그먼트가 없습니다" ) ;
38- _mockMessageBroker . Verify ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) , Times . Never ) ;
38+ _mockMessageBroker . Verify ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) , Times . Never ) ;
3939 }
4040
4141 [ Fact ]
@@ -53,7 +53,7 @@ public async Task HandleAsync_WithValidSegments_ShouldSendAllMessages()
5353 await _handler . HandleAsync ( context ) ;
5454
5555 _mockMessageBroker . Verify (
56- x => x . SendToUserAsync ( context . UserId . ToString ( ) , It . IsAny < WebSocketMessage > ( ) ) ,
56+ x => x . SendToUserAsync ( context . UserId . ToString ( ) , It . IsAny < object > ( ) ) ,
5757 Times . Exactly ( 3 ) ) ;
5858
5959 VerifyDebugLogged ( "채팅 결과 전송 완료" ) ;
@@ -74,7 +74,7 @@ public async Task HandleAsync_WithMixedValidAndEmptySegments_ShouldOnlySendValid
7474 await _handler . HandleAsync ( context ) ;
7575
7676 _mockMessageBroker . Verify (
77- x => x . SendToUserAsync ( context . UserId . ToString ( ) , It . IsAny < WebSocketMessage > ( ) ) ,
77+ x => x . SendToUserAsync ( context . UserId . ToString ( ) , It . IsAny < object > ( ) ) ,
7878 Times . Exactly ( 2 ) ) ;
7979 }
8080
@@ -88,8 +88,8 @@ public async Task HandleAsync_WithAudioSegment_ShouldIncludeAudioData()
8888 context . SetResponse ( "Test" , new List < ChatSegment > { segment } , 0.0 ) ;
8989
9090 WebSocketMessage ? sentMessage = null ;
91- _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) )
92- . Callback < string , WebSocketMessage > ( ( _ , message ) => sentMessage = message ) ;
91+ _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) )
92+ . Callback < string , object > ( ( _ , message ) => sentMessage = message as WebSocketMessage ) ;
9393
9494 await _handler . HandleAsync ( context ) ;
9595
@@ -108,15 +108,15 @@ public async Task HandleAsync_WithWebSocketFailure_ShouldThrowImmediately()
108108 var segment = ChatSegment . CreateText ( "Test message" ) ;
109109 context . SetResponse ( "Test" , new List < ChatSegment > { segment } , 0.0 ) ;
110110
111- _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) )
111+ _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) )
112112 . ThrowsAsync ( new Exception ( "Connection failed" ) ) ;
113113
114114 var act = async ( ) => await _handler . HandleAsync ( context ) ;
115115
116116 await act . Should ( ) . ThrowAsync < Exception > ( ) . WithMessage ( "Connection failed" ) ;
117117
118118 _mockMessageBroker . Verify (
119- x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) ,
119+ x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) ,
120120 Times . Once ) ;
121121
122122 VerifyErrorLogged ( "채팅 결과 전송 중 오류 발생" ) ;
@@ -135,8 +135,12 @@ public async Task HandleAsync_WithCorrectOrder_ShouldSendInOrderedSequence()
135135 context . SetResponse ( "Test" , segments , 0.0 ) ;
136136
137137 var sentMessages = new List < WebSocketMessage > ( ) ;
138- _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) )
139- . Callback < string , WebSocketMessage > ( ( _ , message ) => sentMessages . Add ( message ) ) ;
138+ _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) )
139+ . Callback < string , object > ( ( _ , message ) =>
140+ {
141+ if ( message is WebSocketMessage wsMessage )
142+ sentMessages . Add ( wsMessage ) ;
143+ } ) ;
140144
141145 await _handler . HandleAsync ( context ) ;
142146
@@ -160,8 +164,12 @@ public async Task HandleAsync_WithDifferentSegmentTypes_ShouldAllUseChatType()
160164 context . SetResponse ( "Test" , segments , 0.0 ) ;
161165
162166 var sentMessages = new List < WebSocketMessage > ( ) ;
163- _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) )
164- . Callback < string , WebSocketMessage > ( ( _ , message ) => sentMessages . Add ( message ) ) ;
167+ _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) )
168+ . Callback < string , object > ( ( _ , message ) =>
169+ {
170+ if ( message is WebSocketMessage wsMessage )
171+ sentMessages . Add ( wsMessage ) ;
172+ } ) ;
165173
166174 await _handler . HandleAsync ( context ) ;
167175
@@ -183,8 +191,8 @@ public async Task HandleAsync_ShouldIncludeRequestIdInMessages()
183191 context . SetResponse ( "Test" , new List < ChatSegment > { segment } , 0.0 ) ;
184192
185193 WebSocketMessage ? sentMessage = null ;
186- _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) )
187- . Callback < string , WebSocketMessage > ( ( _ , message ) => sentMessage = message ) ;
194+ _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) )
195+ . Callback < string , object > ( ( _ , message ) => sentMessage = message as WebSocketMessage ) ;
188196
189197 await _handler . HandleAsync ( context ) ;
190198
@@ -206,8 +214,12 @@ public async Task HandleAsync_ShouldUseConsistentWebSocketMessageType()
206214 context . SetResponse ( "Test" , segments , 0.0 ) ;
207215
208216 var sentMessages = new List < WebSocketMessage > ( ) ;
209- _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < WebSocketMessage > ( ) ) )
210- . Callback < string , WebSocketMessage > ( ( _ , message ) => sentMessages . Add ( message ) ) ;
217+ _mockMessageBroker . Setup ( x => x . SendToUserAsync ( It . IsAny < string > ( ) , It . IsAny < object > ( ) ) )
218+ . Callback < string , object > ( ( _ , message ) =>
219+ {
220+ if ( message is WebSocketMessage wsMessage )
221+ sentMessages . Add ( wsMessage ) ;
222+ } ) ;
211223
212224 await _handler . HandleAsync ( context ) ;
213225
0 commit comments