Updated tests.

ej2 · ej2 · commit 870e7deff408 · 2016-04-04T09:17:07.000-05:00
diff --git a/tests/integration/test_bill.py b/tests/integration/test_bill.py
@@ -1,13 +1,23 @@
+import os
 import unittest
 from datetime import datetime
 
+from quickbooks.client import QuickBooks
 from quickbooks.objects.base import Ref
 from quickbooks.objects.bill import Bill, BillLine, AccountBasedExpenseLineDetail
 from quickbooks.objects.vendor import Vendor
 
 
 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/test_billpayment.py b/tests/integration/test_billpayment.py
@@ -1,6 +1,8 @@
+import os
 import unittest
 from datetime import datetime
 
+from quickbooks.client import QuickBooks
 from quickbooks.objects.account import Account
 from quickbooks.objects.bill import Bill
 from quickbooks.objects.billpayment import BillPayment, BillPaymentLine, CheckPayment
@@ -9,6 +11,14 @@
 
 class BillPaymentTest(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/test_invoice.py b/tests/integration/test_invoice.py
@@ -1,5 +1,7 @@
+import os
 import unittest
 
+from quickbooks.client import QuickBooks
 from quickbooks.objects.customer import Customer
 from quickbooks.objects.detailline import SaleItemLine, SalesItemLineDetail
 from quickbooks.objects.invoice import Invoice
@@ -8,6 +10,16 @@
 
 
 class InvoiceTest(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')
+        )
+
     def test_create(self):
         invoice = Invoice()
 
diff --git a/tests/integration/test_item.py b/tests/integration/test_item.py
@@ -1,12 +1,23 @@
+import os
 import unittest
 from datetime import datetime
 
+from quickbooks.client import QuickBooks
 from quickbooks.objects.account import Account
 from quickbooks.objects.item import Item
 
 
 class ItemTest(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 Item {0}".format(self.account_number)
 
