changhe batch
This commit is contained in:
@@ -18,11 +18,18 @@ def test_batch():
|
||||
with processor(records=records, handler=record_handler) as p:
|
||||
processed_messages = p.process()
|
||||
|
||||
assert len(processed_messages) == 3
|
||||
assert processed_messages == (
|
||||
Result(status=Status.SUCCESS, input_record=True, output=True, cause=None),
|
||||
Result(status=Status.SUCCESS, input_record=True, output=True, cause=None),
|
||||
Result(
|
||||
status=Status.FAIL,
|
||||
input_record=False,
|
||||
output=None,
|
||||
cause={'type': 'ValueError', 'message': 'Invalid record'},
|
||||
),
|
||||
)
|
||||
|
||||
fail_record = processed_messages[2]
|
||||
assert isinstance(fail_record.cause, ValueError)
|
||||
assert str(fail_record.cause) == 'Invalid record'
|
||||
assert len(processed_messages) == 3
|
||||
|
||||
assert processor.successes == [True, True]
|
||||
assert processor.failures == [False]
|
||||
@@ -48,7 +55,7 @@ def test_batch_context():
|
||||
processed_messages = processor.process()
|
||||
|
||||
assert processed_messages == (
|
||||
Result(Status.SUCCESS, output=4, input_record=2),
|
||||
Result(Status.SUCCESS, output=6, input_record=3),
|
||||
Result(Status.SUCCESS, output=8, input_record=4),
|
||||
Result(status=Status.SUCCESS, input_record=2, output=4, cause=None),
|
||||
Result(status=Status.SUCCESS, input_record=3, output=6, cause=None),
|
||||
Result(status=Status.SUCCESS, input_record=4, output=8, cause=None),
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user