Test wizard completion notice on private platforms

Author: rsmithlal

app/controllers/concerns/better_together/wizard_methods.rb

@@ -15,9 +15,8 @@ def determine_wizard_outcome # rubocop:todo Metrics/AbcSize
       raise StandardError, "Wizard #{wizard_identifier} was not found. Have you run the seeds?" unless wizard
 
       if wizard.completed?
-        flash[:notice] = wizard.success_message
-        # TODO: This needs to be adjusted for private platforms. Flash message is not retained after wizard completion
-        redirect_to wizard.success_path
+        flash.keep(:notice)
+        redirect_to wizard.success_path, notice: wizard.success_message
       else
         next_step_path, flash_key, message = wizard_next_step_info
         flash[flash_key] = message if message

spec/requests/better_together/wizard_completion_spec.rb

@@ -0,0 +1,22 @@
+# frozen_string_literal: true
+
+require 'rails_helper'
+
+RSpec.describe 'Setup Wizard completion', type: :request do
+  let!(:platform) { create(:better_together_platform, :host, privacy: 'private') }
+  let!(:wizard) { BetterTogether::Wizard.find_by!(identifier: 'host_setup') }
+  let!(:user) { create(:better_together_user, :confirmed, :platform_manager) }
+
+  before do
+    wizard.mark_completed
+    login_as(user, scope: :user)
+  end
+
+  it 'redirects to the success path with notice preserved' do
+    get better_together.setup_wizard_path(locale: I18n.locale)
+    expect(response).to redirect_to(wizard.success_path)
+
+    follow_redirect!
+    expect(response.body).to include(wizard.success_message)
+  end
+end