feat(wallet): Refresh wallet ongoing balance on subscription termination

getlago · Dec 4, 2024 · e10fcef · e10fcef
1 parent 10d265b
commit e10fcef
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/app/services/subscriptions/terminate_service.rb b/app/services/subscriptions/terminate_service.rb
@@ -36,7 +36,12 @@ def call
         # NOTE: We should bill subscription and generate invoice for all cases except for the upgrade
         #       For upgrade we will create only one invoice for termination charges and for in advance charges
         #       It is handled in subscriptions/create_service.rb
-        bill_subscription unless upgrade
+        unless upgrade
+          bill_subscription
+          Wallets::Balance::RefreshOngoingService.call(
+            wallet: subscription.customer.wallets.active.first
+          )
+        end
       end
 
       # NOTE: Pending next subscription should be canceled as well