refactor ManagingOrganisation question

app/models/form/lettings/questions/managing_organisation.rb

@@ -1,48 +1,45 @@
 class Form::Lettings::Questions::ManagingOrganisation < ::Form::Question
-  attr_accessor :current_user, :log
-
   def initialize(id, hsh, page)
     super
     @id = "managing_organisation_id"
     @check_answer_label = "Managing agent"
     @header = "Which organisation manages this letting?"
     @type = "select"
-    @answer_options = answer_options
   end
 
-  def answer_options
+  def answer_options(log = nil, user = nil)
     opts = { "" => "Select an option" }
 
     return opts unless ActiveRecord::Base.connected?
-    return opts unless current_user
+    return opts unless user
     return opts unless log
 
     if log.managing_organisation.present?
       opts = opts.merge({ log.managing_organisation.id => log.managing_organisation.name })
     end
 
-    if current_user.support?
+    if user.support?
       if log.owning_organisation.holds_own_stock?
         opts[log.owning_organisation.id] = "#{log.owning_organisation.name} (Owning organisation)"
       end
     else
-      opts[current_user.organisation.id] = "#{current_user.organisation.name} (Your organisation)"
+      opts[user.organisation.id] = "#{user.organisation.name} (Your organisation)"
     end
 
-    opts.merge(managing_organisations_answer_options)
+    orgs = if user.support?
+             log.owning_organisation
+           else
+             user.organisation
+           end.managing_agents.pluck(:id, :name).to_h
+
+    opts.merge(orgs)
   end
 
   def displayed_answer_options(log, user)
-    @current_user = user
-    @log = log
-
-    answer_options
+    answer_options(log, user)
   end
 
-  def label_from_value(value, log = nil, user = nil)
-    @log = log
-    @current_user = user
-
+  def label_from_value(value, _log = nil, _user = nil)
     return unless value
 
     answer_options[value]
@@ -53,25 +50,20 @@ class Form::Lettings::Questions::ManagingOrganisation < ::Form::Question
   end
 
   def hidden_in_check_answers?(log, user = nil)
-    @current_user = user
-    @current_user.nil? || !@page.routed_to?(log, user)
+    user.nil? || !@page.routed_to?(log, user)
   end
 
   def enabled
     true
   end
 
+  def answer_label(log, _current_user = nil)
+    Organisation.find_by(id: log.managing_organisation_id)&.name
+  end
+
 private
 
   def selected_answer_option_is_derived?(_log)
     true
   end
-
-  def managing_organisations_answer_options
-    if current_user.support?
-      log.owning_organisation
-    else
-      current_user.organisation
-    end.managing_agents.pluck(:id, :name).to_h
-  end
 end

spec/models/form/lettings/questions/managing_organisation_spec.rb

@@ -184,4 +184,36 @@ RSpec.describe Form::Lettings::Questions::ManagingOrganisation, type: :model do
       end
     end
   end
+
+  describe "#answer_label" do
+    context "when answered" do
+      let(:managing_organisation) { create(:organisation) }
+      let(:log) { create(:lettings_log, managing_organisation:) }
+
+      it "returns org name" do
+        expect(question.answer_label(log)).to eq(managing_organisation.name)
+      end
+    end
+
+    context "when unanswered" do
+      let(:log) { create(:lettings_log, managing_organisation: nil) }
+
+      it "returns nil" do
+        expect(question.answer_label(log)).to be_nil
+      end
+    end
+
+    context "when org does not exist" do
+      let(:managing_organisation) { create(:organisation) }
+      let(:log) { create(:lettings_log, managing_organisation:) }
+
+      before do
+        allow(Organisation).to receive(:find_by).and_return(nil)
+      end
+
+      it "returns nil" do
+        expect(question.answer_label(log)).to be_nil
+      end
+    end
+  end
 end