feat(override-plan): Take into consideration children for customers_count

app/graphql/types/plans/object.rb

@@ -40,10 +40,6 @@ def charges_count
         object.charges.count
       end
 
-      def customers_count
-        object.subscriptions.active.select(:customer_id).distinct.count
-      end
-
       def subscriptions_count
         object.subscriptions.count
       end

app/models/plan.rb

@@ -70,6 +70,15 @@ def yearly_amount_cents
     amount_cents * 52
   end
 
+  def customers_count
+    count = subscriptions.active.select(:customer_id).distinct.count
+    return count unless children
+
+    count + children.joins(:subscriptions).where(
+      subscriptions: { status: :active },
+    ).select('distinct(customer_id)').count
+  end
+
   private
 
   def validate_code_unique

app/serializers/v1/plan_serializer.rb

@@ -16,6 +16,7 @@ def serialize
         trial_period: model.trial_period,
         pay_in_advance: model.pay_in_advance,
         bill_charges_monthly: model.bill_charges_monthly,
+        customers_count: model.customers_count,
         active_subscriptions_count:,
         draft_invoices_count:,
         parent_id: model.parent_id,

spec/models/plan_spec.rb

@@ -76,4 +76,18 @@
       end
     end
   end
+
+  describe '#customers_count' do
+    let(:customer) { create(:customer) }
+    let(:plan) { create(:plan) }
+
+    it 'returns the number of impacted customers' do
+      create(:subscription, customer:, plan:)
+      overridden_plan = create(:plan, parent_id: plan.id)
+      customer2 = create(:customer, organization: plan.organization)
+      create(:subscription, customer: customer2, plan: overridden_plan)
+
+      expect(plan.customers_count).to eq(2)
+    end
+  end
 end

spec/serializers/v1/plan_serializer_spec.rb

@@ -6,11 +6,17 @@
   subject(:serializer) { described_class.new(plan, root_name: 'plan', includes: %i[charges taxes]) }
 
   let(:plan) { create(:plan) }
+  let(:customer) { create(:customer, organization: plan.organization) }
+  let(:subscription) { create(:subscription, customer:, plan:) }
   let(:charge) { create(:standard_charge, plan:) }
 
-  before { charge }
+  before { subscription && charge }
 
   it 'serializes the object', :aggregate_failures do
+    overridden_plan = create(:plan, parent_id: plan.id)
+    customer2 = create(:customer, organization: plan.organization)
+    create(:subscription, customer: customer2, plan: overridden_plan)
+
     result = JSON.parse(serializer.to_json)
 
     expect(result['plan']).to include(
@@ -26,7 +32,8 @@
       'trial_period' => plan.trial_period,
       'pay_in_advance' => plan.pay_in_advance,
       'bill_charges_monthly' => plan.bill_charges_monthly,
-      'active_subscriptions_count' => 0,
+      'customers_count' => 2,
+      'active_subscriptions_count' => 1,
       'draft_invoices_count' => 0,
       'parent_id' => nil,
       'taxes' => [],