Пример #1
0
  @Test
  public void migrateHtmlWithRelatedMessage() throws Exception {
    SQLiteDatabase db = createV50Database();
    insertHtmlWithRelatedMessage(db);
    db.close();

    LocalStore localStore = LocalStore.getInstance(account, RuntimeEnvironment.application);

    LocalMessage msg = localStore.getFolder("dev").getMessage("10");
    FetchProfile fp = new FetchProfile();
    fp.add(FetchProfile.Item.BODY);
    localStore.getFolder("dev").fetch(Collections.singletonList(msg), fp, null);

    Assert.assertEquals(9, msg.getId());
    Assert.assertEquals(11, msg.getHeaderNames().size());
    Assert.assertEquals("multipart/mixed", msg.getMimeType());
    Assert.assertEquals(1, msg.getAttachmentCount());

    Multipart msgBody = (Multipart) msg.getBody();
    Assert.assertEquals("------------050707070308090509030605", msgBody.getBoundary());

    Multipart multipartAlternativePart = (Multipart) msgBody.getBodyPart(0).getBody();
    BodyPart htmlPart = multipartAlternativePart.getBodyPart(1);
    String msgTextContent = MessageExtractor.getTextFromPart(htmlPart);
    Assert.assertNotNull(msgTextContent);
    Assert.assertTrue(msgTextContent.contains("cid:[email protected]"));

    Assert.assertEquals("image/jpeg", msgBody.getBodyPart(1).getMimeType());
  }
Пример #2
0
  @Test
  public void migrateTextPlain() throws Exception {
    SQLiteDatabase db = createV50Database();
    insertSimplePlaintextMessage(db);
    db.close();

    LocalStore localStore = LocalStore.getInstance(account, RuntimeEnvironment.application);

    LocalMessage msg = localStore.getFolder("dev").getMessage("3");
    FetchProfile fp = new FetchProfile();
    fp.add(FetchProfile.Item.BODY);
    localStore.getFolder("dev").fetch(Collections.singletonList(msg), fp, null);

    Assert.assertEquals("text/plain", msg.getMimeType());
    Assert.assertEquals(2, msg.getId());
    Assert.assertEquals(13, msg.getHeaderNames().size());
    Assert.assertEquals(0, msg.getAttachmentCount());

    Assert.assertEquals(1, msg.getHeader("User-Agent").length);
    Assert.assertEquals("Mutt/1.5.24 (2015-08-30)", msg.getHeader("User-Agent")[0]);
    Assert.assertEquals(1, msg.getHeader(MimeHeader.HEADER_CONTENT_TYPE).length);
    Assert.assertEquals(
        "text/plain",
        MimeUtility.getHeaderParameter(msg.getHeader(MimeHeader.HEADER_CONTENT_TYPE)[0], null));
    Assert.assertEquals(
        "utf-8",
        MimeUtility.getHeaderParameter(
            msg.getHeader(MimeHeader.HEADER_CONTENT_TYPE)[0], "charset"));

    Assert.assertTrue(msg.getBody() instanceof BinaryMemoryBody);

    String msgTextContent = MessageExtractor.getTextFromPart(msg);
    Assert.assertEquals("nothing special here.\r\n", msgTextContent);
  }
Пример #3
0
  @Test
  public void testShouldSanitizeOutputHtml() throws MessagingException {
    // Create text/plain body
    TextBody body = new TextBody(BODY_TEXT);

    // Create message
    MimeMessage message = new MimeMessage();
    MimeMessageHelper.setBody(message, body);

    // Prepare fixture
    HtmlSanitizer htmlSanitizer = mock(HtmlSanitizer.class);
    MessageViewInfoExtractor messageViewInfoExtractor =
        new MessageViewInfoExtractor(context, null, htmlSanitizer);
    String value = "--sanitized html--";
    when(htmlSanitizer.sanitize(any(String.class))).thenReturn(value);

    // Extract text
    List<Part> outputNonViewableParts = new ArrayList<>();
    ArrayList<Viewable> outputViewableParts = new ArrayList<>();
    MessageExtractor.findViewablesAndAttachments(
        message, outputViewableParts, outputNonViewableParts);
    ViewableExtractedText viewableExtractedText =
        messageViewInfoExtractor.extractTextFromViewables(outputViewableParts);

    assertSame(value, viewableExtractedText.html);
  }
Пример #4
0
  @Test
  public void testSimpleHtmlMessage() throws MessagingException {
    String bodyText = "<strong>K-9 Mail</strong> rocks :&gt;";

    // Create text/plain body
    TextBody body = new TextBody(bodyText);

    // Create message
    MimeMessage message = new MimeMessage();
    message.setHeader(MimeHeader.HEADER_CONTENT_TYPE, "text/html");
    MimeMessageHelper.setBody(message, body);

    // Extract text
    ArrayList<Viewable> outputViewableParts = new ArrayList<>();
    MessageExtractor.findViewablesAndAttachments(message, outputViewableParts, null);
    assertEquals(outputViewableParts.size(), 1);
    ViewableExtractedText container =
        messageViewInfoExtractor.extractTextFromViewables(outputViewableParts);

    String expectedText = BODY_TEXT;
    String expectedHtml = bodyText;

    assertEquals(expectedText, container.text);
    assertEquals(expectedHtml, getHtmlBodyText(container.html));
  }
Пример #5
0
  private boolean containsInlinePgpEncryptedText(Message message) {
    Part textPart = textPartFinder.findFirstTextPart(message);
    if (!isUsableTextPart(textPart)) {
      return false;
    }

    String text = MessageExtractor.getTextFromPart(textPart);
    if (text == null) {
      return false;
    }

    return PGP_MESSAGE_PATTERN.matcher(text).find();
  }
Пример #6
0
  @Test
  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 = MimeMultipart.newInstance();
    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();
    MimeMessageHelper.setBody(message, multipart);

    // Extract text
    List<Part> outputNonViewableParts = new ArrayList<>();
    ArrayList<Viewable> outputViewableParts = new ArrayList<>();
    MessageExtractor.findViewablesAndAttachments(
        message, outputViewableParts, outputNonViewableParts);
    ViewableExtractedText container =
        messageViewInfoExtractor.extractTextFromViewables(outputViewableParts);

    String expectedText =
        bodyText1
            + "\r\n\r\n"
            + "------------------------------------------------------------------------\r\n\r\n"
            + bodyText2;
    String expectedHtml =
        "<pre class=\"k9mail\">"
            + bodyText1
            + "</pre>"
            + "<p style=\"margin-top: 2.5em; margin-bottom: 1em; "
            + "border-bottom: 1px solid #000\"></p>"
            + "<pre class=\"k9mail\">"
            + bodyText2
            + "</pre>";

    assertEquals(expectedText, container.text);
    assertEquals(expectedHtml, getHtmlBodyText(container.html));
  }
Пример #7
0
  @Test
  public void migratePgpInlineClearsignedMessage() throws Exception {
    SQLiteDatabase db = createV50Database();
    insertPgpInlineClearsignedMessage(db);
    db.close();

    LocalStore localStore = LocalStore.getInstance(account, RuntimeEnvironment.application);

    LocalMessage msg = localStore.getFolder("dev").getMessage("8");
    FetchProfile fp = new FetchProfile();
    fp.add(FetchProfile.Item.BODY);
    localStore.getFolder("dev").fetch(Collections.singletonList(msg), fp, null);

    Assert.assertEquals(7, msg.getId());
    Assert.assertEquals(12, msg.getHeaderNames().size());
    Assert.assertEquals("text/plain", msg.getMimeType());
    Assert.assertEquals(0, msg.getAttachmentCount());
    Assert.assertTrue(msg.getBody() instanceof BinaryMemoryBody);

    String msgTextContent = MessageExtractor.getTextFromPart(msg);
    Assert.assertEquals(
        OpenPgpUtils.PARSE_RESULT_SIGNED_MESSAGE, OpenPgpUtils.parseMessage(msgTextContent));
  }
Пример #8
0
  @Test
  public void testSimplePlainTextMessage() throws MessagingException {
    // Create text/plain body
    TextBody body = new TextBody(BODY_TEXT);

    // Create message
    MimeMessage message = new MimeMessage();
    message.setHeader(MimeHeader.HEADER_CONTENT_TYPE, "text/plain");
    MimeMessageHelper.setBody(message, body);

    // Extract text
    List<Part> outputNonViewableParts = new ArrayList<>();
    ArrayList<Viewable> outputViewableParts = new ArrayList<>();
    MessageExtractor.findViewablesAndAttachments(
        message, outputViewableParts, outputNonViewableParts);
    ViewableExtractedText container =
        messageViewInfoExtractor.extractTextFromViewables(outputViewableParts);

    String expectedText = BODY_TEXT;
    String expectedHtml = "<pre class=\"k9mail\">" + "K-9 Mail rocks :&gt;" + "</pre>";

    assertEquals(expectedText, container.text);
    assertEquals(expectedHtml, getHtmlBodyText(container.html));
  }
Пример #9
0
  @Test
  public void testMultipartDigestWithMessages() throws Exception {
    String data =
        "Content-Type: multipart/digest; boundary=\"bndry\"\r\n"
            + "\r\n"
            + "--bndry\r\n"
            + "\r\n"
            + "Content-Type: text/plain\r\n"
            + "\r\n"
            + "text body of first message\r\n"
            + "\r\n"
            + "--bndry\r\n"
            + "\r\n"
            + "Subject: subject of second message\r\n"
            + "Content-Type: multipart/alternative; boundary=\"bndry2\"\r\n"
            + "\r\n"
            + "--bndry2\r\n"
            + "Content-Type: text/plain\r\n"
            + "\r\n"
            + "text part of second message\r\n"
            + "\r\n"
            + "--bndry2\r\n"
            + "Content-Type: text/html\"\r\n"
            + "\r\n"
            + "html part of second message\r\n"
            + "\r\n"
            + "--bndry2--\r\n"
            + "\r\n"
            + "--bndry--\r\n";
    MimeMessage message =
        MimeMessage.parseMimeMessage(new ByteArrayInputStream(data.getBytes()), false);

    // Extract text
    List<Part> outputNonViewableParts = new ArrayList<>();
    ArrayList<Viewable> outputViewableParts = new ArrayList<>();
    MessageExtractor.findViewablesAndAttachments(
        message, outputViewableParts, outputNonViewableParts);

    String expectedExtractedText =
        "Subject: (No subject)\r\n"
            + "\r\n"
            + "text body of first message\r\n"
            + "\r\n"
            + "\r\n"
            + "------------------------------------------------------------------------\r\n"
            + "\r\n"
            + "Subject: subject of second message\r\n"
            + "\r\n"
            + "text part of second message\r\n";
    String expectedHtmlText =
        "<table style=\"border: 0\">"
            + "<tr><th style=\"text-align: left; vertical-align: top;\">Subject:</th><td>(No subject)</td></tr>"
            + "</table>"
            + "<pre class=\"k9mail\">text body of first message<br /></pre>"
            + "<p style=\"margin-top: 2.5em; margin-bottom: 1em; border-bottom: 1px solid #000\"></p>"
            + "<table style=\"border: 0\">"
            + "<tr><th style=\"text-align: left; vertical-align: top;\">Subject:</th><td>subject of second message</td></tr>"
            + "</table>"
            + "<pre class=\"k9mail\">text part of second message<br /></pre>";

    assertEquals(4, outputViewableParts.size());
    assertEquals(
        "subject of second message",
        ((MessageHeader) outputViewableParts.get(2)).getMessage().getSubject());

    ViewableExtractedText firstMessageExtractedText =
        messageViewInfoExtractor.extractTextFromViewables(outputViewableParts);
    assertEquals(expectedExtractedText, firstMessageExtractedText.text);
    assertEquals(expectedHtmlText, getHtmlBodyText(firstMessageExtractedText.html));
  }
Пример #10
0
  @Test
  public void testTextPlusRfc822Message() throws MessagingException {
    K9ActivityCommon.setLanguage(context, "en");
    Locale.setDefault(Locale.US);
    TimeZone.setDefault(TimeZone.getTimeZone("GMT+01:00"));

    String innerBodyText = "Hey there. I'm inside a message/rfc822 (inline) attachment.";

    // Create text/plain body
    TextBody textBody = new TextBody(BODY_TEXT);

    // 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), false);
    innerMessage.setRecipients(RecipientType.TO, new Address[] {new Address("*****@*****.**")});
    innerMessage.setSubject("Subject");
    innerMessage.setFrom(new Address("*****@*****.**"));
    MimeMessageHelper.setBody(innerMessage, innerBody);

    // Create multipart/mixed part
    MimeMultipart multipart = MimeMultipart.newInstance();
    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();
    MimeMessageHelper.setBody(message, multipart);

    // Extract text
    List<Part> outputNonViewableParts = new ArrayList<Part>();
    ArrayList<Viewable> outputViewableParts = new ArrayList<>();
    MessageExtractor.findViewablesAndAttachments(
        message, outputViewableParts, outputNonViewableParts);
    ViewableExtractedText container =
        messageViewInfoExtractor.extractTextFromViewables(outputViewableParts);

    String expectedText =
        BODY_TEXT
            + "\r\n\r\n"
            + "----- message.eml ------------------------------------------------------"
            + "\r\n\r\n"
            + "From: [email protected]"
            + "\r\n"
            + "To: [email protected]"
            + "\r\n"
            + "Sent: Sat Mar 17 00:00:00 GMT+01:00 2012"
            + "\r\n"
            + "Subject: Subject"
            + "\r\n"
            + "\r\n"
            + innerBodyText;
    String expectedHtml =
        "<pre class=\"k9mail\">"
            + BODY_TEXT_HTML
            + "</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>[email protected]</td>"
            + "</tr><tr>"
            + "<th style=\"text-align: left; vertical-align: top;\">To:</th>"
            + "<td>[email protected]</td>"
            + "</tr><tr>"
            + "<th style=\"text-align: left; vertical-align: top;\">Sent:</th>"
            + "<td>Sat Mar 17 00:00:00 GMT+01:00 2012</td>"
            + "</tr><tr>"
            + "<th style=\"text-align: left; vertical-align: top;\">Subject:</th>"
            + "<td>Subject</td>"
            + "</tr>"
            + "</table>"
            + "<pre class=\"k9mail\">"
            + innerBodyText
            + "</pre>";

    assertEquals(expectedText, container.text);
    assertEquals(expectedHtml, getHtmlBodyText(container.html));
  }