Merge pull request #6 from Uscreen-video/improve-support-to-13-0

Improve support to 13.0

Author: lutchobandeira

lib/stripe_mock/data.rb

@@ -248,6 +248,19 @@ def self.mock_charge(params={})
         amount: 0,
         application_fee: nil,
         application_fee_amount: nil,
+        billing_details: {
+          address: {
+            city: 'New Orleans',
+            country: 'US',
+            line1: 'Bourbon Street 23',
+            line2: nil,
+            postal_code: '10000',
+            state: nil
+          },
+          email: 'foo@bar.com',
+          name: 'John Dolton',
+          phone: nil
+        },
         currency: currency,
         destination: nil,
         fraud_details: {},
@@ -1252,6 +1265,7 @@ def self.mock_balance_transaction(params = {})
         created: 1461880226,
         currency: currency,
         description: nil,
+        exchange_rate: nil,
         fee: 320,
         fee_details: [
           {

lib/stripe_mock/instance.rb

@@ -108,6 +108,7 @@ def initialize
       @dispute_counter = 0
       @conversion_rate = 1.0
       @account_balance = 10000
+      @account_currency = "usd"
 
       # This is basically a cache for ParamValidators
       @base_strategy = TestStrategies::Base.new
@@ -231,7 +232,9 @@ def new_balance_transaction(prefix, params = {})
         calculate_fees(params) unless params[:fee]
         params[:net] = amount - params[:fee]
         params[:amount] = amount * @conversion_rate
+        params[:currency] = @account_currency
       end
+      params[:exchange_rate] = @conversion_rate if @conversion_rate != 1.0
       @balance_transactions[id] = Data.mock_balance_transaction(params.merge(id: id))
       id
     end

lib/stripe_mock/request_handlers/charges.rb

@@ -180,7 +180,7 @@ def non_positive_charge_amount?(params)
       end
 
       def allowed_params(params)
-        allowed = [:description, :metadata, :receipt_email, :fraud_details, :shipping, :destination]
+        allowed = [:description, :metadata, :receipt_email, :fraud_details, :shipping, :destination, :billing_details]
 
         # This is a workaround for the way the Stripe API sends params even when they aren't modified.
         # Stipe will include those params even when they aren't modified.

lib/stripe_mock/request_handlers/customers.rb

@@ -135,6 +135,14 @@ def get_customer(route, method_url, params, headers)
           end
         end
 
+        if params[:expand].is_a?(Array) && params[:expand].include?('invoice_credit_balance')
+          customer[:invoice_credit_balance] = if customer[:balance_transactions]
+            customer[:balance_transactions].group_by { |bt| bt[:currency] }.transform_values { |v| v.sum { |t| t[:amount] } }
+          else
+            {}
+          end
+        end
+
         customer
       end
 

lib/stripe_mock/request_handlers/payment_intents.rb

@@ -68,6 +68,11 @@ def get_payment_intent(route, method_url, params, headers)
         payment_intent = assert_existence :payment_intent, payment_intent_id, payment_intents[payment_intent_id]
 
         payment_intent = payment_intent.clone
+
+        if params[:expand].is_a?(Array) && params[:expand].include?('latest_charge')
+          payment_intent[:latest_charge] = charges.values.first
+        end
+
         payment_intent
       end
 
@@ -180,7 +185,11 @@ def last_payment_error_generator(code: nil, message: nil, decline_code: nil)
 
       def succeeded_payment_intent(payment_intent)
         payment_intent[:status] = 'succeeded'
-        btxn = new_balance_transaction('txn', { source: payment_intent[:id] })
+        btxn = new_balance_transaction('txn', {
+          source: payment_intent[:id],
+          amount: payment_intent[:amount],
+          currency: payment_intent[:currency]
+        })
 
         charge_id = new_id('ch')
 

spec/shared_stripe_examples/customer_examples.rb

@@ -332,6 +332,25 @@ def gen_card_tk
     expect(customer.default_source).to be_a(Stripe::Card)
   end
 
+  it "can expand invoice_credit_balance" do
+    original = Stripe::Customer.create({
+      email: 'johnny@appleseed.com',
+      source: gen_card_tk
+    })
+
+    Stripe::Customer.create_balance_transaction(
+      original.id,
+      { amount: -100, currency: 'usd' }
+    )
+
+    customer = Stripe::Customer.retrieve(
+      id: original.id,
+      expand: ['invoice_credit_balance']
+    )
+
+    expect(customer.invoice_credit_balance.to_h).to eq({ usd: -100 })
+  end
+
   it "cannot retrieve a customer that doesn't exist" do
     expect { Stripe::Customer.retrieve('nope') }.to raise_error {|e|
       expect(e).to be_a Stripe::InvalidRequestError

spec/shared_stripe_examples/payment_intent_examples.rb

@@ -77,6 +77,12 @@
     expect(payment_intent.metadata.to_hash).to eq(original.metadata.to_hash)
   end
 
+  it "can expand latest_charge" do
+    original = Stripe::PaymentIntent.create(amount:  100, currency: "usd", confirm: true)
+    payment_intent = Stripe::PaymentIntent.retrieve({ id: original.id, expand: ["latest_charge"] })
+    expect(payment_intent.latest_charge.object).to eq("charge")
+  end
+
   it "cannot retrieve a payment_intent that doesn't exist" do
     expect { Stripe::PaymentIntent.retrieve('nope') }.to raise_error {|e|
       expect(e).to be_a Stripe::InvalidRequestError