/*
 * Copyright 2016-2017, Nokia Corporation
 *
 * 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.
 */
package org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.direct.notification;

import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import com.nokia.cbam.lcm.v32.model.AffectedVirtualStorage;
import com.nokia.cbam.lcm.v32.model.AffectedVnfc;
import com.nokia.cbam.lcm.v32.model.ResourceHandle;
import org.junit.Before;
import org.junit.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.onap.aai.domain.yang.v11.*;
import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.direct.AAIRestApiProvider;
import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.TestBase;

import java.util.ArrayList;
import java.util.List;
import java.util.NoSuchElementException;

import static junit.framework.TestCase.assertEquals;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.when;
import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.direct.AAIRestApiProvider.AAIService.CLOUD;
import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.direct.notification.AbstractManager.buildRelationshipData;
import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.direct.notification.TestGenericVnfManager.assertRelation;
import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.LifecycleManager.getCloudOwner;
import static org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.LifecycleManager.getRegionName;
import static org.springframework.test.util.ReflectionTestUtils.setField;

public class TestVserverManager extends TestBase {
    private ObjectFactory OBJECT_FACTORY = new ObjectFactory();
    private ArgumentCaptor<Vserver> payload = ArgumentCaptor.forClass(Vserver.class);

    @Mock
    private AAIRestApiProvider aaiRestApiProvider;
    private VserverManager vserverManager;

    @Before
    public void init() {
        vserverManager = new VserverManager(aaiRestApiProvider, cbamRestApiProvider, driverProperties);
        setField(VserverManager.class, "logger", logger);
    }

    /**
     * test update
     */
    @Test
    public void testUpdate() throws Exception {
        AffectedVnfc affectedVnfc = new AffectedVnfc();
        affectedVnfc.setComputeResource(new ResourceHandle());
        affectedVnfc.getComputeResource().setResourceId("serverProviderId");
        JsonObject additionalData = new JsonObject();
        additionalData.addProperty("name", "serverName");
        additionalData.addProperty("tenantId", "myTenantId");
        JsonArray links = new JsonArray();
        links.add(new JsonObject());
        JsonObject nonRelSelfink = new JsonObject();
        nonRelSelfink.addProperty("rel", "self2");
        nonRelSelfink.addProperty("href", "url");
        links.add(nonRelSelfink);

        JsonObject link = new JsonObject();
        link.addProperty("rel", "self");
        link.addProperty("href", "url");
        links.add(link);
        additionalData.add("links", links);
        affectedVnfc.getComputeResource().setAdditionalData(additionalData);
        affectedVnfc.setId("vnfcId");
        List<AffectedVirtualStorage> affectedStorages = new ArrayList<>();
        affectedStorages.add(new AffectedVirtualStorage());
        affectedStorages.get(0).setId("sId");
        affectedStorages.get(0).setResource(new ResourceHandle());
        affectedStorages.get(0).getResource().setResourceId("storageProviderId");
        affectedVnfc.setStorageResourceIds(new ArrayList<>());
        affectedVnfc.getStorageResourceIds().add("sId");

        String url = "/cloud-regions/cloud-region/myCloudOwnerId/myRegionName/tenants/tenant/myTenantId/vservers/vserver/serverProviderId";

        Vserver existingVserver = OBJECT_FACTORY.createVserver();
        existingVserver.setVolumes(new Volumes());
        when(aaiRestApiProvider.get(eq(logger), eq(CLOUD), eq(url), eq(Vserver.class))).thenReturn(existingVserver);
        when(aaiRestApiProvider.put(eq(logger), eq(CLOUD), eq(url), payload.capture(), eq(Void.class))).thenReturn(null);
        //when
        vserverManager.update(VIM_ID, VNF_ID, affectedVnfc, affectedStorages, true);
        //verify
        Vserver vserver = payload.getValue();
        assertEquals("serverProviderId", vserver.getVserverId());
        assertEquals("active", vserver.getProvStatus());
        assertEquals("serverName", vserver.getVserverName());
        assertEquals("url", vserver.getVserverSelflink());
        assertEquals(1, vserver.getVolumes().getVolume().size());
        assertEquals("storageProviderId", vserver.getVolumes().getVolume().get(0).getVolumeId());
    }

    /**
     * test missing links
     */
    @Test
    public void testUpdateEmptyLinks() throws Exception {
        AffectedVnfc affectedVnfc = new AffectedVnfc();
        affectedVnfc.setComputeResource(new ResourceHandle());
        affectedVnfc.getComputeResource().setResourceId("serverProviderId");
        JsonObject additionalData = new JsonObject();
        additionalData.addProperty("name", "serverName");
        additionalData.addProperty("tenantId", "myTenantId");
        JsonArray links = new JsonArray();
        additionalData.add("links", links);
        affectedVnfc.getComputeResource().setAdditionalData(additionalData);
        affectedVnfc.setId("vnfcId");
        List<AffectedVirtualStorage> affectedStorages = new ArrayList<>();
        affectedStorages.add(new AffectedVirtualStorage());
        affectedStorages.get(0).setId("sId");
        affectedStorages.get(0).setResource(new ResourceHandle());
        affectedStorages.get(0).getResource().setResourceId("storageProviderId");
        affectedVnfc.setStorageResourceIds(new ArrayList<>());
        affectedVnfc.getStorageResourceIds().add("sId");

        String url = "/cloud-regions/cloud-region/myCloudOwnerId/myRegionName/tenants/tenant/myTenantId/vservers/vserver/serverProviderId";
        when(aaiRestApiProvider.get(eq(logger), eq(CLOUD), eq(url), eq(Vserver.class))).thenThrow(new NoSuchElementException());
        when(aaiRestApiProvider.put(eq(logger), eq(CLOUD), eq(url), payload.capture(), eq(Void.class))).thenReturn(null);
        //when
        vserverManager.update(VIM_ID, VNF_ID, affectedVnfc, affectedStorages, true);
        //verify
        Vserver vserver = payload.getValue();
        assertEquals("serverProviderId", vserver.getVserverId());
        assertEquals("active", vserver.getProvStatus());
        assertEquals("serverName", vserver.getVserverName());
        assertEquals("unknown", vserver.getVserverSelflink());
        assertEquals(1, vserver.getVolumes().getVolume().size());
        assertEquals("storageProviderId", vserver.getVolumes().getVolume().get(0).getVolumeId());
    }

    /**
     * test update when links is not present on vServer
     */
    @Test
    public void testUpdateWithNoLinks() throws Exception {
        AffectedVnfc affectedVnfc = new AffectedVnfc();
        affectedVnfc.setComputeResource(new ResourceHandle());
        affectedVnfc.getComputeResource().setResourceId("serverProviderId");
        JsonObject additionalData = new JsonObject();
        additionalData.addProperty("name", "serverName");
        additionalData.addProperty("tenantId", "myTenantId");
        affectedVnfc.getComputeResource().setAdditionalData(additionalData);
        affectedVnfc.setId("vnfcId");
        List<AffectedVirtualStorage> affectedStorages = new ArrayList<>();
        affectedStorages.add(new AffectedVirtualStorage());
        affectedStorages.get(0).setId("sId");
        affectedStorages.get(0).setResource(new ResourceHandle());
        affectedStorages.get(0).getResource().setResourceId("storageProviderId");
        affectedVnfc.setStorageResourceIds(new ArrayList<>());
        affectedVnfc.getStorageResourceIds().add("sId");

        String url = "/cloud-regions/cloud-region/myCloudOwnerId/myRegionName/tenants/tenant/myTenantId/vservers/vserver/serverProviderId";
        when(aaiRestApiProvider.get(eq(logger), eq(CLOUD), eq(url), eq(Vserver.class))).thenThrow(new NoSuchElementException());
        when(aaiRestApiProvider.put(eq(logger), eq(CLOUD), eq(url), payload.capture(), eq(Void.class))).thenReturn(null);
        //when
        vserverManager.update(VIM_ID, VNF_ID, affectedVnfc, affectedStorages, true);
        //verify
        Vserver vserver = payload.getValue();
        assertEquals("serverProviderId", vserver.getVserverId());
        assertEquals("active", vserver.getProvStatus());
        assertEquals("serverName", vserver.getVserverName());
        assertEquals("unknown", vserver.getVserverSelflink());
        assertEquals(1, vserver.getVolumes().getVolume().size());
        assertEquals("storageProviderId", vserver.getVolumes().getVolume().get(0).getVolumeId());
    }

    /**
     * test update when removing volumes
     */
    @Test
    public void testUpdateWithRemovingVolumes() throws Exception {
        AffectedVnfc affectedVnfc = new AffectedVnfc();
        affectedVnfc.setComputeResource(new ResourceHandle());
        affectedVnfc.getComputeResource().setResourceId("serverProviderId");
        JsonObject additionalData = new JsonObject();
        additionalData.addProperty("name", "serverName");
        additionalData.addProperty("tenantId", "myTenantId");
        affectedVnfc.getComputeResource().setAdditionalData(additionalData);
        affectedVnfc.setId("vnfcId");
        List<AffectedVirtualStorage> affectedStorages = new ArrayList<>();
        String url = "/cloud-regions/cloud-region/myCloudOwnerId/myRegionName/tenants/tenant/myTenantId/vservers/vserver/serverProviderId";
        when(aaiRestApiProvider.get(eq(logger), eq(CLOUD), eq(url), eq(Vserver.class))).thenThrow(new NoSuchElementException());
        when(aaiRestApiProvider.put(eq(logger), eq(CLOUD), eq(url), payload.capture(), eq(Void.class))).thenReturn(null);
        //when
        vserverManager.update(VIM_ID, VNF_ID, affectedVnfc, affectedStorages, true);
        //verify
        Vserver vserver = payload.getValue();
        assertEquals("serverProviderId", vserver.getVserverId());
        assertEquals("active", vserver.getProvStatus());
        assertEquals("serverName", vserver.getVserverName());
        assertEquals("unknown", vserver.getVserverSelflink());
        assertEquals(0, vserver.getVolumes().getVolume().size());
    }

    /**
     * test delete
     */
    @Test
    public void testDelete() throws Exception {
        AffectedVnfc affectedVnfc = new AffectedVnfc();
        affectedVnfc.setComputeResource(new ResourceHandle());
        affectedVnfc.getComputeResource().setResourceId("serverProviderId");
        JsonObject additionalData = new JsonObject();
        additionalData.addProperty("name", "serverName");
        additionalData.addProperty("tenantId", "myTenantId");
        affectedVnfc.getComputeResource().setAdditionalData(additionalData);
        affectedVnfc.setId("vnfcId");
        //when
        vserverManager.delete(VIM_ID, affectedVnfc);
        //verify
        String url = "/cloud-regions/cloud-region/myCloudOwnerId/myRegionName/tenants/tenant/myTenantId/vservers/vserver/serverProviderId";
        aaiRestApiProvider.delete(logger, CLOUD, url);
    }

    @Test
    public void testLinks() {
        Relationship relationship = VserverManager.linkTo(VIM_ID, "myTenantPrivderId", "serverProviderId");
        RelationshipList relationships = new RelationshipList();
        relationships.getRelationship().add(relationship);
        assertRelation(relationships, "vserver",
                buildRelationshipData("cloud-region.cloud-owner", getCloudOwner(VIM_ID)),
                buildRelationshipData("cloud-region.cloud-region-id", getRegionName(VIM_ID)),
                buildRelationshipData("tenant.tenant-id", "myTenantPrivderId"),
                buildRelationshipData("vserver.vserver-id", "serverProviderId"));
    }

    /**
     * test inheritence
     */
    @Test
    public void testInheritence() {
        assertEquals(logger, vserverManager.getLogger());
    }


}