Updated attachable and added test for file upload.

ej2 · ej2 · commit 4d5bd28a1cb1 · 2016-07-21T13:53:42.000-04:00
diff --git a/README.rst b/README.rst
@@ -245,8 +245,22 @@ Review results for batch operation:
 
 Attachments
 ----------------
+See `Attachable documentation`_ for list of valid file types
+Attaching a note:
 
-Adding attachments (See `Attachable documentation`_ for list of valid file types):
+::
+
+     attachment = Attachable()
+
+     attachable_ref = AttachableRef()
+     attachable_ref = .EntityRef = entity.to_ref()
+
+     attachment.AttachableRef.append(attachable_ref)
+
+     attachment.Note = 'This is a note'
+     attachment.save(qb=qb)
+
+Attaching a file:
 
 ::
 
diff --git a/quickbooks/objects/attachable.py b/quickbooks/objects/attachable.py
@@ -49,3 +49,21 @@ def to_ref(self):
         ref.type = self.qbo_object_name
         ref.value = self.Id
         return ref
+
+    def save(self, qb=None):
+        if not qb:
+            qb = QuickBooks()
+
+        if self.Id and self.Id > 0:
+            json_data = qb.update_object(self.qbo_object_name, self.to_json(), _file_path=self._FilePath)
+        else:
+            json_data = qb.create_object(self.qbo_object_name, self.to_json(), _file_path=self._FilePath)
+
+        if self.FileName:
+            obj = type(self).from_json(json_data['AttachableResponse'][0]['Attachable'])
+        else:
+            obj = type(self).from_json(json_data['Attachable'])
+
+        self.Id = obj.Id
+
+        return obj
diff --git a/tests/integration/test_attachable.py b/tests/integration/test_attachable.py
@@ -46,3 +46,21 @@ def test_update_note(self):
 
         query_attachable = Attachable.get(attachable.Id, qb=self.qb_client)
         self.assertEquals(query_attachable.Note, "Note updated on {}".format(self.time.strftime("%Y-%m-%d %H:%M:%S")))
+
+    def test_create_file(self):
+        attachable = Attachable()
+
+        vendor = Vendor.all(max_results=1, qb=self.qb_client)[0]
+
+        attachable_ref = AttachableRef()
+        attachable_ref.EntityRef = vendor.to_ref()
+        attachable.AttachableRef.append(attachable_ref)
+
+        attachable.FileName = 'TestFileName'
+        attachable._FilePath = __file__
+        attachable.ContentType = 'application/txt'
+
+        attachable.save(qb=self.qb_client)
+        query_attachable = Attachable.get(attachable.Id, qb=self.qb_client)
+
+        self.assertEquals(query_attachable.AttachableRef[0].EntityRef.value, vendor.Id)
diff --git a/tests/unit/objects/test_attachable.py b/tests/unit/objects/test_attachable.py
@@ -0,0 +1,30 @@
+import unittest
+
+from quickbooks import QuickBooks
+from quickbooks.objects.attachable import Attachable
+
+
+class AttachableTests(unittest.TestCase):
+    def test_unicode(self):
+        attachable = Attachable()
+        attachable.FileName = "test"
+
+        self.assertEquals(str(attachable), "test")
+
+    def test_to_ref(self):
+        attachable = Attachable()
+        attachable.FileName = "test"
+        attachable.Id = 12
+
+        ref = attachable.to_ref()
+
+        self.assertEquals(ref.name, "test")
+        self.assertEquals(ref.type, "Attachable")
+        self.assertEquals(ref.value, 12)
+
+    def test_valid_object_name(self):
+        attachable = Attachable()
+        client = QuickBooks()
+        result = client.isvalid_object_name(attachable.qbo_object_name)
+
+        self.assertTrue(result)
