/*- 
 * ============LICENSE_START======================================================= 
 * ONAP - SO 
 * ================================================================================ 
 * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved. 
 * ================================================================================
 * Modifications Copyright (c) 2019 Samsung
 * ================================================================================
 * 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.bpmn.common;

import static com.github.tomakehurst.wiremock.client.WireMock.aResponse;
import static com.github.tomakehurst.wiremock.client.WireMock.containing;
import static com.github.tomakehurst.wiremock.client.WireMock.put;
import static com.github.tomakehurst.wiremock.client.WireMock.urlMatching;
import java.util.HashMap;
import java.util.Map;
import java.util.UUID;
import org.junit.Assert;
import org.junit.Test;
import org.onap.so.BaseIntegrationTest;
import org.onap.so.bpmn.core.WorkflowException;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import com.github.tomakehurst.wiremock.WireMockServer;

/**
 * Unit test for CreateAAIVfModule.bpmn.
 */

public class CreateAAIVfModuleIT extends BaseIntegrationTest {

    Logger logger = LoggerFactory.getLogger(CreateAAIVfModuleIT.class);

    @Test
    public void TestCreateGenericVnfSuccess_200() {

        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);

        Map<String, Object> variables = new HashMap<>();
        variables.put("mso-request-id", UUID.randomUUID().toString());
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfName", "STMTN5MMSC22");
        variables.put("serviceId", "00000000-0000-0000-0000-000000000000");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC22-MMSC::module-0-0");
        variables.put("vfModuleModelName", "MMSC::module-0");

        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        String response = BPMNUtil.getVariable(processEngine, "CreateAAIVfModule",
                "CAAIVfMod_createVfModuleResponseCode", processId);
        String responseCode = BPMNUtil.getVariable(processEngine, "CreateAAIVfModule",
                "CAAIVfMod_createVfModuleResponseCode", processId);
        Assert.assertEquals("201", responseCode);
        logger.debug(response);
    }

    @Test
    public void TestCreateVfModuleSuccess_200() {
        // create Add-on VF Module for existing Generic VNF
        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);
        Map<String, Object> variables = new HashMap<>();
        variables.put("mso-request-id", UUID.randomUUID().toString());
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfId", "a27ce5a9-29c4-4c22-a017-6615ac73c721");
        variables.put("serviceId", "00000000-0000-0000-0000-000000000000");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC21-MMSC::module-1-0");
        variables.put("vfModuleModelName", "STMTN5MMSC21-MMSC::model-1-0");
        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        String response = BPMNUtil.getVariable(processEngine, "CreateAAIVfModule",
                "CAAIVfMod_createVfModuleResponseCode", processId);
        String responseCode = BPMNUtil.getVariable(processEngine, "CreateAAIVfModule",
                "CAAIVfMod_createVfModuleResponseCode", processId);
        Assert.assertEquals("201", responseCode);
        logger.debug(response);
    }

    @Test
    public void TestQueryGenericVnfFailure_5000() {
        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);

        Map<String, Object> variables = new HashMap<>();
        variables.put("mso-request-id", UUID.randomUUID().toString());
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfName", "STMTN5MMSC23");
        variables.put("serviceId", "00000000-0000-0000-0000-000000000000");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC23-MMSC::module-0-0");
        variables.put("vfModuleModelName", "MMSC::module-0");
        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        WorkflowException exception =
                BPMNUtil.getRawVariable(processEngine, "CreateAAIVfModule", "WorkflowException", processId);
        Assert.assertEquals(500, exception.getErrorCode());
        Assert.assertEquals(true, exception.getErrorMessage().contains("Error occurred attempting to query AAI"));
        logger.debug(exception.getErrorMessage());
    }

    @Test
    public void TestCreateDupGenericVnfFailure_1002() {
        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);

        Map<String, Object> variables = new HashMap<>();
        variables.put("mso-request-id", UUID.randomUUID().toString());
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfName", "STMTN5MMSC21");
        variables.put("serviceId", "00000000-0000-0000-0000-000000000000");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC21-MMSC::module-0-0");
        variables.put("vfModuleModelName", "MMSC::module-0");
        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        WorkflowException exception =
                BPMNUtil.getRawVariable(processEngine, "CreateAAIVfModule", "WorkflowException", processId);
        Assert.assertEquals(1002, exception.getErrorCode());
        Assert.assertEquals(true,
                exception.getErrorMessage().contains("Invalid request for new Generic VNF which already exists"));
        logger.debug(exception.getErrorMessage());
    }

    @Test
    public void TestCreateDupVfModuleFailure_1002() {
        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);

        Map<String, Object> variables = new HashMap<>();
        variables.put("mso-request-id", UUID.randomUUID().toString());
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfId", "2f6aee38-1e2a-11e6-82d1-ffc7d9ee8aa4");
        variables.put("serviceId", "00000000-0000-0000-0000-000000000000");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC20-MMSC::module-1-0");
        variables.put("vfModuleModelName", "STMTN5MMSC20-MMSC::model-1-0");
        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        WorkflowException exception =
                BPMNUtil.getRawVariable(processEngine, "CreateAAIVfModule", "WorkflowException", processId);
        Assert.assertEquals(1002, exception.getErrorCode());
        Assert.assertEquals(true, exception.getErrorMessage().contains("already exists for Generic VNF"));
        logger.debug(exception.getErrorMessage());
    }

    @Test
    public void TestCreateGenericVnfFailure_5000() {
        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);

        Map<String, Object> variables = new HashMap<>();
        variables.put("mso-request-id", UUID.randomUUID().toString());
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfName", "STMTN5MMSC22");
        variables.put("serviceId", "99999999-9999-9999-9999-999999999999");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC22-PCRF::module-1-0");
        variables.put("vfModuleModelName", "PCRF::module-0");
        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        WorkflowException exception =
                BPMNUtil.getRawVariable(processEngine, "CreateAAIVfModule", "WorkflowException", processId);
        Assert.assertEquals(5000, exception.getErrorCode());
        Assert.assertEquals(true, exception.getErrorMessage().contains("<messageId>SVC3002</messageId>"));
        logger.debug(exception.getErrorMessage());
    }

    @Test
    public void TestCreateGenericVnfFailure_1002() {
        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);

        Map<String, Object> variables = new HashMap<>();
        variables.put("mso-request-id", UUID.randomUUID().toString());
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfId", "768073c7-f41f-4822-9323-b75962763d74");
        variables.put("serviceId", "00000000-0000-0000-0000-000000000000");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC22-PCRF::module-1-0");
        variables.put("vfModuleModelName", "PCRF::module-0");
        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        WorkflowException exception =
                BPMNUtil.getRawVariable(processEngine, "CreateAAIVfModule", "WorkflowException", processId);
        Assert.assertEquals(1002, exception.getErrorCode());
        Assert.assertEquals(true, exception.getErrorMessage().contains("Generic VNF Not Found"));
        logger.debug(exception.getErrorMessage());
    }

    @Test
    public void TestCreateVfModuleFailure_5000() {
        new MockAAIGenericVnfSearch(wireMockServer);
        MockAAICreateGenericVnf(wireMockServer);
        MockAAIVfModulePUT(wireMockServer, true);

        Map<String, Object> variables = new HashMap<>();
        variables.put("isDebugLogEnabled", "true");
        variables.put("isVidRequest", "false");
        variables.put("vnfId", "a27ce5a9-29c4-4c22-a017-6615ac73c721");
        variables.put("serviceId", "99999999-9999-9999-9999-999999999999");
        variables.put("personaModelId", "973ed047-d251-4fb9-bf1a-65b8949e0a73");
        variables.put("personaModelVersion", "1.0");
        variables.put("vfModuleName", "STMTN5MMSC21-PCRF::module-1-0");
        variables.put("vfModuleModelName", "STMTN5MMSC21-PCRF::model-1-0");
        variables.put("mso-request-id", UUID.randomUUID().toString());
        String processId = invokeSubProcess("CreateAAIVfModule", variables);
        WorkflowException exception =
                BPMNUtil.getRawVariable(processEngine, "CreateAAIVfModule", "WorkflowException", processId);
        Assert.assertEquals(5000, exception.getErrorCode());
        Assert.assertEquals(true, exception.getErrorMessage().contains("<messageId>SVC3002</messageId>"));
        logger.debug(exception.getErrorMessage());
    }

    public static void MockAAICreateGenericVnf(WireMockServer wireMockServer) {
        wireMockServer.stubFor(put(urlMatching("/aai/v[0-9]+/network/generic-vnfs/generic-vnf/.*"))
                .withRequestBody(containing("<service-id>00000000-0000-0000-0000-000000000000</service-id>"))
                .willReturn(aResponse().withStatus(201)));
        wireMockServer.stubFor(put(urlMatching("/aai/v[0-9]+/network/generic-vnfs/generic-vnf/.*"))
                .withRequestBody(containing("<service-id>99999999-9999-9999-9999-999999999999</service-id>"))
                .willReturn(aResponse().withStatus(500).withHeader("Content-Type", "text/xml")
                        .withBodyFile("aaiFault.xml")));
    }

    // start of mocks used locally and by other VF Module unit tests
    public static void MockAAIVfModulePUT(WireMockServer wireMockServer, boolean isCreate) {
        wireMockServer
                .stubFor(put(urlMatching("/aai/v[0-9]+/network/generic-vnfs/generic-vnf/.*/vf-modules/vf-module/.*"))
                        .withRequestBody(containing("MMSC")).willReturn(aResponse().withStatus(isCreate ? 201 : 200)));
        wireMockServer
                .stubFor(put(urlMatching("/aai/v[0-9]+/network/generic-vnfs/generic-vnf/.*/vf-modules/vf-module/.*"))
                        .withRequestBody(containing("PCRF")).willReturn(aResponse().withStatus(500)
                                .withHeader("Content-Type", "text/xml").withBodyFile("aaiFault.xml")));
        wireMockServer.stubFor(
                put(urlMatching("/aai/v[0-9]+/network/generic-vnfs/generic-vnf/a27ce5a9-29c4-4c22-a017-6615ac73c721"))
                        .willReturn(aResponse().withStatus(200)));
    }

}