Reassign vaccination records, batches and sessions

When splitting the East of England team in to six teams, we need to make
sure that the batches, sessions and vaccination records get reassigned
to the most approriate team.

Jira-Issue: MAV-1664

Author: thomasleese

app/lib/team_migration/east_of_england.rb

@@ -81,6 +81,22 @@ def process_row(urn, sen, team)
 
     programmes = (sen ? [flu_programme, hpv_programme] : [hpv_programme])
     add_school_year_groups(location, programmes, sen:)
+
+    location.sessions.find_each do |session|
+      log("Reassigning session to #{team.workgroup}")
+      session.update!(team:)
+
+      session
+        .vaccination_records
+        .includes(:batch)
+        .find_each do |vaccination_record|
+          batch = vaccination_record.batch
+          if batch.team_id != team.id
+            log("Reassigning batch #{batch.name} to #{team.workgroup}")
+            batch.update!(team:)
+          end
+        end
+    end
   end
 
   def school_rows

spec/lib/team_migration/east_of_england_spec.rb

@@ -18,6 +18,30 @@
     )
   end
 
+  let!(:existing_school) { create(:school, urn: "109464", team: current_team) }
+
+  let!(:existing_session) do
+    create(
+      :session,
+      location: existing_school,
+      team: current_team,
+      programmes: [hpv_programme]
+    )
+  end
+
+  let!(:existing_batch) do
+    create(:batch, team: current_team, vaccine: hpv_programme.vaccines.first)
+  end
+
+  let!(:existing_vaccination_record) do
+    create(
+      :vaccination_record,
+      session: existing_session,
+      batch: existing_batch,
+      programme: hpv_programme
+    )
+  end
+
   let(:csv_data) { <<-CSV }
 URN,SEN,ICB Shortname
 109464,,BLMK ICB
@@ -39,7 +63,8 @@
 
     allow(CSV).to receive(:foreach).and_return(parsed_csv_data)
 
-    parsed_csv_data.each do |row|
+    # The first school is `existing_school`.
+    parsed_csv_data[1..].each do |row|
       create(
         :school,
         urn: row.fetch("URN"),
@@ -93,6 +118,22 @@
     expect { current_team.reload }.to raise_error(ActiveRecord::RecordNotFound)
   end
 
+  it "reassigns batches, sessions and vaccination records" do
+    expect(existing_school.team).to eq(current_team)
+    expect(existing_session.team).to eq(current_team)
+    expect(existing_batch.team).to eq(current_team)
+    expect(existing_vaccination_record.team).to eq(current_team)
+
+    call
+    new_team =
+      organisation.teams.find_by(workgroup: "bedslutonmiltonkeynessais")
+
+    expect(existing_school.reload.team).to eq(new_team)
+    expect(existing_session.reload.team).to eq(new_team)
+    expect(existing_batch.reload.team).to eq(new_team)
+    expect(existing_vaccination_record.reload.team).to eq(new_team)
+  end
+
   context "with patients in community clinic" do
     let(:current_session) do
       current_team.generic_clinic_session(academic_year: AcademicYear.current)