Updated tests

deathnfudge · deathnfudge · commit a4a04d95cba5 · 2017-08-17T09:49:49.000-07:00
diff --git a/tests/integration/test_estimate.py b/tests/integration/test_estimate.py
@@ -4,13 +4,9 @@
 
 from quickbooks.objects.detailline import DetailLine, SalesItemLineDetail, \
     DiscountLineDetail, SalesItemLine
-
 from quickbooks.objects.tax import TxnTaxDetail
-
 from quickbooks.objects.customer import Customer
-
 from quickbooks.objects.base import Address, EmailAddress, CustomerMemo, Ref
-
 from quickbooks.objects.estimate import Estimate
 
 from quickbooks import QuickBooks
@@ -36,7 +32,6 @@ def test_create(self):
         estimate.EmailStatus = "NotSet"
 
         estimate.BillAddr = Address()
-        # estimate.BillAddr.Id = "1" + datetime.now().strftime('%d%H%M')
         estimate.BillAddr.Line1 = "65 Ocean Dr."
         estimate.BillAddr.City = "Half Moon Bay"
         estimate.BillAddr.CountrySubDivisionCode = "CA"
@@ -89,17 +84,16 @@ def test_create(self):
         line2 = DetailLine()
         line2.Amount = 3.5
 
-        discount_line_detail = DiscountLineDetail()
-        discount_line_detail.PercentBased = True
-        discount_line_detail.DiscountPercent = 10
+        line2.DiscountLineDetail = DiscountLineDetail()
+        line2.DiscountLineDetail.PercentBased = True
+        line2.DiscountLineDetail.DiscountPercent = 10
 
-        discount_account_ref = Ref()
-        discount_account_ref.value = "86"
-        discount_account_ref.name = "Discounts given"
-        discount_line_detail.DiscountAccountRef = discount_account_ref
+        line2.DiscountLineDetail.DiscountAccountRef = Ref()
+        line2.DiscountLineDetail.DiscountAccountRef.value = "86"
+        line2.DiscountLineDetail.DiscountAccountRef.name = "Discounts given"
 
         line2.DetailType = "DiscountLineDetail"
-        line2.DiscountLineDetail = discount_line_detail
+
         estimate.Line.append(line2)
 
         estimate.save(qb=self.qb_client)
@@ -127,3 +121,15 @@ def test_create(self):
         self.assertEqual(query_estimate.Line[0].LineNum, estimate.Line[0].LineNum)
         self.assertEqual(query_estimate.Line[0].Description, estimate.Line[0].Description)
         self.assertEqual(query_estimate.Line[0].Amount, estimate.Line[0].Amount)
+        self.assertEqual(query_estimate.Line[0].SalesItemLineDetail.UnitPrice,
+                         estimate.Line[0].SalesItemLineDetail.UnitPrice)
+        self.assertEqual(query_estimate.Line[0].SalesItemLineDetail.Qty, estimate.Line[0].SalesItemLineDetail.Qty)
+        self.assertEqual(query_estimate.Line[2].Amount, estimate.Line[1].Amount)
+        self.assertEqual(query_estimate.Line[2].DiscountLineDetail.PercentBased,
+                         estimate.Line[1].DiscountLineDetail.PercentBased)
+        self.assertEqual(query_estimate.Line[2].DiscountLineDetail.DiscountPercent,
+                         estimate.Line[1].DiscountLineDetail.DiscountPercent)
+        self.assertEqual(query_estimate.Line[2].DiscountLineDetail.DiscountAccountRef.value,
+                         estimate.Line[1].DiscountLineDetail.DiscountAccountRef.value)
+        self.assertEqual(query_estimate.Line[2].DiscountLineDetail.DiscountAccountRef.name,
+                         estimate.Line[1].DiscountLineDetail.DiscountAccountRef.name)
