1 |
|
|
2 |
|
|
3 |
|
|
4 |
|
|
5 |
|
|
6 |
|
|
7 |
|
|
8 |
|
|
9 |
|
|
10 |
|
|
11 |
|
|
12 |
|
|
13 |
|
|
14 |
|
|
15 |
|
|
16 |
|
|
17 |
|
|
18 |
|
|
19 |
|
|
20 |
|
package org.xwiki.mail.internal.thread; |
21 |
|
|
22 |
|
import java.util.Arrays; |
23 |
|
import java.util.Iterator; |
24 |
|
import java.util.Properties; |
25 |
|
import java.util.UUID; |
26 |
|
|
27 |
|
import javax.inject.Provider; |
28 |
|
import javax.mail.Session; |
29 |
|
import javax.mail.internet.MimeMessage; |
30 |
|
|
31 |
|
import org.junit.Before; |
32 |
|
import org.junit.Rule; |
33 |
|
import org.junit.Test; |
34 |
|
import org.mockito.Mockito; |
35 |
|
import org.mockito.invocation.InvocationOnMock; |
36 |
|
import org.mockito.stubbing.Answer; |
37 |
|
import org.xwiki.component.util.DefaultParameterizedType; |
38 |
|
import org.xwiki.context.ExecutionContext; |
39 |
|
import org.xwiki.mail.ExtendedMimeMessage; |
40 |
|
import org.xwiki.mail.MailContentStore; |
41 |
|
import org.xwiki.mail.MailListener; |
42 |
|
import org.xwiki.mail.MailState; |
43 |
|
import org.xwiki.mail.MailStatus; |
44 |
|
import org.xwiki.mail.MailStatusResult; |
45 |
|
import org.xwiki.mail.MailStoreException; |
46 |
|
import org.xwiki.mail.internal.MemoryMailListener; |
47 |
|
import org.xwiki.mail.internal.UpdateableMailStatusResult; |
48 |
|
import org.xwiki.test.AllLogRule; |
49 |
|
import org.xwiki.test.annotation.ComponentList; |
50 |
|
import org.xwiki.test.mockito.MockitoComponentMockingRule; |
51 |
|
|
52 |
|
import com.xpn.xwiki.XWikiContext; |
53 |
|
|
54 |
|
import static org.junit.Assert.assertEquals; |
55 |
|
import static org.junit.Assert.assertFalse; |
56 |
|
import static org.junit.Assert.assertNotNull; |
57 |
|
import static org.junit.Assert.assertTrue; |
58 |
|
import static org.mockito.ArgumentMatchers.any; |
59 |
|
import static org.mockito.Mockito.doAnswer; |
60 |
|
import static org.mockito.Mockito.doThrow; |
61 |
|
import static org.mockito.Mockito.when; |
62 |
|
|
63 |
|
|
64 |
|
@link |
65 |
|
|
66 |
|
@version |
67 |
|
@since |
68 |
|
|
69 |
|
@ComponentList({MemoryMailListener.class, PrepareMailQueueManager.class}) |
|
|
| 99.1% |
Uncovered Elements: 1 (113) |
Complexity: 10 |
Complexity Density: 0.1 |
|
70 |
|
public class PrepareMailRunnableTest |
71 |
|
{ |
72 |
|
@Rule |
73 |
|
public AllLogRule logRule = new AllLogRule(); |
74 |
|
|
75 |
|
@Rule |
76 |
|
public MockitoComponentMockingRule<PrepareMailRunnable> mocker = new MockitoComponentMockingRule<>( |
77 |
|
PrepareMailRunnable.class); |
78 |
|
|
|
|
| 100% |
Uncovered Elements: 0 (2) |
Complexity: 1 |
Complexity Density: 0.5 |
|
79 |
2 |
@Before... |
80 |
|
public void setUp() throws Exception |
81 |
|
{ |
82 |
2 |
Provider<XWikiContext> xwikiContextProvider = this.mocker.registerMockComponent(XWikiContext.TYPE_PROVIDER); |
83 |
2 |
when(xwikiContextProvider.get()).thenReturn(Mockito.mock(XWikiContext.class)); |
84 |
|
} |
85 |
|
|
|
|
| 100% |
Uncovered Elements: 0 (42) |
Complexity: 1 |
Complexity Density: 0.02 |
1PASS
|
|
86 |
1 |
@Test... |
87 |
|
public void prepareMailWhenContentStoreFails() throws Exception |
88 |
|
{ |
89 |
1 |
Properties properties = new Properties(); |
90 |
1 |
Session session = Session.getDefaultInstance(properties); |
91 |
|
|
92 |
1 |
MimeMessage message1 = new MimeMessage(session); |
93 |
1 |
message1.setText("Content1"); |
94 |
|
|
95 |
1 |
MimeMessage message2 = new MimeMessage(session); |
96 |
1 |
message2.setText("Content2"); |
97 |
|
|
98 |
1 |
String batchId1 = UUID.randomUUID().toString(); |
99 |
1 |
String batchId2 = UUID.randomUUID().toString(); |
100 |
|
|
101 |
1 |
ExecutionContext context1 = new ExecutionContext(); |
102 |
1 |
XWikiContext xContext1 = new XWikiContext(); |
103 |
1 |
xContext1.setWikiId("wiki1"); |
104 |
1 |
context1.setProperty(XWikiContext.EXECUTIONCONTEXT_KEY, xContext1); |
105 |
|
|
106 |
1 |
ExecutionContext context2 = new ExecutionContext(); |
107 |
1 |
XWikiContext xContext2 = new XWikiContext(); |
108 |
1 |
xContext2.setWikiId("wiki2"); |
109 |
1 |
context2.setProperty(XWikiContext.EXECUTIONCONTEXT_KEY, xContext2); |
110 |
|
|
111 |
1 |
MemoryMailListener listener1 = this.mocker.getInstance(MailListener.class, "memory"); |
112 |
1 |
PrepareMailQueueItem item1 = |
113 |
|
new PrepareMailQueueItem(Arrays.asList(message1), session, listener1, batchId1, context1); |
114 |
1 |
MemoryMailListener listener2 = this.mocker.getInstance(MailListener.class, "memory"); |
115 |
1 |
PrepareMailQueueItem item2 = |
116 |
|
new PrepareMailQueueItem(Arrays.asList(message2), session, listener2, batchId2, context2); |
117 |
|
|
118 |
1 |
MailQueueManager mailQueueManager = |
119 |
|
this.mocker.getInstance(new DefaultParameterizedType(null, MailQueueManager.class, |
120 |
|
PrepareMailQueueItem.class)); |
121 |
|
|
122 |
|
|
123 |
1 |
MailContentStore contentStore = this.mocker.getInstance(MailContentStore.class, "filesystem"); |
124 |
1 |
doThrow(new MailStoreException("error")).when(contentStore).save(any(String.class), any(ExtendedMimeMessage.class)); |
125 |
|
|
126 |
|
|
127 |
|
|
128 |
1 |
mailQueueManager.addToQueue(item1); |
129 |
1 |
mailQueueManager.addToQueue(item2); |
130 |
|
|
131 |
1 |
MailRunnable runnable = this.mocker.getComponentUnderTest(); |
132 |
1 |
Thread thread = new Thread(runnable); |
133 |
1 |
thread.start(); |
134 |
|
|
135 |
|
|
136 |
1 |
try { |
137 |
1 |
listener1.getMailStatusResult().waitTillProcessed(10000L); |
138 |
1 |
listener2.getMailStatusResult().waitTillProcessed(10000L); |
139 |
|
} finally { |
140 |
1 |
runnable.stopProcessing(); |
141 |
1 |
thread.interrupt(); |
142 |
1 |
thread.join(); |
143 |
|
} |
144 |
|
|
145 |
1 |
MailStatusResult result1 = listener1.getMailStatusResult(); |
146 |
1 |
MailStatusResult result2 = listener2.getMailStatusResult(); |
147 |
|
|
148 |
|
|
149 |
1 |
assertTrue(result1.isProcessed()); |
150 |
1 |
assertTrue(result2.isProcessed()); |
151 |
|
|
152 |
|
|
153 |
1 |
MailStatus status1 = result1.getByState(MailState.PREPARE_ERROR).next(); |
154 |
1 |
assertEquals("MailStoreException: error", status1.getErrorSummary()); |
155 |
1 |
MailStatus status2 = result2.getByState(MailState.PREPARE_ERROR).next(); |
156 |
1 |
assertEquals("MailStoreException: error", status2.getErrorSummary()); |
157 |
|
} |
158 |
|
|
|
|
| 100% |
Uncovered Elements: 0 (45) |
Complexity: 1 |
Complexity Density: 0.02 |
1PASS
|
|
159 |
1 |
@Test... |
160 |
|
public void prepareMailWhenIteratorFails() throws Exception |
161 |
|
{ |
162 |
1 |
Properties properties = new Properties(); |
163 |
1 |
Session session = Session.getDefaultInstance(properties); |
164 |
|
|
165 |
1 |
final MimeMessage message1 = new MimeMessage(session); |
166 |
1 |
message1.setText("Content1"); |
167 |
|
|
168 |
1 |
MimeMessage message2 = new MimeMessage(session); |
169 |
1 |
message2.setText("Content2"); |
170 |
|
|
171 |
1 |
String batchId1 = UUID.randomUUID().toString(); |
172 |
1 |
String batchId2 = UUID.randomUUID().toString(); |
173 |
|
|
174 |
1 |
ExecutionContext context1 = new ExecutionContext(); |
175 |
1 |
XWikiContext xContext1 = new XWikiContext(); |
176 |
1 |
xContext1.setWikiId("wiki1"); |
177 |
1 |
context1.setProperty(XWikiContext.EXECUTIONCONTEXT_KEY, xContext1); |
178 |
|
|
179 |
1 |
ExecutionContext context2 = new ExecutionContext(); |
180 |
1 |
XWikiContext xContext2 = new XWikiContext(); |
181 |
1 |
xContext2.setWikiId("wiki2"); |
182 |
1 |
context2.setProperty(XWikiContext.EXECUTIONCONTEXT_KEY, xContext2); |
183 |
|
|
184 |
1 |
MemoryMailListener listener1 = this.mocker.getInstance(MailListener.class, "memory"); |
185 |
1 |
PrepareMailQueueItem item1 = |
186 |
|
new PrepareMailQueueItem(new Iterable<MimeMessage>() |
187 |
|
{ |
|
|
| 100% |
Uncovered Elements: 0 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
188 |
1 |
@Override... |
189 |
|
public Iterator<MimeMessage> iterator() |
190 |
|
{ |
191 |
1 |
return new Iterator<MimeMessage>() |
192 |
|
{ |
193 |
|
int index = 0; |
194 |
|
|
|
|
| 100% |
Uncovered Elements: 0 (1) |
Complexity: 1 |
Complexity Density: 1 |
|
195 |
2 |
@Override... |
196 |
|
public boolean hasNext() |
197 |
|
{ |
198 |
2 |
return true; |
199 |
|
} |
200 |
|
|
|
|
| 100% |
Uncovered Elements: 0 (5) |
Complexity: 2 |
Complexity Density: 0.67 |
|
201 |
2 |
@Override... |
202 |
|
public MimeMessage next() |
203 |
|
{ |
204 |
2 |
if (index++ == 0) { |
205 |
1 |
return message1; |
206 |
|
} |
207 |
1 |
throw new RuntimeException("Iterator failure"); |
208 |
|
} |
209 |
|
|
|
|
| - |
Uncovered Elements: 0 (0) |
Complexity: 1 |
Complexity Density: - |
|
210 |
0 |
@Override... |
211 |
|
public void remove() |
212 |
|
{ |
213 |
|
|
214 |
|
} |
215 |
|
}; |
216 |
|
} |
217 |
|
}, session, listener1, batchId1, context1); |
218 |
1 |
MemoryMailListener listener2 = this.mocker.getInstance(MailListener.class, "memory"); |
219 |
1 |
PrepareMailQueueItem item2 = |
220 |
|
new PrepareMailQueueItem(Arrays.asList(message2), session, listener2, batchId2, context2); |
221 |
|
|
222 |
1 |
MailQueueManager prepareMailQueueManager = |
223 |
|
this.mocker.getInstance(new DefaultParameterizedType(null, MailQueueManager.class, |
224 |
|
PrepareMailQueueItem.class)); |
225 |
|
|
226 |
1 |
MailQueueManager sendMailQueueManager = |
227 |
|
this.mocker.getInstance(new DefaultParameterizedType(null, MailQueueManager.class, |
228 |
|
SendMailQueueItem.class)); |
229 |
|
|
230 |
1 |
MailContentStore contentStore = this.mocker.getInstance(MailContentStore.class, "filesystem"); |
231 |
1 |
doAnswer(new Answer<Object>() |
232 |
|
{ |
|
|
| 100% |
Uncovered Elements: 0 (4) |
Complexity: 1 |
Complexity Density: 0.25 |
|
233 |
2 |
@Override... |
234 |
|
public Object answer(InvocationOnMock invocationOnMock) throws Throwable |
235 |
|
{ |
236 |
2 |
Object[] args = invocationOnMock.getArguments(); |
237 |
2 |
MimeMessage message = (MimeMessage) args[1]; |
238 |
2 |
message.saveChanges(); |
239 |
2 |
return null; |
240 |
|
} |
241 |
|
}).when(contentStore).save(any(String.class), any(ExtendedMimeMessage.class)); |
242 |
|
|
243 |
1 |
doAnswer(new Answer<Object>() |
244 |
|
{ |
|
|
| 100% |
Uncovered Elements: 0 (4) |
Complexity: 1 |
Complexity Density: 0.25 |
|
245 |
2 |
@Override... |
246 |
|
public Object answer(InvocationOnMock invocationOnMock) throws Throwable |
247 |
|
{ |
248 |
2 |
Object[] args = invocationOnMock.getArguments(); |
249 |
2 |
SendMailQueueItem item = (SendMailQueueItem) args[0]; |
250 |
2 |
((UpdateableMailStatusResult)item.getListener().getMailStatusResult()).incrementCurrentSize(); |
251 |
2 |
return null; |
252 |
|
} |
253 |
|
}).when(sendMailQueueManager).addToQueue(any(SendMailQueueItem.class)); |
254 |
|
|
255 |
|
|
256 |
|
|
257 |
|
|
258 |
1 |
prepareMailQueueManager.addToQueue(item1); |
259 |
1 |
prepareMailQueueManager.addToQueue(item2); |
260 |
|
|
261 |
1 |
MailRunnable runnable = this.mocker.getComponentUnderTest(); |
262 |
1 |
Thread thread = new Thread(runnable); |
263 |
1 |
thread.start(); |
264 |
|
|
265 |
|
|
266 |
1 |
try { |
267 |
1 |
listener1.getMailStatusResult().waitTillProcessed(10000L); |
268 |
1 |
listener2.getMailStatusResult().waitTillProcessed(10000L); |
269 |
|
} finally { |
270 |
1 |
runnable.stopProcessing(); |
271 |
1 |
thread.interrupt(); |
272 |
1 |
thread.join(); |
273 |
|
} |
274 |
|
|
275 |
1 |
MailStatusResult result1 = listener1.getMailStatusResult(); |
276 |
1 |
MailStatusResult result2 = listener2.getMailStatusResult(); |
277 |
|
|
278 |
|
|
279 |
1 |
assertTrue(result1.isProcessed()); |
280 |
1 |
assertTrue(result2.isProcessed()); |
281 |
|
|
282 |
|
|
283 |
1 |
assertNotNull(listener1.getMailStatusResult().getByState(MailState.PREPARE_SUCCESS).next()); |
284 |
1 |
assertNotNull(listener2.getMailStatusResult().getByState(MailState.PREPARE_SUCCESS).next()); |
285 |
1 |
assertFalse(listener1.getMailStatusResult().getByState(MailState.PREPARE_ERROR).hasNext()); |
286 |
1 |
assertFalse(listener2.getMailStatusResult().getByState(MailState.PREPARE_ERROR).hasNext()); |
287 |
|
|
288 |
1 |
assertEquals("Failure during preparation phase of thread [" + batchId1 + "]", logRule.getMessage(0)); |
289 |
|
} |
290 |
|
} |