org.obm.push.spushnik.resources.FolderSyncScenarioTest.java Source code

Java tutorial

Introduction

Here is the source code for org.obm.push.spushnik.resources.FolderSyncScenarioTest.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.push.spushnik.resources;

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

import java.io.InputStream;

import javax.servlet.http.HttpServletResponse;

import org.apache.commons.io.IOUtils;
import org.apache.http.HttpResponse;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.entity.ByteArrayEntity;
import org.apache.http.entity.ContentType;
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.opush.Users;
import org.obm.guice.GuiceRunner;
import org.obm.opush.env.CassandraServer;
import org.obm.push.OpushServer;
import org.obm.push.spushnik.SpushnikModule;
import org.obm.push.spushnik.SpushnikScenarioTestUtils;
import org.obm.push.spushnik.SpushnikServer;
import org.obm.push.spushnik.SpushnikTestUtils;

import com.google.common.base.Charsets;
import com.google.common.io.ByteStreams;
import com.google.inject.Inject;

@RunWith(GuiceRunner.class)
@GuiceModule(ScenarioTestModule.class)
public class FolderSyncScenarioTest {

    @Inject
    private Users users;
    @Inject
    private OpushServer opushServer;
    @Inject
    private IMocksControl mocksControl;
    @Inject
    private Configuration configuration;
    @Inject
    private PolicyConfigurationProvider policyConfigurationProvider;
    @Inject
    private CassandraServer cassandraServer;
    @Inject
    private SpushnikScenarioTestUtils spushnikScenarioTestUtils;

    private CloseableHttpClient httpClient;
    private SpushnikServer spushnikServer;
    private String baseURL;

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

    @Before
    public void setUp() throws Exception {
        spushnikServer = new SpushnikServer(SpushnikServer.JETTY_SELECTED_PORT, SpushnikModule.class, 4, 1);
        spushnikServer.start();
        baseURL = "http://localhost:" + spushnikServer.getPort() + "/spushnik/";
        expect(policyConfigurationProvider.get()).andReturn("fakeConfiguration").anyTimes();
        spushnikScenarioTestUtils.mockWorkingFolderSync(users.jaures);
        mocksControl.replay();
        cassandraServer.start();
        opushServer.start();
        httpClient = HttpClientBuilder.create().build();
    }

    @Test
    public void testFolderSyncScenarioWithNiceCertificate() throws Exception {
        HttpResponse httpResponse = folderSyncScenarioWithRequest(baseURL, "request_jaures_certificate_good.txt");

        InputStream content = httpResponse.getEntity().getContent();
        assertThat(httpResponse.getStatusLine().getStatusCode()).isEqualTo(HttpServletResponse.SC_OK);
        assertThat(IOUtils.toString(content, Charsets.UTF_8)).isEqualTo("{\"status\":0,\"messages\":[]}");
    }

    @Test
    public void testFolderSyncScenarioWithNiceCertificateWrongPwd() throws Exception {
        HttpResponse httpResponse = folderSyncScenarioWithRequest(baseURL,
                "request_jaures_certificate_wrong_pwd.txt");

        InputStream content = httpResponse.getEntity().getContent();
        assertThat(httpResponse.getStatusLine().getStatusCode()).isEqualTo(HttpServletResponse.SC_OK);
        assertThat(IOUtils.toString(content, Charsets.UTF_8)).startsWith(
                "{\"status\":2,\"messages\":[\"org.obm.push.spushnik.service.InvalidCredentialsException: Invalid certificate")
                .endsWith("\"]}");
    }

    @Test
    public void testFolderSyncScenarioWithBadCertificate() throws Exception {
        HttpResponse httpResponse = folderSyncScenarioWithRequest(baseURL, "request_jaures_certificate_bad.txt");

        InputStream content = httpResponse.getEntity().getContent();
        assertThat(httpResponse.getStatusLine().getStatusCode()).isEqualTo(HttpServletResponse.SC_OK);
        String string = IOUtils.toString(content, Charsets.UTF_8);
        assertThat(string).startsWith(
                "{\"status\":2,\"messages\":[\"org.obm.push.spushnik.service.InvalidCredentialsException: Invalid certificate")
                .endsWith("\"]}");
    }

    @Test
    public void testFolderSyncScenarioWithNullCertificate() throws Exception {
        HttpResponse httpResponse = folderSyncScenarioWithRequest(baseURL, "request_jaures_certificate_null.txt");

        InputStream content = httpResponse.getEntity().getContent();
        assertThat(httpResponse.getStatusLine().getStatusCode()).isEqualTo(HttpServletResponse.SC_OK);
        assertThat(IOUtils.toString(content, Charsets.UTF_8)).isEqualTo("{\"status\":0,\"messages\":[]}");
    }

    @Test
    public void testFolderSyncScenarioWithoutCertificate() throws Exception {
        HttpResponse httpResponse = folderSyncScenarioWithRequest(baseURL, "request_jaures_certificate_none.txt");

        InputStream content = httpResponse.getEntity().getContent();
        assertThat(httpResponse.getStatusLine().getStatusCode()).isEqualTo(HttpServletResponse.SC_OK);
        assertThat(IOUtils.toString(content, Charsets.UTF_8)).isEqualTo("{\"status\":0,\"messages\":[]}");
    }

    private HttpResponse folderSyncScenarioWithRequest(String baseURL, String certificate) throws Exception {
        InputStream requestInputStream = ClassLoader.getSystemClassLoader().getResourceAsStream(certificate);
        byte[] requestContent = ByteStreams.toByteArray(requestInputStream);

        HttpPost httpPost = new HttpPost(buildRequestUrl(baseURL));
        httpPost.setEntity(new ByteArrayEntity(requestContent, ContentType.APPLICATION_JSON));
        return httpClient.execute(httpPost);
    }

    private String buildRequestUrl(String baseURL) {
        return baseURL + "foldersync?serviceUrl=" + SpushnikTestUtils.buildServiceUrl(opushServer.getHttpPort());
    }
}