/*
 * ============LICENSE_START===================================================
 * Copyright (c) 2018 Amdocs
 * ============================================================================
 * 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.pomba.contextbuilder.sdnc.unittest.service;

import static com.github.tomakehurst.wiremock.client.WireMock.get;
import static com.github.tomakehurst.wiremock.client.WireMock.okJson;
import static com.github.tomakehurst.wiremock.core.WireMockConfiguration.wireMockConfig;
import static org.junit.Assert.assertEquals;
import static org.junit.Assert.assertTrue;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;

import com.github.jknack.handlebars.internal.Files;
import com.github.tomakehurst.wiremock.junit.WireMockRule;
import java.io.File;
import java.io.IOException;
import java.util.Collections;
import java.util.UUID;
import javax.servlet.http.HttpServletRequest;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MultivaluedHashMap;
import javax.ws.rs.core.MultivaluedMap;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.Response.Status;
import org.junit.After;
import org.junit.Before;
import org.junit.Rule;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.onap.aai.restclient.client.RestClient;
import org.onap.pomba.contextbuilder.sdnc.Application;
import org.onap.pomba.contextbuilder.sdnc.model.ServiceEntity;
import org.onap.pomba.contextbuilder.sdnc.service.rs.RestService;
import org.onap.pomba.contextbuilder.sdnc.util.RestUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.autoconfigure.EnableAutoConfiguration;
import org.springframework.boot.autoconfigure.jdbc.DataSourceAutoConfiguration;
import org.springframework.boot.autoconfigure.orm.jpa.HibernateJpaAutoConfiguration;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.TestPropertySource;
import org.springframework.test.context.junit4.SpringJUnit4ClassRunner;
import org.springframework.test.context.web.WebAppConfiguration;

@RunWith(SpringJUnit4ClassRunner.class)
@EnableAutoConfiguration(exclude = {DataSourceAutoConfiguration.class, HibernateJpaAutoConfiguration.class})
@WebAppConfiguration
@SpringBootTest (classes = Application.class)
@TestPropertySource(properties = {"sdnc.serviceName=localhost",
                                  "sdnc.servicePort=30202",
                                  "aai.httpProtocol=http",
                                  "aai.serviceName=localhost",
                                  "aai.servicePort=9808"})

public class SdncContextBuilderTest {

    private String testRestHeaders = "testRestHeaders";
    private String servicePath = "/service-subscriptions/service-subscription/vFW/service-instances/service-instance/";
    private String genericVnfPath = "/aai/v11/network/generic-vnfs/generic-vnf/";
    private String genericResourcePath = "/restconf/config/GENERIC-RESOURCE-API:services/service/";
    private String portMirrorConfigurationsResourcePath = "/restconf/config/GENERIC-RESOURCE-API:port-mirror-configurations/port-mirror-configuration/3c368d8d-efda-49d4-bbb5-a6465330a230/configuration-data/configuration-operation-information/port-mirror-configuration-request-input";
    private String vnfPath = "/restconf/config/VNF-API:vnfs/vnf-list/";
    private String serviceInstanceIdVfw = "7d518257-49bd-40ac-8d17-017a726ec12a"; // customerData.json
    private String serviceInstanceIdVcpe = "68352304-7bba-4609-8551-0d0b819376c3"; // queryNodeDataVcpe.json
    private String customerIdVfw = "DemoCust_651800ed-2a3c-45f5-b920-85c1ed155fc2"; // customerData.json
    private String customerIdVcpe = "SDN-ETHERNET-INTERNET";
    private String genericVnfId = "d94daff6-7d5b-4d2e-bc99-c9af0754b59d";
    private String moduleId = "2c3f8902-f278-4ee3-93cf-9d2364cbafca";

    HttpServletRequest httpServletRequest = mock(HttpServletRequest.class);

    @Autowired
    RestService service;
    @Autowired
    private String sdncCtxBuilderBasicAuthorization;
    //AAI related
    @Autowired
    private String aaiBasicAuthorization;
    @Autowired
    private RestClient aaiClient;
    @Autowired
    private String aaiBaseUrl;
    @Autowired
    private String aaiPathToSearchNodeQuery;
    @Autowired
    private String aaiPathToCustomerQuery;
    @Rule
    public WireMockRule aaiEnricherRule = new WireMockRule(wireMockConfig().port(9808));

    @Rule
    public WireMockRule sdncRule = new WireMockRule(wireMockConfig().port(30202));

    @Before
    public void setUp() throws Exception {
    }

    @After
    public void tearDown() throws Exception {
    }

    @Test
    public void testRestHeaders() throws Exception {
        HttpHeaders mockHttpHeaders = mock( HttpHeaders.class);

        // Test with No Partner Name
        final MultivaluedMap<String, String> multivaluedMapImpl = buildHeaders(
                null, testRestHeaders, sdncCtxBuilderBasicAuthorization);

        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl);
        Response response = service.getContext(httpServletRequest, mockHttpHeaders, serviceInstanceIdVfw);
        assertTrue(response.getEntity().toString().contains("Missing header parameter: " + RestUtil.FROM_APP_ID));

        // Test with no Authorization
        final MultivaluedMap<String, String> multivaluedMapImpl1 = buildHeaders(
                "test2", testRestHeaders, null);

        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl1);
        response = service.getContext(httpServletRequest, mockHttpHeaders, serviceInstanceIdVfw);
        assertTrue(response.getEntity().toString().contains("Missing header parameter: " + RestUtil.AUTHORIZATION));

        // Test with garbage Authorization
        final MultivaluedMap<String, String> multivaluedMapImpl2 = buildHeaders(
                "test3", testRestHeaders, "garbage");

        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl2);
        response = service.getContext(httpServletRequest, mockHttpHeaders, serviceInstanceIdVfw);
        assertTrue(response.getEntity().toString().contains("Failed Basic " + RestUtil.AUTHORIZATION));
    }

    @Test
    public void testRestParameterServiceInstanceId() throws Exception {
        HttpHeaders mockHttpHeaders = mock( HttpHeaders.class);
        final MultivaluedMap<String, String> multivaluedMapImpl = buildHeaders(
                "testRestParameterServiceInstanceId", "test1", sdncCtxBuilderBasicAuthorization);

        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl);
        Response response = service.getContext(httpServletRequest, mockHttpHeaders, null);
        assertTrue(response
                   .getEntity()
                   .toString()
                   .contains("Invalid request URL, missing parameter: serviceInstanceId"));
    }

    @Test
    public void testVerifySdncContextBuilder() throws Exception {


        HttpHeaders mockHttpHeaders = mock( HttpHeaders.class);
        final MultivaluedMap<String, String> multivaluedMapImpl = buildHeaders(
                "testVerifyServiceDecomposition", "test1", sdncCtxBuilderBasicAuthorization);
        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl);

        // First try a vFW service instance

        String queryNodeVfwUrl = aaiPathToSearchNodeQuery + serviceInstanceIdVfw;
        addResponse(queryNodeVfwUrl, "junit/queryNodeDataVfw.json", aaiEnricherRule);

        String customerVfwUrl = aaiPathToCustomerQuery + customerIdVfw;
        addResponse(customerVfwUrl, "junit/customerData.json", aaiEnricherRule);

        String serviceInstanceUrl = aaiPathToCustomerQuery
                                    + customerIdVfw
                                    + servicePath
                                    + serviceInstanceIdVfw;
        addResponse(serviceInstanceUrl, "junit/serviceInstance.json", aaiEnricherRule);

        String vnfApiUrl = vnfPath + moduleId;
        addResponse(vnfApiUrl, "junit/vnfApiResponse.json", sdncRule);

        String genericVnfUrl = genericVnfPath + genericVnfId + "?depth=all";
        addResponse(genericVnfUrl, "junit/genericVnf.json", aaiEnricherRule);
        Response response = this.service.getContext(httpServletRequest, mockHttpHeaders, serviceInstanceIdVfw);
        assertEquals(Status.OK.getStatusCode(), response.getStatus());

        // Now try a vCPE service instance

        String queryNodeUrlVcpe = aaiPathToSearchNodeQuery + serviceInstanceIdVcpe;
        addResponse(queryNodeUrlVcpe, "junit/queryNodeDataVcpe.json", aaiEnricherRule);

        String customerVcpeUrl = aaiPathToCustomerQuery + customerIdVcpe;
        addResponse(customerVcpeUrl, "junit/customerData.json", aaiEnricherRule);


        String urlStr = genericResourcePath + serviceInstanceIdVcpe;
        addResponse(urlStr, "junit/sdncGenericResponse.json", sdncRule);

        addResponse(portMirrorConfigurationsResourcePath, "junit/portMirrorConfigurationsResponse.json", sdncRule);

        response = this.service.getContext(httpServletRequest, mockHttpHeaders, serviceInstanceIdVcpe);
        assertEquals(Status.OK.getStatusCode(), response.getStatus());

        // Try again with no transcactionId
        final MultivaluedMap<String, String> multivaluedMapImpl1 = buildHeaders(
                "testVerifyServiceDecomposition", null, sdncCtxBuilderBasicAuthorization);
        when(mockHttpHeaders.getRequestHeaders()).thenReturn(multivaluedMapImpl1);
        response = this.service.getContext(httpServletRequest, mockHttpHeaders, serviceInstanceIdVcpe);
        assertEquals(Status.OK.getStatusCode(), response.getStatus());
    }


    //AAI related

    @Test
    public void testObtainResouceLinkBasedOnServiceInstanceFromAai() throws Exception {
        String transactionId = UUID.randomUUID().toString();
        String queryNodeUrl = aaiPathToSearchNodeQuery + serviceInstanceIdVfw;
        addResponse(queryNodeUrl, "junit/queryNodeDataVfw.json", aaiEnricherRule);
        String customerUrl = aaiPathToCustomerQuery + customerIdVfw;
        addResponse(customerUrl, "junit/customerData.json", aaiEnricherRule);

        ServiceEntity serviceEntity = RestUtil.getServiceEntity(aaiClient,
                                                                aaiBaseUrl,
                                                                aaiBasicAuthorization,
                                                                aaiPathToSearchNodeQuery,
                                                                aaiPathToCustomerQuery,
                                                                serviceInstanceIdVfw,
                                                                transactionId);

        assertEquals(serviceInstanceIdVfw, serviceEntity.getServiceInstanceId());
        assertEquals("vFW", serviceEntity.getServiceType());  // customerData.json
        assertEquals(customerIdVfw, serviceEntity.getCustomerId());  // queryNodeData-1.json
        assertEquals("DemoCust_651800ed-2a3c-45f5-b920-85c1ed155fc2",
                     serviceEntity.getCustomerName());  //  customerData.json
        assertEquals("CUST", serviceEntity.getCustomerType()); // customerData.json
    }

    @Test
    public void testObtainResouceLinkBasedOnServiceInstanceFromAaiNullResourceLink() throws Exception {
        String transactionId = UUID.randomUUID().toString();
        String queryNodeUrl = aaiPathToSearchNodeQuery + serviceInstanceIdVfw;
        addResponse(queryNodeUrl, "junit/queryNodeDataNullResourceLink.json", aaiEnricherRule);

        try {
            RestUtil.getServiceEntity(aaiClient,
                                      aaiBaseUrl,
                                      aaiBasicAuthorization,
                                      aaiPathToSearchNodeQuery,
                                      aaiPathToCustomerQuery,
                                      serviceInstanceIdVfw,
                                      transactionId);
        } catch (Exception e) {
            assertTrue(e.getMessage().contains("JSONObject[\"resource-link\"] not found"));
        }
    }

    @Test
    public void testObtainResouceLinkBasedOnServiceInstanceFromAaiNullCustomerType() throws Exception {
        String transactionId = UUID.randomUUID().toString();
        String queryNodeUrl = aaiPathToSearchNodeQuery + serviceInstanceIdVfw;
        addResponse(queryNodeUrl, "junit/queryNodeDataVfw.json", aaiEnricherRule);
        String customerUrl = aaiPathToCustomerQuery + customerIdVfw;
        addResponse(customerUrl, "junit/customerDataCustomerIdNotFound.json", aaiEnricherRule);

        try {
            RestUtil.getServiceEntity(aaiClient,
                                      aaiBaseUrl,
                                      aaiBasicAuthorization,
                                      aaiPathToSearchNodeQuery,
                                      aaiPathToCustomerQuery,
                                      serviceInstanceIdVfw,
                                      transactionId);
        } catch (Exception e) {
            assertTrue(e.getMessage().contains("Customer ID cannot be found from AAI"));
        }
    }

    private void addResponse(String url, String responseFile, WireMockRule thisMock) throws IOException {
        File file = new File(ClassLoader.getSystemResource(responseFile).getFile());
        String payload = Files.read(file);
        thisMock.stubFor(get(url).willReturn(okJson(payload)));
    }

    private static MultivaluedMap<String, String> buildHeaders(
            String partnerName, String transactionId, String authorization) {

        MultivaluedMap<String, String> headers = new MultivaluedHashMap<>();
        headers.put(RestUtil.FROM_APP_ID, Collections.singletonList(partnerName));
        headers.put(RestUtil.TRANSACTION_ID, Collections.singletonList(transactionId));
        if (null != authorization) {
            headers.put(RestUtil.AUTHORIZATION, Collections.singletonList(authorization));
        }
        return headers;
    }

}