only display toggle scheme/location for active and deactivted schemes

app/helpers/locations_helper.rb

@@ -53,6 +53,11 @@ module LocationsHelper
     availability.strip
   end
 
+  def toggle_location_link(location)
+    return govuk_button_link_to "Deactivate this location", scheme_location_new_deactivation_path(location.scheme, location), warning: true if location.active?
+    return govuk_button_link_to "Reactivate this location", scheme_location_new_reactivation_path(location.scheme, location) if location.deactivated?
+  end
+
 private
 
   ActivePeriod = Struct.new(:from, :to)

app/helpers/schemes_helper.rb

@@ -42,6 +42,11 @@ module SchemesHelper
     availability.strip
   end
 
+  def toggle_scheme_link(scheme)
+    return govuk_button_link_to "Deactivate this scheme", scheme_new_deactivation_path(scheme), warning: true if scheme.active?
+    return govuk_button_link_to "Reactivate this scheme", scheme_new_reactivation_path(scheme) if scheme.deactivated?
+  end
+
 private
 
   ActivePeriod = Struct.new(:from, :to)

app/models/location.rb

@@ -398,6 +398,10 @@ class Location < ApplicationRecord
     status == :active
   end
 
+  def deactivated?
+    status == :deactivated
+  end
+
   def reactivating_soon?
     status == :reactivating_soon
   end

app/models/scheme.rb

@@ -241,4 +241,8 @@ class Scheme < ApplicationRecord
   def reactivating_soon?
     status == :reactivating_soon
   end
+
+  def deactivated?
+    status == :deactivated
+  end
 end

app/views/locations/show.html.erb

@@ -24,9 +24,5 @@
     </div>
 </div>
 <% if FeatureToggle.location_toggle_enabled? %>
-    <% if @location.active? || @location.reactivating_soon? %>
+    <%= toggle_location_link(@location) %>
-        <%= govuk_button_link_to "Deactivate this location", scheme_location_new_deactivation_path(@scheme, @location), warning: true %>
-    <% else %>
-        <%= govuk_button_link_to "Reactivate this location", scheme_location_new_reactivation_path(@scheme, @location) %>
-    <% end %>
 <% end %>

app/views/schemes/show.html.erb

@@ -34,9 +34,5 @@
 <% end %>
 
 <% if FeatureToggle.scheme_toggle_enabled? %>
-  <% if @scheme.active? || @scheme.reactivating_soon? %>
+  <%= toggle_scheme_link(@scheme) %>
-    <%= govuk_button_link_to "Deactivate this scheme", scheme_new_deactivation_path(@scheme), warning: true %>
-  <% else %>
-    <%= govuk_button_link_to "Reactivate this scheme", scheme_new_reactivation_path(@scheme) %>
-  <% end %>
 <% end %>

spec/requests/locations_controller_spec.rb

@@ -1519,18 +1519,20 @@ RSpec.describe LocationsController, type: :request do
       context "with location that's deactivating soon" do
         let(:location_deactivation_period) { FactoryBot.create(:location_deactivation_period, deactivation_date: Time.zone.local(2022, 10, 12), location:) }
 
-        it "renders reactivate this location" do
+        it "does not render toggle location link" do
           expect(response).to have_http_status(:ok)
-          expect(page).to have_link("Reactivate this location", href: "/schemes/#{scheme.id}/locations/#{location.id}/new-reactivation")
+          expect(page).not_to have_link("Reactivate this location")
+          expect(page).not_to have_link("Deactivate this location")
         end
       end
 
       context "with location that's reactivating soon" do
         let(:location_deactivation_period) { FactoryBot.create(:location_deactivation_period, deactivation_date: Time.zone.local(2022, 4, 12), reactivation_date: Time.zone.local(2022, 10, 12), location:) }
 
-        it "renders reactivate this location" do
+        it "does not render toggle location link" do
           expect(response).to have_http_status(:ok)
-          expect(page).to have_link("Deactivate this location", href: "/schemes/#{scheme.id}/locations/#{location.id}/new-deactivation")
+          expect(page).not_to have_link("Reactivate this location")
+          expect(page).not_to have_link("Deactivate this location")
         end
       end
     end

spec/requests/schemes_controller_spec.rb

@@ -292,9 +292,10 @@ RSpec.describe SchemesController, type: :request do
         context "with scheme that's deactivating soon" do
           let(:scheme_deactivation_period) { FactoryBot.create(:scheme_deactivation_period, deactivation_date: Time.zone.local(2022, 10, 12), scheme:) }
 
-          it "renders reactivate this scheme" do
+          it "does not render toggle scheme link" do
             expect(response).to have_http_status(:ok)
-            expect(page).to have_link("Reactivate this scheme", href: "/schemes/#{scheme.id}/new-reactivation")
+            expect(page).not_to have_link("Reactivate this scheme")
+            expect(page).not_to have_link("Deactivate this scheme")
           end
         end
       end