Updated Employee.

ej2 · ej2 · commit 571ceaf5d659 · 2015-09-10T11:47:08.000-05:00
diff --git a/quickbooks/objects/employee.py b/quickbooks/objects/employee.py
@@ -1,7 +1,7 @@
-from base import Address, PhoneNumber, QuickbooksManagedObject
+from base import Address, PhoneNumber, QuickbooksManagedObject, QuickbooksTransactionEntity, Ref
 
 
-class Employee(QuickbooksManagedObject):
+class Employee(QuickbooksManagedObject, QuickbooksTransactionEntity):
     """
     QBO definition: Employee represents the people who are working for the company.
     """
@@ -16,14 +16,34 @@ class Employee(QuickbooksManagedObject):
     def __init__(self):
         super(Employee, self).__init__()
         self.SSN = ""
-        self.BillableTime = ""
+
         self.GivenName = ""
         self.FamilyName = ""
+        self.MiddleName = ""
         self.DisplayName = ""
+        self.Suffix = ""
         self.PrintOnCheckName = ""
+        self.EmployeeNumber = ""
+        self.Title = ""
+        self.BillRate = 0
+        self.BirthDate = ""
+        self.Gender = ""
+        self.HiredDate = ""
+        self.ReleasedDate = ""
         self.Active = True
+        self.Organization = False
+        self.BillableTime = False
 
         self.PrimaryAddr = None
 
     def __unicode__(self):
         return self.DisplayName
+
+    def to_ref(self):
+        ref = Ref()
+
+        ref.name = self.DisplayName
+        ref.type = self.qbo_object_name
+        ref.value = self.Id
+
+        return ref
diff --git a/tests/unit/objects/test_employee.py b/tests/unit/objects/test_employee.py
@@ -9,3 +9,14 @@ def test_unicode(self):
         employee.DisplayName = "test"
 
         self.assertEquals(unicode(employee), "test")
+
+    def test_to_ref(self):
+        employee = Employee()
+        employee.DisplayName = "test"
+        employee.Id = 100
+
+        ref = employee.to_ref()
+
+        self.assertEquals(ref.name, "test")
+        self.assertEquals(ref.type, "Employee")
+        self.assertEquals(ref.value, 100)
