|
1 | 1 | package org.nhindirect.common.mail.dsn; |
2 | 2 |
|
3 | 3 | import static org.junit.Assert.assertNotNull; |
| 4 | +import static org.junit.Assert.assertThat; |
4 | 5 | import static org.junit.Assert.assertTrue; |
| 6 | +import static org.hamcrest.CoreMatchers.containsString; |
5 | 7 | import static org.junit.Assert.assertEquals; |
6 | 8 |
|
7 | 9 | import java.util.ArrayList; |
| 10 | +import java.util.Collections; |
8 | 11 | import java.util.List; |
9 | 12 | import java.util.UUID; |
10 | 13 |
|
11 | 14 | import javax.mail.Address; |
| 15 | +import javax.mail.Header; |
12 | 16 | import javax.mail.internet.InternetAddress; |
13 | 17 | import javax.mail.internet.MimeBodyPart; |
14 | 18 | import javax.mail.internet.MimeMessage; |
| 19 | +import javax.mail.internet.MimeMultipart; |
15 | 20 |
|
16 | 21 | import org.junit.Test; |
17 | 22 | import org.nhindirect.common.mail.MailStandard; |
@@ -62,4 +67,46 @@ public void testCreateDSNMessage_createGeneralDSNMessage() throws Exception |
62 | 67 | assertTrue(!MailStandard.getHeader(dsnMessage, MailStandard.Headers.Date).isEmpty()); |
63 | 68 |
|
64 | 69 | } |
| 70 | + |
| 71 | + @Test |
| 72 | + public void testCreateDSNMessage_createDSNMessageWithOriginalHeaders() throws Exception |
| 73 | + { |
| 74 | + final DSNGenerator dsnGenerator = new DSNGenerator("Not Delivered:"); |
| 75 | + |
| 76 | + final DSNRecipientHeaders dsnRecipHeaders = |
| 77 | + new DSNRecipientHeaders(DSNAction.FAILED, |
| 78 | + DSNStatus.getStatus(DSNStatus.PERMANENT, DSNStatus.UNDEFINED_STATUS), new InternetAddress("ah4626@test.com")); |
| 79 | + |
| 80 | + final List<DSNRecipientHeaders> dsnHeaders = new ArrayList<DSNRecipientHeaders>(); |
| 81 | + dsnHeaders.add(dsnRecipHeaders); |
| 82 | + |
| 83 | + final String originalMessageId = UUID.randomUUID().toString(); |
| 84 | + final DSNMessageHeaders messageDSNHeaders = new DSNMessageHeaders("DirectJUNIT", originalMessageId, MtaNameType.DNS); |
| 85 | + |
| 86 | + List<Address> faileRecips = new ArrayList<Address>(); |
| 87 | + faileRecips.add(new InternetAddress("ah4626@test.com")); |
| 88 | + |
| 89 | + List<Header> originalMessageHeaders = new ArrayList<Header>(); |
| 90 | + originalMessageHeaders.add(new Header("Date", "Tue, 11 Jun 2015 02:43:38 -0500 (CDT)")); |
| 91 | + originalMessageHeaders.add(new Header("From", "\"Smith, John\" <from@test.com>")); |
| 92 | + originalMessageHeaders.add(new Header("To", "\"Ben & Jerry\" <benandjerry@test.com>")); |
| 93 | + originalMessageHeaders.add(new Header("Subject", "subject \0goes here")); // this subject contains a NUL character, which should get removed |
| 94 | + |
| 95 | + final DefaultDSNFailureTextBodyPartGenerator textGenerator = new DefaultDSNFailureTextBodyPartGenerator("", "%headers_tag%", "", |
| 96 | + "", "", HumanReadableTextAssemblerFactory.getInstance()); |
| 97 | + |
| 98 | + |
| 99 | + |
| 100 | + final MimeBodyPart textBodyPart = textGenerator.generate(new InternetAddress("gm2552@test.com"), faileRecips, Collections.enumeration(originalMessageHeaders)); |
| 101 | + |
| 102 | + MimeMessage dsnMessage = dsnGenerator.createDSNMessage(new InternetAddress("gm2552@test.com"), "test", new InternetAddress("postmaster@test.com"), |
| 103 | + dsnHeaders, messageDSNHeaders, textBodyPart); |
| 104 | + |
| 105 | + assertNotNull(dsnMessage); |
| 106 | + MimeBodyPart htmlBodyPart = (MimeBodyPart) ((MimeMultipart) dsnMessage.getContent()).getBodyPart(0); |
| 107 | + String htmlContent = (String) htmlBodyPart.getContent(); |
| 108 | + assertThat(htmlContent, containsString("subject goes here")); // NUL is removed |
| 109 | + assertThat(htmlContent, containsString("Ben & Jerry")); // ampersand is encoded |
| 110 | + assertThat(htmlContent, containsString("<from@test.com>")); // <> are encoded |
| 111 | + } |
65 | 112 | } |
0 commit comments