org.obm.opush.command.email.SmartReplyHandlerTest.java Source code

Java tutorial

Introduction

Here is the source code for org.obm.opush.command.email.SmartReplyHandlerTest.java

Source

/* ***** BEGIN LICENSE BLOCK *****
 * 
 * Copyright (C) 2011-2014  Linagora
 *
 * This program is free software: you can redistribute it and/or 
 * modify it under the terms of the GNU Affero General Public License as 
 * published by the Free Software Foundation, either version 3 of the 
 * License, or (at your option) any later version, provided you comply 
 * with the Additional Terms applicable for OBM connector by Linagora 
 * pursuant to Section 7 of the GNU Affero General Public License, 
 * subsections (b), (c), and (e), pursuant to which you must notably (i) retain 
 * the Message sent thanks to OBM, Free Communication by Linagora? 
 * signature notice appended to any and all outbound messages 
 * (notably e-mail and meeting requests), (ii) retain all hypertext links between 
 * OBM and obm.org, as well as between Linagora and linagora.com, and (iii) refrain 
 * from infringing Linagora intellectual property rights over its trademarks 
 * and commercial brands. Other Additional Terms apply, 
 * see <http://www.linagora.com/licenses/> for more details. 
 *
 * This program is distributed in the hope that it will be useful, 
 * but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY 
 * or FITNESS FOR A PARTICULAR PURPOSE. See the GNU Affero General Public License 
 * for more details. 
 *
 * You should have received a copy of the GNU Affero General Public License 
 * and its applicable Additional Terms for OBM along with this program. If not, 
 * see <http://www.gnu.org/licenses/> for the GNU Affero General Public License version 3 
 * and <http://www.linagora.com/licenses/> for the Additional Terms applicable to 
 * OBM connectors. 
 * 
 * ***** END LICENSE BLOCK ***** */
package org.obm.opush.command.email;

import static org.assertj.core.api.Assertions.assertThat;
import static org.easymock.EasyMock.expect;
import static org.obm.DateUtils.date;

import java.util.Date;

import javax.mail.internet.InternetAddress;

import org.apache.http.impl.client.CloseableHttpClient;
import org.apache.http.impl.client.HttpClientBuilder;
import org.assertj.core.util.Files;
import org.easymock.IMocksControl;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.obm.Configuration;
import org.obm.ConfigurationModule.PolicyConfigurationProvider;
import org.obm.guice.GuiceModule;
import org.obm.guice.GuiceRunner;
import org.obm.opush.IntegrationTestUtils;
import org.obm.opush.IntegrationUserAccessUtils;
import org.obm.opush.MailBackendTestModule;
import org.obm.opush.Users;
import org.obm.opush.Users.OpushUser;
import org.obm.opush.env.CassandraServer;
import org.obm.opush.env.OpushStaticConfiguration;
import org.obm.push.OpushServer;
import org.obm.push.bean.FolderType;
import org.obm.push.bean.ServerId;
import org.obm.push.bean.change.hierarchy.BackendFolder.BackendId;
import org.obm.push.bean.change.hierarchy.Folder;
import org.obm.push.bean.change.hierarchy.FolderSnapshot;
import org.obm.push.bean.change.hierarchy.MailboxPath;
import org.obm.push.configuration.OpushEmailConfiguration;
import org.obm.push.protocol.bean.CollectionId;
import org.obm.push.service.FolderSnapshotDao;
import org.obm.push.state.FolderSyncKey;
import org.obm.sync.client.user.UserClient;
import org.obm.sync.date.DateProvider;
import org.obm.sync.push.client.OPClient;

import com.google.common.base.Optional;
import com.google.common.collect.ImmutableSet;
import com.google.inject.Inject;
import com.icegreen.greenmail.imap.ImapHostManager;
import com.icegreen.greenmail.store.MailFolder;
import com.icegreen.greenmail.store.SimpleStoredMessage;
import com.icegreen.greenmail.user.GreenMailUser;
import com.icegreen.greenmail.util.GreenMail;

@RunWith(GuiceRunner.class)
@GuiceModule(MailBackendTestModule.class)
public class SmartReplyHandlerTest {

    @Inject
    private Users users;
    @Inject
    private OpushServer opushServer;
    @Inject
    private IMocksControl mocksControl;
    @Inject
    private Configuration configuration;
    @Inject
    private GreenMail greenMail;
    @Inject
    private PolicyConfigurationProvider policyConfigurationProvider;
    @Inject
    private DateProvider dateProvider;
    @Inject
    private CassandraServer cassandraServer;
    @Inject
    private IntegrationTestUtils testUtils;
    @Inject
    private IntegrationUserAccessUtils userAccessUtils;
    @Inject
    private FolderSnapshotDao folderSnapshotDao;
    @Inject
    private UserClient userClient;
    @Inject
    private OpushStaticConfiguration.Email emailConfiguration;

    private OpushUser user;
    private GreenMailUser greenMailUser;
    private MailboxPath inboxPath;
    private CollectionId inboxCollectionId;
    private Folder folder;
    private MailFolder inboxFolder;
    private MailFolder sentFolder;
    private ServerId serverId;
    private CloseableHttpClient httpClient;
    private ImapHostManager imapHostManager;

    @Before
    public void setUp() throws Exception {
        httpClient = HttpClientBuilder.create().build();
        cassandraServer.start();
        user = users.jaures;
        greenMail.start();
        greenMailUser = greenMail.setUser(user.user.getLoginAtDomain(), String.valueOf(user.password));
        imapHostManager = greenMail.getManagers().getImapHostManager();
        sentFolder = imapHostManager.createMailbox(greenMailUser, OpushEmailConfiguration.IMAP_SENT_NAME);
        inboxFolder = imapHostManager.getInbox(greenMailUser);

        inboxPath = MailboxPath.of(OpushEmailConfiguration.IMAP_INBOX_NAME);
        inboxCollectionId = CollectionId.of(1234);
        folder = Folder.builder().backendId(inboxPath).collectionId(inboxCollectionId)
                .parentBackendIdOpt(Optional.<BackendId>absent()).displayName("INBOX")
                .folderType(FolderType.DEFAULT_INBOX_FOLDER).build();
        serverId = inboxCollectionId.serverId(1);

        FolderSyncKey syncKey = new FolderSyncKey("4fd6280c-cbaa-46aa-a859-c6aad00f1ef3");
        folderSnapshotDao.create(user.user, user.device, syncKey,
                FolderSnapshot.nextId(2).folders(ImmutableSet.of(folder)));

        userAccessUtils.mockUsersAccess(user);
        expect(userClient.getUserEmail(user.accessToken)).andReturn(user.user.getLoginAtDomain()).anyTimes();
        expect(policyConfigurationProvider.get()).andReturn("fakeConfiguration").anyTimes();
    }

    @After
    public void shutdown() throws Exception {
        opushServer.stop();
        cassandraServer.stop();
        httpClient.close();
        Files.delete(configuration.dataDir);
    }

    @Test
    public void testReplyWithAlternativeToAnAlternative() throws Exception {
        testUtils.appendToINBOX(greenMailUser, "eml/multipartAlternative.eml");
        assertThat(sentFolder.getMessageCount()).isEqualTo(0);

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/multipartAlternative.eml"),
                inboxCollectionId, serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(sentFolder.getMessageCount()).isEqualTo(1);
    }

    @Test
    public void testReplyWithTextToAnAlternative() throws Exception {
        testUtils.appendToINBOX(greenMailUser, "eml/multipartAlternative.eml");
        assertThat(sentFolder.getMessageCount()).isEqualTo(0);

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/textPlain.eml"), inboxCollectionId,
                serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(sentFolder.getMessageCount()).isEqualTo(1);
    }

    @Test
    public void testReplyWithHtmlToAnAlternative() throws Exception {
        testUtils.appendToINBOX(greenMailUser, "eml/multipartAlternative.eml");
        assertThat(sentFolder.getMessageCount()).isEqualTo(0);

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/textHtml.eml"), inboxCollectionId,
                serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(sentFolder.getMessageCount()).isEqualTo(1);
    }

    @Test
    public void testOBMFULL4924() throws Exception {
        testUtils.appendToINBOX(greenMailUser, "eml/OBMFULL-4924-inboxEmail.eml");
        assertThat(sentFolder.getMessageCount()).isEqualTo(0);

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/OBMFULL-4924-replyingEmail.eml"),
                inboxCollectionId, serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(sentFolder.getMessageCount()).isEqualTo(1);
    }

    @Test
    public void testQuotaExceededErrorMail() throws Exception {
        testUtils.appendToINBOX(greenMailUser, "eml/quotaExceeded.eml");
        assertThat(inboxFolder.getMessageCount()).isEqualTo(1);

        Date expectedDate = date("2012-05-04T11:30:12");
        expect(dateProvider.getDate()).andReturn(expectedDate);

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/quotaExceeded.eml"), inboxCollectionId,
                serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(inboxFolder.getMessageCount()).isEqualTo(2);
        SimpleStoredMessage inboxMessage = inboxFolder.getMessages().get(1);
        assertThat(inboxMessage.getMimeMessage().getSentDate()).isEqualTo(expectedDate);
    }

    @Test
    public void testReplySendsReportEmailWhenError() throws Exception {
        testUtils.appendToINBOX(greenMailUser, "eml/multipartAlternative.eml");

        Date expectedDate = date("2012-05-04T11:30:12");
        expect(dateProvider.getDate()).andReturn(expectedDate);

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/badToSyntax.eml"), inboxCollectionId,
                serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(sentFolder.getMessageCount()).isEqualTo(0);
        assertThat(inboxFolder.getMessages().size()).isEqualTo(2);
        SimpleStoredMessage inboxMessage = inboxFolder.getMessages().get(1);
        assertThat(inboxMessage.getMimeMessage().getSentDate()).isEqualTo(expectedDate);
    }

    @Test
    public void smartReplyShouldNotFailWhenNoSentFolder() throws Exception {
        imapHostManager.deleteMailbox(greenMailUser, OpushEmailConfiguration.IMAP_SENT_NAME);
        testUtils.appendToINBOX(greenMailUser, "eml/multipartAlternative.eml");

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/textPlain.eml"), inboxCollectionId,
                serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(inboxFolder.getMessages().size()).isEqualTo(2);
    }

    @Test
    public void smartReplyShouldFindSpecialSentFolder() throws Exception {
        String sentPath = "INBOX/Sent";
        emailConfiguration.configuration.imapMailboxSent = sentPath;
        imapHostManager.deleteMailbox(greenMailUser, OpushEmailConfiguration.IMAP_SENT_NAME);
        MailFolder sentFolder = imapHostManager.createMailbox(greenMailUser, sentPath);
        testUtils.appendToINBOX(greenMailUser, "eml/multipartAlternative.eml");

        mocksControl.replay();
        opushServer.start();
        opClient().emailReply(testUtils.loadEmail("eml/textPlain.eml"), inboxCollectionId, serverId);
        mocksControl.verify();

        assertThat(sentFolder.getMessages().size()).isEqualTo(1);
    }

    @Test
    public void testSentEmailAttribute() throws Exception {
        testUtils.appendToINBOX(greenMailUser, "eml/OBMFULL-4924-inboxEmail.eml");

        mocksControl.replay();
        opushServer.start();
        boolean success = opClient().emailReply(testUtils.loadEmail("eml/OBMFULL-4924-replyingEmail.eml"),
                inboxCollectionId, serverId);
        mocksControl.verify();

        assertThat(success).isTrue();
        assertThat(sentFolder.getMessageCount()).isEqualTo(1);
        InternetAddress fromAddress = (InternetAddress) sentFolder.getMessages().get(0).getMimeMessage()
                .getFrom()[0];
        assertThat(fromAddress.toString()).isEqualTo("Jean Jaures <jaures@sfio.fr>");

    }

    private OPClient opClient() {
        return testUtils.buildWBXMLOpushClient(user, opushServer.getHttpPort(), httpClient);
    }
}