mirror of
https://github.com/moparisthebest/k-9
synced 2024-12-02 14:02:17 -05:00
Added tests for MimeUtility.extractTextAndAttachments()
This commit is contained in:
cketti
parent
b9803ece19
commit
03d4cee14a
188
tests/src/com/fsck/k9/mail/internet/ViewablesTest.java
Normal file
188
tests/src/com/fsck/k9/mail/internet/ViewablesTest.java
Normal file
@ -0,0 +1,188 @@
|
|||||||
|
package com.fsck.k9.mail.internet;
|
||||||
|
|
||||||
|
import java.util.Date;
|
||||||
|
import java.util.Locale;
|
||||||
|
import java.util.TimeZone;
|
||||||
|
import android.test.AndroidTestCase;
|
||||||
|
import com.fsck.k9.mail.Address;
|
||||||
|
import com.fsck.k9.mail.MessagingException;
|
||||||
|
import com.fsck.k9.mail.Message.RecipientType;
|
||||||
|
import com.fsck.k9.mail.internet.MimeUtility.ViewableContainer;
|
||||||
|
|
||||||
|
public class ViewablesTest extends AndroidTestCase {
|
||||||
|
|
||||||
|
public void testSimplePlainTextMessage() throws MessagingException {
|
||||||
|
String bodyText = "K-9 Mail rocks :>";
|
||||||
|
|
||||||
|
// Create text/plain body
|
||||||
|
TextBody body = new TextBody(bodyText);
|
||||||
|
|
||||||
|
// Create message
|
||||||
|
MimeMessage message = new MimeMessage();
|
||||||
|
message.setBody(body);
|
||||||
|
|
||||||
|
// Extract text
|
||||||
|
ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message);
|
||||||
|
|
||||||
|
String expectedText = bodyText;
|
||||||
|
String expectedHtml =
|
||||||
|
"<html><head/><body>" +
|
||||||
|
"<pre style=\"white-space: pre-wrap; word-wrap:break-word; " +
|
||||||
|
"font-family: sans-serif\">" +
|
||||||
|
"K-9 Mail rocks :>" +
|
||||||
|
"</pre>" +
|
||||||
|
"</body></html>";
|
||||||
|
|
||||||
|
assertEquals(expectedText, container.text);
|
||||||
|
assertEquals(expectedHtml, container.html);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void testSimpleHtmlMessage() throws MessagingException {
|
||||||
|
String bodyText = "<strong>K-9 Mail</strong> rocks :>";
|
||||||
|
|
||||||
|
// Create text/plain body
|
||||||
|
TextBody body = new TextBody(bodyText);
|
||||||
|
|
||||||
|
// Create message
|
||||||
|
MimeMessage message = new MimeMessage();
|
||||||
|
message.setHeader("Content-Type", "text/html");
|
||||||
|
message.setBody(body);
|
||||||
|
|
||||||
|
// Extract text
|
||||||
|
ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message);
|
||||||
|
|
||||||
|
String expectedText = "K-9 Mail rocks :>";
|
||||||
|
String expectedHtml =
|
||||||
|
"<html><head/><body>" +
|
||||||
|
bodyText +
|
||||||
|
"</body></html>";
|
||||||
|
|
||||||
|
assertEquals(expectedText, container.text);
|
||||||
|
assertEquals(expectedHtml, container.html);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void testMultipartPlainTextMessage() throws MessagingException {
|
||||||
|
String bodyText1 = "text body 1";
|
||||||
|
String bodyText2 = "text body 2";
|
||||||
|
|
||||||
|
// Create text/plain bodies
|
||||||
|
TextBody body1 = new TextBody(bodyText1);
|
||||||
|
TextBody body2 = new TextBody(bodyText2);
|
||||||
|
|
||||||
|
// Create multipart/mixed part
|
||||||
|
MimeMultipart multipart = new MimeMultipart();
|
||||||
|
MimeBodyPart bodyPart1 = new MimeBodyPart(body1, "text/plain");
|
||||||
|
MimeBodyPart bodyPart2 = new MimeBodyPart(body2, "text/plain");
|
||||||
|
multipart.addBodyPart(bodyPart1);
|
||||||
|
multipart.addBodyPart(bodyPart2);
|
||||||
|
|
||||||
|
// Create message
|
||||||
|
MimeMessage message = new MimeMessage();
|
||||||
|
message.setBody(multipart);
|
||||||
|
|
||||||
|
// Extract text
|
||||||
|
ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message);
|
||||||
|
|
||||||
|
String expectedText =
|
||||||
|
bodyText1 + "\n\n" +
|
||||||
|
"------------------------------------------------------------------------\n\n" +
|
||||||
|
bodyText2;
|
||||||
|
String expectedHtml =
|
||||||
|
"<html><head/><body>" +
|
||||||
|
"<pre style=\"white-space: pre-wrap; word-wrap:break-word; " +
|
||||||
|
"font-family: sans-serif\">" +
|
||||||
|
bodyText1 +
|
||||||
|
"</pre>" +
|
||||||
|
"<p style=\"margin-top: 2.5em; margin-bottom: 1em; " +
|
||||||
|
"border-bottom: 1px solid #000\"></p>" +
|
||||||
|
"<pre style=\"white-space: pre-wrap; word-wrap:break-word; " +
|
||||||
|
"font-family: sans-serif\">" +
|
||||||
|
bodyText2 +
|
||||||
|
"</pre>" +
|
||||||
|
"</body></html>";
|
||||||
|
|
||||||
|
|
||||||
|
assertEquals(expectedText, container.text);
|
||||||
|
assertEquals(expectedHtml, container.html);
|
||||||
|
}
|
||||||
|
|
||||||
|
public void testTextPlusRfc822Message() throws MessagingException {
|
||||||
|
Locale.setDefault(Locale.US);
|
||||||
|
TimeZone.setDefault(TimeZone.getTimeZone("UTC"));
|
||||||
|
|
||||||
|
String bodyText = "Some text here";
|
||||||
|
String innerBodyText = "Hey there. I'm inside a message/rfc822 (inline) attachment.";
|
||||||
|
|
||||||
|
// Create text/plain body
|
||||||
|
TextBody textBody = new TextBody(bodyText);
|
||||||
|
|
||||||
|
// Create inner text/plain body
|
||||||
|
TextBody innerBody = new TextBody(innerBodyText);
|
||||||
|
|
||||||
|
// Create message/rfc822 body
|
||||||
|
MimeMessage innerMessage = new MimeMessage();
|
||||||
|
innerMessage.addSentDate(new Date(112, 02, 17));
|
||||||
|
innerMessage.setRecipients(RecipientType.TO, new Address[] { new Address("to@example.com") });
|
||||||
|
innerMessage.setSubject("Subject");
|
||||||
|
innerMessage.setFrom(new Address("from@example.com"));
|
||||||
|
innerMessage.setBody(innerBody);
|
||||||
|
|
||||||
|
// Create multipart/mixed part
|
||||||
|
MimeMultipart multipart = new MimeMultipart();
|
||||||
|
MimeBodyPart bodyPart1 = new MimeBodyPart(textBody, "text/plain");
|
||||||
|
MimeBodyPart bodyPart2 = new MimeBodyPart(innerMessage, "message/rfc822");
|
||||||
|
bodyPart2.setHeader("Content-Disposition", "inline; filename=\"message.eml\"");
|
||||||
|
multipart.addBodyPart(bodyPart1);
|
||||||
|
multipart.addBodyPart(bodyPart2);
|
||||||
|
|
||||||
|
// Create message
|
||||||
|
MimeMessage message = new MimeMessage();
|
||||||
|
message.setBody(multipart);
|
||||||
|
|
||||||
|
// Extract text
|
||||||
|
ViewableContainer container = MimeUtility.extractTextAndAttachments(getContext(), message);
|
||||||
|
|
||||||
|
String expectedText =
|
||||||
|
bodyText +
|
||||||
|
"\n\n" +
|
||||||
|
"----- message.eml ------------------------------------------------------" +
|
||||||
|
"\n\n" +
|
||||||
|
"From: from@example.com" + "\n" +
|
||||||
|
"To: to@example.com" + "\n" +
|
||||||
|
"Sent: Sat Mar 17 00:00:00 GMT+00:00 2012" + "\n" +
|
||||||
|
"Subject: Subject" + "\n" +
|
||||||
|
"\n" +
|
||||||
|
innerBodyText;
|
||||||
|
String expectedHtml =
|
||||||
|
"<html><head/><body>" +
|
||||||
|
"<pre style=\"white-space: pre-wrap; word-wrap:break-word; " +
|
||||||
|
"font-family: sans-serif\">" +
|
||||||
|
bodyText +
|
||||||
|
"</pre>" +
|
||||||
|
"<p style=\"margin-top: 2.5em; margin-bottom: 1em; border-bottom: " +
|
||||||
|
"1px solid #000\">message.eml</p>" +
|
||||||
|
"<table style=\"border: 0\">" +
|
||||||
|
"<tr>" +
|
||||||
|
"<th style=\"text-align: left; vertical-align: top;\">From:</th>" +
|
||||||
|
"<td>from@example.com</td>" +
|
||||||
|
"</tr><tr>" +
|
||||||
|
"<th style=\"text-align: left; vertical-align: top;\">To:</th>" +
|
||||||
|
"<td>to@example.com</td>" +
|
||||||
|
"</tr><tr>" +
|
||||||
|
"<th style=\"text-align: left; vertical-align: top;\">Sent:</th>" +
|
||||||
|
"<td>Sat Mar 17 00:00:00 GMT+00:00 2012</td>" +
|
||||||
|
"</tr><tr>" +
|
||||||
|
"<th style=\"text-align: left; vertical-align: top;\">Subject:</th>" +
|
||||||
|
"<td>Subject</td>" +
|
||||||
|
"</tr>" +
|
||||||
|
"</table>" +
|
||||||
|
"<pre style=\"white-space: pre-wrap; word-wrap:break-word; " +
|
||||||
|
"font-family: sans-serif\">" +
|
||||||
|
innerBodyText +
|
||||||
|
"</pre>" +
|
||||||
|
"</body></html>";
|
||||||
|
|
||||||
|
assertEquals(expectedText, container.text);
|
||||||
|
assertEquals(expectedHtml, container.html);
|
||||||
|
}
|
||||||
|
}
|
||||||
Loading…
Reference in New Issue
Block a user