Transfers exports include quantity moved for each item

Author: jp524

app/controllers/transfers_controller.rb

@@ -15,7 +15,7 @@ def index
     @to_storage_locations = StorageLocation.with_transfers_to(current_organization)
     respond_to do |format|
       format.html
-      format.csv { send_data Exports::ExportTransfersCSVService.new(transfers: @transfers, organization: current_organization).generate_csv, filename: "Transfers-#{Time.zone.today}.csv" }
+      format.csv { send_data Exports::ExportTransfersCSVService.new(transfers: @transfers.includes(line_items: :item), organization: current_organization).generate_csv, filename: "Transfers-#{Time.zone.today}.csv" }
     end
   end
 

app/services/exports/export_transfers_csv_service.rb

@@ -27,7 +27,7 @@ def generate_csv_data
     private
 
     def headers
-      base_headers
+      base_headers + item_headers
     end
 
     def base_table
@@ -51,8 +51,28 @@ def base_headers
       base_table.keys
     end
 
+    def item_headers
+      @item_headers ||= @organization.items.select("DISTINCT ON (LOWER(name)) items.name").order("LOWER(name) ASC").map(&:name)
+    end
+
+    def headers_with_indexes
+      @headers_with_indexes ||= headers.each_with_index.to_h
+    end
+
     def build_row_data(transfer)
-      base_table.values.map { |closure| closure.call(transfer) }
+      row = base_table.values.map { |closure| closure.call(transfer) }
+
+      row += Array.new(item_headers.size, 0)
+
+      transfer.line_items.each do |line_item|
+        item_name = line_item.item.name
+        item_column_idx = headers_with_indexes[item_name]
+        next unless item_column_idx
+
+        row[item_column_idx] += line_item.quantity
+      end
+
+      row
     end
   end
 end

spec/services/exports/export_transfers_csv_service_spec.rb

@@ -0,0 +1,102 @@
+RSpec.describe Exports::ExportTransfersCSVService do
+  let!(:organization) { create(:organization) }
+
+  describe "#generate_csv_data" do
+    subject { described_class.new(transfers:, organization:).generate_csv_data }
+
+    let(:from_location) { create(:storage_location, name: "From Location") }
+    let(:to_location) { create(:storage_location, name: "To Location") }
+    let(:duplicate_item) { create(:item, name: "Dupe Item") }
+
+    let(:items_lists) do # Used to created four transfers
+      [
+        [
+          [duplicate_item, 5],
+          [create(:item), 7],
+          [duplicate_item, 3]
+        ],
+
+        *(Array.new(3) do |i|
+          [[create(:item), i + 1]]
+        end)
+      ]
+    end
+
+    let(:transfers) do
+      items_lists.map do |items|
+        transfer = create(:transfer, from: from_location, to: to_location)
+
+        items.each do |(item, quantity)|
+          transfer.line_items << create(:line_item, quantity:, item:)
+        end
+
+        transfer
+      end
+    end
+
+    let(:all_org_items) { organization.items.sort_by { |item| item.name.downcase } }
+
+    let(:total_item_quantities) do
+      template = all_org_items.pluck(:name).index_with(0)
+
+      items_lists.map do |items_list|
+        row = template.dup
+        items_list.each do |(item, quantity)|
+          row[item.name] += quantity
+        end
+        row.values
+      end
+    end
+
+    let(:non_item_headers) { ["From", "To", "Comment", "Total Moved"] }
+    let(:expected_headers) { non_item_headers + all_org_items.pluck(:name) }
+
+    it "should match the expected content for the csv" do
+      expect(subject[0]).to eq(expected_headers)
+
+      transfers.zip(total_item_quantities).each_with_index do |(transfer, total_item_quantity), idx|
+        row = [
+          transfer.from.name,
+          transfer.to.name,
+          transfer.comment,
+          transfer.line_items.total
+        ]
+        row += total_item_quantity
+
+        expect(subject[idx + 1]).to eq(row)
+      end
+    end
+
+    context "when a new item is added to the organization" do
+      let!(:new_item) { create(:item, name: "New Item") }
+
+      it "should be included as the last column of the csv" do
+        expect(subject[0]).to eq(expected_headers).and end_with(new_item.name)
+      end
+
+      it "should have a quantity of 0 if this item isn't part of any transfer" do
+        transfers.zip(total_item_quantities).each_with_index do |(transfer, total_item_quantity), idx|
+          row = [
+            transfer.from.name,
+            transfer.to.name,
+            transfer.comment,
+            transfer.line_items.total
+          ]
+          row += total_item_quantity
+
+          expect(subject[idx + 1]).to eq(row).and end_with(0)
+        end
+      end
+    end
+
+    context "when there are no transfers but the report is requested" do
+      let(:transfers) { Transfer.none }
+
+      it "returns a csv with only headers and no rows" do
+        expect(subject.size).to eq(1)
+        header_row = subject[0]
+        expect(header_row).to eq(expected_headers)
+      end
+    end
+  end
+end