Merge pull request routablehq#37 from sidecars/bill_to_ref

ej2 · web-flow · commit 80f0ca8d8249 · 2016-07-21T14:23:05.000-04:00
Added to_ref method to the Bill object and added a test for it.
diff --git a/quickbooks/objects/bill.py b/quickbooks/objects/bill.py
@@ -134,3 +134,13 @@ def to_linked_txn(self):
         linked_txn.TxnLineId = 1
 
         return linked_txn
+
+    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_bill.py b/tests/unit/objects/test_bill.py
@@ -36,6 +36,17 @@ def test_valid_object_name(self):
 
         self.assertTrue(result)
 
+    def test_to_ref(self):
+        bill = Bill()
+        bill.DisplayName = "test"
+        bill.Id = 100
+
+        ref = bill.to_ref()
+
+        self.assertEquals(ref.name, "test")
+        self.assertEquals(ref.type, "Bill")
+        self.assertEquals(ref.value, 100)
+
 
 class BillLineTests(unittest.TestCase):
     def test_unicode(self):
