update batch
This commit is contained in:
@@ -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),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user