From f4c1f63351b14969fe9cbf210fa849423f3b871a Mon Sep 17 00:00:00 2001
From: str4d <str4d@mail.i2p>
Date: Thu, 1 Jan 2015 10:36:33 +0000
Subject: [PATCH] Fixed attachment creation in tests

---
 .../provider/AttachmentProviderTests.java     | 35 ++++++++++++++++---
 1 file changed, 30 insertions(+), 5 deletions(-)

diff --git a/app/src/androidTest/java/i2p/bote/android/provider/AttachmentProviderTests.java b/app/src/androidTest/java/i2p/bote/android/provider/AttachmentProviderTests.java
index 2f5a5a1..77afdce 100644
--- a/app/src/androidTest/java/i2p/bote/android/provider/AttachmentProviderTests.java
+++ b/app/src/androidTest/java/i2p/bote/android/provider/AttachmentProviderTests.java
@@ -26,6 +26,8 @@ import java.io.OutputStream;
 import java.util.ArrayList;
 import java.util.List;
 
+import javax.activation.DataHandler;
+import javax.activation.DataSource;
 import javax.mail.Part;
 import javax.mail.internet.MimeBodyPart;
 
@@ -132,17 +134,40 @@ public class AttachmentProviderTests extends ProviderTestCase2<AttachmentProvide
     }
 
     private ContentAttachment createTextAttachment() throws Exception {
-        Part part = new MimeBodyPart(new ByteArrayInputStream("Test file content".getBytes()));
-        part.setFileName("test.txt");
-        return new ContentAttachment(getMockContext(), part);
+        return createAttachment("test.txt", "text/plain", "Test file content".getBytes());
     }
 
     private ContentAttachment createImageAttachment() throws Exception {
         Bitmap bitmap = BitmapFactory.decodeResource(getContext().getResources(), R.drawable.intro_1);
         ByteArrayOutputStream out = new ByteArrayOutputStream();
         bitmap.compress(Bitmap.CompressFormat.PNG, 100, out);
-        Part part = new MimeBodyPart(new ByteArrayInputStream(out.toByteArray()));
-        part.setFileName("intro_1.png");
+        return createAttachment("intro_1.png", "image/png", out.toByteArray());
+    }
+
+    private ContentAttachment createAttachment(final String fileName, final String mimeType, final byte[] content) throws Exception {
+        Part part = new MimeBodyPart();
+        part.setDataHandler(new DataHandler(new DataSource() {
+            @Override
+            public InputStream getInputStream() throws IOException {
+                return new ByteArrayInputStream(content);
+            }
+
+            @Override
+            public OutputStream getOutputStream() throws IOException {
+                throw new IOException("Cannot write to attachments");
+            }
+
+            @Override
+            public String getContentType() {
+                return mimeType;
+            }
+
+            @Override
+            public String getName() {
+                return fileName;
+            }
+        }));
+        part.setFileName(fileName);
         return new ContentAttachment(getMockContext(), part);
     }