diff --git a/spec/helpers/form_page_helper_spec.rb b/spec/helpers/form_page_helper_spec.rb index fc304fd89..aeded6f8d 100644 --- a/spec/helpers/form_page_helper_spec.rb +++ b/spec/helpers/form_page_helper_spec.rb @@ -3,13 +3,26 @@ require "rails_helper" RSpec.describe FormPageHelper do describe "#action_href" do let(:lettings_log) { FactoryBot.create(:lettings_log) } + let(:sales_log) { FactoryBot.create(:sales_log) } - it "has an update answer link href helper" do - expect(action_href(lettings_log, "net_income")).to eq("/lettings-logs/#{lettings_log.id}/net-income?referrer=check_answers") + context "with a lettings log" do + it "has an update answer link href helper" do + expect(action_href(lettings_log, "net_income")).to eq("/lettings-logs/#{lettings_log.id}/net-income?referrer=check_answers") + end + + it "returns a correct referrer in the url" do + expect(action_href(lettings_log, "retirement_value_check", "interruption_screen")).to eq("/lettings-logs/#{lettings_log.id}/retirement-value-check?referrer=interruption_screen") + end end - it "returns a correct referrer in the url" do - expect(action_href(lettings_log, "retirement_value_check", "interruption_screen")).to eq("/lettings-logs/#{lettings_log.id}/retirement-value-check?referrer=interruption_screen") + context "with a sales log" do + it "has an update answer link href helper" do + expect(action_href(sales_log, "buyer_1_age")).to eq("/sales-logs/#{sales_log.id}/buyer-1-age?referrer=check_answers") + end + + it "returns a correct referrer in the url" do + expect(action_href(sales_log, "grant_value_check", "interruption_screen")).to eq("/sales-logs/#{sales_log.id}/grant-value-check?referrer=interruption_screen") + end end end end