Merge pull request #284 from code-corps/add-token

Add Token endpoints and tests

lib/stripe/core_resources/token.ex

@@ -8,12 +8,11 @@ defmodule Stripe.Token do
   - Create a token with all options - Only for Unit Tests with Stripe
   - Retrieve a token
 
-  Does not yet render lists or take options.
-
   Stripe API reference: https://stripe.com/docs/api#token
 
   """
   use Stripe.Entity
+  import Stripe.Request
 
   @type token_bank_account :: %{
     id: Stripe.id,
@@ -58,12 +57,12 @@ defmodule Stripe.Token do
   @type t :: %__MODULE__{
     id: Stripe.id,
     object: String.t,
-    bank_account: token_bank_account,
-    card: token_card,
+    bank_account: token_bank_account | nil,
+    card: token_card | nil,
     client_ip: String.t | nil,
     created: Stripe.timestamp,
     livemode: boolean,
-    type: :card | :bank_account,
+    type: :card | :bank_account | :pii,
     used: boolean
   }
 
@@ -82,59 +81,31 @@ defmodule Stripe.Token do
   @plural_endpoint "tokens"
 
   @doc """
-  Create a token for a Connect customer with a card belonging to the platform
-  customer.
-
-  You must pass in the account number for the Stripe Connect account in `opts`.
-  """
-  @spec create_on_connect_account(String.t, String.t, Keyword.t) :: {:ok, t} | {
-    :error,
-    Stripe.api_error_struct
-  }
-  def create_on_connect_account(customer_id, customer_card_id, opts = [connect_account: _]) do
-    body = %{
-      card: customer_card_id,
-      customer: customer_id
-    }
-    Stripe.Request.create(@plural_endpoint, body, opts)
-  end
-
-  @doc """
-  Create a token for a Connect customer using the default card.
-
-  You must pass in the account number for the Stripe Connect account in `opts`.
-  """
-  @spec create_with_default_card(String.t, Keyword.t) :: {:ok, t} | {
-    :error,
-    Stripe.api_error_struct
-  }
-  def create_with_default_card(customer_id, opts \\ []) do
-    body = %{
-      customer: customer_id
-    }
-    Stripe.Request.create(@plural_endpoint, body, opts)
-  end
-
-  @doc """
-  Create a token.
-
-  WARNING: This function is mainly for testing purposes only, you should not
-  use it on a production server, unless you are able to transfer and store
-  credit card data on your server in a PCI compliant way.
+  Creates a single use token that wraps the details of a credit card. This
+  token can be used in place of a credit card dictionary with any API method.
+  These tokens can only be used once: by creating a new charge object, or
+  attaching them to a customer.
 
-  Use the Stripe.js library on the client device instead.
+  In most cases, you should create tokens client-side using Checkout, Elements,
+  or Stripe's mobile libraries, instead of using the API.
   """
-  @spec create(map, Keyword.t) :: {:ok, t} | {:error, Stripe.api_error_struct}
-  def create(changes, opts \\ []) do
-    Stripe.Request.create(@plural_endpoint, changes, opts)
+  @spec create(map, Stripe.options) :: {:ok, t} | {:error, Stripe.Error.t}
+  def create(params, opts \\ []) do
+    new_request(opts)
+    |> put_endpoint(@plural_endpoint)
+    |> put_params(params)
+    |> put_method(:post)
+    |> make_request()
   end
 
   @doc """
   Retrieve a token.
   """
-  @spec retrieve(binary, Keyword.t) :: {:ok, t} | {:error, Stripe.api_error_struct}
+  @spec retrieve(Stripe.id | t, Stripe.options) :: {:ok, t} | {:error, Stripe.Error.t}
   def retrieve(id, opts \\ []) do
-    endpoint = @plural_endpoint <> "/" <> id
-    Stripe.Request.retrieve(endpoint, opts)
+    new_request(opts)
+    |> put_endpoint(@plural_endpoint <> "/#{get_id!(id)}")
+    |> put_method(:get)
+    |> make_request()
   end
 end

test/stripe/core_resources/token_test.exs

@@ -0,0 +1,47 @@
+defmodule Stripe.TokenTest do
+  use Stripe.StripeCase, async: true
+
+  @card %{
+    number: "4242424242424242",
+    exp_month: 11,
+    exp_year: 2024,
+    cvc: "123"
+  }
+
+  @bank_account %{
+    country: "US",
+    currency: "usd",
+    account_holder_name: "Ella Garcia",
+    account_holder_type: "individual",
+    routing_number: "110000000",
+    account_number: "000123456789"
+  }
+
+  @pii %{
+    personal_id_number: "000000000"
+  }
+
+  describe "create/2" do
+    test "creates a card token" do
+      assert {:ok, %Stripe.Token{}} = Stripe.Token.create(%{card: @card})
+      assert_stripe_requested :post, "/v1/tokens"
+    end
+
+    test "creates a bank account token" do
+      assert {:ok, %Stripe.Token{}} = Stripe.Token.create(%{bank_account: @bank_account})
+      assert_stripe_requested :post, "/v1/tokens"
+    end
+
+    test "creates a PII token" do
+      assert {:ok, %Stripe.Token{}} = Stripe.Token.create(%{pii: @pii})
+      assert_stripe_requested :post, "/v1/tokens"
+    end
+  end
+
+  describe "retrieve/2" do
+    test "retrieves a token" do
+      assert {:ok, %Stripe.Token{}} = Stripe.Token.retrieve("tok_123")
+      assert_stripe_requested :get, "/v1/tokens/tok_123"
+    end
+  end
+end