Fix session selection logic

`Session`s which are `generic_clinic`s use `send_invitations_at`
instead of `send_consent_requests_at`. Both need notifications
to be sent, so must be extracted separately in the DB query.

Author: alistair-white-horne-tng

app/jobs/enqueue_vaccinations_search_in_nhs_job.rb

@@ -8,11 +8,16 @@ def perform(sessions = nil)
       begin
         flu = Programme.flu.sole
         Session
-          .includes(:session_dates)
+          .includes(:session_dates, :location)
           .has_programmes([flu])
-          .where("sessions.send_consent_requests_at <= ?", 2.days.from_now)
+          .where(
+            "(locations.type IN (?) AND sessions.send_invitations_at <= ?) OR " \
+            "(locations.type IN (?, ?) AND sessions.send_consent_requests_at <= ?)",
+            Location.types[:generic_clinic], 2.days.from_now,
+            Location.types[:community_clinic], Location.types[:school], 2.days.from_now
+          )
           .where("session_dates.value >= ?", Time.zone.today)
-          .references(:session_dates)
+          .references(:session_dates, :locations)
       end
 
     patient_ids =

spec/jobs/enqueue_vaccinations_search_in_nhs_job_spec.rb

@@ -14,92 +14,250 @@
 
     before { allow(SearchVaccinationRecordsInNHSJob).to receive(:perform_bulk) }
 
-    let(:send_consent_requests_at) {}
-    let(:days_before_consent_reminders) { 7 }
-    let!(:session) do
-      create(
-        :session,
-        programmes: [flu],
-        academic_year: AcademicYear.pending,
-        dates:,
-        send_consent_requests_at:,
-        days_before_consent_reminders:,
-        team:,
-        location:,
-        patients: [patient]
-      )
-    end
+    context "with a normal school session" do
+      let(:send_consent_requests_at) {}
+      let(:days_before_consent_reminders) { 7 }
+      let!(:session) do
+        create(
+          :session,
+          programmes: [flu],
+          academic_year: AcademicYear.pending,
+          dates:,
+          send_consent_requests_at:,
+          days_before_consent_reminders:,
+          team:,
+          location:,
+          patients: [patient]
+        )
+      end
 
-    context "with an specific, unscheduled session" do
-      before { described_class.perform_now([session]) }
+      context "with an specific, unscheduled session" do
+        before { described_class.perform_now([session]) }
 
-      let(:dates) { [] }
-      let(:days_before_consent_reminders) { nil }
+        let(:dates) { [] }
+        let(:days_before_consent_reminders) { nil }
 
-      it { should have_received(:perform_bulk).once.with([[patient.id]]) }
-    end
+        it { should have_received(:perform_bulk).once.with([[patient.id]]) }
+      end
 
-    context "session with dates in the future" do
-      before { described_class.perform_now }
+      context "session with dates in the future" do
+        before { described_class.perform_now }
 
-      let(:dates) { [7.days.from_now] }
-      let(:send_consent_requests_at) { 14.days.ago }
+        let(:dates) { [7.days.from_now] }
+        let(:send_consent_requests_at) { 14.days.ago }
 
-      it { should have_received(:perform_bulk).once.with([[patient.id]]) }
+        it { should have_received(:perform_bulk).once.with([[patient.id]]) }
 
-      context "community clinic session" do
-        let(:location) { create(:community_clinic, team:, programmes: [flu]) }
-        let(:school) { create(:school, team:, programmes: [flu]) }
+        context "community clinic session" do
+          let(:location) { create(:community_clinic, team:, programmes: [flu]) }
+          let(:school) { create(:school, team:, programmes: [flu]) }
 
-        it { should have_received(:perform_bulk).exactly(:once) }
+          it { should have_received(:perform_bulk).exactly(:once) }
+        end
+
+        context "generic clinic session" do
+          let(:location) { create(:generic_clinic, team:, programmes: [flu]) }
+          let(:school) { create(:school, team:, programmes: [flu]) }
+
+          it { should have_received(:perform_bulk).exactly(:once) }
+        end
       end
 
-      context "generic clinic session" do
-        let(:location) { create(:generic_clinic, team:, programmes: [flu]) }
-        let(:school) { create(:school, team:, programmes: [flu]) }
+      context "session with dates in the past",
+              within_academic_year: {
+                from_start: 7.days
+              } do
+        before { described_class.perform_now }
+
+        let(:dates) { [7.days.ago] }
+        let(:send_consent_requests_at) { 30.days.ago }
+
+        it { should_not have_received(:perform_bulk) }
+      end
+
+      context "session with dates in the past and the future",
+              within_academic_year: {
+                from_start: 7.days
+              } do
+        before { described_class.perform_now }
+
+        let(:send_consent_requests_at) { 28.days.ago }
+        let(:dates) { [7.days.ago, 7.days.from_now] }
 
         it { should have_received(:perform_bulk).exactly(:once) }
       end
-    end
 
-    context "session with dates in the past",
-            within_academic_year: {
-              from_start: 7.days
-            } do
-      before { described_class.perform_now }
+      context "session with send_invitations_at in the future" do
+        before { described_class.perform_now }
 
-      let(:dates) { [7.days.ago] }
-      let(:send_consent_requests_at) { 30.days.ago }
+        let(:send_consent_requests_at) { 2.days.from_now }
+        let(:dates) { [17.days.from_now] }
+
+        it { should have_received(:perform_bulk).exactly(:once) }
+      end
+
+      context "session with send_consent_requests_at too far in the future" do
+      let(:send_consent_requests_at) { 3.days.from_now }
+      let(:dates) { [17.days.from_now] }
 
       it { should_not have_received(:perform_bulk) }
+      end
     end
 
-    context "session with dates in the past and the future",
-            within_academic_year: {
-              from_start: 7.days
-            } do
-      before { described_class.perform_now }
+    shared_examples "notification date logic" do |notification_date_field|
+      context "with notification date within threshold" do
+        let!(:session) do
+          session_attributes = {
+            programmes: [flu],
+            academic_year: AcademicYear.pending,
+            dates:,
+            team:,
+            location:,
+            patients: [patient]
+          }
+          
+          # Set the appropriate notification date field
+          if notification_date_field == :send_invitations_at
+            session_attributes[:send_invitations_at] = 1.day.from_now
+            session_attributes[:send_consent_requests_at] = nil
+          else
+            session_attributes[:send_invitations_at] = nil
+            session_attributes[:send_consent_requests_at] = 1.day.from_now
+            session_attributes[:days_before_consent_reminders] = 7
+          end
+          
+          create(:session, session_attributes)
+        end
 
-      let(:send_consent_requests_at) { 28.days.ago }
-      let(:dates) { [7.days.ago, 7.days.from_now] }
+        before { described_class.perform_now }
 
-      it { should have_received(:perform_bulk).exactly(:once) }
-    end
+        it "includes the session" do
+          expect(SearchVaccinationRecordsInNHSJob).to have_received(:perform_bulk).with(
+            [patient.id].zip
+          )
+        end
+      end
 
-    context "session with send_invitations_at in the future" do
-      before { described_class.perform_now }
+      context "with notification date too far in future" do
+        let!(:session) do
+          session_attributes = {
+            programmes: [flu],
+            academic_year: AcademicYear.pending,
+            dates:,
+            team:,
+            location:,
+            patients: [patient]
+          }
+          
+          # Set the appropriate notification date field
+          if notification_date_field == :send_invitations_at
+            session_attributes[:send_invitations_at] = 3.days.from_now
+            session_attributes[:send_consent_requests_at] = nil
+          else
+            session_attributes[:send_invitations_at] = nil
+            session_attributes[:send_consent_requests_at] = 3.days.from_now
+            session_attributes[:days_before_consent_reminders] = 7
+          end
+          
+          create(:session, session_attributes)
+        end
 
-      let(:send_consent_requests_at) { 2.days.from_now }
-      let(:dates) { [17.days.from_now] }
+        before { described_class.perform_now }
 
-      it { should have_received(:perform_bulk).exactly(:once) }
+        it "excludes the session" do
+          expect(SearchVaccinationRecordsInNHSJob).not_to have_received(:perform_bulk)
+        end
+      end
     end
 
-    context "session with send_consent_requests_at too far in the future" do
-      let(:send_consent_requests_at) { 3.days.from_now }
-      let(:dates) { [17.days.from_now] }
+    context "testing notification date logic for different location types" do
+      let(:dates) { [30.days.from_now] }
 
-      it { should_not have_received(:perform_bulk) }
+      context "generic clinic sessions" do
+        let(:location) { create(:generic_clinic, team:, programmes: [flu]) }
+        let(:school) { create(:school, team:, programmes: [flu]) }
+
+        include_examples "notification date logic", :send_invitations_at
+      end
+
+      context "community clinic sessions" do
+        let(:location) { create(:community_clinic, team:, programmes: [flu]) }
+        let(:school) { create(:school, team:, programmes: [flu]) }
+
+        include_examples "notification date logic", :send_consent_requests_at
+      end
+
+      context "school sessions" do
+        let(:location) { create(:school, team:, programmes: [flu]) }
+        let(:school) { location }
+
+        include_examples "notification date logic", :send_consent_requests_at
+      end
+
+      context "mixed session types" do
+        let(:generic_clinic) { create(:generic_clinic, team:, programmes: [flu]) }
+        let(:community_clinic) { create(:community_clinic, team:, programmes: [flu]) }
+        let(:school_location) { create(:school, team:, programmes: [flu]) }
+
+        let!(:first_patient) { create(:patient, team:, school:) }
+        let!(:second_patient) { create(:patient, team:, school:) }
+        let!(:third_patient) { create(:patient, team:, school:) }
+
+        let!(:generic_session) do
+          create(
+            :session,
+            programmes: [flu],
+            academic_year: AcademicYear.pending,
+            dates:,
+            send_invitations_at: 1.day.from_now,
+            send_consent_requests_at: nil,
+            days_before_consent_reminders: nil,
+            team:,
+            location: generic_clinic,
+            patients: [first_patient]
+          )
+        end
+
+        let!(:community_session) do
+          create(
+            :session,
+            programmes: [flu],
+            academic_year: AcademicYear.pending,
+            dates:,
+            send_invitations_at: nil,
+            send_consent_requests_at: 1.day.from_now,
+            days_before_consent_reminders: 7,
+            team:,
+            location: community_clinic,
+            patients: [second_patient]
+          )
+        end
+
+        let!(:school_session) do
+          create(
+            :session,
+            programmes: [flu],
+            academic_year: AcademicYear.pending,
+            dates:,
+            send_invitations_at: nil,
+            send_consent_requests_at: 1.day.from_now,
+            days_before_consent_reminders: 7,
+            team:,
+            location: school_location,
+            patients: [third_patient]
+          )
+        end
+
+        before { described_class.perform_now }
+
+        it "includes all sessions when their respective notification dates are within threshold" do
+          expect(SearchVaccinationRecordsInNHSJob).to have_received(:perform_bulk).once
+
+          expect(SearchVaccinationRecordsInNHSJob).to have_received(:perform_bulk).with(
+            [first_patient.id, second_patient.id, third_patient.id].zip
+          )
+        end
+      end
     end
   end
 end