Only reset NR objects created before a team's cutoff date

Because we weren't able to reset the piloting national reporting teams
before they onboarded, we must now tweak this CLI tool to ensure that
we don't affect information which the teams have uploaded since they
onboarded for real, on their `national_reporting_cut_off_date`.

Jira-Issue: MAV-7080

Author: alistair-white-horne-tng

app/lib/mavis_cli/teams/reset_national_reporting.rb

@@ -29,7 +29,8 @@ def call(workgroup: nil, force: false, **)
         puts "Found #{teams.count} national reporting team(s) to reset:"
         teams.each do |team|
           puts "  - #{team.name} (#{team.workgroup})"
-          puts "    - Immunisation imports: #{ImmunisationImport.where(team:).count}"
+          puts "    - Cut-off date: #{team.national_reporting_cut_off_date}"
+          puts "    - Immunisation imports (before cut-off): #{find_immunisation_imports_for_team(team).count}"
           puts "    - Total patients: #{find_patients_for_team(team).count}"
 
           vaccination_records = find_vaccination_records_for_team(team)
@@ -92,8 +93,9 @@ def reset_team(team)
         patient_ids_to_update = Set.new
 
         ActiveRecord::Base.transaction do
-          immunisation_imports = ImmunisationImport.where(team:)
-          puts "  - Found #{immunisation_imports.count} immunisation import(s)"
+          immunisation_imports = find_immunisation_imports_for_team(team)
+          puts "  - Found #{immunisation_imports.count} immunisation import(s) created before team's cut off date: " \
+                 "#{team.national_reporting_cut_off_date}"
 
           patient_ids = find_patients_for_team(team).ids
           puts "  - Found #{patient_ids.count} patient(s) in this team"
@@ -169,14 +171,21 @@ def reset_team(team)
         PatientStatusUpdaterJob.perform_bulk(patient_ids_to_update.zip)
       end
 
+      def find_immunisation_imports_for_team(team)
+        ImmunisationImport.where(team:).where(
+          "immunisation_imports.created_at < ?",
+          team.national_reporting_cut_off_date
+        )
+      end
+
       def find_patients_for_team(team)
         Patient.joins(:patient_teams).where(patient_teams: { team: }).distinct
       end
 
       def find_vaccination_records_for_team(team)
         VaccinationRecord.joins(:immunisation_imports).where(
           immunisation_imports: {
-            team:
+            id: find_immunisation_imports_for_team(team)
           }
         )
       end

spec/features/cli_teams_reset_national_reporting_spec.rb

@@ -28,6 +28,7 @@
       given_a_national_reporting_team_exists
       and_i_upload_some_vaccination_records
       and_i_view_a_patient_record
+      and_the_cut_off_date_is_set_to_tomorrow
 
       when_i_run_the_command_for_single_team
 
@@ -109,6 +110,18 @@
       and_the_other_archive_reasons_are_deleted
       and_the_other_patients_are_deleted
     end
+
+    it "only operates on imports created before the cut-off date" do
+      given_a_national_reporting_team_exists
+      and_the_national_reporting_team_has_immunisation_imports_both_before_and_after_the_cut_off_date
+
+      when_i_run_the_command_for_single_team
+
+      then_the_imports_before_the_cut_off_date_are_deleted
+      and_the_vaccination_records_from_imports_before_the_cut_off_date_are_deleted
+      and_the_imports_on_or_after_the_cut_off_date_are_not_deleted
+      and_the_vaccination_records_from_imports_on_or_after_the_cut_off_date_are_not_deleted
+    end
   end
 
   context "when resetting all national_reporting teams" do
@@ -250,8 +263,22 @@ def and_i_view_the_patient_that_is_associated_with_both_teams
   end
 
   def and_the_national_reporting_team_has_immunisation_imports_with_vaccination_records
-    @import1 = create(:immunisation_import, team: @national_reporting_team)
-    @import2 = create(:immunisation_import, team: @national_reporting_team)
+    cut_off_date =
+      @national_reporting_team.national_reporting_cut_off_date ||
+        Time.zone.today
+
+    @import1 =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team,
+        created_at: cut_off_date - 2.days
+      )
+    @import2 =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team,
+        created_at: cut_off_date - 1.day
+      )
 
     @vaccination_record1 =
       create(
@@ -279,7 +306,13 @@ def and_the_national_reporting_team_has_immunisation_imports_with_vaccination_re
   end
 
   def and_a_patient_is_associated_with_both_teams
-    @import = create(:immunisation_import, team: @national_reporting_team)
+    cut_off_date = @national_reporting_team.national_reporting_cut_off_date
+    @import =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team,
+        created_at: cut_off_date - 1.day
+      )
     @shared_patient = create(:patient)
     @national_reporting_vaccination_record =
       create(
@@ -319,8 +352,19 @@ def given_multiple_national_reporting_teams_exist
   end
 
   def and_each_team_has_immunisation_imports
-    @import1 = create(:immunisation_import, team: @national_reporting_team1)
-    @import2 = create(:immunisation_import, team: @national_reporting_team2)
+    cut_off_date = @national_reporting_team1.national_reporting_cut_off_date
+    @import1 =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team1,
+        created_at: cut_off_date - 1.day
+      )
+    @import2 =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team2,
+        created_at: cut_off_date - 1.day
+      )
 
     @vaccination_record1 =
       create(
@@ -536,7 +580,83 @@ def and_the_status_updater_job_is_enqueued
     )
   end
 
+  def and_the_cut_off_date_is_set_to_tomorrow
+    @national_reporting_team.update!(
+      national_reporting_cut_off_date: Time.zone.tomorrow
+    )
+  end
+
   def and_the_access_log_entry_is_not_deleted
     expect(AccessLogEntry.where(patient: @shared_patient)).to exist
   end
+
+  def and_the_national_reporting_team_has_immunisation_imports_both_before_and_after_the_cut_off_date
+    cut_off_date = @national_reporting_team.national_reporting_cut_off_date
+
+    @import_before =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team,
+        created_at: cut_off_date - 1.day
+      )
+    @import_on_cut_off =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team,
+        created_at: cut_off_date
+      )
+    @import_after =
+      create(
+        :immunisation_import,
+        team: @national_reporting_team,
+        created_at: cut_off_date + 1.day
+      )
+
+    @vaccination_record_before =
+      create(
+        :vaccination_record,
+        :sourced_from_national_reporting,
+        immunisation_import: @import_before,
+        team: @national_reporting_team,
+        performed_at: cut_off_date - 2.days
+      )
+    @vaccination_record_on_cut_off =
+      create(
+        :vaccination_record,
+        :sourced_from_national_reporting,
+        immunisation_import: @import_on_cut_off,
+        team: @national_reporting_team,
+        performed_at: cut_off_date
+      )
+    @vaccination_record_after =
+      create(
+        :vaccination_record,
+        :sourced_from_national_reporting,
+        immunisation_import: @import_after,
+        team: @national_reporting_team,
+        performed_at: cut_off_date + 2.days
+      )
+  end
+
+  def then_the_imports_before_the_cut_off_date_are_deleted
+    expect(ImmunisationImport.where(id: @import_before.id)).to be_empty
+  end
+
+  def and_the_vaccination_records_from_imports_before_the_cut_off_date_are_deleted
+    expect(
+      VaccinationRecord.where(id: @vaccination_record_before.id)
+    ).to be_empty
+  end
+
+  def and_the_imports_on_or_after_the_cut_off_date_are_not_deleted
+    expect(ImmunisationImport.where(id: @import_on_cut_off.id)).to exist
+    expect(ImmunisationImport.where(id: @import_after.id)).to exist
+  end
+
+  def and_the_vaccination_records_from_imports_on_or_after_the_cut_off_date_are_not_deleted
+    expect(
+      VaccinationRecord.where(id: @vaccination_record_on_cut_off.id)
+    ).to exist
+    expect(VaccinationRecord.where(id: @vaccination_record_after.id)).to exist
+  end
 end