improve price test

ckc/stripe/serializers.py

@@ -72,12 +72,14 @@ class Meta:
             'unit_amount',
             'currency',
             'recurring',
+            'nickname',
         )
         read_only_fields = (
             'id',
             'unit_amount',
             'currency',
             'recurring',
+            'nickname',
         )
 
 

ckc/stripe/views.py

@@ -1,4 +1,4 @@
-from djstripe.models import PaymentMethod, Price, Plan, Customer
+from djstripe.models import PaymentMethod, Price, Customer
 from rest_framework import viewsets, mixins
 from rest_framework.decorators import action
 from rest_framework.permissions import IsAuthenticated, AllowAny

tests/integration/test_payment_processing.py

@@ -2,7 +2,7 @@
 
 import stripe
 from django.urls import reverse
-from djstripe.models import PaymentMethod, Customer
+from djstripe.models import PaymentMethod, Customer, Price
 from djstripe.sync import sync_subscriber
 # from djstripe.core import Price
 from rest_framework.test import APITestCase
@@ -117,9 +117,16 @@ def test_subscriptions(self):
 
     def test_subscription_plan_list(self):
         for i in range(3):
-            create_price(2000 + i, "month", product_name=f"Sample Product Name: {i}", currency="usd")
+            prod_name = f"Sample Product Name: {i}"
+            create_price(2000 + i, "month", product_name=prod_name, nickname=prod_name, currency="usd")
 
         url = reverse('prices-list')
         resp = self.client.get(url)
         assert resp.status_code == 200
         assert len(resp.data) == 3
+        from pprint import pprint
+        pprint(resp.data)
+
+        for i in range(3):
+            assert resp.data[i]['unit_amount'] / 100 == 2000 + i
+            assert resp.data[i]['nickname'] == f"Sample Product Name: {i}"