Clean up

soitgoes · Aug 19, 2016 · ed52868 · ed52868
1 parent 64f0991
commit ed52868
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 1 deletion.
diff --git a/LoveSeat.IntegrationTest/PagingHelperTest.cs b/LoveSeat.IntegrationTest/PagingHelperTest.cs
@@ -1,5 +1,4 @@
 using LoveSeat.Interfaces;
-using LoveSeat.Support;
 using Moq;
 using NUnit.Framework;
 

diff --git a/LoveSeat/Document.cs b/LoveSeat/Document.cs
@@ -16,6 +16,8 @@ public Document(T obj) : base(objectSerializer.Serialize<T>(obj)) {
         public Document(T obj, IObjectSerializer objectSerializer) : base(objectSerializer.Serialize<T>(obj))
         {            
         }
+
+        public T Item { get { return this.ToObject<T>(); } }        
     }
 
     #region Bulk documents
@@ -119,6 +121,14 @@ public bool HasAttachment
             get { return this["_attachments"] != null; }
         }
 
+        public void AddAttachment(string filename, byte[] data)
+        {
+            var jobj = this.GetValue("_attachments") as JObject ?? new JObject();
+            jobj[filename] = new JObject();
+            jobj[filename]["data"] = Convert.ToBase64String(data);
+            this["_attachments"] = jobj;
+        }
+
         public IEnumerable<string> GetAttachmentNames()
         {
             var attachment = this["_attachments"];