Added tests for invoice and billpayment.

ej2 · ej2 · commit aa6db2a95cdb · 2015-11-12T18:28:20.000-06:00
diff --git a/quickbooks/objects/bill.py b/quickbooks/objects/bill.py
@@ -98,7 +98,6 @@ class Bill(QuickbooksManagedObject, QuickbooksTransactionEntity, LinkedTxnMixin)
 
     list_dict = {
         "Line": BillLine,
-        "LinkedTxn": LinkedTxn
     }
 
     qbo_object_name = "Bill"
@@ -127,3 +126,11 @@ def __init__(self):
 
     def __str__(self):
         return str(self.Balance)
+
+    def to_linked_txn(self):
+        linked_txn = LinkedTxn()
+        linked_txn.TxnId = self.Id
+        linked_txn.TxnType = "Bill"
+        linked_txn.TxnLineId = 1
+
+        return linked_txn
diff --git a/quickbooks/objects/billpayment.py b/quickbooks/objects/billpayment.py
@@ -13,7 +13,7 @@ class CheckPayment(QuickbooksBaseObject):
 
     def __init__(self):
         super(CheckPayment, self).__init__()
-        self.PrintStatus = ""
+        self.PrintStatus = "NotSet"
         self.BankAccountRef = None
 
     def __str__(self):
@@ -80,10 +80,9 @@ def __init__(self):
         super(BillPayment, self).__init__()
         self.PayType = ""
         self.TotalAmt = 0
-        self.TxnDate = ""
         self.PrivateNote = ""
         self.DocNumber = ""
-        self.ProcessBillPayment = False
+        #self.ProcessBillPayment = False
 
         self.VendorRef = None
         self.CheckPayment = None
diff --git a/tests/integration_tests/test_bill.py b/tests/integration_tests/test_bill.py
@@ -1,22 +1,13 @@
-from datetime import datetime
-import os
 import unittest
-from quickbooks.client import QuickBooks
-from quickbooks.objects.bill import Bill, BillLine, AccountBasedExpenseLineDetail
+from datetime import datetime
+
 from quickbooks.objects.base import Ref
+from quickbooks.objects.bill import Bill, BillLine, AccountBasedExpenseLineDetail
 from quickbooks.objects.vendor import Vendor
 
 
-class AccountTest(unittest.TestCase):
+class BillTest(unittest.TestCase):
     def setUp(self):
-        QuickBooks(
-            sandbox=True,
-            consumer_key=os.environ.get('CONSUMER_KEY'),
-            consumer_secret=os.environ.get('CONSUMER_SECRET'),
-            access_token=os.environ.get('ACCESS_TOKEN'),
-            access_token_secret=os.environ.get('ACCESS_TOKEN_SECRET'),
-            company_id=os.environ.get('COMPANY_ID')
-        )
 
         self.account_number = datetime.now().strftime('%d%H%M')
         self.name = "Test Account {0}".format(self.account_number)
diff --git a/tests/integration_tests/test_billpayment.py b/tests/integration_tests/test_billpayment.py
@@ -0,0 +1,52 @@
+import unittest
+from datetime import datetime
+
+from quickbooks.objects.creditcardpayment import CreditCardPayment
+from quickbooks.objects.account import Account
+from quickbooks.objects.bill import Bill
+from quickbooks.objects.billpayment import BillPayment, BillPaymentLine
+from quickbooks.objects.base import Ref
+from quickbooks.objects.vendor import Vendor
+
+
+class BillPaymentTest(unittest.TestCase):
+    def setUp(self):
+
+        self.account_number = datetime.now().strftime('%d%H%M')
+        self.name = "Test Account {0}".format(self.account_number)
+
+    def test_create(self):
+        bill_payment = BillPayment()
+
+        bill_payment.PayType = "CreditCard"
+        bill_payment.TotalAmt = 200
+        bill_payment.PrivateNote = "Private Note"
+
+        vendor = Vendor.all(max_results=1)[0]
+        bill_payment.VendorRef = vendor.to_ref()
+
+        bill_payment.CreditCardPayment = CreditCardPayment()
+
+        #account = Account.all(max_results=1)[0]
+        #bill_payment.CreditCardPayment.CCAccountRef = account.to_ref()
+
+        ap_account = Account.where("AccountSubType = 'AccountsPayable'")[0]
+        bill_payment.APAccountRef = ap_account.to_ref()
+
+        bill = Bill.all(max_results=1)[0]
+
+        line = BillPaymentLine()
+        line.LinkedTxn.append(bill.to_linked_txn())
+        line.Amount = 200
+
+        bill_payment.Line.append(line)
+        bill_payment.save()
+
+        query_bill_payment = BillPayment.get(bill_payment.Id)
+
+        self.assertEquals(query_bill_payment.PayType, "Check")
+        self.assertEquals(query_bill_payment.TotalAmt, 200.0)
+        self.assertEquals(query_bill_payment.PrivateNote,"Private Note")
+
+        self.assertEquals(len(query_bill_payment.Line), 1)
+        self.assertEquals(query_bill_payment.Line[0].Amount, 200.0)
diff --git a/tests/integration_tests/test_invoice.py b/tests/integration_tests/test_invoice.py
@@ -0,0 +1,36 @@
+import unittest
+
+from quickbooks.objects.customer import Customer
+from quickbooks.objects.detailline import SaleItemLine, SalesItemLineDetail
+from quickbooks.objects.invoice import Invoice
+from quickbooks.objects.item import Item
+from quickbooks.objects.base import CustomerMemo
+
+
+class InvoiceTest(unittest.TestCase):
+    def test_create(self):
+        invoice = Invoice()
+
+        line = SaleItemLine()
+        line.LineNum = 1
+        line.Description = "description"
+        line.Amount = 100
+        line.SalesItemLineDetail = SalesItemLineDetail()
+        item = Item.all(max_results=1)[0]
+
+        line.SalesItemLineDetail.ItemRef = item.to_ref()
+        invoice.Line.append(line)
+
+        customer = Customer.all(max_results=1)[0]
+        invoice.CustomerRef = customer.to_ref()
+
+        invoice.CustomerMemo = CustomerMemo()
+        invoice.CustomerMemo.value = "Customer Memo"
+        invoice.save()
+
+        query_invoice = Invoice.get(invoice.Id)
+
+        self.assertEquals(query_invoice.CustomerRef.name, customer.DisplayName)
+        self.assertEquals(query_invoice.CustomerMemo.value, "Customer Memo")
+        self.assertEquals(query_invoice.Line[0].Description, "description")
+        self.assertEquals(query_invoice.Line[0].Amount, 100.0)
diff --git a/tests/unit/objects/test_bill.py b/tests/unit/objects/test_bill.py
@@ -18,6 +18,16 @@ def test_unicode(self):
 
         self.assertEquals(str(bill), "1000")
 
+    def test_to_LinkedTxn(self):
+        bill = Bill()
+        bill.Id = 10
+
+        linked_txn = bill.to_linked_txn()
+
+        self.assertEquals(linked_txn.TxnId, bill.Id)
+        self.assertEquals(linked_txn.TxnType, "Bill")
+        self.assertEquals(linked_txn.TxnLineId, 1)
+
 
 class BillLineTests(unittest.TestCase):
     def test_unicode(self):
