Merge pull request #772 from ngyv/add-balance-transaction-when-creati…

…ng-transfer

Add balance transaction when creating transfer

lib/stripe_mock/request_handlers/transfers.rb

@@ -45,7 +45,18 @@ def new_transfer(route, method_url, params, headers)
           raise Stripe::InvalidRequestError.new("Invalid integer: #{params[:amount]}", 'amount', http_status: 400)
         end
 
-        transfers[id] = Data.mock_transfer(params.merge :id => id)
+        bal_trans_params = { amount: params[:amount].to_i, source: id }
+
+        balance_transaction_id = new_balance_transaction('txn', bal_trans_params)
+
+        transfers[id] = Data.mock_transfer(params.merge(id: id, balance_transaction: balance_transaction_id))
+
+        transfer = transfers[id].clone
+        if params[:expand] == ['balance_transaction']
+          transfer[:balance_transaction] = balance_transactions[balance_transaction_id]
+        end
+
+        transfer
       end
 
       def get_transfer(route, method_url, params, headers)

spec/shared_stripe_examples/transfer_examples.rb

@@ -9,7 +9,7 @@
     expect(transfer.id).to match /^test_tr/
     expect(transfer.amount).to eq(100)
     expect(transfer.amount_reversed).to eq(0)
-    expect(transfer.balance_transaction).to eq('txn_2dyYXXP90MN26R')
+    expect(transfer.balance_transaction).to eq('test_txn_1')
     expect(transfer.created).to eq(1304114826)
     expect(transfer.currency).to eq('usd')
     expect(transfer.description).to eq('Transfer description')
@@ -30,6 +30,15 @@
     expect(transfer.transfer_group).to eq("group_ch_164xRv2eZvKYlo2Clu1sIJWB")
   end
 
+  it "creates a balance transaction" do
+    destination = Stripe::Account.create(type: "custom", email: "#{SecureRandom.uuid}@example.com", id: "acct_12345", requested_capabilities: ['card_payments', 'platform_payments'])
+    transfer = Stripe::Transfer.create(amount: 100, currency: "usd", destination: destination.id)
+
+    bal_trans = Stripe::BalanceTransaction.retrieve(transfer.balance_transaction)
+    expect(bal_trans.amount).to eq(100)
+    expect(bal_trans.source).to eq(transfer.id)
+  end
+
   describe "listing transfers" do
     let(:destination) { Stripe::Account.create(type: "custom", email: "#{SecureRandom.uuid}@example.com", business_name: "MyCo") }