From 5ad800075acf35b9e8f7dd81b42c94e9ba891dbb Mon Sep 17 00:00:00 2001
From: Kat <katrina@kosiak.co.uk>
Date: Wed, 25 Oct 2023 11:01:46 +0100
Subject: [PATCH] Infer lar and irproduct

---
 .../lettings_log_variables.rb                 | 13 +++++
 spec/models/lettings_log_spec.rb              | 52 +++++++++++++++++++
 2 files changed, 65 insertions(+)

diff --git a/app/models/derived_variables/lettings_log_variables.rb b/app/models/derived_variables/lettings_log_variables.rb
index 9679a9839..7d1274540 100644
--- a/app/models/derived_variables/lettings_log_variables.rb
+++ b/app/models/derived_variables/lettings_log_variables.rb
@@ -45,6 +45,8 @@ module DerivedVariables::LettingsLogVariables
     end
     self.renttype = RENT_TYPE_MAPPING[rent_type]
     self.lettype = get_lettype
+    self.lar = get_lar
+    self.irproduct = get_irproduct
     self.totchild = get_totchild
     self.totelder = get_totelder
     self.totadult = get_totadult
@@ -307,4 +309,15 @@ private
     self.town_or_city = nil
     self.county = nil
   end
+
+  def get_lar
+    return 1 if rent_type == 2
+    return 2 if rent_type == 1
+  end
+
+  def get_irproduct
+    return 1 if rent_type == 3
+    return 2 if rent_type == 4
+    return 3 if rent_type == 5
+  end
 end
diff --git a/spec/models/lettings_log_spec.rb b/spec/models/lettings_log_spec.rb
index 82cc254cb..6c4d9b1e0 100644
--- a/spec/models/lettings_log_spec.rb
+++ b/spec/models/lettings_log_spec.rb
@@ -2270,6 +2270,58 @@ RSpec.describe LettingsLog do
         expect(lettings_log["housingneeds_g"]).to eq(1)
       end
     end
+
+    context "when saving rent_type" do
+      it "derives lar as yes (1) if rent_type is london affordable rent" do
+        lettings_log.update!({ rent_type: 2 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["lar"]).to eq(1)
+      end
+
+      it "derives lar as no (2) if rent_type is affordable rent" do
+        lettings_log.update!({ rent_type: 1 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["lar"]).to eq(2)
+      end
+
+      it "clears previously set lar if rent_type is not affordable rent" do
+        lettings_log.update!({ rent_type: 2 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["lar"]).to eq(1)
+
+        lettings_log.update!({ rent_type: 3 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["lar"]).to eq(nil)
+      end
+
+      it "derives irproduct as rent_to_buy (1) if rent_type is rent_to_buy (3)" do
+        lettings_log.update!({ rent_type: 3 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["irproduct"]).to eq(1)
+      end
+
+      it "derives irproduct as london_living_rent (2) if rent_type is london_living_rent (4)" do
+        lettings_log.update!({ rent_type: 4 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["irproduct"]).to eq(2)
+      end
+
+      it "derives irproduct as other_intermediate_rent_product (3) if rent_type is other_intermediate_rent_product (5)" do
+        lettings_log.update!({ rent_type: 5, irproduct_other: "other" })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["irproduct"]).to eq(3)
+      end
+
+      it "clears previously set irproduct if rent_type is intermediate rent" do
+        lettings_log.update!({ rent_type: 4 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["irproduct"]).to eq(2)
+
+        lettings_log.update!({ rent_type: 2 })
+        record_from_db = described_class.find(lettings_log.id)
+        expect(record_from_db["irproduct"]).to eq(nil)
+      end
+    end
   end
 
   describe "optional fields" do