Refactored test for new apply_async

department-of-veterans-affairs · Dec 3, 2024 · 5d8d26a · 5d8d26a
1 parent 73f35af
commit 5d8d26a
Showing 1 changed file with 3 additions and 3 deletions.
diff --git 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