CLDC-3014 scheme csv download improvements (#2091)

* feat: optimise locations count

* feat: use each to preserve ordering, and optimise scheme downloads

app/helpers/schemes_helper.rb

@@ -60,10 +60,10 @@ module SchemesHelper
     when "schemes"
       "You've selected #{pluralize(scheme_count, 'scheme')}."
     when "locations"
-      location_count = schemes.map(&:locations).flatten.count
+      location_count = Location.where(scheme: schemes).count
       "You've selected #{pluralize(location_count, 'location')} from #{pluralize(scheme_count, 'scheme')}."
     when "combined"
-      location_count = schemes.map(&:locations).flatten.count
+      location_count = Location.where(scheme: schemes).count
       "You've selected #{pluralize(scheme_count, 'scheme')} with #{pluralize(location_count, 'location')}. The CSV will have one location per row with scheme details listed for each location."
     end
   end

app/services/csv/lettings_log_csv_service.rb

@@ -10,7 +10,7 @@ module Csv
       CSV.generate(headers: true) do |csv|
         csv << @attributes
 
-        logs.find_each do |log|
+        logs.each do |log|
           csv << @attributes.map { |attribute| value(attribute, log) }
         end
       end

app/services/csv/sales_log_csv_service.rb

@@ -9,7 +9,7 @@ module Csv
       CSV.generate(headers: true) do |csv|
         csv << @attributes
 
-        logs.find_each do |log|
+        logs.each do |log|
           csv << @attributes.map { |attribute| value(attribute, log) }
         end
       end

app/services/csv/scheme_csv_service.rb

@@ -10,22 +10,27 @@ module Csv
     def prepare_csv(schemes)
       CSV.generate(headers: true) do |csv|
         csv << attributes
-        schemes.find_each do |scheme|
+
-          if @download_type == "schemes"
-            csv << scheme_attributes.map { |attribute| scheme_value(attribute, scheme) }
-          else
-            scheme.locations.each do |location|
         case @download_type
+        when "schemes"
+          schemes.each do |scheme|
+            csv << scheme_attributes.map { |attribute| scheme_value(attribute, scheme) }
+          end
         when "locations"
+          schemes.each do |scheme|
+            scheme.locations.each do |location|
               csv << [scheme.id_to_display] + location_attributes.map { |attribute| location_value(attribute, location) }
+            end
+          end
         when "combined"
+          schemes.each do |scheme|
+            scheme.locations.each do |location|
               csv << scheme_attributes.map { |attribute| scheme_value(attribute, scheme) } + location_attributes.map { |attribute| location_value(attribute, location) }
             end
           end
         end
       end
     end
-    end
 
   private