Updated BillPayment.

ej2 · ej2 · commit 34bd69670c47 · 2015-09-10T09:32:13.000-05:00
diff --git a/quickbooks/objects/billpayment.py b/quickbooks/objects/billpayment.py
@@ -1,4 +1,5 @@
-from base import QuickbooksBaseObject, Ref, LinkedTxn, QuickbooksManagedObject, LinkedTxnMixin
+from base import QuickbooksBaseObject, Ref, LinkedTxn, QuickbooksManagedObject, LinkedTxnMixin, \
+    QuickbooksTransactionEntity
 
 
 class CheckPayment(QuickbooksBaseObject):
@@ -17,6 +18,18 @@ def __unicode__(self):
         return self.PrintStatus
 
 
+class BillPaymentCreditCard(QuickbooksBaseObject):
+    class_dict = {
+        "CCAccountRef": Ref
+    }
+
+    qbo_object_name = "BillPaymentCreditCard"
+
+    def __init__(self):
+        super(BillPaymentCreditCard, self).__init__()
+        self.CCAccountRef = None
+
+
 class BillPaymentLine(QuickbooksBaseObject):
     list_dict = {
         "LinkedTxn": LinkedTxn
@@ -33,7 +46,7 @@ def __unicode__(self):
         return str(self.Amount)
 
 
-class BillPayment(QuickbooksManagedObject, LinkedTxnMixin):
+class BillPayment(QuickbooksManagedObject, QuickbooksTransactionEntity, LinkedTxnMixin):
     """
     QBO definition: A BillPayment entity represents the financial transaction of payment of bills that the
     business owner receives from a vendor for goods or services purchased from the vendor. QuickBooks Online
@@ -47,7 +60,10 @@ class BillPayment(QuickbooksManagedObject, LinkedTxnMixin):
     class_dict = {
         "VendorRef": Ref,
         "CheckPayment": CheckPayment,
+        "CreditCardPayment": BillPaymentCreditCard,
         "APAccountRef": Ref,
+        "DepartmentRef": Ref,
+
     }
 
     list_dict = {
@@ -62,12 +78,16 @@ def __init__(self):
         self.TotalAmt = 0
         self.TxnDate = ""
         self.PrivateNote = ""
+        self.DocNumber = ""
+        self.ProcessBillPayment = False
 
         self.VendorRef = None
         self.CheckPayment = None
         self.APAccountRef = None
+        self.DepartmentRef = None
+        self.CreditCardPayment = None
 
         self.Line = []
 
     def __unicode__(self):
-        return str(self.TotalAmt)
+        return str(self.TotalAmt)
diff --git a/tests/unit/objects/test_billpayment.py b/tests/unit/objects/test_billpayment.py
@@ -1,6 +1,6 @@
 import unittest
 
-from quickbooks.objects.billpayment import BillPayment, BillPaymentLine, CheckPayment
+from quickbooks.objects.billpayment import BillPayment, BillPaymentLine, CheckPayment, BillPaymentCreditCard
 
 
 class CheckPaymentTests(unittest.TestCase):
@@ -25,3 +25,10 @@ def test_unicode(self):
         bill_payment.TotalAmt = 1000
 
         self.assertEquals(unicode(bill_payment), "1000")
+
+
+class BillPaymentCreditCardTests(unittest.TestCase):
+    def test_init(self):
+        bill_payment_cc = BillPaymentCreditCard()
+
+        self.assertEquals(bill_payment_cc.CCAccountRef, None)
