Skip to content

Commit 7a62f64

Browse files
fix assertion in racy test
1 parent 6af89ee commit 7a62f64

File tree

1 file changed

+1
-5
lines changed

1 file changed

+1
-5
lines changed

tests/integration_tests/test_streaming.py

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -120,15 +120,11 @@ async def test_stream_failure_async(test_native_async_client):
120120

121121
stream = await test_native_async_client.query_row_block_stream(query)
122122

123-
with pytest.raises(StreamFailureError) as excinfo:
123+
with pytest.raises(StreamFailureError):
124124
async with stream:
125125
async for _ in stream:
126126
pass
127127

128-
error_msg = str(excinfo.value).lower()
129-
# Race condition: may get actual ClickHouse error or generic connection closed
130-
assert 'division by zero' in error_msg or 'connection closed' in error_msg
131-
132128

133129
def test_raw_stream(param_client, call, consume_stream):
134130
"""Test raw_stream for streaming response."""

0 commit comments

Comments
 (0)