tests: align EventTranslator streaming expectations

Author: contextablemark

typescript-sdk/integrations/adk-middleware/python/tests/test_event_translator_comprehensive.py

@@ -195,17 +195,18 @@ async def test_translate_text_content_basic(self, translator, mock_adk_event_wit
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events.append(event)
 
-        assert len(events) == 3  # START, CONTENT , END
+        assert len(events) == 2  # START, CONTENT
         assert isinstance(events[0], TextMessageStartEvent)
         assert isinstance(events[1], TextMessageContentEvent)
-        assert isinstance(events[2], TextMessageEndEvent)
 
         # Check content
         assert events[1].delta == "Test content"
 
         # Check message IDs are consistent
         message_id = events[0].message_id
         assert events[1].message_id == message_id
+        # No END event emitted for non-final, non-turn-complete content
+        assert all(not isinstance(event, TextMessageEndEvent) for event in events)
 
     @pytest.mark.asyncio
     async def test_translate_text_content_multiple_parts(self, translator, mock_adk_event):
@@ -222,9 +223,10 @@ async def test_translate_text_content_multiple_parts(self, translator, mock_adk_
         async for event in translator.translate(mock_adk_event, "thread_1", "run_1"):
             events.append(event)
 
-        assert len(events) == 3  # START, CONTENT , END
+        assert len(events) == 2  # START, CONTENT
         assert isinstance(events[1], TextMessageContentEvent)
         assert events[1].delta == "First partSecond part"  # Joined without newlines
+        assert all(not isinstance(event, TextMessageEndEvent) for event in events)
 
     @pytest.mark.asyncio
     async def test_translate_text_content_partial_streaming(self, translator, mock_adk_event_with_content):
@@ -236,9 +238,10 @@ async def test_translate_text_content_partial_streaming(self, translator, mock_a
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events.append(event)
 
-        assert len(events) == 3  # START, CONTENT , END
+        assert len(events) == 2  # START, CONTENT
         assert isinstance(events[0], TextMessageStartEvent)
         assert isinstance(events[1], TextMessageContentEvent)
+        assert all(not isinstance(event, TextMessageEndEvent) for event in events)
 
     @pytest.mark.asyncio
     async def test_translate_text_content_final_response_callable(self, translator, mock_adk_event_with_content):
@@ -304,7 +307,7 @@ async def test_translate_text_content_final_response_from_agent_callback(self, t
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events.append(event)
 
-        assert len(events) == 3  # START, CONTENT , END
+        assert len(events) == 3  # START, CONTENT, END (non-streaming final response)
         assert isinstance(events[0], TextMessageStartEvent)
         assert isinstance(events[1], TextMessageContentEvent)
         assert events[1].delta == mock_adk_event_with_content.content.parts[0].text
@@ -360,8 +363,9 @@ async def test_translate_text_content_mixed_text_parts(self, translator, mock_ad
         async for event in translator.translate(mock_adk_event, "thread_1", "run_1"):
             events.append(event)
 
-        assert len(events) == 3  # START, CONTENT , END
+        assert len(events) == 2  # START, CONTENT
         assert events[1].delta == "Valid textMore text"
+        assert all(not isinstance(event, TextMessageEndEvent) for event in events)
 
     @pytest.mark.asyncio
     async def test_translate_function_calls_basic(self, translator, mock_adk_event):
@@ -647,21 +651,23 @@ async def test_streaming_state_management(self, translator, mock_adk_event_with_
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events1.append(event)
 
-        assert len(events1) == 3  # START, CONTENT, END
+        assert len(events1) == 2  # START, CONTENT (stream stays open)
         message_id = events1[0].message_id
 
-        # streaming is stoped after TextMessageEndEvent
-        assert translator._is_streaming is False
-        # since the streaming is stopped
-        assert translator._streaming_message_id == None
+        # Stream remains active until completion signal
+        assert translator._is_streaming is True
+        assert translator._streaming_message_id == message_id
 
-        # Second event should continue streaming (same message ID)
+        # Second event should append to the same stream (content only)
         events2 = []
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events2.append(event)
 
-        assert len(events2) == 3  # New Streaming (START , CONTENT ,END)
-        assert events2[0].message_id != message_id  # Same message ID
+        assert len(events2) == 1  # CONTENT only
+        assert isinstance(events2[0], TextMessageContentEvent)
+        assert events2[0].message_id == message_id
+        assert translator._is_streaming is True
+        assert translator._streaming_message_id == message_id
 
     @pytest.mark.asyncio
     async def test_complex_event_with_multiple_features(self, translator, mock_adk_event):
@@ -685,16 +691,16 @@ async def test_complex_event_with_multiple_features(self, translator, mock_adk_e
         async for event in translator.translate(mock_adk_event, "thread_1", "run_1"):
             events.append(event)
 
-        # Should have text events, state delta, and custom event
-        assert len(events) == 5  # START, CONTENT, STATE_DELTA, CUSTOM , END
+        # Should have text events, state delta, and custom event (stream remains open)
+        assert len(events) == 4  # START, CONTENT, STATE_DELTA, CUSTOM
 
         # Check event types
         event_types = [type(event) for event in events]
         assert TextMessageStartEvent in event_types
         assert TextMessageContentEvent in event_types
         assert StateDeltaEvent in event_types
         assert CustomEvent in event_types
-        assert TextMessageEndEvent in event_types
+        assert TextMessageEndEvent not in event_types
 
     @pytest.mark.asyncio
     async def test_event_logging_coverage(self, translator, mock_adk_event_with_content):
@@ -765,8 +771,8 @@ async def test_partial_streaming_continuation(self, translator, mock_adk_event_w
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events1.append(event)
 
-        assert len(events1) == 3  # START, CONTENT , END
-        assert translator._is_streaming is False
+        assert len(events1) == 2  # START, CONTENT
+        assert translator._is_streaming is True
         message_id = events1[0].message_id
 
         # Second partial event (should continue streaming)
@@ -777,9 +783,11 @@ async def test_partial_streaming_continuation(self, translator, mock_adk_event_w
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events2.append(event)
 
-        assert len(events2) == 3  # Will start from begining (START , CONTENT , END)
-        assert isinstance(events2[1], TextMessageContentEvent)
-        assert events2[0].message_id != message_id  # Not the same message ID Because its a new streaming
+        assert len(events2) == 1  # Additional CONTENT chunk
+        assert isinstance(events2[0], TextMessageContentEvent)
+        assert events2[0].message_id == message_id  # Same stream continues
+        assert translator._is_streaming is True
+        assert translator._streaming_message_id == message_id
 
         # Final event (should end streaming - requires is_final_response=True)
         mock_adk_event_with_content.partial = False
@@ -790,7 +798,9 @@ async def test_partial_streaming_continuation(self, translator, mock_adk_event_w
         async for event in translator.translate(mock_adk_event_with_content, "thread_1", "run_1"):
             events3.append(event)
 
-        assert len(events3) == 0  # No more message (turn Complete)
+        assert len(events3) == 1  # Final END to close the stream
+        assert isinstance(events3[0], TextMessageEndEvent)
+        assert events3[0].message_id == message_id
 
         # Should reset streaming state
         assert translator._is_streaming is False