feat(purchase_order_number): Subscriptions upgrade/downgrade

Author: D1353L

app/services/subscriptions/plan_downgrade_service.rb

@@ -42,6 +42,7 @@ def call
           ending_at: params.key?(:ending_at) ? params[:ending_at] : current_subscription.ending_at,
           progressive_billing_disabled: params[:progressive_billing_disabled] || false,
           consolidate_invoice: params.key?(:consolidate_invoice) ? params[:consolidate_invoice] : current_subscription.consolidate_invoice,
+          purchase_order_number: params.key?(:purchase_order_number) ? params[:purchase_order_number] : current_subscription.purchase_order_number,
           billing_entity_id: current_subscription.billing_entity_id
         )
 

app/services/subscriptions/plan_upgrade_service.rb

@@ -62,6 +62,7 @@ def new_subscription_with_overrides
         billing_time: current_subscription.billing_time,
         ending_at: params.key?(:ending_at) ? params[:ending_at] : current_subscription.ending_at,
         consolidate_invoice: params.key?(:consolidate_invoice) ? params[:consolidate_invoice] : current_subscription.consolidate_invoice,
+        purchase_order_number: params.key?(:purchase_order_number) ? params[:purchase_order_number] : current_subscription.purchase_order_number,
         billing_entity_id: resolved_entity&.id || current_subscription.billing_entity_id
       )
 

spec/services/subscriptions/plan_downgrade_service_spec.rb

@@ -132,6 +132,33 @@
       end
     end
 
+    context "when subscription has a purchase_order_number" do
+      let(:subscription) do
+        create(
+          :subscription,
+          customer:,
+          plan: old_plan,
+          subscription_at: Time.current,
+          external_id: SecureRandom.uuid,
+          purchase_order_number: "PO-OLD"
+        )
+      end
+
+      it "inherits the purchase_order_number on the new subscription" do
+        expect(result).to be_success
+        expect(result.subscription.next_subscription.purchase_order_number).to eq("PO-OLD")
+      end
+
+      context "when params override purchase_order_number" do
+        let(:params) { {name: subscription_name, purchase_order_number: "PO-NEW"} }
+
+        it "applies the override on the new subscription" do
+          expect(result).to be_success
+          expect(result.subscription.next_subscription.purchase_order_number).to eq("PO-NEW")
+        end
+      end
+    end
+
     context "with plan overrides", :premium do
       let(:params) do
         {

spec/services/subscriptions/plan_upgrade_service_spec.rb

@@ -103,6 +103,35 @@
       end
     end
 
+    context "when subscription has a purchase_order_number" do
+      let(:subscription) do
+        create(
+          :subscription,
+          customer:,
+          plan: old_plan,
+          status: :active,
+          subscription_at: Time.current,
+          started_at: Time.current,
+          external_id: SecureRandom.uuid,
+          purchase_order_number: "PO-OLD"
+        )
+      end
+
+      it "inherits the purchase_order_number on the new subscription" do
+        expect(result).to be_success
+        expect(result.subscription.purchase_order_number).to eq("PO-OLD")
+      end
+
+      context "when params override purchase_order_number" do
+        let(:params) { {name: subscription_name, purchase_order_number: "PO-NEW"} }
+
+        it "applies the override on the new subscription" do
+          expect(result).to be_success
+          expect(result.subscription.purchase_order_number).to eq("PO-NEW")
+        end
+      end
+    end
+
     context "with payment method" do
       let(:payment_method) { create(:payment_method, organization: subscription.organization, customer: subscription.customer) }
       let(:params) do