Added organization argument to ExportPartnersCSVService so which partials to show can be determined even when no partials are present

Author: Benjamin-Couey

app/controllers/partners_controller.rb

@@ -15,7 +15,7 @@ def index
 
     respond_to do |format|
       format.html
-      format.csv { send_data Exports::ExportPartnersCSVService.new(@partners.unscope(:includes)).generate_csv, filename: "Partners-#{Time.zone.today}.csv" }
+      format.csv { send_data Exports::ExportPartnersCSVService.new(@partners.unscope(:includes), current_organization).generate_csv, filename: "Partners-#{Time.zone.today}.csv" }
     end
   end
 

app/services/exports/export_partners_csv_service.rb

@@ -4,12 +4,12 @@ class ExportPartnersCSVService
     # to this service object; this is happening within the same request/response cycle, so it's already
     # in memory, so we can pass that collection in directly. Should this be moved to a background / async
     # job, we will need to pass in a collection of IDs instead.
-    def initialize(partners)
+    def initialize(partners, organization)
       @partners = partners.includes(:profile)
       # Assumes that all of the partners belong to the same organization. This is true for the time being
       # and, were that to change, base_table would need to be reworked anyway to account for partners from
       # different orgs having different partials enabled, and thus different columns in the CSV.
-      @partials_to_show = partners.first.partials_to_show
+      @partials_to_show = organization.partials_to_show
     end
 
     def generate_csv

spec/services/exports/export_partners_csv_service_spec.rb

@@ -1,6 +1,6 @@
 RSpec.describe Exports::ExportPartnersCSVService do
   describe "#generate_csv" do
-    subject { CSV.parse(described_class.new(partners).generate_csv) }
+    subject { CSV.parse(described_class.new(partners, organization).generate_csv) }
 
     let(:organization) { create(:organization) }
 
@@ -392,7 +392,7 @@
       partial_to_headers.keys.each do |partial|
         organization.update(partner_form_fields: [partial])
         partners.reload
-        limited_export = CSV.parse(described_class.new(partners).generate_csv)
+        limited_export = CSV.parse(described_class.new(partners, organization).generate_csv)
         expect(limited_export[0]).to eq(headers_base + partial_to_headers[partial])
         expect(limited_export[1]).to eq(values_base + partial_to_values[partial])
       end