diff --git a/tests/test_views_multipart.py b/tests/test_views_multipart.py index 3b4774e..0b57ca6 100644 --- a/tests/test_views_multipart.py +++ b/tests/test_views_multipart.py @@ -484,7 +484,7 @@ def _mock_celery_result(): if res.status_code == 200: data = get_json(res) assert data["completed"] is True - assert task.called_with(str(multipart.upload_id)) + assert task.delay.call_args.args == (str(multipart.upload_id),) # Two whitespaces expected to have been sent to client before # JSON was sent. assert res.data.startswith(b" {")