Merge pull request #526 from blueyed/charge-repr-pk-upstream

Charge: add pk to __repr__
pinax · Dec 1, 2017 · dd87314 · dd87314
2 parents 257a9f2 + 79c61fd
commit dd87314
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/pinax/stripe/models.py b/pinax/stripe/models.py
@@ -489,7 +489,8 @@ class Charge(StripeAccountFromCustomerMixin, StripeObject):
     objects = ChargeManager()
 
     def __repr__(self):
-        return "Charge(customer={!r}, source={!r}, amount={!r}, captured={!r}, paid={!r}, stripe_id={!r})".format(
+        return "Charge(pk={!r}, customer={!r}, source={!r}, amount={!r}, captured={!r}, paid={!r}, stripe_id={!r})".format(
+            self.pk,
             self.customer,
             self.source,
             self.amount,

diff --git a/pinax/stripe/tests/test_models.py b/pinax/stripe/tests/test_models.py
@@ -137,9 +137,9 @@ def test_connected_customer_str_and_repr(self):
     def test_charge_repr(self):
         charge = Charge()
         if PY2:
-            self.assertEquals(repr(charge), "Charge(customer=None, source=u'', amount=None, captured=None, paid=None, stripe_id=u'')")
+            self.assertEquals(repr(charge), "Charge(pk=None, customer=None, source=u'', amount=None, captured=None, paid=None, stripe_id=u'')")
         else:
-            self.assertEquals(repr(charge), "Charge(customer=None, source='', amount=None, captured=None, paid=None, stripe_id='')")
+            self.assertEquals(repr(charge), "Charge(pk=None, customer=None, source='', amount=None, captured=None, paid=None, stripe_id='')")
 
     def test_plan_display_invoiceitem(self):
         p = Plan(amount=decimal.Decimal("5"), name="My Plan", interval="monthly", interval_count=1)