From 5d8d26a5ce8823a75f82eef3af02cd2b7699831b Mon Sep 17 00:00:00 2001 From: Kyle MacMillan Date: Tue, 3 Dec 2024 12:58:42 -0500 Subject: [PATCH] Refactored test for new apply_async --- tests/app/celery/test_scheduled_tasks.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/app/celery/test_scheduled_tasks.py b/tests/app/celery/test_scheduled_tasks.py index 6b89995e7c..3fcd240704 100644 --- a/tests/app/celery/test_scheduled_tasks.py +++ b/tests/app/celery/test_scheduled_tasks.py @@ -321,11 +321,11 @@ def test_replay_created_notifications( ) replay_created_notifications() - _, kwargs = mocked.call_args.args - assert kwargs['notification_id'] == str(old_notification.id) - assert mocked.call_args.kwargs['queue'] == f'send-{notification_type}-tasks' mocked.assert_called_once() + args, kwargs = mocked.call_args + assert args[1] == {'notification_id': str(old_notification.id), 'sms_sender_id': None} + assert kwargs['queue'] == f'send-{notification_type}-tasks' @pytest.mark.serial