def test_replay_letters_in_error_for_one_file(notify_api, mocker): mockObject = boto3.resource('s3').Object('ERROR', 'ERROR/file_name') mocker.patch("app.celery.letters_pdf_tasks.get_file_names_from_error_bucket", return_value=[mockObject]) mock_move = mocker.patch("app.celery.letters_pdf_tasks.move_error_pdf_to_scan_bucket") mock_celery = mocker.patch("app.celery.letters_pdf_tasks.notify_celery.send_task") replay_letters_in_error("file_name") mock_move.assert_called_once_with('file_name') mock_celery.assert_called_once_with(name='scan-file', kwargs={'filename': 'file_name'}, queue='antivirus-tasks')
def test_replay_letters_in_error_for_one_file(notify_api, mocker): mockObject = boto3.resource("s3").Object("ERROR", "ERROR/file_name") mocker.patch( "app.celery.letters_pdf_tasks.get_file_names_from_error_bucket", return_value=[mockObject], ) mock_move = mocker.patch( "app.celery.letters_pdf_tasks.move_error_pdf_to_scan_bucket") mock_celery = mocker.patch( "app.celery.letters_pdf_tasks.notify_celery.send_task") replay_letters_in_error("file_name") mock_move.assert_called_once_with("file_name") mock_celery.assert_called_once_with(name="scan-file", kwargs={"filename": "file_name"}, queue="antivirus-tasks")