Add tests for the prices API.

Author: astex

localstripe/resources.py

@@ -2581,8 +2581,8 @@ class Price(StripeObject):
     _id_prefix = 'price_'
 
     def __init__(self, id=None, active=None, currency=None, metadata=None,
-                 nickname=None, product=None, recurring=None, unit_amount=None,
-                 **kwargs):
+                 nickname=None, product=None, product_data=None,
+                 recurring=None, unit_amount=None, **kwargs):
         if kwargs:
             raise UserError(400, 'Unexpected ' + ', '.join(kwargs.keys()))
 
@@ -2609,6 +2609,10 @@ def __init__(self, id=None, active=None, currency=None, metadata=None,
             # TODO: Add support for "meter" and "usage_type".
         if product is not None:
             Product._api_retrieve(product)  # to return 404 if not existant
+        if product_data is not None:
+            assert isinstance(product_data, dict)
+            assert 'name' in product_data
+            product = Product(name=product_data['name']).id
 
         super().__init__(id)
 
@@ -3357,8 +3361,10 @@ def _update(self, metadata=None, items=None, trial_end=None,
         ) or (
             self.price and
             (
-                self.price.interval != old_price.interval or
-                self.price.interval_count != old_price.interval_count
+                self.price.recurring.get('interval') !=
+                old_price.recurring.get('interval') or
+                self.price.recurring.get('interval_count') !=
+                old_price.recurring.get('interval_count')
             )
         )
         if create_an_invoice:
@@ -3458,7 +3464,7 @@ def _current_period(self):
                 'week': timedelta(weeks=1),
                 'month': relativedelta(months=1),
                 'year': relativedelta(years=1)
-            }[self.price.interval]
+            }[self.price.recurring['interval']]
         if self.price is None:
             assert self.plan is not None
             if self.plan.interval == 'day':

test.sh

@@ -1279,3 +1279,24 @@ inv=$(curl -sSfg -u $SK: $HOST/v1/subscriptions \
 total=$(curl -sSfg -u $SK: $HOST/v1/invoices/$inv \
         | grep -oP '"total": \K([0-9]+)' )
 [ "$total" -eq 16383 ]
+
+# Create a price under an existing product.
+curl -sSfg -u $SK: $HOST/v1/prices \
+  -d id=price_abc001 \
+  -d currency=usd \
+  -d unit_amount=1000 \
+  -d recurring[interval]=month \
+  -d product=PRODUCT1234
+
+# Create a price and create a new product as a side-effect.
+curl -sSfg -u $SK: $HOST/v1/prices \
+  -d id=price_abc002 \
+  -d currency=usd \
+  -d unit_amount=10000 \
+  -d recurring[interval]=year \
+  -d product_data[name]=Gold\ Plan
+
+# Subscribe a user to an existing price.
+curl -sSfg -u $SK: $HOST/v1/subscriptions \
+  -d customer=$cus \
+  -d items[0][price]=price_abc001