diff --git a/tests/laborious/activities/test_model_import_status.py b/tests/laborious/activities/test_model_import_status.py index 1960880..893d945 100644 --- a/tests/laborious/activities/test_model_import_status.py +++ b/tests/laborious/activities/test_model_import_status.py @@ -450,9 +450,10 @@ async def test_a_refused_claim_statement_is_reported_and_reraised(): write_errors=[IntegrityError('UPDATE ...', {}, Exception('refused'))], ) activity = build_activity(engine) + payload = claim_input() with raises(ImportStatusRejectedError) as failure: - await activity.claim_import_status(claim_input()) + await activity.claim_import_status(payload) assert failure.value.non_retryable is True logged = activity.logger.custom_error.call_args_list[0].args[0]