/*
 * ============LICENSE_START=======================================================
 * aaf-certservice-client
 * ================================================================================
 * Copyright (C) 2020 Nokia. 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.aaf.certservice.client.httpclient;

import org.apache.http.HttpEntity;
import org.apache.http.StatusLine;
import org.apache.http.client.methods.CloseableHttpResponse;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.impl.client.CloseableHttpClient;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.onap.aaf.certservice.client.httpclient.exception.CertServiceApiResponseException;
import org.onap.aaf.certservice.client.httpclient.exception.HttpClientException;
import org.onap.aaf.certservice.client.httpclient.model.CertServiceResponse;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.util.List;

import static java.net.HttpURLConnection.HTTP_BAD_REQUEST;
import static java.net.HttpURLConnection.HTTP_OK;
import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
import static org.mockito.Mockito.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import static org.onap.aaf.certservice.client.CerServiceRequestTestData.CA_NAME;
import static org.onap.aaf.certservice.client.CerServiceRequestTestData.CORRECT_RESPONSE;
import static org.onap.aaf.certservice.client.CerServiceRequestTestData.CSR;
import static org.onap.aaf.certservice.client.CerServiceRequestTestData.EXPECTED_FIRST_ELEMENT_OF_CERTIFICATE_CHAIN;
import static org.onap.aaf.certservice.client.CerServiceRequestTestData.EXPECTED_FIRST_ELEMENT_OF_TRUSTED_CERTIFICATES;
import static org.onap.aaf.certservice.client.CerServiceRequestTestData.MISSING_PK_RESPONSE;
import static org.onap.aaf.certservice.client.CerServiceRequestTestData.PK;

class HttpClientTest {

    private HttpClient httpClient;
    private CloseableHttpClient closeableHttpClient;
    private HttpEntity httpEntity;
    private StatusLine statusLine;
    private CloseableHttpResponse httpResponse;

    @BeforeEach
    void setUp() {

        closeableHttpClient = mock(CloseableHttpClient.class);
        httpEntity = mock(HttpEntity.class);
        statusLine = mock(StatusLine.class);
        httpResponse = mock(CloseableHttpResponse.class);

        CloseableHttpClientProvider httpClientProvider = mock(CloseableHttpClientProvider.class);

        when(httpClientProvider.getClient()).thenReturn(closeableHttpClient);
        String testCertServiceAddress = "";
        httpClient = new HttpClient(httpClientProvider, testCertServiceAddress);
    }

    @Test
    void shouldReturnCorrectListsOfCertificatedChainsAndTrustedCertificates_WhenRequestDataIsCorrect()
            throws Exception {

        // given
        mockServerResponse(HTTP_OK, CORRECT_RESPONSE);

        // when
        CertServiceResponse certServiceResponse =
                httpClient.retrieveCertServiceData(CA_NAME, CSR, PK);
        List<String> certificateChain = certServiceResponse.getCertificateChain();
        List<String> trustedCertificate = certServiceResponse.getTrustedCertificates();

        // then
        assertThat(certServiceResponse).isNotNull();

        final int expectedTwoElements = 2;

        assertThat(certificateChain).hasSize(expectedTwoElements);
        assertThat(trustedCertificate).hasSize(expectedTwoElements);

        assertThat(certificateChain.get(0)).isEqualTo(EXPECTED_FIRST_ELEMENT_OF_CERTIFICATE_CHAIN);
        assertThat(trustedCertificate.get(0)).isEqualTo(EXPECTED_FIRST_ELEMENT_OF_TRUSTED_CERTIFICATES);
    }

    @Test
    void shouldThrowCertServiceApiResponseException_WhenPkHeaderIsMissing() throws Exception {

        //given
        mockServerResponse(HTTP_BAD_REQUEST, MISSING_PK_RESPONSE);

        //when //then
        assertThatExceptionOfType(CertServiceApiResponseException.class)
                .isThrownBy(()->httpClient.retrieveCertServiceData(CA_NAME, CSR, ""));
    }

    @Test
    void shouldThrowHttpClientException_WhenCannotExecuteRequestToAPI() throws Exception {

        //given
        when(closeableHttpClient.execute(any(HttpGet.class))).thenThrow(IOException.class);

        //when //then
        assertThatExceptionOfType(HttpClientException.class)
                .isThrownBy(()->httpClient.retrieveCertServiceData(CA_NAME, CSR, ""));
    }

    @Test
    void shouldThrowHttpClientException_WhenCannotParseResponseToString() throws Exception {

        //given
        mockServerResponse(HTTP_OK, CORRECT_RESPONSE);
        when(httpEntity.getContent()).thenThrow(IOException.class);

        //when //then
        assertThatExceptionOfType(HttpClientException.class)
                .isThrownBy(()->httpClient.retrieveCertServiceData(CA_NAME, CSR, ""));
    }

    private void mockServerResponse(int serverCodeResponse, String stringResponse)
            throws IOException {
        when(statusLine.getStatusCode()).thenReturn(serverCodeResponse);
        when(httpResponse.getStatusLine()).thenReturn(statusLine);
        when(httpResponse.getEntity()).thenReturn(httpEntity);
        when(closeableHttpClient.execute(any(HttpGet.class))).thenReturn(httpResponse);

        when(httpEntity.getContent()).thenReturn(new ByteArrayInputStream(stringResponse.getBytes()));
    }
}