MAILET-107 Provide tests for HasAttachment matcher
Project: http://git-wip-us.apache.org/repos/asf/james-project/repo Commit: http://git-wip-us.apache.org/repos/asf/james-project/commit/16829c1e Tree: http://git-wip-us.apache.org/repos/asf/james-project/tree/16829c1e Diff: http://git-wip-us.apache.org/repos/asf/james-project/diff/16829c1e Branch: refs/heads/master Commit: 16829c1e3ee94f1c89c32d1e6cb793afff1e472c Parents: 829dfc6 Author: Benoit Tellier <[email protected]> Authored: Wed Aug 17 15:06:09 2016 +0700 Committer: Benoit Tellier <[email protected]> Committed: Mon Aug 29 19:29:08 2016 +0700 ---------------------------------------------------------------------- .../transport/matchers/HasAttachmentTest.java | 108 +++++++++++++++++++ 1 file changed, 108 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/james-project/blob/16829c1e/mailet/standard/src/test/java/org/apache/james/transport/matchers/HasAttachmentTest.java ---------------------------------------------------------------------- diff --git a/mailet/standard/src/test/java/org/apache/james/transport/matchers/HasAttachmentTest.java b/mailet/standard/src/test/java/org/apache/james/transport/matchers/HasAttachmentTest.java new file mode 100644 index 0000000..0d47689 --- /dev/null +++ b/mailet/standard/src/test/java/org/apache/james/transport/matchers/HasAttachmentTest.java @@ -0,0 +1,108 @@ +/**************************************************************** + * Licensed to the Apache Software Foundation (ASF) under one * + * or more contributor license agreements. See the NOTICE file * + * distributed with this work for additional information * + * regarding copyright ownership. The ASF licenses this file * + * to you under the Apache License, Version 2.0 (the * + * "License"); you may not use this file except in compliance * + * with the License. You may obtain a copy of the License at * + * * + * http://www.apache.org/licenses/LICENSE-2.0 * + * * + * Unless required by applicable law or agreed to in writing, * + * software distributed under the License is distributed on an * + * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY * + * KIND, either express or implied. See the License for the * + * specific language governing permissions and limitations * + * under the License. * + ****************************************************************/ + +package org.apache.james.transport.matchers; + +import static org.assertj.core.api.Assertions.assertThat; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.when; + +import java.util.Properties; + +import javax.mail.Session; +import javax.mail.internet.MimeBodyPart; +import javax.mail.internet.MimeMessage; +import javax.mail.internet.MimeMultipart; + +import org.apache.mailet.Mail; +import org.apache.mailet.MailAddress; +import org.apache.mailet.base.test.FakeMail; +import org.junit.Before; +import org.junit.Test; + +import com.google.common.collect.ImmutableList; + +public class HasAttachmentTest { + + private HasAttachment testee; + private MimeMessage mimeMessage; + private Mail mail; + + @Before + public void setUp() throws Exception { + testee = new HasAttachment(); + + mimeMessage = new MimeMessage(Session.getDefaultInstance(new Properties())); + mail = new FakeMail(); + mail.setRecipients(ImmutableList.of(new MailAddress("[email protected]"))); + mail.setMessage(mimeMessage); + } + + @Test + public void textMailsShouldNotBeMatched() throws Exception { + mimeMessage.setText("A simple text message"); + + assertThat(testee.match(mail)).isNull(); + } + + @Test + public void emptyMultipartShouldNotBeMatched() throws Exception { + mimeMessage.setContent(new MimeMultipart()); + + assertThat(testee.match(mail)).isNull(); + } + + @Test + public void inlinedOnlyMultipartShouldNotBeMatched() throws Exception { + MimeMultipart mimeMultipart = new MimeMultipart(); + MimeBodyPart part = new MimeBodyPart(); + part.setDisposition(MimeMessage.INLINE); + part.setFileName("bahamas.png"); + mimeMultipart.addBodyPart(part); + mimeMessage.setContent(mimeMultipart); + + assertThat(testee.match(mail)).isNull(); + } + + @Test + public void multipartWithOneAttachmentShouldBeMatched() throws Exception { + MimeMultipart mimeMultipart = new MimeMultipart(); + MimeBodyPart textPart = new MimeBodyPart(); + textPart.setDisposition(MimeMessage.INLINE); + mimeMultipart.addBodyPart(textPart); + MimeBodyPart attachmentPart = new MimeBodyPart(); + attachmentPart.setDisposition(MimeMessage.ATTACHMENT); + mimeMultipart.addBodyPart(attachmentPart); + mimeMessage.setContent(mimeMultipart); + + assertThat(testee.match(mail)).containsAll(mail.getRecipients()); + } + + @Test + public void attachmentMailsShouldBeMatched() throws Exception { + MimeMessage mimeMessage = mock(MimeMessage.class); + when(mimeMessage.getContent()).thenReturn(new Object()); + when(mimeMessage.getDisposition()).thenReturn(MimeMessage.ATTACHMENT); + when(mimeMessage.getContentType()).thenReturn("application/json"); + mail.setMessage(mimeMessage); + + assertThat(testee.match(mail)).containsAll(mail.getRecipients()); + } + +} --------------------------------------------------------------------- To unsubscribe, e-mail: [email protected] For additional commands, e-mail: [email protected]
