/*
 * ============LICENSE_START=======================================================
 * PNF-REGISTRATION-HANDLER
 * ================================================================================
 * Copyright (C) 2018 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.pnfsimulator.simulator.client;

import org.apache.http.Header;
import org.apache.http.HttpHeaders;
import org.apache.http.HttpResponse;
import org.apache.http.HttpStatus;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpPost;
import org.apache.http.conn.socket.PlainConnectionSocketFactory;
import org.apache.http.conn.ssl.SSLConnectionSocketFactory;
import org.apache.http.message.BasicHeader;
import org.apache.tomcat.util.codec.binary.Base64;
import org.junit.jupiter.api.BeforeEach;
import org.junit.jupiter.api.Test;
import org.mockito.ArgumentCaptor;
import org.onap.pnfsimulator.simulator.client.utils.ssl.SslAuthenticationHelper;

import java.io.IOException;
import java.net.MalformedURLException;
import java.nio.charset.StandardCharsets;
import java.security.GeneralSecurityException;
import java.util.List;

import static org.assertj.core.api.Assertions.assertThat;
import static org.junit.jupiter.api.Assertions.assertEquals;
import static org.junit.jupiter.api.Assertions.assertThrows;
import static org.mockito.ArgumentMatchers.any;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.doThrow;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.onap.pnfsimulator.simulator.client.HttpTestUtils.createMockedHttpEntity;
import static org.onap.pnfsimulator.simulator.client.HttpTestUtils.createStatusLine;

class HttpClientAdapterImplTest {

    private static final String HTTPS_URL = "https://0.0.0.0:8443/";
    private static final String HTTP_URL = "http://0.0.0.0:8000/";

    private HttpClient httpClient;
    private HttpResponse httpResponse;

    @BeforeEach
    void setup() {
        httpClient = mock(HttpClient.class);
        httpResponse = mock(HttpResponse.class);
    }

    @Test
    void sendShouldSuccessfullySendRequestGivenValidUrl() throws IOException {
        assertAdapterSentRequest("http://valid-url:8080",
            HttpStatus.SC_FORBIDDEN, HttpTestUtils.HTTP_MESSAGE_FORBIDDEN);
    }

    @Test
    void sendShouldSuccessfullySendRequestGivenValidUrlUsingHttps() throws IOException {
        assertAdapterSentRequest("https://valid-url:8443",
            HttpStatus.SC_ACCEPTED, HttpTestUtils.HTTP_MESSAGE_ACCEPTER);
    }

    @Test
    void sendShouldSuccessfullySendRequestUsingBasicAuth() throws IOException {
        String testUserInfo = "user1:pass1";
        Header authorizationHeader = createAuthorizationHeader(testUserInfo);
        assertAdapterSentRequest("https://" + testUserInfo + "@valid-url:8443",
            HttpStatus.SC_ACCEPTED, HttpTestUtils.HTTP_MESSAGE_ACCEPTER,
            List.of(authorizationHeader));
    }

    @Test
    void sendShouldFailToSendRequestGivenInvalidUrlUsingAdnShouldInformUser() throws IOException {
        assertAdapterInformsUserWhenServiceIsUnavailable("https://invalid-url:8080");
    }

    @Test
    void shouldThrowExceptionWhenMalformedVesUrlPassed() {
        assertThrows(MalformedURLException.class, () -> new HttpClientAdapterImpl("http://blablabla:VES-PORT", new SslAuthenticationHelper()));
    }

    @Test
    void shouldCreateAdapterWithClientNotSupportingSslConnection() throws IOException, GeneralSecurityException {
        HttpClientAdapter adapterWithHttps = new HttpClientAdapterImpl(HTTPS_URL, new SslAuthenticationHelper());
        try {
            adapterWithHttps.send("sample");
        } catch (Exception actualException) {
            assertThat(actualException).hasStackTraceContaining(SSLConnectionSocketFactory.class.toString());
        }
    }

    @Test
    void shouldCreateAdapterWithClientSupportingPlainConnectionOnly() throws IOException, GeneralSecurityException {
        HttpClientAdapter adapterWithHttps = new HttpClientAdapterImpl(HTTP_URL, new SslAuthenticationHelper());
        try {
            adapterWithHttps.send("sample");
        } catch (Exception actualException) {
            assertThat(actualException).hasStackTraceContaining(PlainConnectionSocketFactory.class.toString());
        }
    }

    private Header createAuthorizationHeader(String testUserInfo) {
        String encodedUserInfo = new String(Base64.encodeBase64(testUserInfo.getBytes(StandardCharsets.UTF_8)));
        return new BasicHeader(HttpHeaders.AUTHORIZATION, "Basic " + encodedUserInfo);
    }

    private void assertAdapterSentRequest(String targetUrl, int responseCode, String responseMessage) throws IOException {
        assertAdapterSentRequest(targetUrl, responseCode, responseMessage, List.of());
    }

    private void assertAdapterSentRequest(String targetUrl, int responseCode, String responseMessage, List<Header> expectedHeaders) throws IOException {
        HttpClientAdapter adapter = new HttpClientAdapterImpl(httpClient, targetUrl);
        doReturn(httpResponse).when(httpClient).execute(any());
        doReturn(createStatusLine(responseCode)).when(httpResponse).getStatusLine();
        doReturn(createMockedHttpEntity(responseMessage)).when(httpResponse).getEntity();

        HttpResponseAdapter response = adapter.send("test-msg");

        ArgumentCaptor<HttpPost> httpPostCaptor = ArgumentCaptor.forClass(HttpPost.class);
        verify(httpClient).execute(httpPostCaptor.capture());
        Header[] headers = httpPostCaptor.getValue().getAllHeaders();
        assertEquals(responseCode, response.getCode());
        assertEquals(responseMessage, response.getMessage());
        assertThat(headers).usingFieldByFieldElementComparator().containsAll(expectedHeaders);
    }

    private void assertAdapterInformsUserWhenServiceIsUnavailable(String targetUrl) throws IOException {
        HttpClientAdapter adapter = new HttpClientAdapterImpl(httpClient, targetUrl);
        String exceptionMessage = "test message";
        doThrow(new IOException(exceptionMessage)).when(httpClient).execute(any());

        HttpResponseAdapter response = adapter.send("test-msg");

        verify(httpClient).execute(any());
        assertEquals(421, response.getCode());
        assertEquals(String.format("Fail to connect with ves: %s", exceptionMessage), response.getMessage());
    }

}