Implement retrieve token. Closes stripe-ruby-mock#54

Author: gilbert

lib/stripe_mock/request_handlers/tokens.rb

@@ -3,15 +3,31 @@ module RequestHandlers
  module Tokens
 
  def Tokens.included(klass)
- klass.add_handler 'post /v1/tokens', :create_token
+ klass.add_handler 'post /v1/tokens', :create_token
+ klass.add_handler 'get /v1/tokens/(.*)', :get_token
  end
 
  def create_token(route, method_url, params, headers)
- card = Data.mock_card(params[:card].merge(last4: params[:card][:number][-4,4], customer: nil))
- id = generate_card_token(card)
- Data.mock_token(:id => id, :card => card)
+ # "Sanitize" card number
+ params[:card][:last4] = params[:card][:number][-4,4]
+ token_id = generate_card_token(params[:card])
+ card = @card_tokens[token_id]
+
+ Data.mock_token(params.merge :id => token_id, :card => card)
  end
 
+ def get_token(route, method_url, params, headers)
+ route =~ method_url
+ # A Stripe token can be either a bank token or a card token
+ bank_or_card = @bank_tokens[$1] || @card_tokens[$1]
+ assert_existance :token, $1, bank_or_card
+
+ if bank_or_card[:object] == 'card'
+ Data.mock_token(:id => $1, :card => bank_or_card)
+ elsif bank_or_card[:object] == 'bank_account'
+ Data.mock_token(:id => $1, :bank_account => bank_or_card)
+ end
+ end
  end
  end
 end

spec/shared_stripe_examples/bank_token_examples.rb

@@ -36,5 +36,16 @@
  expect(recipient.active_account.bank_name).to eq("Bank Token Mocking")
  end
 
-end
+ it "retrieves a created token" do
+ bank_token = StripeMock.generate_bank_token(
+ :bank_name => "Cha-ching Banking",
+ :last4 => "3939"
+ )
+ token = Stripe::Token.retrieve(bank_token)
 
+ expect(token.id).to eq(bank_token)
+ expect(token.bank_account.last4).to eq("3939")
+ expect(token.bank_account.bank_name).to eq("Cha-ching Banking")
+ end
+
+end

spec/shared_stripe_examples/card_token_examples.rb

@@ -26,6 +26,16 @@
  expect(card.exp_month).to eq(11)
  expect(card.exp_year).to eq(2099)
  end
+
+ it "retrieves a created token" do
+ card_token = StripeMock.generate_card_token(last4: "2323", exp_month: 33, exp_year: 2222)
+ token = Stripe::Token.retrieve(card_token)
+
+ expect(token.id).to eq(card_token)
+ expect(token.card.last4).to eq("2323")
+ expect(token.card.exp_month).to eq(33)
+ expect(token.card.exp_year).to eq(2222)
+ end
  end
 
  describe 'Stripe::Token' do
@@ -47,7 +57,7 @@
  expect(card.exp_year).to eq(2017)
  end
 
- it "generated and reads a card token for update customer" do
+ it "generates and reads a card token for update customer" do
  card_token = Stripe::Token.create({
  card: {
  number: "1111222233334444",