Skip to content

Commit eb838a2

Browse files
fix: update tests to use ConstantFlightServerDelayed for timeout scenarios
[EMPTY] trigger CI
1 parent c35f372 commit eb838a2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_query.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -436,7 +436,7 @@ async def fibo(iters):
436436
@asyncio_run
437437
async def test_query_async_timeout(self):
438438
with pytest.raises(FlightTimedOutError):
439-
with ConstantFlightServer() as server:
439+
with ConstantFlightServerDelayed() as server:
440440
connection_string = f"grpc://localhost:{server.port}"
441441
token = "my_token"
442442
database = "my_database"
@@ -452,7 +452,7 @@ async def test_query_async_timeout(self):
452452

453453
def test_query_timeout_per_call_override(self):
454454
with pytest.raises(FlightTimedOutError):
455-
with ConstantFlightServer() as server:
455+
with ConstantFlightServerDelayed() as server:
456456
connection_string = f"grpc://localhost:{server.port}"
457457
token = "my_token"
458458
database = "my_database"

0 commit comments

Comments
 (0)