update batch

This commit is contained in:
2025-05-30 15:32:31 -03:00
parent efd962cdba
commit b048febbd5
24 changed files with 455 additions and 69 deletions

View File

@@ -18,11 +18,11 @@ def test_batch():
with processor(records=records, handler=record_handler) as p:
processed_messages = p.process()
assert processed_messages == (
Result(Status.SUCCESS, True, True),
Result(Status.SUCCESS, True, True),
Result(Status.FAIL, 'ValueError: Invalid record', False),
)
assert len(processed_messages) == 3
fail_record = processed_messages[2]
assert isinstance(fail_record.cause, ValueError)
assert str(fail_record.cause) == 'Invalid record'
assert processor.successes == [True, True]
assert processor.failures == [False]
@@ -30,9 +30,7 @@ def test_batch():
with processor(records=(False,), handler=record_handler):
processed_messages = processor.process()
assert processed_messages == (
Result(Status.FAIL, 'ValueError: Invalid record', False),
)
assert processed_messages[0].status == Status.FAIL
assert processor.successes == []
assert processor.failures == [False]
@@ -50,7 +48,7 @@ def test_batch_context():
processed_messages = processor.process()
assert processed_messages == (
Result(Status.SUCCESS, 4, 2),
Result(Status.SUCCESS, 6, 3),
Result(Status.SUCCESS, 8, 4),
Result(Status.SUCCESS, output=4, input_record=2),
Result(Status.SUCCESS, output=6, input_record=3),
Result(Status.SUCCESS, output=8, input_record=4),
)