Add school_id column

This adds a `school_id` column to the `patient_locations` table which
will eventually replace the `location_id` column. We're doing this
because the plan is for the patient-location association to belong an
explicit patient-school association, while the association between
patients and clinics is built using a different mechanism.

Jira-Issue: MAV-7114
Jira-Issue: MAV-7113

Author: thomasleese

app/models/patient_location.rb

@@ -11,18 +11,24 @@
 #  updated_at    :datetime         not null
 #  location_id   :bigint           not null
 #  patient_id    :bigint           not null
+#  school_id     :bigint
 #
 # Indexes
 #
 #  idx_on_location_id_academic_year_patient_id_3237b32fa0    (location_id,academic_year,patient_id) UNIQUE
 #  idx_on_patient_id_location_id_academic_year_08a1dc4afe    (patient_id,location_id,academic_year) UNIQUE
+#  idx_on_patient_id_school_id_academic_year_652216fa07      (patient_id,school_id,academic_year) UNIQUE
+#  idx_on_school_id_academic_year_patient_id_c647e75f26      (school_id,academic_year,patient_id) UNIQUE
 #  index_patient_locations_on_location_id                    (location_id)
 #  index_patient_locations_on_location_id_and_academic_year  (location_id,academic_year)
+#  index_patient_locations_on_school_id                      (school_id)
+#  index_patient_locations_on_school_id_and_academic_year    (school_id,academic_year)
 #
 # Foreign Keys
 #
 #  fk_rails_...  (location_id => locations.id)
 #  fk_rails_...  (patient_id => patients.id)
+#  fk_rails_...  (school_id => locations.id)
 #
 
 class PatientLocation < ApplicationRecord

db/migrate/20260427122013_add_school_to_patient_locations.rb

@@ -0,0 +1,27 @@
+# frozen_string_literal: true
+
+class AddSchoolToPatientLocations < ActiveRecord::Migration[8.1]
+  disable_ddl_transaction!
+
+  def change
+    add_reference :patient_locations,
+                  :school,
+                  foreign_key: {
+                    to_table: :locations
+                  }
+
+    add_index :patient_locations,
+              %i[school_id academic_year],
+              algorithm: :concurrently
+
+    add_index :patient_locations,
+              %i[school_id academic_year patient_id],
+              unique: true,
+              algorithm: :concurrently
+
+    add_index :patient_locations,
+              %i[patient_id school_id academic_year],
+              unique: true,
+              algorithm: :concurrently
+  end
+end

db/schema.rb

@@ -652,11 +652,16 @@
     t.daterange "date_range", default: -::Float::INFINITY...::Float::INFINITY, null: false
     t.bigint "location_id", null: false
     t.bigint "patient_id", null: false
+    t.bigint "school_id"
     t.datetime "updated_at", null: false
     t.index ["location_id", "academic_year", "patient_id"], name: "idx_on_location_id_academic_year_patient_id_3237b32fa0", unique: true
     t.index ["location_id", "academic_year"], name: "index_patient_locations_on_location_id_and_academic_year"
     t.index ["location_id"], name: "index_patient_locations_on_location_id"
     t.index ["patient_id", "location_id", "academic_year"], name: "idx_on_patient_id_location_id_academic_year_08a1dc4afe", unique: true
+    t.index ["patient_id", "school_id", "academic_year"], name: "idx_on_patient_id_school_id_academic_year_652216fa07", unique: true
+    t.index ["school_id", "academic_year", "patient_id"], name: "idx_on_school_id_academic_year_patient_id_c647e75f26", unique: true
+    t.index ["school_id", "academic_year"], name: "index_patient_locations_on_school_id_and_academic_year"
+    t.index ["school_id"], name: "index_patient_locations_on_school_id"
   end
 
   create_table "patient_merge_log_entries", force: :cascade do |t|
@@ -1166,6 +1171,7 @@
   add_foreign_key "patient_changesets", "locations", column: "school_id"
   add_foreign_key "patient_changesets", "patients"
   add_foreign_key "patient_locations", "locations"
+  add_foreign_key "patient_locations", "locations", column: "school_id"
   add_foreign_key "patient_locations", "patients"
   add_foreign_key "patient_merge_log_entries", "patients", on_delete: :cascade
   add_foreign_key "patient_merge_log_entries", "users"

spec/factories/patient_locations.rb

@@ -11,18 +11,24 @@
 #  updated_at    :datetime         not null
 #  location_id   :bigint           not null
 #  patient_id    :bigint           not null
+#  school_id     :bigint
 #
 # Indexes
 #
 #  idx_on_location_id_academic_year_patient_id_3237b32fa0    (location_id,academic_year,patient_id) UNIQUE
 #  idx_on_patient_id_location_id_academic_year_08a1dc4afe    (patient_id,location_id,academic_year) UNIQUE
+#  idx_on_patient_id_school_id_academic_year_652216fa07      (patient_id,school_id,academic_year) UNIQUE
+#  idx_on_school_id_academic_year_patient_id_c647e75f26      (school_id,academic_year,patient_id) UNIQUE
 #  index_patient_locations_on_location_id                    (location_id)
 #  index_patient_locations_on_location_id_and_academic_year  (location_id,academic_year)
+#  index_patient_locations_on_school_id                      (school_id)
+#  index_patient_locations_on_school_id_and_academic_year    (school_id,academic_year)
 #
 # Foreign Keys
 #
 #  fk_rails_...  (location_id => locations.id)
 #  fk_rails_...  (patient_id => patients.id)
+#  fk_rails_...  (school_id => locations.id)
 #
 FactoryBot.define do
   factory :patient_location do

spec/models/patient_location_spec.rb

@@ -11,18 +11,24 @@
 #  updated_at    :datetime         not null
 #  location_id   :bigint           not null
 #  patient_id    :bigint           not null
+#  school_id     :bigint
 #
 # Indexes
 #
 #  idx_on_location_id_academic_year_patient_id_3237b32fa0    (location_id,academic_year,patient_id) UNIQUE
 #  idx_on_patient_id_location_id_academic_year_08a1dc4afe    (patient_id,location_id,academic_year) UNIQUE
+#  idx_on_patient_id_school_id_academic_year_652216fa07      (patient_id,school_id,academic_year) UNIQUE
+#  idx_on_school_id_academic_year_patient_id_c647e75f26      (school_id,academic_year,patient_id) UNIQUE
 #  index_patient_locations_on_location_id                    (location_id)
 #  index_patient_locations_on_location_id_and_academic_year  (location_id,academic_year)
+#  index_patient_locations_on_school_id                      (school_id)
+#  index_patient_locations_on_school_id_and_academic_year    (school_id,academic_year)
 #
 # Foreign Keys
 #
 #  fk_rails_...  (location_id => locations.id)
 #  fk_rails_...  (patient_id => patients.id)
+#  fk_rails_...  (school_id => locations.id)
 #
 
 describe PatientLocation do