Added tests and unicodes for Invoice and Item.

ej2 · ej2 · commit b3aafa1e3431 · 2015-08-16T09:06:45.000-05:00
diff --git a/quickbooks/objects/invoice.py b/quickbooks/objects/invoice.py
@@ -13,6 +13,9 @@ def __init__(self):
         self.DiscountPercent = 0
         self.DiscountAccountRef = None
 
+    def __unicode__(self):
+        return self.DiscountPercent
+
 
 class SalesItemLineDetail(QuickbooksBaseObject):
     class_dict = {
@@ -25,6 +28,9 @@ def __init__(self):
         self.UnitPrice = 0
         self.Qty = 0
 
+    def __unicode__(self):
+        return self.UnitPrice
+
 
 class InvoiceDetail(QuickbooksBaseObject):
     class_dict = {
@@ -39,6 +45,8 @@ def __init__(self):
         self.Amount = ""
         self.DetailType = ""
 
+    def __unicode__(self):
+        return "[{0}] {1} {2}".format(self.LineNum, self.Description, self.Amount)
 
 class Invoice(QuickbooksManagedObject):
     """
@@ -91,3 +99,6 @@ def __init__(self):
         self.CustomerMemo = None
 
         self.CustomField = []
+
+    def __unicode__(self):
+        return self.TotalAmt
diff --git a/tests/unit/objects/test_estimate.py b/tests/unit/objects/test_estimate.py
@@ -4,9 +4,6 @@
 
 
 class EstimateTests(unittest.TestCase):
-    def setUp(self):
-        pass
-
     def test_unicode(self):
         estimate = Estimate()
         estimate.TotalAmt = 10
@@ -15,9 +12,6 @@ def test_unicode(self):
 
 
 class EstimatelineTests(unittest.TestCase):
-    def setUp(self):
-        pass
-
     def test_unicode(self):
         estimateline = EstimateLine()
         estimateline.Amount = 100
@@ -26,9 +20,6 @@ def test_unicode(self):
 
 
 class SalesItemLineDetailTests(unittest.TestCase):
-    def setUp(self):
-        pass
-
     def test_unicode(self):
         sales_detail = SalesItemLineDetail()
         sales_detail.UnitPrice = 10
diff --git a/tests/unit/objects/test_invoice.py b/tests/unit/objects/test_invoice.py
@@ -0,0 +1,37 @@
+import unittest
+
+from quickbooks.objects.invoice import Invoice, InvoiceDetail, SalesItemLineDetail, DiscountLineDetail
+
+
+class InvoiceTests(unittest.TestCase):
+    def test_unicode(self):
+        invoice = Invoice()
+        invoice.TotalAmt = 10
+
+        self.assertEquals(invoice.__unicode__(), 10)
+
+
+class InvoiceDetailTests(unittest.TestCase):
+    def test_unicode(self):
+        detail = InvoiceDetail()
+        detail.LineNum = 1
+        detail.Description = "Product Description"
+        detail.Amount = 100
+
+        self.assertEquals(detail.__unicode__(), "[1] Product Description 100")
+
+
+class SalesItemLineDetailTests(unittest.TestCase):
+    def test_unicode(self):
+        detail = SalesItemLineDetail()
+        detail.UnitPrice = 100
+
+        self.assertEquals(detail.__unicode__(), 100)
+
+
+class DiscountLineDetailTests(unittest.TestCase):
+    def test_unicode(self):
+        detail = DiscountLineDetail()
+        detail.DiscountPercent = 5
+
+        self.assertEquals(detail.__unicode__(), 5)
diff --git a/tests/unit/objects/test_item.py b/tests/unit/objects/test_item.py
@@ -0,0 +1,14 @@
+import unittest
+
+from quickbooks.objects.item import Item
+
+
+class ItemTests(unittest.TestCase):
+    def setUp(self):
+        pass
+
+    def test_unicode(self):
+        item = Item()
+        item.Name = "test"
+
+        self.assertEquals(item.__unicode__(), "test")
