/**
 * ============LICENSE_START=======================================================
 * org.onap.aai
 * ================================================================================
 * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
 * Copyright © 2017-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.crud.service;

import static org.assertj.core.api.Assertions.assertThat;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.when;
import java.security.cert.X509Certificate;
import java.util.Collections;
import java.util.List;
import java.util.Map.Entry;
import javax.security.auth.x500.X500Principal;
import javax.ws.rs.core.EntityTag;
import javax.ws.rs.core.HttpHeaders;
import javax.ws.rs.core.MultivaluedHashMap;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriInfo;
import org.apache.commons.lang3.tuple.ImmutablePair;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.mockito.Mockito;
import org.onap.crud.exception.CrudException;
import org.onap.crud.util.TestUtil;
import org.onap.schema.OxmModelLoader;
import org.springframework.mock.web.MockHttpServletRequest;
import com.att.aft.dme2.internal.jersey.api.client.ClientResponse.Status;

public class TestResourceServiceEdgeOperations {

    private MockHttpServletRequest servletRequest;
    private UriInfo uriInfo;
    private HttpHeaders headers;
    private AbstractGraphDataService graphDataService;

    @BeforeClass
    public static void setUpBeforeClass() throws Exception {
        System.setProperty("CONFIG_HOME", "src/test/resources");
        System.setProperty("AJSC_HOME", ".");
        System.setProperty("BUNDLECONFIG_DIR", "src/test/resources/bundleconfig-local");

        OxmModelLoader.loadModels();
    }

    @Before
    public void setUp() throws Exception {
        graphDataService = mock(CrudGraphDataService.class);
        uriInfo = mock(UriInfo.class);

        mockHeaderRequests(getCreateHeaders());

        servletRequest = new MockHttpServletRequest();
        servletRequest.setSecure(true);
        servletRequest.setScheme("https");
        servletRequest.setServerPort(9520);
        servletRequest.setServerName("localhost");
        servletRequest.setRequestURI("/services/inventory/relationships/");

        setUser("CN=ONAP, OU=ONAP, O=ONAP, L=Ottawa, ST=Ontario, C=CA");

        servletRequest.setAttribute("javax.servlet.request.cipher_suite", "");
    }

    private MultivaluedHashMap<String, String> getCreateHeaders() {
        MultivaluedHashMap<String, String> headersMap = new MultivaluedHashMap<>();
        headersMap.put("X-TransactionId", createSingletonList("transaction-id"));
        headersMap.put("X-FromAppId", createSingletonList("app-id"));
        headersMap.put("Host", createSingletonList("hostname"));
        return headersMap;
    }

    private void mockHeaderRequests(MultivaluedHashMap<String, String> headersMap) {
        headers = Mockito.mock(HttpHeaders.class);
        for (Entry<String, List<String>> entry : headersMap.entrySet()) {
            when(headers.getRequestHeader(entry.getKey())).thenReturn(entry.getValue());
        }
        when(headers.getRequestHeaders()).thenReturn(headersMap);
    }

    @Test
    public void testCreateRelationship() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-auto-props.json");

        Response response = callCreateRelationship(postEdgeBody);

        assertThat(response.getStatus()).isEqualTo(Status.CREATED.getStatusCode());
    }

    @Test
    public void testCreateRelationshipWithMatchingType() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-auto-props.json");
        String type = "tosca.relationships.HostedOn";

        Response response = createRelationshipWithType(postEdgeBody, type);

        assertThat(response.getStatus()).isEqualTo(Status.CREATED.getStatusCode());
    }

    @Test
    public void testCreateRelationshipNoMatchingType() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-auto-props.json");
        String type = "type.does.not.match";

        Response response = createRelationshipWithType(postEdgeBody, type);

        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testCreateRelationshipWithTypeNullPropsIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-null-props.json");
        String type = "tosca.relationships.HostedOn";

        Response response = createRelationshipWithType(postEdgeBody, type);

        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testCreateRelationshipWithTypeWithIdIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-with-id.json");
        String type = "tosca.relationships.HostedOn";

        Response response = createRelationshipWithType(postEdgeBody, type);

        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testInvalidUser() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-auto-props.json");

        setUser("CN=INVALID, OU=INVALID, O=INVALID, L=Ottawa, ST=Ontario, C=CA");

        Response response = callCreateRelationship(postEdgeBody);

        assertThat(response.getStatus()).isEqualTo(Status.FORBIDDEN.getStatusCode());
    }

    @Test
    public void testNullPropertiesIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-null-props.json");

        Response response = callCreateRelationship(postEdgeBody);

        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testNoPropertiesIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-no-props.json");

        Response response = callCreateRelationship(postEdgeBody);

        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testCreateWithIdIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-with-id.json");

        Response response = callCreateRelationship(postEdgeBody);

        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testNoTypeIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-no-type.json");

        Response response = callCreateRelationship(postEdgeBody);

        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testPatchEdge() throws Exception {
        MultivaluedHashMap<String, String> headersMap = getCreateHeaders();
        headersMap.put(AaiResourceService.HTTP_PATCH_METHOD_OVERRIDE, createSingletonList("PATCH"));
        mockHeaderRequests(headersMap);

        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-upsert.json");
        String type = "tosca.relationships.HostedOn";
        String id = "12345";
        EntityTag entityTag = new EntityTag("1234");
        
        when(graphDataService.patchEdge(any(), any(), any(), any())).thenReturn(new ImmutablePair<EntityTag, String>(entityTag, "dummy output"));
        AaiResourceService aaiResourceService = new AaiResourceService(graphDataService);
        Response response =
                aaiResourceService.upsertEdge(postEdgeBody, type, id, "uri", headers, uriInfo, servletRequest);
        assertThat(response.getStatus()).isEqualTo(Status.OK.getStatusCode());
    }

    @Test
    public void testUpdateEdge() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-upsert.json");
        String type = "tosca.relationships.HostedOn";
        String id = "12345";
        EntityTag entityTag = new EntityTag("1234");

        when(graphDataService.updateEdge(any(), any(), any(), any())).thenReturn(new ImmutablePair<EntityTag, String>(entityTag, "dummy output"));
        AaiResourceService aaiResourceService = new AaiResourceService(graphDataService);
        Response response =
                aaiResourceService.upsertEdge(postEdgeBody, type, id, "uri", headers, uriInfo, servletRequest);
        assertThat(response.getStatus()).isEqualTo(Status.OK.getStatusCode());
    }

    @Test
    public void testUpdateEdgeNullPropsIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-null-props.json");
        String type = "tosca.relationships.HostedOn";
        String id = "12345";
        EntityTag entityTag = new EntityTag("1234");

        when(graphDataService.updateEdge(any(), any(), any(), any())).thenReturn(new ImmutablePair<EntityTag, String>(entityTag, "dummy output"));
        AaiResourceService aaiResourceService = new AaiResourceService(graphDataService);
        Response response =
                aaiResourceService.upsertEdge(postEdgeBody, type, id, "uri", headers, uriInfo, servletRequest);
        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    @Test
    public void testUpdateEdgeWithMismatchIdIsBadRequest() throws Exception {
        String postEdgeBody = TestUtil.getFileAsString("aai-resource-service/post-edge-upsert.json");
        String type = "tosca.relationships.HostedOn";
        String id = "mismatch";
        EntityTag entityTag = new EntityTag("1234");

        when(graphDataService.updateEdge(any(), any(), any(), any())).thenReturn(new ImmutablePair<EntityTag, String>(entityTag, "dummy output"));
        AaiResourceService aaiResourceService = new AaiResourceService(graphDataService);
        Response response =
                aaiResourceService.upsertEdge(postEdgeBody, type, id, "uri", headers, uriInfo, servletRequest);
        assertThat(response.getStatus()).isEqualTo(Status.BAD_REQUEST.getStatusCode());
    }

    private Response createRelationshipWithType(String postEdgeBody, String type) throws CrudException, Exception {
        EntityTag entityTag = new EntityTag("1234");
        
        when(graphDataService.addEdge(any(), any(), any())).thenReturn(new ImmutablePair<EntityTag, String>(entityTag, "dummy output"));
        AaiResourceService aaiResourceService = new AaiResourceService(graphDataService);
        Response response =
                aaiResourceService.createRelationship(postEdgeBody, type, "uri", headers, uriInfo, servletRequest);
        return response;
    }

    private Response callCreateRelationship(String postEdgeBody) throws CrudException, Exception {
        EntityTag entityTag = new EntityTag("1234");
        
        when(graphDataService.addEdge(any(), any(), any())).thenReturn(new ImmutablePair<EntityTag, String>(entityTag, "dummy output"));
        AaiResourceService aaiResourceService = new AaiResourceService(graphDataService);
        Response response =
                aaiResourceService.createRelationship(postEdgeBody, "uri", headers, uriInfo, servletRequest);
        return response;
    }

    private void setUser(String user) {
        X509Certificate mockCertificate = Mockito.mock(X509Certificate.class);
        when(mockCertificate.getSubjectX500Principal()).thenReturn(new X500Principal(user));
        servletRequest.setAttribute("javax.servlet.request.X509Certificate", new X509Certificate[] {mockCertificate});
    }

    private List<String> createSingletonList(String listItem) {
        return Collections.<String>singletonList(listItem);
    }
}