Add indicator for period supplies in partner export

Author: Sukhpreet-s

app/models/distribution.rb

@@ -48,6 +48,7 @@ class Distribution < ApplicationRecord
   enum delivery_method: { pick_up: 0, delivery: 1, shipped: 2 }
   scope :active, -> { joins(:line_items).joins(:items).where(items: { active: true }) }
   scope :with_diapers, -> { joins(line_items: :item).merge(Item.disposable.or(Item.cloth_diapers)) }
+  scope :with_period_supplies, -> { joins(line_items: :item).merge(Item.period_supplies) }
   # add item_id scope to allow filtering distributions by item
   scope :by_item_id, ->(item_id) { includes(:items).where(items: { id: item_id }) }
   # partner scope to allow filtering by partner

app/models/partner.rb

@@ -186,7 +186,8 @@ def self.csv_export_headers
       "Contact Phone",
       "Contact Email",
       "Notes",
-      "Providing Diapers"
+      "Providing Diapers",
+      "Providing Period Supplies"
     ]
   end
 
@@ -204,14 +205,19 @@ def csv_export_attributes
       contact_person[:phone],
       contact_person[:email],
       notes,
-      providing_diapers
+      providing_diapers,
+      providing_period_supplies
     ]
   end
 
   def providing_diapers
     distributions.with_diapers.any? ? "Y" : "N"
   end
 
+  def providing_period_supplies
+    distributions.with_period_supplies.any? ? "Y" : "N"
+  end
+
   def contact_person
     return @contact_person if @contact_person
 

spec/models/partner_spec.rb

@@ -300,6 +300,7 @@
     let(:other_agency_type) { "Another Agency Name" }
     let(:notes) { "Some notes" }
     let(:providing_diapers) { "N" }
+    let(:providing_period_supplies) { "N" }
 
     before do
       partner.profile.update({
@@ -332,12 +333,14 @@
         contact_phone,
         contact_email,
         notes,
-        providing_diapers
+        providing_diapers,
+        providing_period_supplies
       ])
     end
 
     context "when partner has a distribution" do
       let(:providing_diapers) { "Y" }
+      let(:providing_period_supplies) { "Y" }
       let(:distribution) { create(:distribution, partner: partner) }
       let(:item) { create(:item) }
 
@@ -365,6 +368,18 @@
       context "with a cloth diaper item" do
         include_examples "providing_diapers check", :cloth_diapers
       end
+
+      context "with a period supplies item" do
+        before do
+          create(:line_item, item: item, itemizable: distribution)
+          # Mock the given scope to at least satisfy one item
+          allow(Item).to receive(:period_supplies) { Item.where(id: item.id) }
+        end
+
+        it "should have Y as providing_period_supplies" do
+          expect(partner.csv_export_attributes[13]).to eq(providing_period_supplies)
+        end
+      end
     end
   end