/*
 * Copyright (c) 2016, 2017 Ericsson India Global Services Pvt Ltd. and others.  All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */

package org.opendaylight.ovsdb.hwvtepsouthbound;

import com.google.common.collect.Lists;
import com.google.common.util.concurrent.MoreExecutors;
import org.apache.commons.lang3.reflect.FieldUtils;
import org.junit.After;
import org.junit.Before;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.mockito.Matchers;
import org.opendaylight.ovsdb.hwvtepsouthbound.transact.DependencyQueue;
import org.opendaylight.ovsdb.lib.operations.Operations;
import org.opendaylight.ovsdb.lib.schema.DatabaseSchema;
import org.opendaylight.ovsdb.lib.schema.typed.TypedBaseTable;
import org.opendaylight.ovsdb.schema.hardwarevtep.LogicalSwitch;
import org.opendaylight.ovsdb.schema.hardwarevtep.McastMacsRemote;
import org.opendaylight.ovsdb.schema.hardwarevtep.UcastMacsRemote;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.LogicalSwitches;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteMcastMacs;
import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.ovsdb.hwvtep.rev150901.hwvtep.global.attributes.RemoteUcastMacs;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.node.TerminationPoint;
import org.opendaylight.yangtools.yang.binding.DataObject;
import org.powermock.core.classloader.annotations.PrepareForTest;
import org.powermock.modules.junit4.PowerMockRunner;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.Field;
import java.lang.reflect.Modifier;
import java.util.Iterator;
import java.util.List;

import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.verify;
import static org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType.CONFIGURATION;
import static org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType.OPERATIONAL;

/**
 * Unit tests for the data-tree change listener.
 */
@RunWith(PowerMockRunner.class)
@PrepareForTest({HwvtepConnectionInstance.class, HwvtepConnectionManager.class, Operations.class})
public class HwvtepDataChangeListenerTest extends DataChangeListenerTestBase {

    static Logger LOG = LoggerFactory.getLogger(HwvtepDataChangeListenerTest.class);

    String[][] ucastMacs = new String[][]{
            {"20:00:00:00:00:01", "11.10.10.1", "192.168.122.20", "ls0"},
            {"20:00:00:00:00:02", "11.10.10.2", "192.168.122.20", "ls0"},
            {"20:00:00:00:00:03", "11.10.10.3", "192.168.122.30", "ls1"},
            {"20:00:00:00:00:04", "11.10.10.4", "192.168.122.30", "ls1"}
    };

    String[][] logicalSwitches = new String[][]{
            {"ls0", "100"},
            {"ls1", "200"},
    };

    String[][] terminationPoints = new String[][]{
            {"192.168.122.10"},
            {"192.168.122.20"},
            {"192.168.122.30"},
            {"192.168.122.40"}
    };

    String[][] mcastMacs = new String[][]{
            {"FF:FF:FF:FF:FF:FF", "ls0", "192.168.122.20", "192.168.122.30"},
            {"FF:FF:FF:FF:FF:FF", "ls1", "192.168.122.10", "192.168.122.30"}
    };

    String[][] mcastMac2 = new String[][]{
            {"FF:FF:FF:FF:FF:FF", "ls0", "192.168.122.20", "192.168.122.10"},
            {"FF:FF:FF:FF:FF:FF", "ls1", "192.168.122.10", "192.168.122.20"}
    };

    String[][] mcastMac3WithZeroLocators = new String[][]{
            {"FF:FF:FF:FF:FF:FF", "ls0"},
            {"FF:FF:FF:FF:FF:FF", "ls1"}
    };

    HwvtepOperationalDataChangeListener opDataChangeListener;

    @Before
    public void setupListener() throws Exception {
        setFinalStatic(DependencyQueue.class, "executorService", MoreExecutors.newDirectExecutorService());
        opDataChangeListener = new HwvtepOperationalDataChangeListener(dataBroker, hwvtepConnectionManager, connectionInstance);
    }

    @After
    public void cleanupListener() {
        try {
            opDataChangeListener.close();
        } catch (Exception e) {
        }
    }

    void setFinalStatic(Class cls, String fieldName, Object newValue) throws Exception {
        Field fields[] = FieldUtils.getAllFields(cls);
        for (Field field : fields) {
            if (fieldName.equals(field.getName())) {
                field.setAccessible(true);
                Field modifiersField = Field.class.getDeclaredField("modifiers");
                modifiersField.setAccessible(true);
                modifiersField.setInt(field, field.getModifiers() & ~Modifier.FINAL);
                field.set(null, newValue);
                break;
            }
        }
    }

    @Test
    public <T extends DataObject> void testLogicalSwitchAdd() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        verifyThatLogicalSwitchCreated();
    }

    @Test
    public <T extends DataObject> void testLogicalSwitchDelete() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        deleteData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        verify(Operations.op,  times(2)).delete(any());
    }

    @Test
    public <T extends DataObject> void testUcastMacAdd() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        //4 ucast macs + 2 termination points
        verify(Operations.op,  times(6)).insert(any(UcastMacsRemote.class));
        //TODO add finer grained validation
    }

    @Test
    public <T extends DataObject> void testUcastMacAddWithoutConfigTep() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        //4 ucast macs + 2 termination points
        verify(Operations.op,  times(6)).insert(any(UcastMacsRemote.class));
        //TODO add finer grained validation
    }

    @Test
    public <T extends DataObject> void testUcastMacDelete() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        addData(OPERATIONAL, RemoteUcastMacs.class, ucastMacs);
        addData(OPERATIONAL, TerminationPoint.class, terminationPoints);

        resetOperations();
        deleteData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        verify(Operations.op,  times(4)).delete(any());
        //TODO add finer grained validation
    }

    @Test
    public <T extends DataObject> void testMcastMacAdd() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        //2 mcast macs + 2 locator sets + 3 termination points
        verify(Operations.op,  times(7)).insert(Matchers.<McastMacsRemote>any());
    }

    @Test
    public <T extends DataObject> void testMcastMacAddWithoutConfigTep() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        //2 mcast macs + 2 locator sets + 3 termination points
        verify(Operations.op,  times(7)).insert(Matchers.<McastMacsRemote>any());
    }

    @Test
    public <T extends DataObject> void testMcastMacDelete() throws Exception {
        addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        addData(OPERATIONAL, TerminationPoint.class, terminationPoints);
        addData(OPERATIONAL, RemoteMcastMacs.class, mcastMacs);

        resetOperations();
        deleteData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        verify(Operations.op,  times(2)).delete(Matchers.any());
    }

    @Test
    public <T extends DataObject> void testAddMacs() throws Exception {
        Node node = addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        verify(Operations.op,  times(6)).insert(any(UcastMacsRemote.class));

        addData(OPERATIONAL, TerminationPoint.class, terminationPoints);
        addData(OPERATIONAL, RemoteUcastMacs.class, ucastMacs);
        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        //2 mcast mac + 2 locator sets ( termination point already added )
        verify(Operations.op,  times(4)).insert(Matchers.<McastMacsRemote>any());
    }

    @Test
    public <T extends DataObject> void testUpdateMacs() throws Exception {
        Node node = addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        verify(Operations.op,  times(6)).insert(any(UcastMacsRemote.class));

        addData(OPERATIONAL, TerminationPoint.class, terminationPoints);
        addData(OPERATIONAL, RemoteUcastMacs.class, ucastMacs);
        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        verify(Operations.op,  times(4)).insert(Matchers.<McastMacsRemote>any());
        addData(OPERATIONAL, RemoteMcastMacs.class, mcastMacs);

        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMac2);
        verify(Operations.op,  times(2)).insert(Matchers.<McastMacsRemote>any());
        verify(Operations.op,  times(2)).update(Matchers.<McastMacsRemote>any());
        verify(Operations.op,  times(0)).delete(Matchers.any());
    }

    @Test
    public <T extends DataObject> void testUpdateMacsWithZeroLocators() throws Exception {
        Node node = addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        verify(Operations.op,  times(6)).insert(any(UcastMacsRemote.class));

        addData(OPERATIONAL, TerminationPoint.class, terminationPoints);
        addData(OPERATIONAL, RemoteUcastMacs.class, ucastMacs);
        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        verify(Operations.op,  times(4)).insert(Matchers.<McastMacsRemote>any());
        addData(OPERATIONAL, RemoteMcastMacs.class, mcastMacs);

        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMac3WithZeroLocators);
        verify(Operations.op,  times(2)).delete(Matchers.any());
    }

    @Test
    public <T extends DataObject> void testBackToBackMacsUpdate() throws Exception {
        Node node = addData(CONFIGURATION, LogicalSwitches.class, logicalSwitches);
        addData(OPERATIONAL, LogicalSwitches.class, logicalSwitches);
        resetOperations();
        addData(CONFIGURATION, TerminationPoint.class, terminationPoints);
        addData(CONFIGURATION, RemoteUcastMacs.class, ucastMacs);
        verify(Operations.op,  times(6)).insert(any(UcastMacsRemote.class));

        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMacs);
        //2 mcast mac + 2 locator sets ( termination point already added )
        verify(Operations.op,  times(0)).insert(Matchers.<McastMacsRemote>any());
        resetOperations();
        addData(OPERATIONAL, TerminationPoint.class, terminationPoints);
        addData(OPERATIONAL, RemoteUcastMacs.class, ucastMacs);
        connectionInstance.getDeviceInfo().onOperDataAvailable();
        //2 mcast mac + 2 locator sets ( termination point already added )
        verify(Operations.op,  times(4)).insert(Matchers.<McastMacsRemote>any());

        resetOperations();
        addData(CONFIGURATION, RemoteMcastMacs.class, mcastMac2);
        verify(Operations.op,  times(0)).insert(Matchers.<McastMacsRemote>any());
        addData(OPERATIONAL, RemoteMcastMacs.class, mcastMacs);
        connectionInstance.getDeviceInfo().onOperDataAvailable();
        verify(Operations.op,  times(2)).insert(Matchers.<McastMacsRemote>any());
        verify(Operations.op,  times(2)).update(Matchers.<McastMacsRemote>any());
    }

    private void verifyThatLogicalSwitchCreated() {
        verify(ovsdbClient, times(1)).transact(any(DatabaseSchema.class), any(List.class));
        verify(Operations.op, times(2)).insert(any(LogicalSwitch.class));

        assertNotNull(insertOpCapture.getAllValues());
        assertTrue(insertOpCapture.getAllValues().size() == 2);

        List<String> expected = Lists.newArrayList("ls0", "ls1");
        Iterator<TypedBaseTable> it = insertOpCapture.getAllValues().iterator();
        while (it.hasNext()) {
            TypedBaseTable table = it.next();
            assertTrue(table instanceof LogicalSwitch);
            LogicalSwitch ls = (LogicalSwitch)table;
            assertTrue(expected.contains(ls.getName()));
            expected.remove(ls.getName());
            it.next();
        }
    }
}