Merge pull request #3901 from nhsuk/improve-test-patient-creation

Add progress bar to the CLI command `mavis generate cohort-imports`

Author: thomasleese

app/lib/mavis_cli/generate/cohort_imports.rb

@@ -15,7 +15,17 @@ class CohortImports < Dry::CLI::Command
       def call(patients:)
         MavisCLI.load_rails
 
-        ::Generate::CohortImports.call(patient_count: patients.to_i)
+        patient_count = patients.to_i
+        puts "Generating cohort import with #{patient_count} patients..."
+        progress_bar = MavisCLI.progress_bar(patient_count)
+
+        result =
+          ::Generate::CohortImports.call(
+            patient_count: patient_count,
+            progress_bar: progress_bar
+          )
+
+        puts "\nCohort import CSV generated: #{result}"
       end
     end
   end

lib/generate/cohort_imports.rb

@@ -36,14 +36,16 @@ class CohortImports
                 :programme,
                 :urns,
                 :patient_count,
-                :school_year_groups
+                :school_year_groups,
+                :progress_bar
 
     def initialize(
       ods_code: "A9A5A",
       programme: "hpv",
       urns: nil,
       school_year_groups: nil,
-      patient_count: 10
+      patient_count: 10,
+      progress_bar: nil
     )
       @organisation = Organisation.find_by(ods_code:)
       @programme = Programme.find_by(type: programme)
@@ -56,6 +58,7 @@ def initialize(
             .pluck(:urn)
       @school_year_groups = school_year_groups
       @patient_count = patient_count
+      @progress_bar = progress_bar
       @nhs_numbers = Set.new
     end
 
@@ -121,6 +124,7 @@ def write_cohort_import_csv
             patient.parent_relationships.second&.type,
             patient.school.urn
           ]
+          progress_bar&.increment
         end
       end
       cohort_import_csv_filepath.to_s