feat: Allow SaaS users to manage subscription within the dashboard (#5059)

spec/enterprise/services/enterprise/billing/create_session_service_spec.rb

@@ -0,0 +1,22 @@
+require 'rails_helper'
+
+describe Enterprise::Billing::CreateSessionService do
+  subject(:create_session_service) { described_class }
+
+  describe '#perform' do
+    it 'calls stripe billing portal session' do
+      customer_id = 'cus_random_number'
+      return_url = 'https://www.chatwoot.com'
+      allow(Stripe::BillingPortal::Session).to receive(:create).with({ customer: customer_id, return_url: return_url })
+
+      create_session_service.new.create_session(customer_id, return_url)
+
+      expect(Stripe::BillingPortal::Session).to have_received(:create).with(
+        {
+          customer: customer_id,
+          return_url: return_url
+        }
+      )
+    end
+  end
+end

spec/enterprise/services/enterprise/billing/create_stripe_customer_service_spec.rb

@@ -0,0 +1,81 @@
+require 'rails_helper'
+
+describe Enterprise::Billing::CreateStripeCustomerService do
+  subject(:create_stripe_customer_service) { described_class }
+
+  let(:account) { create(:account) }
+  let!(:admin1) { create(:user, account: account, role: :administrator) }
+  let(:admin2) { create(:user, account: account, role: :administrator) }
+
+  describe '#perform' do
+    before do
+      create(
+        :installation_config,
+        { name: 'CHATWOOT_CLOUD_PLANS', value: [
+          { 'name' => 'A Plan Name', 'product_id' => ['prod_hacker_random'], 'price_ids' => ['price_hacker_random'] }
+        ] }
+      )
+    end
+
+    it 'does not call stripe methods if customer id is present' do
+      account.update!(custom_attributes: { stripe_customer_id: 'cus_random_number' })
+
+      allow(Stripe::Customer).to receive(:create)
+      allow(Stripe::Subscription).to receive(:create)
+        .and_return(
+          {
+            plan: { id: 'price_random_number', product: 'prod_random_number' },
+            quantity: 2
+          }.with_indifferent_access
+        )
+
+      create_stripe_customer_service.new(account: account).perform
+
+      expect(Stripe::Customer).not_to have_received(:create)
+      expect(Stripe::Subscription)
+        .to have_received(:create)
+        .with({ customer: 'cus_random_number', items: [{ price: 'price_hacker_random', quantity: 2 }] })
+
+      expect(account.reload.custom_attributes).to eq(
+        {
+          stripe_customer_id: 'cus_random_number',
+          stripe_price_id: 'price_random_number',
+          stripe_product_id: 'prod_random_number',
+          subscribed_quantity: 2,
+          plan_name: 'A Plan Name'
+        }.with_indifferent_access
+      )
+    end
+
+    it 'calls stripe methods to create a customer and updates the account' do
+      customer = double
+      allow(Stripe::Customer).to receive(:create).and_return(customer)
+      allow(customer).to receive(:id).and_return('cus_random_number')
+      allow(Stripe::Subscription)
+        .to receive(:create)
+        .and_return(
+          {
+            plan: { id: 'price_random_number', product: 'prod_random_number' },
+            quantity: 2
+          }.with_indifferent_access
+        )
+
+      create_stripe_customer_service.new(account: account).perform
+
+      expect(Stripe::Customer).to have_received(:create).with({ name: account.name, email: admin1.email })
+      expect(Stripe::Subscription)
+        .to have_received(:create)
+        .with({ customer: customer.id, items: [{ price: 'price_hacker_random', quantity: 2 }] })
+
+      expect(account.reload.custom_attributes).to eq(
+        {
+          stripe_customer_id: customer.id,
+          stripe_price_id: 'price_random_number',
+          stripe_product_id: 'prod_random_number',
+          subscribed_quantity: 2,
+          plan_name: 'A Plan Name'
+        }.with_indifferent_access
+      )
+    end
+  end
+end