From b0a187f7a10e18e9bdf47f9d9176587c9f52216f Mon Sep 17 00:00:00 2001
From: Harvey Hartwell <harvey@ckcollab.com>
Date: Thu, 11 Jan 2024 11:14:08 -0800
Subject: [PATCH] improve price test

---
 ckc/stripe/serializers.py                    |  2 ++
 ckc/stripe/views.py                          |  2 +-
 tests/integration/test_payment_processing.py | 11 +++++++++--
 3 files changed, 12 insertions(+), 3 deletions(-)

diff --git a/ckc/stripe/serializers.py b/ckc/stripe/serializers.py
index 81f17a2..2cddd71 100644
--- a/ckc/stripe/serializers.py
+++ b/ckc/stripe/serializers.py
@@ -72,12 +72,14 @@ class Meta:
             'unit_amount',
             'currency',
             'recurring',
+            'nickname',
         )
         read_only_fields = (
             'id',
             'unit_amount',
             'currency',
             'recurring',
+            'nickname',
         )
 
 
diff --git a/ckc/stripe/views.py b/ckc/stripe/views.py
index bbc6c71..9eb0af8 100644
--- a/ckc/stripe/views.py
+++ b/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
diff --git a/tests/integration/test_payment_processing.py b/tests/integration/test_payment_processing.py
index 092e261..28c0c2e 100644
--- a/tests/integration/test_payment_processing.py
+++ b/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}"