Merge branch 'add-stripe-token-information' into develop

mysociety · Sep 29, 2023 · da3e283 · da3e283
2 parents 60cde68 + dd41d8e
commit da3e283
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 9 deletions.
diff --git a/app/controllers/alaveteli_pro/payment_methods_controller.rb b/app/controllers/alaveteli_pro/payment_methods_controller.rb
@@ -6,12 +6,13 @@ def update
       @token = Stripe::Token.retrieve(params[:stripe_token])
 
       @pro_account = current_user.pro_account ||= current_user.build_pro_account
-      @pro_account.source = @token.id
+      @pro_account.token = @token
       @pro_account.update_stripe_customer
 
       flash[:notice] = _('Your payment details have been updated')
 
-    rescue Stripe::CardError => e
+    rescue ProAccount::CardError,
+           Stripe::CardError => e
       flash[:error] = e.message
 
     rescue Stripe::RateLimitError,

diff --git a/app/controllers/alaveteli_pro/subscriptions_controller.rb b/app/controllers/alaveteli_pro/subscriptions_controller.rb
@@ -46,7 +46,7 @@ def create
 
       @token = Stripe::Token.retrieve(params[:stripe_token])
 
-      @pro_account.source = @token.id
+      @pro_account.token = @token
       @pro_account.update_stripe_customer
 
       @subscription = @pro_account.subscriptions.build
@@ -60,7 +60,8 @@ def create
 
       @subscription.save
 
-    rescue Stripe::CardError => e
+    rescue ProAccount::CardError,
+           Stripe::CardError => e
       flash[:error] = e.message
 
     rescue Stripe::RateLimitError,

diff --git a/app/models/pro_account.rb b/app/models/pro_account.rb
@@ -14,7 +14,9 @@
 class ProAccount < ApplicationRecord
   include AlaveteliFeatures::Helpers
 
-  attr_writer :source
+  CardError = Class.new(StandardError)
+
+  attr_writer :token
 
   belongs_to :user,
              inverse_of: :pro_account
@@ -65,9 +67,9 @@ def update_email
   end
 
   def update_source
-    return unless @source
+    return unless @token
 
-    stripe_customer.source = @source
+    stripe_customer.source = @token.id
   end
 
   def stripe_customer!

diff --git a/spec/models/pro_account_spec.rb b/spec/models/pro_account_spec.rb
@@ -84,7 +84,7 @@
 
     it 'sets Stripe customer default source' do
       old_source = customer.default_source
-      pro_account.source = stripe_helper.generate_card_token
+      pro_account.token = double(id: stripe_helper.generate_card_token)
 
       expect { pro_account.update_stripe_customer }.to change(
         customer, :default_source
@@ -120,7 +120,7 @@
 
       it 'does not set new Stripe customer default source' do
         with_feature_disabled(:alaveteli_pro) do
-          pro_account.source = stripe_helper.generate_card_token
+          pro_account.token = double(id: stripe_helper.generate_card_token)
           expect { pro_account.update_stripe_customer }.to_not change(
             customer, :default_source
           )