diff --git a/app/main/views/send.py b/app/main/views/send.py index 84100af2d..08af846c1 100644 --- a/app/main/views/send.py +++ b/app/main/views/send.py @@ -547,6 +547,7 @@ def _check_messages(service_id, template_id, upload_id, preview_row, letters_as_ if any(recipients) and not recipients.has_errors: session['file_uploads'][upload_id]['notification_count'] = len(recipients) + session['file_uploads'][upload_id]['template_id'] = str(template_id) session['file_uploads'][upload_id]['valid'] = True else: session['file_uploads'].pop(upload_id) diff --git a/tests/app/main/views/test_send.py b/tests/app/main/views/test_send.py index 0c6795a88..473bdf420 100644 --- a/tests/app/main/views/test_send.py +++ b/tests/app/main/views/test_send.py @@ -1443,9 +1443,9 @@ def test_upload_csvfile_with_valid_phone_shows_all_numbers( follow_redirects=True ) with logged_in_client.session_transaction() as sess: - assert 'template_id' not in sess['file_uploads'][fake_uuid] assert 'original_file_name' not in sess['file_uploads'][fake_uuid] assert sess['file_uploads'][fake_uuid]['notification_count'] == 53 + assert sess['file_uploads'][fake_uuid]['template_id'] == fake_uuid assert sess['file_uploads'][fake_uuid]['valid'] is True content = response.get_data(as_text=True)