1 /*****************************************************************
2 * Licensed to the Apache Software Foundation (ASF) under one *
3 * or more contributor license agreements. See the NOTICE file *
4 * distributed with this work for additional information *
5 * regarding copyright ownership. The ASF licenses this file *
6 * to you under the Apache License, Version 2.0 (the *
7 * "License"); you may not use this file except in compliance *
8 * with the License. You may obtain a copy of the License at *
9 * *
10 * http://www.apache.org/licenses/LICENSE-2.0 *
11 * *
12 * Unless required by applicable law or agreed to in writing, *
13 * software distributed under the License is distributed on an *
14 * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY *
15 * KIND, either express or implied. See the License for the *
16 * specific language governing permissions and limitations *
17 * under the License. *
18 ****************************************************************/
19
20 package org.apache.james.transport.matchers;
21
22 import org.apache.james.test.mock.javaxmail.MockMimeMessage;
23 import org.apache.james.test.mock.mailet.MockMail;
24 import org.apache.james.test.mock.mailet.MockMailContext;
25 import org.apache.james.test.mock.mailet.MockMatcherConfig;
26
27 import org.apache.mailet.MailAddress;
28 import org.apache.mailet.Matcher;
29
30 import javax.mail.MessagingException;
31 import javax.mail.internet.InternetAddress;
32 import javax.mail.internet.MimeMessage;
33 import javax.mail.internet.ParseException;
34 import javax.mail.internet.MimeMessage.RecipientType;
35
36 import java.io.UnsupportedEncodingException;
37 import java.util.Arrays;
38 import java.util.Collection;
39
40 import junit.framework.TestCase;
41
42 public class FetchedFromTest extends TestCase {
43
44 private MimeMessage mockedMimeMessage;
45
46 private MockMail mockedMail;
47
48 private Matcher matcher;
49
50 private final String HEADER_NAME = "X-fetched-from";
51
52 private final String HEADER_VALUE = "james-user";
53
54 private String headerName = "defaultHeaderName";
55
56 private String headerValue = "defaultHeaderValue";
57
58 public FetchedFromTest(String arg0) throws UnsupportedEncodingException {
59 super(arg0);
60 }
61
62 private void setHeaderName(String headerName) {
63 this.headerName = headerName;
64 }
65
66 private void setHeaderValue(String headerValue) {
67 this.headerValue = headerValue;
68 }
69
70 private void setupMockedMimeMessage() throws MessagingException {
71 String sender = "test@james.apache.org";
72 String rcpt = "test2@james.apache.org";
73
74 mockedMimeMessage = new MockMimeMessage();
75 mockedMimeMessage.setFrom(new InternetAddress(sender));
76 mockedMimeMessage.setRecipients(RecipientType.TO, rcpt);
77 mockedMimeMessage.setHeader(headerName, headerValue);
78 mockedMimeMessage.setSubject("testmail");
79 mockedMimeMessage.setText("testtext");
80 mockedMimeMessage.saveChanges();
81
82 }
83
84 private void setupMockedMail(MimeMessage m) throws ParseException {
85 mockedMail = new MockMail();
86 mockedMail.setMessage(m);
87 mockedMail.setRecipients(Arrays.asList(new MailAddress[] {
88 new MailAddress("test@james.apache.org"),
89 new MailAddress("test2@james.apache.org") }));
90
91 }
92
93 private void setupMatcher() throws MessagingException {
94 setupMockedMimeMessage();
95 matcher = new FetchedFrom();
96 MockMatcherConfig mci = new MockMatcherConfig("FetchedFrom="
97 + HEADER_VALUE, new MockMailContext());
98 matcher.init(mci);
99 }
100
101
102 public void testHeaderIsMatched() throws MessagingException {
103 setHeaderName(HEADER_NAME);
104 setHeaderValue(HEADER_VALUE);
105
106 setupMockedMimeMessage();
107 setupMockedMail(mockedMimeMessage);
108 setupMatcher();
109
110 Collection matchedRecipients = matcher.match(mockedMail);
111 assertNotNull(matchedRecipients);
112 assertEquals(matchedRecipients.size(), mockedMail.getRecipients()
113 .size());
114 }
115
116
117 public void testHeaderIsNotMatched() throws MessagingException {
118 setHeaderName(HEADER_NAME);
119 setHeaderValue(headerValue);
120
121 setupMockedMimeMessage();
122 setupMockedMail(mockedMimeMessage);
123 setupMatcher();
124
125 Collection matchedRecipients = matcher.match(mockedMail);
126 assertNull(matchedRecipients);
127 }
128
129
130 public void testHeaderWasRemovedAfterMatched() throws MessagingException {
131 setHeaderName(HEADER_NAME);
132 setHeaderValue(HEADER_VALUE);
133
134 setupMockedMimeMessage();
135 setupMockedMail(mockedMimeMessage);
136 setupMatcher();
137
138 Collection matchedRecipients = matcher.match(mockedMail);
139 Collection matchedRecipients2 = matcher.match(mockedMail);
140
141 assertNotNull(matchedRecipients);
142 assertNull(matchedRecipients2);
143 }
144 }