Skip to content

Fixing integration StreamingResponse tests #629

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Rodrigo Brandão (rodrigobr-msft) merged 3 commits into from
Oct 16, 2026
Merged

Fixing integration StreamingResponse tests #629

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
de67f12
Fixing integration StreamingResponse tests
rodrigobr-msft Oct 15, 2026
3c2f40a
Merge branch 'main' into users/robrandao/int-test-fix
rodrigobr-msft Oct 16, 2026
cfe9f60
Merge branch 'main' into users/robrandao/int-test-fix
rodrigobr-msft Oct 16, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 9 additions & 6 deletions dev/integration/tests/test_streaming_response.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -62,17 +62,18 @@ async def test_basic_streaming_response_non_streaming_channel(agent_client: Agen
# give enough time for all the activities to send
await agent_client.send("/stream", wait=expected_len * 2.0)

# Non-streaming channels drop informative/streaming updates and send a single
# plain message at end_stream, without a streaminfo entity.
stream_activities = agent_client.select().where(
entities=lambda x: any(e["type"] == "streaminfo" for e in x)
).get()

assert len(stream_activities) == 1
assert len(stream_activities) == 0

final_streaminfo = get_streaminfo(stream_activities[0])
messages = agent_client.select().where(type=ActivityTypes.message).get()

assert final_streaminfo.stream_sequence == 1
assert final_streaminfo.stream_type == "final"
assert stream_activities[0].text == FULL_TEXT.replace(" ", "")
assert len(messages) == 1
assert messages[0].text == FULL_TEXT.replace(" ", "")



Expand Down Expand Up @@ -116,6 +117,8 @@ async def test_basic_streaming_response_streaming_channel(agent_client: AgentCli

final_streaminfo = get_streaminfo(stream_activities[-1])

assert final_streaminfo.stream_sequence == len(stream_activities)
# The final message carries no stream_sequence
assert final_streaminfo.stream_sequence is None
assert final_streaminfo.stream_type == "final"
assert stream_activities[-1].type == ActivityTypes.message
assert stream_activities[-1].text == FULL_TEXT.replace(" ", "")
Loading