Repository: james-project Updated Branches: refs/heads/master ce408bf81 -> be37b9c0c
JAMES-2063 Add tests on SelectedMailboxImpl for concurrent events Project: http://git-wip-us.apache.org/repos/asf/james-project/repo Commit: http://git-wip-us.apache.org/repos/asf/james-project/commit/b74db024 Tree: http://git-wip-us.apache.org/repos/asf/james-project/tree/b74db024 Diff: http://git-wip-us.apache.org/repos/asf/james-project/diff/b74db024 Branch: refs/heads/master Commit: b74db024409fea9edeb84c571eb02e4863391ca6 Parents: ce408bf Author: benwa <btell...@linagora.com> Authored: Tue Jun 20 10:49:28 2017 +0700 Committer: benwa <btell...@linagora.com> Committed: Tue Jun 20 11:47:24 2017 +0700 ---------------------------------------------------------------------- protocols/imap/pom.xml | 5 + .../processor/base/SelectedMailboxImplTest.java | 172 +++++++++++++++++++ 2 files changed, 177 insertions(+) ---------------------------------------------------------------------- http://git-wip-us.apache.org/repos/asf/james-project/blob/b74db024/protocols/imap/pom.xml ---------------------------------------------------------------------- diff --git a/protocols/imap/pom.xml b/protocols/imap/pom.xml index c2f590d..12b9826 100644 --- a/protocols/imap/pom.xml +++ b/protocols/imap/pom.xml @@ -75,6 +75,11 @@ <groupId>org.slf4j</groupId> <artifactId>slf4j-api</artifactId> </dependency> + <dependency> + <groupId>org.slf4j</groupId> + <artifactId>slf4j-simple</artifactId> + <scope>test</scope> + </dependency> <dependency> <groupId>commons-codec</groupId> http://git-wip-us.apache.org/repos/asf/james-project/blob/b74db024/protocols/imap/src/test/java/org/apache/james/imap/processor/base/SelectedMailboxImplTest.java ---------------------------------------------------------------------- diff --git a/protocols/imap/src/test/java/org/apache/james/imap/processor/base/SelectedMailboxImplTest.java b/protocols/imap/src/test/java/org/apache/james/imap/processor/base/SelectedMailboxImplTest.java new file mode 100644 index 0000000..a9e7144 --- /dev/null +++ b/protocols/imap/src/test/java/org/apache/james/imap/processor/base/SelectedMailboxImplTest.java @@ -0,0 +1,172 @@ +/**************************************************************** + * 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.imap.processor.base; + +import static org.assertj.core.api.Assertions.assertThat; +import static org.mockito.Matchers.any; +import static org.mockito.Matchers.eq; +import static org.mockito.Mockito.doAnswer; +import static org.mockito.Mockito.mock; +import static org.mockito.Mockito.when; + +import java.util.Date; +import java.util.Iterator; +import java.util.TreeMap; +import java.util.concurrent.ExecutorService; +import java.util.concurrent.Executors; +import java.util.concurrent.atomic.AtomicInteger; + +import javax.mail.Flags; + +import org.apache.james.imap.api.ImapSessionUtils; +import org.apache.james.imap.api.process.ImapSession; +import org.apache.james.mailbox.MailboxListener; +import org.apache.james.mailbox.MailboxManager; +import org.apache.james.mailbox.MailboxSession; +import org.apache.james.mailbox.MessageManager; +import org.apache.james.mailbox.MessageUid; +import org.apache.james.mailbox.model.MailboxConstants; +import org.apache.james.mailbox.model.MailboxPath; +import org.apache.james.mailbox.model.MessageMetaData; +import org.apache.james.mailbox.model.SearchQuery; +import org.apache.james.mailbox.store.SimpleMessageMetaData; +import org.apache.james.mailbox.store.event.EventFactory; +import org.apache.james.mailbox.store.mail.model.DefaultMessageId; +import org.apache.james.mailbox.store.mail.model.Mailbox; +import org.junit.After; +import org.junit.Before; +import org.junit.Ignore; +import org.junit.Test; +import org.mockito.invocation.InvocationOnMock; +import org.mockito.stubbing.Answer; +import org.slf4j.Logger; +import org.slf4j.LoggerFactory; + +import com.google.common.collect.ImmutableList; + + +public class SelectedMailboxImplTest { + + private static final Logger LOGGER = LoggerFactory.getLogger(SelectedMailboxImplTest.class); + public static final MessageUid MESSAGE_UID_5 = MessageUid.of(5); + + private ExecutorService executorService; + private MailboxManager mailboxManager; + private MessageManager messageManager; + private MailboxPath mailboxPath; + private ImapSession imapSession; + private Mailbox mailbox; + + @Before + public void setUp() throws Exception { + executorService = Executors.newFixedThreadPool(1); + + mailboxPath = new MailboxPath(MailboxConstants.USER_NAMESPACE, "tell...@linagora.com", MailboxConstants.INBOX); + mailboxManager = mock(MailboxManager.class); + messageManager = mock(MessageManager.class); + when(mailboxManager.getMailbox(eq(mailboxPath), any(MailboxSession.class))).thenReturn(messageManager); + when(messageManager.getApplicableFlags(any(MailboxSession.class))).thenReturn(new Flags()); + when(messageManager.search(any(SearchQuery.class), any(MailboxSession.class))) + .then(sleepThenSearchAnswer()); + + imapSession = mock(ImapSession.class); + when(imapSession.getAttribute(ImapSessionUtils.MAILBOX_SESSION_ATTRIBUTE_SESSION_KEY)).thenReturn(mock(MailboxSession.class)); + + mailbox = mock(Mailbox.class); + when(mailbox.generateAssociatedPath()).thenReturn(mailboxPath); + } + + @After + public void tearDown() { + executorService.shutdownNow(); + } + + @Ignore + @Test + public void concurrentEventShouldNotSkipUidEmmitedDuringInitialization() throws Exception { + final AtomicInteger success = new AtomicInteger(0); + doAnswer(generateEmitEventAnswer(success)) + .when(mailboxManager) + .addListener(eq(mailboxPath), any(MailboxListener.class), any(MailboxSession.class)); + + SelectedMailboxImpl selectedMailbox = new SelectedMailboxImpl( + mailboxManager, + imapSession, + mailboxPath); + + assertThat(selectedMailbox.getLastUid().get()).isEqualTo(MESSAGE_UID_5); + } + + @Ignore + @Test + public void concurrentEventShouldBeSupportedDuringInitialisation() throws Exception { + final AtomicInteger success = new AtomicInteger(0); + doAnswer(generateEmitEventAnswer(success)) + .when(mailboxManager) + .addListener(eq(mailboxPath), any(MailboxListener.class), any(MailboxSession.class)); + + new SelectedMailboxImpl( + mailboxManager, + imapSession, + mailboxPath); + + assertThat(success.get()) + .as("Get the incremented value in case of successful event processing.") + .isEqualTo(1); + } + + private Answer<Iterator<MessageUid>> sleepThenSearchAnswer() { + return new Answer<Iterator<MessageUid>>() { + @Override + public Iterator<MessageUid> answer(InvocationOnMock invocation) throws Throwable { + Thread.sleep(1000); + return ImmutableList.of(MessageUid.of(1), MessageUid.of(3)).iterator(); + } + }; + } + + private Answer generateEmitEventAnswer(final AtomicInteger success) { + return new Answer() { + @Override + public Object answer(InvocationOnMock invocation) throws Throwable { + Object[] args = invocation.getArguments(); + final MailboxListener mailboxListener = (MailboxListener) args[1]; + executorService.submit(new Runnable() { + @Override + public void run() { + try { + emitEvent(mailboxListener); + success.incrementAndGet(); + } catch (Exception e) { + LOGGER.error("Error while processing event on a concurrent thread", e); + } + } + }); + return null; + } + }; + } + + private void emitEvent(MailboxListener mailboxListener) { + TreeMap<MessageUid, MessageMetaData> result = new TreeMap<MessageUid, MessageMetaData>(); + result.put(MESSAGE_UID_5, new SimpleMessageMetaData(MESSAGE_UID_5, 12, new Flags(), 38, new Date(), new DefaultMessageId())); + mailboxListener.event(new EventFactory().added(mock(MailboxSession.class), result, mailbox)); + } +} --------------------------------------------------------------------- To unsubscribe, e-mail: server-dev-unsubscr...@james.apache.org For additional commands, e-mail: server-dev-h...@james.apache.org
