[BE] add support for updating CC information

HealthByRo · May 30, 2018 · b395a3c · b395a3c
1 parent 4b7eb45
commit b395a3c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/aa_stripe/models.py b/aa_stripe/models.py
@@ -65,6 +65,7 @@ def create_at_stripe(self):
         )
         self.stripe_customer_id = customer["id"]
         self.stripe_response = customer
+        self.sources = customer.sources.data
         self.is_created_at_stripe = True
         self.save()
         return self

diff --git a/tests/test_customers.py b/tests/test_customers.py
@@ -88,10 +88,10 @@ def test_user_create(self):
                         "sources": {
                             "object": "list",
                             "data": [
-
+                                {"id": "card_xyz", "object": "card"}
                             ],
                             "has_more": False,
-                            "total_count": 0,
+                            "total_count": 1,
                             "url": "/v1/customers/cus_9Oop0gQ1R1ATMi/sources"
                         },
                         "subscriptions": {
@@ -127,6 +127,7 @@ def test_user_create(self):
             self.assertEqual(customer.stripe_js_response, self.stripe_js_response)
             self.assertEqual(customer.stripe_customer_id, "cus_9Oop0gQ1R1ATMi")
             self.assertEqual(customer.stripe_response["id"], "cus_9Oop0gQ1R1ATMi")
+            self.assertEqual(customer.sources, [{"id": "card_xyz", "object": "card"}])
 
     def test_change_description(self):
         customer_id = self.stripe_js_response["id"]