improve plan struct (#380)

beam-community · Jun 6, 2018 · 76e5182 · 76e5182
1 parent eb8afe0
commit 76e5182
Show file tree

Hide file tree

Showing 2 changed files with 35 additions and 4 deletions.
diff --git a/lib/stripe/subscriptions/plan.ex b/lib/stripe/subscriptions/plan.ex
@@ -45,7 +45,9 @@ defmodule Stripe.Plan do
   @type t :: %__MODULE__{
           id: Stripe.id(),
           object: String.t(),
-          amount: non_neg_integer,
+          aggregate_usage: String.t() | nil,
+          amount: non_neg_integer | nil,
+          billing_scheme: String.t() | nil,
           created: Stripe.timestamp(),
           currency: String.t(),
           interval: String.t(),
@@ -54,13 +56,20 @@ defmodule Stripe.Plan do
           metadata: Stripe.Types.metadata(),
           name: String.t(),
           nickname: String.t() | nil,
-          product: Stripe.id() | Stripe.Product.t()
+          product: Stripe.id() | Stripe.Product.t(),
+          tiers: Stripe.List.t(map) | nil,
+          tiers_mode: boolean | nil,
+          transform_usage: map | nil,
+          trial_period_days: non_neg_integer | nil,
+          usage_type: String.t() | nil
         }
 
   defstruct [
     :id,
     :object,
+    :aggregate_usage,
     :amount,
+    :billing_scheme,
     :created,
     :currency,
     :interval,
@@ -69,7 +78,12 @@ defmodule Stripe.Plan do
     :metadata,
     :name,
     :nickname,
-    :product
+    :product,
+    :tiers,
+    :tiers_mode,
+    :transform_usage,
+    :trial_period_days,
+    :usage_type
   ]
 
   @plural_endpoint "plans"

diff --git a/test/stripe/subscriptions/plan_test.exs b/test/stripe/subscriptions/plan_test.exs
@@ -9,7 +9,24 @@ defmodule Stripe.PlanTest do
         id: "sapphire-elite",
         interval: "month",
         nickname: "Sapphire elite",
-        product: "abc_123"
+        product: "abc_123",
+        trial_period_days: 10
+      }
+
+      assert {:ok, %Stripe.Plan{}} = Stripe.Plan.create(params)
+      assert_stripe_requested(:post, "/v1/plans")
+    end
+
+    test "creates a Plan for a customer with tiers" do
+      params = %{
+        amount: 5000,
+        billing_scheme: "tiered",
+        currency: "usd",
+        id: "sapphire-elite",
+        interval: "month",
+        nickname: "Sapphire elite",
+        product: "abc_123",
+        tiers: [%{amount: 10, up_to: 12}]
       }
 
       assert {:ok, %Stripe.Plan{}} = Stripe.Plan.create(params)