org.plos.crepo.dao.ContentRepoBaseDaoTest.java Source code

Introduction

Here is the source code for org.plos.crepo.dao.ContentRepoBaseDaoTest.java

Source

/*
 * Copyright 2017 Public Library of Science
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
 * DEALINGS IN THE SOFTWARE.
 */

package org.plos.crepo.dao;

import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.client.methods.HttpRequestBase;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Mock;
import org.plos.crepo.config.ContentRepoAccessConfig;
import org.plos.crepo.exceptions.ContentRepoException;
import org.plos.crepo.exceptions.ErrorType;
import org.plos.crepo.util.HttpResponseUtil;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;

import java.io.IOException;
import java.net.URI;

import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertNull;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.isA;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;

@RunWith(PowerMockRunner.class)
@PrepareForTest(HttpResponseUtil.class)
public class ContentRepoBaseDaoTest extends BaseDaoTest {

    private TestContentRepoBaseDaoImpl contentRepoBaseDao;
    private URI uri = URI.create("http://testUri");

    @Mock
    private ContentRepoAccessConfig repoAccessConfig;

    @Before
    public void setUp() {
        contentRepoBaseDao = new TestContentRepoBaseDaoImpl(repoAccessConfig);
        when(repoAccessConfig.getRepoServer()).thenReturn(REPO_SERVER);
    }

    @Test
    public void executeRequestTest() throws IOException {
        HttpRequestBase httpRequest = mock(HttpRequestBase.class);
        when(httpRequest.getURI()).thenReturn(uri);

        mockCommonCalls(repoAccessConfig, HttpStatus.SC_OK);

        HttpResponse response = contentRepoBaseDao.executeRequest(httpRequest, ErrorType.ErrorFetchingBucketMeta);

        assertNotNull(response);
        assertEquals(mockResponse, response);
        assertEquals(uri, httpRequest.getURI());

        verify(repoAccessConfig).open(httpRequest);
        verify(mockResponse).getStatusLine();
        verify(statusLine).getStatusCode();
        verify(repoAccessConfig).open(httpRequest);

    }

    @Test
    public void executeRequestThrowsExcTest() throws IOException {
        HttpRequestBase httpRequest = mock(HttpRequestBase.class);
        when(httpRequest.getURI()).thenReturn(uri);

        mockCommonCalls(repoAccessConfig, HttpStatus.SC_BAD_REQUEST);
        mockHttpResponseUtilCalls(mockResponse);

        HttpResponse response = null;

        try {
            response = contentRepoBaseDao.executeRequest(httpRequest, ErrorType.ErrorFetchingBucketMeta);
        } catch (ContentRepoException ex) {
            verifyException(ex, response, ErrorType.ErrorFetchingBucketMeta);
        }

        verify(repoAccessConfig).open(httpRequest);
        verify(mockResponse, times(1)).getStatusLine();
        verify(statusLine, times(1)).getStatusCode();
        verify(httpRequest, times(2)).getURI();

        assertNull(response);
        assertEquals(uri, httpRequest.getURI());

    }

    @Test
    public void executeRequestIOThrowsExcTest() throws IOException {
        HttpRequestBase httpRequest = mock(HttpRequestBase.class);
        when(httpRequest.getURI()).thenReturn(uri);
        IOException exception = mock(IOException.class);
        when(repoAccessConfig.open(isA(HttpRequestBase.class))).thenThrow(exception);

        HttpResponse response = null;

        try {
            response = contentRepoBaseDao.executeRequest(httpRequest, ErrorType.ErrorFetchingBucketMeta);
        } catch (ContentRepoException ex) {
            assertNull(response);
            assertEquals(ErrorType.ErrorFetchingBucketMeta, ex.getErrorType());
            assertTrue(ex.getMessage().contains(uri.toString()));
            assertEquals(exception, ex.getCause());
        }

        verify(repoAccessConfig).open(httpRequest);
        verify(httpRequest, times(2)).getURI();

        assertEquals(uri, httpRequest.getURI());

    }

}