From 156f992502af0777d0b5b013cd062459551f112c Mon Sep 17 00:00:00 2001
From: kosiakkatrina <54268893+kosiakkatrina@users.noreply.github.com>
Date: Thu, 19 Oct 2023 08:26:39 +0100
Subject: [PATCH] Correct location admin district (#1986)

---
 .../correct_location_admin_district.rake      |  6 +++
 .../correct_location_admin_district_spec.rb   | 52 +++++++++++++++++++
 2 files changed, 58 insertions(+)
 create mode 100644 lib/tasks/correct_location_admin_district.rake
 create mode 100644 spec/lib/tasks/correct_location_admin_district_spec.rb

diff --git a/lib/tasks/correct_location_admin_district.rake b/lib/tasks/correct_location_admin_district.rake
new file mode 100644
index 000000000..cdc36eebd
--- /dev/null
+++ b/lib/tasks/correct_location_admin_district.rake
@@ -0,0 +1,6 @@
+desc "Infers location admin district for locations from location code where it's missing"
+task correct_location_admin_district: :environment do
+  Location.where.not(location_code: nil).where(location_admin_district: nil).each do |location|
+    location.update(location_admin_district: LocalAuthority.all.active(Time.zone.today).england.find_by(code: location.location_code)&.name)
+  end
+end
diff --git a/spec/lib/tasks/correct_location_admin_district_spec.rb b/spec/lib/tasks/correct_location_admin_district_spec.rb
new file mode 100644
index 000000000..85bc9d054
--- /dev/null
+++ b/spec/lib/tasks/correct_location_admin_district_spec.rb
@@ -0,0 +1,52 @@
+require "rails_helper"
+require "rake"
+
+RSpec.describe "correct_location_admin_district" do
+  describe ":correct_location_admin_district", type: :task do
+    subject(:task) { Rake::Task["correct_location_admin_district"] }
+
+    before do
+      Rake.application.rake_require("tasks/correct_location_admin_district")
+      Rake::Task.define_task(:environment)
+      task.reenable
+    end
+
+    context "when the rake task is run" do
+      let!(:scheme) { create(:scheme) }
+      let!(:location) { create(:location, postcode: "A11AA", location_code: "E06000009", location_admin_district: nil, startdate: nil) }
+
+      it "updates location with missing location admin district and marks it as complete" do
+        expect(location.confirmed).to eq(false)
+        task.invoke
+        location.reload
+        scheme.reload
+        expect(location.confirmed).to eq(true)
+        expect(location.location_admin_district).to eq("Blackpool")
+      end
+
+      it "does not mark location as complete if other fields are missing" do
+        location.update!(units: nil)
+        expect(location.confirmed).to eq(false)
+        task.invoke
+        location.reload
+        scheme.reload
+        expect(location.confirmed).to eq(false)
+        expect(location.location_admin_district).to eq("Blackpool")
+      end
+
+      it "does not override existing location admin district" do
+        location.location_admin_district = "Babergh"
+        location.save!(validate: false)
+        task.invoke
+        expect(location.reload.location_admin_district).to eq("Babergh")
+      end
+
+      it "does not set location admin district if it cannot be found" do
+        location.location_code = "123"
+        location.save!(validate: false)
+        task.invoke
+        expect(location.reload.location_admin_district).to eq(nil)
+      end
+    end
+  end
+end