From a78a27ad4ce85d44231b210322b1084bfd98d3fb Mon Sep 17 00:00:00 2001
From: JG <moarpheus@gmail.com>
Date: Tue, 14 Jun 2022 13:23:43 +0100
Subject: [PATCH] asking for locations in the link

---
 app/views/schemes/show.html.erb | 5 ++++-
 spec/features/schemes_spec.rb   | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/app/views/schemes/show.html.erb b/app/views/schemes/show.html.erb
index 2e7ca22d1..99d677999 100644
--- a/app/views/schemes/show.html.erb
+++ b/app/views/schemes/show.html.erb
@@ -3,8 +3,9 @@
 
 <%= render partial: "organisations/headings", locals: { main: @scheme.service_name, sub: nil } %>
 
+
 <%= render SubNavigationComponent.new(
-  items: scheme_items(request.path, @scheme.id),
+  items: scheme_items(request.path, @scheme.id, @scheme.locations.count),
 ) %>
 
 <div class="govuk-grid-row">
@@ -19,3 +20,5 @@
     <% end %>
   </div>
 </div>
+
+<%= govuk_back_link(href: "#{request.referrer}")%>
diff --git a/spec/features/schemes_spec.rb b/spec/features/schemes_spec.rb
index a09dfe291..18f174890 100644
--- a/spec/features/schemes_spec.rb
+++ b/spec/features/schemes_spec.rb
@@ -130,7 +130,7 @@ RSpec.describe "Supported housing scheme Features" do
           end
 
           it "shows service and locations tab" do
-            expect(page).to have_link("Service")
+            expect(page).to have_link("Scheme")
             expect(page).to have_link("#{schemes.count} locations")
           end
         end