/*
 * 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.nokia.cbam.lcm.v32.model.VnfInfo;
import io.reactivex.Observable;
import java.util.HashSet;
import java.util.List;
import java.util.NoSuchElementException;
import java.util.Set;
import java.util.concurrent.atomic.AtomicLong;
import org.junit.Before;
import org.junit.Test;
import org.mockito.ArgumentCaptor;
import org.mockito.Mock;
import org.mockito.Mockito;
import org.mockito.invocation.InvocationOnMock;
import org.mockito.stubbing.Answer;
import org.onap.aai.api.NetworkApi;
import org.onap.aai.model.GenericVnf;
import org.onap.aai.model.Relationship;
import org.onap.aai.model.RelationshipData;
import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.core.SelfRegistrationManager;
import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.onap.direct.AAIRestApiProvider;
import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.CbamRestApiProvider;
import org.onap.vfc.nfvo.driver.vnfm.svnfm.nokia.vnfm.TestBase;

import static java.lang.Boolean.TRUE;
import static java.util.Optional.of;

import static junit.framework.TestCase.assertEquals;
import static junit.framework.TestCase.fail;
import static org.mockito.Mockito.*;
import static org.springframework.test.util.ReflectionTestUtils.setField;

public class TestGenericVnfManager extends TestBase {
    private ArgumentCaptor<GenericVnf> payload = ArgumentCaptor.forClass(GenericVnf.class);


    @Mock
    private AAIRestApiProvider aaiRestApiProvider;
    @Mock
    private NetworkApi networkApi;
    private GenericVnfManager genericVnfManager;
    private VnfInfo vnfInfo = new VnfInfo();

    static void assertRelation(List<Relationship> relationShips, String relatedTo, RelationshipData... data) {
        for (Relationship relationship : relationShips) {
            if (relationship.getRelatedTo().equals(relatedTo)) {
                assertEquals(data.length, relationship.getRelationshipData().size());
                int i = 0;
                for (RelationshipData c : data) {
                    assertEquals(c.getRelationshipKey(), relationship.getRelationshipData().get(i).getRelationshipKey());
                    assertEquals(c.getRelationshipValue(), relationship.getRelationshipData().get(i).getRelationshipValue());
                    i++;
                }
                return;
            }
        }
        fail();
    }

    @Before
    public void init() {
        when(aaiRestApiProvider.getNetworkApi()).thenReturn(networkApi);
        genericVnfManager = new GenericVnfManager(aaiRestApiProvider, cbamRestApiProviderForSo);
        setField(GenericVnfManager.class, "logger", logger);
        AtomicLong currentTime = new AtomicLong(0L);
        when(systemFunctions.currentTimeMillis()).thenAnswer(new Answer<Long>() {
            @Override
            public Long answer(InvocationOnMock invocation) throws Throwable {
                return currentTime.get();
            }
        });
        Mockito.doAnswer(new Answer() {
            @Override
            public Object answer(InvocationOnMock invocation) throws Throwable {
                currentTime.addAndGet((Long) invocation.getArguments()[0] + 1);
                return null;
            }
        }).when(systemFunctions).sleep(anyLong());
    }

    /**
     * if the VNF does not exist it is created
     */
    @Test
    public void createNonExistingVnf() throws Exception {
        GenericVnf vnfInAaai = new GenericVnf();
        Set<GenericVnf> vnfs = new HashSet<>();
        when(networkApi.getNetworkGenericVnfsGenericVnf(VNF_ID, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null)).thenAnswer((Answer<Observable<GenericVnf>>) invocation -> {
            if (vnfs.size() == 0) {
                throw new NoSuchElementException();
            }
            return buildObservable(vnfs.iterator().next());
        });
        when(cbamRestApiProviderForSo.getCbamLcmApi(VNFM_ID).vnfsVnfInstanceIdGet(VNF_ID, CbamRestApiProvider.NOKIA_LCM_API_VERSION)).thenReturn(buildObservable(vnfInfo));
        when(networkApi.createOrUpdateNetworkGenericVnfsGenericVnf(eq(VNF_ID), payload.capture())).thenAnswer(invocation -> {
            vnfs.add(vnfInAaai);
            return VOID_OBSERVABLE.value();
        });
        vnfInfo.setName("vnfName");
        //when
        genericVnfManager.createOrUpdate(VNF_ID, true, VNFM_ID, of("nsId"));
        //verify
        GenericVnf vnfSentToAai = payload.getValue();
        assertEquals(VNF_ID, vnfSentToAai.getVnfId());
        assertEquals(VNF_ID, vnfSentToAai.getVnfId());
        assertEquals("NokiaVNF", vnfSentToAai.getVnfType());
        assertEquals(SelfRegistrationManager.SERVICE_NAME, vnfSentToAai.getNfType());
        assertEquals(TRUE, vnfSentToAai.isInMaint());
        assertEquals(TRUE, vnfSentToAai.isIsClosedLoopDisabled());
        assertEquals("vnfName", vnfSentToAai.getVnfName());
        verify(systemFunctions, times(10)).sleep(3000);
        verify(networkApi, times(10)).getNetworkGenericVnfsGenericVnf(VNF_ID, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null);
        VOID_OBSERVABLE.assertCalled();
    }

    /**
     * if the VNF exist it is updated
     */
    @Test
    public void testUpdateExistingVnf() throws Exception {
        GenericVnf vnfInAaai = new GenericVnf();
        vnfInAaai.setResourceVersion("v1");
        when(networkApi.getNetworkGenericVnfsGenericVnf(VNF_ID, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null)).thenReturn(buildObservable(vnfInAaai));
        when(cbamRestApiProviderForSo.getCbamLcmApi(VNFM_ID).vnfsVnfInstanceIdGet(VNF_ID, CbamRestApiProvider.NOKIA_LCM_API_VERSION)).thenReturn(buildObservable(vnfInfo));
        when(networkApi.createOrUpdateNetworkGenericVnfsGenericVnf(eq(VNF_ID), payload.capture())).thenReturn(VOID_OBSERVABLE.value());
        vnfInfo.setName("vnfName");
        //when
        genericVnfManager.createOrUpdate(VNF_ID, true, VNFM_ID, of("nsId"));
        //verify
        GenericVnf vnfSentToAai = payload.getValue();
        assertEquals(VNF_ID, vnfSentToAai.getVnfId());
        assertEquals(VNF_ID, vnfSentToAai.getVnfId());
        assertEquals("NokiaVNF", vnfSentToAai.getVnfType());
        assertEquals(TRUE, vnfSentToAai.isInMaint());
        assertEquals(TRUE, vnfSentToAai.isIsClosedLoopDisabled());
        assertEquals("vnfName", vnfSentToAai.getVnfName());
        verify(systemFunctions, never()).sleep(anyLong());
        VOID_OBSERVABLE.assertCalled();
        verify(networkApi, times(1)).getNetworkGenericVnfsGenericVnf(VNF_ID, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null);
    }

    /**
     * error is propagated if unable to query VNF from CBAM
     */
    @Test
    public void testUnableToQueryVnfFromCBAM() throws Exception {
        GenericVnf vnfInAaai = new GenericVnf();
        vnfInAaai.setResourceVersion("v1");
        when(networkApi.getNetworkGenericVnfsGenericVnf(VNF_ID, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null)).thenReturn(buildObservable(vnfInAaai));
        RuntimeException expectedException = new RuntimeException();
        when(cbamRestApiProviderForSo.getCbamLcmApi(VNFM_ID).vnfsVnfInstanceIdGet(VNF_ID, CbamRestApiProvider.NOKIA_LCM_API_VERSION)).thenThrow(expectedException);
        when(networkApi.createOrUpdateNetworkGenericVnfsGenericVnf(eq(VNF_ID), payload.capture())).thenAnswer(invocation -> {
            vnfInAaai.setResourceVersion("v2");
            return null;
        });
        vnfInfo.setName("vnfName");
        //when
        try {
            genericVnfManager.createOrUpdate(VNF_ID, true, VNFM_ID, of("nsId"));
        } catch (Exception e) {
            verify(logger).error("Unable to query VNF with myVnfId identifier from CBAM", expectedException);
            assertEquals("Unable to query VNF with myVnfId identifier from CBAM", e.getMessage());
        }
    }

    /**
     * if the VNF is created after the last attempt to query VNF, but before the
     * the driver creates the VNF it is not created but updated
     */
    @Test
    public void testConcurency1() throws Exception {
        GenericVnf vnfInAaai = new GenericVnf();
        vnfInAaai.setResourceVersion("v3");
        Set<Integer> queryCount = new HashSet<>();
        when(networkApi.getNetworkGenericVnfsGenericVnf(VNF_ID, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null)).thenAnswer((Answer<Observable>) invocationOnMock -> {
            queryCount.add(queryCount.size());
            if (queryCount.size() >= 11) {
                return buildObservable(vnfInAaai);
            }
            throw new NoSuchElementException();
        });
        when(cbamRestApiProviderForSo.getCbamLcmApi(VNFM_ID).vnfsVnfInstanceIdGet(VNF_ID, CbamRestApiProvider.NOKIA_LCM_API_VERSION)).thenReturn(buildObservable(vnfInfo));
        RuntimeException runtimeException = new RuntimeException();
        when(networkApi.createOrUpdateNetworkGenericVnfsGenericVnf(eq(VNF_ID), payload.capture())).thenAnswer(invocation -> {
            GenericVnf vnfSentToAAi = (GenericVnf) invocation.getArguments()[1];
            if (vnfSentToAAi.getResourceVersion() == null) {
                throw runtimeException;
            }
            return VOID_OBSERVABLE.value();
        });
        vnfInfo.setName("vnfName");
        //when
        genericVnfManager.createOrUpdate(VNF_ID, true, VNFM_ID, of("nsId"));
        //verify
        GenericVnf vnfSentToAai = payload.getValue();
        assertEquals(VNF_ID, vnfSentToAai.getVnfId());
        assertEquals(VNF_ID, vnfSentToAai.getVnfId());
        assertEquals("NokiaVNF", vnfSentToAai.getVnfType());
        assertEquals(TRUE, vnfSentToAai.isInMaint());
        assertEquals(TRUE, vnfSentToAai.isIsClosedLoopDisabled());
        assertEquals("vnfName", vnfSentToAai.getVnfName());
        assertEquals("v3", vnfSentToAai.getResourceVersion());
        verify(systemFunctions, times(10)).sleep(3000);
        verify(networkApi, times(11)).getNetworkGenericVnfsGenericVnf(VNF_ID, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null, null);
        verify(networkApi, times(2)).createOrUpdateNetworkGenericVnfsGenericVnf(eq(VNF_ID), any());
        verify(logger).warn(eq("The VNF with myVnfId identifier did not appear in time"), any(NoSuchElementException.class));
        verify(logger).warn("The VNF with myVnfId identifier has been created since after the maximal wait for VNF to appear timeout", runtimeException);
        VOID_OBSERVABLE.assertCalled();
    }

    /**
     * test how entities can refer to a VNF
     */
    @Test
    public void testRelations() {
        //when
        Relationship relationship = GenericVnfManager.linkTo(VNF_ID);
        //verify
        assertEquals("generic-vnf", relationship.getRelatedTo());
        assertEquals(1, relationship.getRelationshipData().size());
        assertEquals("generic-vnf.vnf-id", relationship.getRelationshipData().get(0).getRelationshipKey());
        assertEquals(VNF_ID, relationship.getRelationshipData().get(0).getRelationshipValue());
    }

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