Skip to content

Commit 22ef57b

Browse files
fix: correct ruff formatting and update active_groups test regex to match actual error message
Co-Authored-By: alfredo.garcia@airbyte.io <freddy.garcia7.fg@gmail.com>
1 parent 055655a commit 22ef57b

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

unit_tests/sources/streams/concurrent/test_concurrent_read_processor.py

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff 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

0 commit comments

Comments
 (0)