File tree Expand file tree Collapse file tree
unit_tests/sources/streams/concurrent Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -1492,7 +1492,10 @@ def test_is_done_raises_when_partition_generation_queue_not_empty():
14921492 # Artificially mark the stream as done without removing it from the partition generation queue
14931493 handler ._streams_done .add ("stuck_stream" )
14941494
1495- with pytest .raises (AirbyteTracedException , match = "Partition generation queue is not empty after all streams completed" ):
1495+ with pytest .raises (
1496+ AirbyteTracedException ,
1497+ match = "Partition generation queue is not empty after all streams completed" ,
1498+ ):
14961499 handler .is_done ()
14971500
14981501
@@ -1531,7 +1534,8 @@ def test_is_done_raises_when_active_groups_not_empty():
15311534 handler ._active_groups ["my_group" ] = {"stuck_stream" }
15321535
15331536 with pytest .raises (
1534- AirbyteTracedException , match = "still active after all streams were marked done"
1537+ AirbyteTracedException ,
1538+ match = "Active stream groups are not empty after all streams completed" ,
15351539 ):
15361540 handler .is_done ()
15371541
You can’t perform that action at this time.
0 commit comments