From 896c31bad857c1be10fc9e6ac9a5282d4c3eb470 Mon Sep 17 00:00:00 2001
From: JG <moarpheus@gmail.com>
Date: Wed, 29 Jun 2022 16:42:00 +0100
Subject: [PATCH] flashing test

---
 app/controllers/schemes_controller.rb    | 1 +
 app/views/schemes/check_answers.html.erb | 2 +-
 spec/requests/schemes_controller_spec.rb | 8 ++++++++
 3 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/app/controllers/schemes_controller.rb b/app/controllers/schemes_controller.rb
index 1e3db8de3..cd9065e05 100644
--- a/app/controllers/schemes_controller.rb
+++ b/app/controllers/schemes_controller.rb
@@ -7,6 +7,7 @@ class SchemesController < ApplicationController
   before_action :authenticate_scope!
 
   def index
+    flash[:notice] = ("#{Scheme.find(params[:scheme_id].to_i).service_name} has been created.") if params[:scheme_id]
     redirect_to schemes_organisation_path(current_user.organisation) unless current_user.support?
     all_schemes = Scheme.all
 
diff --git a/app/views/schemes/check_answers.html.erb b/app/views/schemes/check_answers.html.erb
index 64f64a77c..78fff8674 100644
--- a/app/views/schemes/check_answers.html.erb
+++ b/app/views/schemes/check_answers.html.erb
@@ -62,4 +62,4 @@
   <% end %>
 <% end %>
 
-<%= govuk_button_link_to "Create scheme", schemes_path, html: { method: :get } %>
+<%= govuk_button_link_to "Create scheme", schemes_path(scheme_id: @scheme.id), html: { method: :get } %>
diff --git a/spec/requests/schemes_controller_spec.rb b/spec/requests/schemes_controller_spec.rb
index ef9b0a707..ccdeb580b 100644
--- a/spec/requests/schemes_controller_spec.rb
+++ b/spec/requests/schemes_controller_spec.rb
@@ -37,6 +37,14 @@ RSpec.describe SchemesController, type: :request do
         get "/schemes"
       end
 
+      context "params scheme_id is present" do
+        it "shows a success banner" do
+          get "/schemes", params: { scheme_id: schemes.first.id }
+          follow_redirect!
+          expect(page).to have_css(".govuk-notification-banner.govuk-notification-banner--success")
+        end
+      end
+
       it "redirects to the organisation schemes path" do
         follow_redirect!
         expect(path).to match("/organisations/#{user.organisation.id}/schemes")