[so.git] /

/*-
 * ============LICENSE_START=======================================================
 * ONAP - SO
 * ================================================================================
 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
 * ================================================================================
 * Licensed 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.
 * ============LICENSE_END=========================================================
 */

package org.onap.so.apihandlerinfra.exceptions;


import static org.hamcrest.CoreMatchers.equalTo;
import static org.hamcrest.MatcherAssert.assertThat;
import static org.hamcrest.core.StringStartsWith.startsWith;
import static org.junit.Assert.assertEquals;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.ArgumentMatchers.anyObject;
import static org.mockito.ArgumentMatchers.eq;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.when;
import java.io.IOException;
import java.io.Writer;
import java.util.Arrays;
import java.util.List;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.xml.bind.JAXBException;
import javax.xml.bind.Marshaller;
import org.apache.http.HttpStatus;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.ArgumentMatchers;
import org.mockito.InjectMocks;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.junit.MockitoJUnitRunner;
import org.onap.so.apihandler.common.ErrorNumbers;
import org.onap.so.apihandlerinfra.exceptions.ApiException;
import org.onap.so.apihandlerinfra.exceptions.ApiExceptionMapper;
import org.onap.so.apihandlerinfra.exceptions.BPMNFailureException;
import org.onap.so.apihandlerinfra.exceptions.ClientConnectionException;
import org.onap.so.apihandlerinfra.exceptions.DuplicateRequestException;
import org.onap.so.apihandlerinfra.exceptions.ValidateException;
import org.onap.so.apihandlerinfra.exceptions.VfModuleNotFoundException;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.ObjectMapper;


@RunWith(MockitoJUnitRunner.class)
public class ApiExceptionMapperTest {

    @Mock
    private HttpHeaders headers;
    @Mock
    private Marshaller marshaller;

    @InjectMocks
    ApiExceptionMapper mapper = new ApiExceptionMapper();


    @Before
    public void setUp() {
        when(headers.getAcceptableMediaTypes()).thenReturn(Arrays.asList(MediaType.APPLICATION_JSON_TYPE));
    }

    @Test
    public void testObjectMapperError() throws JsonProcessingException {
        ObjectMapper mockedMapper = Mockito.mock(ObjectMapper.class);
        Mockito.when(mockedMapper.writeValueAsString(anyObject())).thenThrow(JsonProcessingException.class);
        ValidateException validateException = new ValidateException.Builder("Test", 0, null).build();
        ApiExceptionMapper mockedException = Mockito.spy(mapper);
        Mockito.doReturn(mockedMapper).when(mockedException).createObjectMapper();
        Response resp = mockedException.toResponse((ApiException) validateException);

        /// assertEquals(resp.getStatus(), HttpStatus.SC_BAD_REQUEST);
        assertThat(resp.getEntity().toString(),
                startsWith("Exception in buildServiceErrorResponse writing exceptionType to string"));
    }

    @Test
    public void testValidateResponse() {
        ValidateException validateException =
                new ValidateException.Builder("Test Message", HttpStatus.SC_BAD_REQUEST, ErrorNumbers.SVC_BAD_PARAMETER)
                        .build();
        Response resp = mapper.toResponse((ApiException) validateException);

        assertEquals(resp.getStatus(), HttpStatus.SC_BAD_REQUEST);
    }

    @Test
    public void testBPMNFailureResponse() {
        BPMNFailureException bpmnException = new BPMNFailureException.Builder("Test Message", HttpStatus.SC_NOT_FOUND,
                ErrorNumbers.SVC_BAD_PARAMETER).build();
        Response resp = mapper.toResponse((ApiException) bpmnException);

        assertEquals(resp.getStatus(), HttpStatus.SC_NOT_FOUND);
    }

    @Test
    public void testClientConnectionResponse() {
        ClientConnectionException clientConnectionException = new ClientConnectionException.Builder("test",
                HttpStatus.SC_INTERNAL_SERVER_ERROR, ErrorNumbers.SVC_BAD_PARAMETER).build();
        Response resp = mapper.toResponse((ApiException) clientConnectionException);

        assertEquals(resp.getStatus(), HttpStatus.SC_INTERNAL_SERVER_ERROR);
    }

    @Test
    public void testVFModuleResponse() {
        VfModuleNotFoundException vfModuleException = new VfModuleNotFoundException.Builder("Test Message",
                HttpStatus.SC_CONFLICT, ErrorNumbers.SVC_BAD_PARAMETER).build();
        Response resp = mapper.toResponse((ApiException) vfModuleException);

        assertEquals(resp.getStatus(), HttpStatus.SC_CONFLICT);
    }

    @Test
    public void testDuplicateRequestResponse() throws IOException {
        DuplicateRequestException duplicateRequestException = new DuplicateRequestException.Builder("Test1", "Test2",
                "Test3", "Test4", HttpStatus.SC_BAD_GATEWAY, ErrorNumbers.SVC_BAD_PARAMETER).build();
        Response resp = mapper.toResponse((ApiException) duplicateRequestException);

        assertEquals(resp.getStatus(), HttpStatus.SC_BAD_GATEWAY);
    }

    @Test
    public void verifyXMLPath() throws JAXBException {
        when(headers.getAcceptableMediaTypes()).thenReturn(Arrays.asList(MediaType.APPLICATION_XML_TYPE));
        BPMNFailureException bpmnException = new BPMNFailureException.Builder("Test Message", HttpStatus.SC_NOT_FOUND,
                ErrorNumbers.SVC_BAD_PARAMETER).build();
        ApiExceptionMapper mapperSpy = Mockito.spy(mapper);
        doReturn(marshaller).when(mapperSpy).getMarshaller();
        Response resp = mapperSpy.toResponse((ApiException) bpmnException);
        verify(marshaller, times(1)).marshal(any(Object.class), any(Writer.class));
    }

    @Test
    public void verifyMediaType() {
        ApiExceptionMapper mapperSpy = Mockito.spy(mapper);
        BPMNFailureException bpmnException = new BPMNFailureException.Builder("Test Message", HttpStatus.SC_NOT_FOUND,
                ErrorNumbers.SVC_BAD_PARAMETER).build();
        when(headers.getAcceptableMediaTypes())
                .thenReturn(Arrays.asList(MediaType.APPLICATION_XML_TYPE.withCharset("UTF-8")));
        mapperSpy.toResponse(bpmnException);
        verify(mapperSpy, times(1)).buildServiceErrorResponse(any(String.class), any(String.class),
                ArgumentMatchers.isNull(), eq(MediaType.APPLICATION_XML_TYPE));
        when(headers.getAcceptableMediaTypes())
                .thenReturn(Arrays.asList(MediaType.APPLICATION_JSON_TYPE.withCharset("UTF-8")));
        mapperSpy = Mockito.spy(mapper);
        mapperSpy.toResponse(bpmnException);
        verify(mapperSpy, times(1)).buildServiceErrorResponse(any(String.class), any(String.class),
                ArgumentMatchers.isNull(), eq(MediaType.APPLICATION_JSON_TYPE));
        when(headers.getAcceptableMediaTypes()).thenReturn(null);
        mapperSpy = Mockito.spy(mapper);
        mapperSpy.toResponse(bpmnException);
        verify(mapperSpy, times(1)).buildServiceErrorResponse(any(String.class), any(String.class),
                ArgumentMatchers.isNull(), eq(MediaType.APPLICATION_JSON_TYPE));
    }
}