Merge branch 'integration_tests' of https://github.com/sidecars/python-quickbooks into integration_tests

Jason Otero · Jason Otero · commit 576d73a4b4e3 · 2016-07-22T10:40:41.000-04:00
diff --git a/README.rst b/README.rst
@@ -8,7 +8,7 @@ A Python library for accessing the Quickbooks API. Complete rework of
 
 These instructions were written for a Django application. Make sure to
 change it to whatever framework/method you’re using.
-You can find additioinal examples of usage in `Integration tests folder`_.
+You can find additional examples of usage in `Integration tests folder`_.
 
 Connecting your application to Quickbooks Online
 ------------------------------------------------
diff --git a/tests/integration/test_trackingclass.py b/tests/integration/test_trackingclass.py
@@ -0,0 +1,42 @@
+import os
+import unittest
+from datetime import datetime
+
+from quickbooks.client import QuickBooks
+from quickbooks.objects.trackingclass import Class
+
+
+class ClassTest(unittest.TestCase):
+    def setUp(self):
+        self.qb_client = 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.name = "Test Class {0}".format(datetime.now().strftime('%d%H%M'))
+
+    def test_create(self):
+        tracking_class = Class()
+        tracking_class.Name = self.name
+        tracking_class.save(qb=self.qb_client)
+
+        query_tracking_class = Class.get(tracking_class.Id, qb=self.qb_client)
+
+        self.assertEquals(query_tracking_class.Id, tracking_class.Id)
+        self.assertEquals(query_tracking_class.Name, self.name)
+
+    def test_update(self):
+        updated_name = "Updated {}".format(self.name)
+
+        tracking_class = Class.all(max_results=1, qb=self.qb_client)[0]
+        tracking_class.Name = updated_name
+        tracking_class.save(qb=self.qb_client)
+
+        query_tracking_class = Class.get(tracking_class.Id, qb=self.qb_client)
+
+        self.assertEquals(query_tracking_class.Id, tracking_class.Id)
+        self.assertEquals(query_tracking_class.Name, updated_name)
