From e3a90a54e4848fc1d493f08a8c40200b6a60209d Mon Sep 17 00:00:00 2001
From: Kat <katrina@kosiak.co.uk>
Date: Fri, 2 Feb 2024 09:23:32 +0000
Subject: [PATCH] Update routing for LaNominations

---
 app/models/form/sales/pages/la_nominations.rb |  6 +++
 .../form/sales/pages/la_nominations_spec.rb   | 53 +++++++++++++++++++
 2 files changed, 59 insertions(+)

diff --git a/app/models/form/sales/pages/la_nominations.rb b/app/models/form/sales/pages/la_nominations.rb
index e756e3071..0ac2cde75 100644
--- a/app/models/form/sales/pages/la_nominations.rb
+++ b/app/models/form/sales/pages/la_nominations.rb
@@ -9,4 +9,10 @@ class Form::Sales::Pages::LaNominations < ::Form::Page
       Form::Sales::Questions::LaNominations.new(nil, nil, self),
     ]
   end
+
+  def routed_to?(log, _current_user)
+    return false if log.staircase == 1 && log.form.start_year_after_2024?
+
+    super
+  end
 end
diff --git a/spec/models/form/sales/pages/la_nominations_spec.rb b/spec/models/form/sales/pages/la_nominations_spec.rb
index b7b40c2dd..ab41d7f94 100644
--- a/spec/models/form/sales/pages/la_nominations_spec.rb
+++ b/spec/models/form/sales/pages/la_nominations_spec.rb
@@ -3,10 +3,16 @@ require "rails_helper"
 RSpec.describe Form::Sales::Pages::LaNominations, type: :model do
   subject(:page) { described_class.new(page_id, page_definition, subsection) }
 
+  let(:log) { create(:sales_log, :completed) }
+
   let(:page_id) { nil }
   let(:page_definition) { nil }
   let(:subsection) { instance_double(Form::Subsection) }
 
+  before do
+    allow(subsection).to receive(:depends_on).and_return(nil)
+  end
+
   it "has correct subsection" do
     expect(page.subsection).to eq(subsection)
   end
@@ -26,4 +32,51 @@ RSpec.describe Form::Sales::Pages::LaNominations, type: :model do
   it "has the correct description" do
     expect(page.description).to be_nil
   end
+
+  context "with 23/24 log" do
+    before do
+      Timecop.freeze(Time.zone.local(2023, 4, 2))
+      Singleton.__init__(FormHandler)
+    end
+
+    after do
+      Timecop.return
+    end
+
+    it "has correct routed to" do
+      log.staircase = 1
+      expect(page.routed_to?(log, nil)).to eq(true)
+    end
+  end
+
+  context "with 24/25 log" do
+    before do
+      Timecop.freeze(Time.zone.local(2024, 4, 2))
+      Singleton.__init__(FormHandler)
+    end
+
+    after do
+      Timecop.return
+    end
+
+    it "has correct routed to when staircase is yes" do
+      log.staircase = 1
+      expect(page.routed_to?(log, nil)).to eq(false)
+    end
+
+    it "has correct routed to when staircase is nil" do
+      log.staircase = nil
+      expect(page.routed_to?(log, nil)).to eq(true)
+    end
+
+    it "has correct routed to when staircase is no" do
+      log.staircase = 2
+      expect(page.routed_to?(log, nil)).to eq(true)
+    end
+
+    it "has correct routed to when staircase is don't know" do
+      log.staircase = 3
+      expect(page.routed_to?(log, nil)).to eq(true)
+    end
+  end
 end