invoiceitem test complete

boulderdave · Jun 10, 2015 · 5878d13 · 5878d13
1 parent 3e9ae97
commit 5878d13
Show file tree

Hide file tree

Showing 2 changed files with 21 additions and 3 deletions.
diff --git a/djstripe/models.py b/djstripe/models.py
@@ -768,8 +768,9 @@ def handle_event(cls, event):
 
 
 class InvoiceItem(TimeStampedModel):
-    """ Not inherited from StripeObject because there can be multiple invoice
-        items for a single stripe_id.
+    """
+    Not inherited from StripeObject because there can be multiple invoice
+    items for a single stripe_id.
     """
 
     stripe_id = models.CharField(max_length=50)
@@ -785,7 +786,6 @@ class InvoiceItem(TimeStampedModel):
     quantity = models.IntegerField(null=True)
 
     def plan_display(self):
-        # TODO - needs test
         return djstripe_settings.PAYMENTS_PLANS[self.plan]["name"]
 
 

diff --git a/tests/test_invoiceitem.py b/tests/test_invoiceitem.py
@@ -0,0 +1,18 @@
+"""
+.. module:: dj-stripe.tests.test_invoiceitem
+   :synopsis: dj-stripe InvoiceItem Model Tests.
+
+.. moduleauthor:: Alex Kavanaugh (@kavdev)
+
+"""
+
+from django.test.testcases import TestCase
+
+from djstripe.models import InvoiceItem
+
+
+class InvoiceItemTest(TestCase):
+
+    def test_plan_display(self):
+        invoiceitem = InvoiceItem(plan="test")
+        self.assertEqual("Test Plan 1", invoiceitem.plan_display())