Add TrackingClass object

amacneil · amacneil · commit 51521c105b99 · 2016-02-09T16:14:21.000-08:00
diff --git a/quickbooks/objects/trackingclass.py b/quickbooks/objects/trackingclass.py
@@ -0,0 +1,38 @@
+from six import python_2_unicode_compatible
+from .base import QuickbooksManagedObject, QuickbooksTransactionEntity, Ref
+
+
+@python_2_unicode_compatible
+class TrackingClass(QuickbooksManagedObject, QuickbooksTransactionEntity):
+    """
+    QBO definition: Classes provide a way to track different segments of the business so they're
+    not tied to a particular client or project. For example, you can define classes to break down
+    the income and expenses for each business segment. Classes are applied to individual detail
+    lines of a transaction. This is in contrast to Department objects, which are applied to the
+    entire transaction.
+    """
+
+    class_dict = {
+        "ParentRef": Ref
+    }
+
+    qbo_object_name = "Class"
+
+    def __init__(self):
+        super(TrackingClass, self).__init__()
+        self.Name = ""
+        self.SubClass = False
+        self.FullyQualifiedName = ""
+        self.Active = True
+
+    def __str__(self):
+        return self.Name
+
+    def to_ref(self):
+        ref = Ref()
+
+        ref.name = self.Name
+        ref.type = self.qbo_object_name
+        ref.value = self.Id
+
+        return ref
diff --git a/tests/unit/objects/test_trackingclass.py b/tests/unit/objects/test_trackingclass.py
@@ -0,0 +1,22 @@
+import unittest
+
+from quickbooks.objects.trackingclass import TrackingClass
+
+
+class TrackingClassTests(unittest.TestCase):
+    def test_unicode(self):
+        cls = TrackingClass()
+        cls.Name = "test"
+
+        self.assertEquals(str(cls), "test")
+
+    def test_to_ref(self):
+        cls = TrackingClass()
+        cls.Name = "test"
+        cls.Id = 100
+
+        dept_ref = cls.to_ref()
+
+        self.assertEquals(dept_ref.name, "test")
+        self.assertEquals(dept_ref.type, "Class")
+        self.assertEquals(dept_ref.value, 100)
