/*
 * Copyright (c) 2015 Cisco Systems, Inc. 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.controller.messagebus.app.impl;

import com.google.common.base.Optional;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
import org.opendaylight.controller.config.yang.messagebus.app.impl.NamespaceToStream;
import org.opendaylight.controller.md.sal.binding.api.BindingService;
import org.opendaylight.controller.md.sal.binding.api.DataBroker;
import org.opendaylight.controller.md.sal.binding.api.MountPoint;
import org.opendaylight.controller.md.sal.binding.api.MountPointService;
import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
import org.opendaylight.controller.md.sal.dom.api.DOMNotificationPublishService;
import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.NotificationsService;
import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeFields;
import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.fields.AvailableCapabilities;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
import org.opendaylight.yangtools.concepts.ListenerRegistration;
import org.opendaylight.yangtools.yang.binding.DataObject;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertTrue;
import static org.mockito.Matchers.any;
import static org.mockito.Matchers.eq;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.verify;
import static org.mockito.Mockito.times;
import static org.mockito.Mockito.notNull;

public class NetconfEventSourceManagerTest {

    NetconfEventSourceManager netconfEventSourceManager;
    ListenerRegistration listenerRegistrationMock;
    DOMMountPointService domMountPointServiceMock;
    MountPointService mountPointServiceMock;
    EventSourceTopology eventSourceTopologyMock;
    AsyncDataChangeEvent asyncDataChangeEventMock;

    @BeforeClass
    public static void initTestClass() throws IllegalAccessException, InstantiationException {
    }

    @Before
    public void setUp() throws Exception {
        DataBroker dataBrokerMock = mock(DataBroker.class);
        DOMNotificationPublishService domNotificationPublishServiceMock = mock(DOMNotificationPublishService.class);
        domMountPointServiceMock = mock(DOMMountPointService.class);
        mountPointServiceMock = mock(MountPointService.class);
        eventSourceTopologyMock = mock(EventSourceTopology.class);
        List<NamespaceToStream> namespaceToStreamList = new ArrayList<>();

        listenerRegistrationMock = mock(ListenerRegistration.class);
        doReturn(listenerRegistrationMock).when(dataBrokerMock).registerDataChangeListener(eq(LogicalDatastoreType.OPERATIONAL), any(InstanceIdentifier.class), any(NetconfEventSourceManager.class), eq(AsyncDataBroker.DataChangeScope.SUBTREE));

        netconfEventSourceManager = new NetconfEventSourceManager(dataBrokerMock, domNotificationPublishServiceMock, domMountPointServiceMock,
                mountPointServiceMock, eventSourceTopologyMock, namespaceToStreamList);
    }

    @Test
    public void constructorTest() {
        assertNotNull("Instance has not been created correctly.", netconfEventSourceManager);
    }

    @Test
    public void onDataChangedTest() {
        AsyncDataChangeEvent asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
        Map<InstanceIdentifier, DataObject> map = new HashMap<>();
        InstanceIdentifier instanceIdentifierMock = mock(InstanceIdentifier.class);
        Node dataObjectMock = mock(Node.class);
        map.put(instanceIdentifierMock, dataObjectMock);
        doReturn(map).when(asyncDataChangeEventMock).getCreatedData();
        doReturn(map).when(asyncDataChangeEventMock).getUpdatedData();
        netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
        verify(dataObjectMock, times(2)).getAugmentation(NetconfNode.class);
    }

    @Test
    public void onDataChangedCreateEventSourceTest() {
        onDataChangedCreateEventSourceTestHelper();
        netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
        verify(eventSourceTopologyMock, times(1)).register(any(Node.class), any(NetconfEventSource.class));
    }

    private void onDataChangedCreateEventSourceTestHelper(){
        asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
        Map<InstanceIdentifier, DataObject> map = new HashMap<>();
        InstanceIdentifier instanceIdentifierMock = mock(InstanceIdentifier.class);
        Node dataObjectMock = mock(Node.class);
        map.put(instanceIdentifierMock, dataObjectMock);
        doReturn(map).when(asyncDataChangeEventMock).getCreatedData();
        doReturn(map).when(asyncDataChangeEventMock).getUpdatedData();

        NetconfNode netconfNodeMock = mock(NetconfNode.class);
        AvailableCapabilities availableCapabilitiesMock = mock(AvailableCapabilities.class);
        doReturn(netconfNodeMock).when(dataObjectMock).getAugmentation(NetconfNode.class);
        doReturn(availableCapabilitiesMock).when(netconfNodeMock).getAvailableCapabilities();
        List<String> availableCapabilityList = new ArrayList<>();
        availableCapabilityList.add("(urn:ietf:params:xml:ns:netconf:notification_availableCapabilityString1");
        doReturn(availableCapabilityList).when(availableCapabilitiesMock).getAvailableCapability();

        doReturn(NetconfNodeFields.ConnectionStatus.Connected).when(netconfNodeMock).getConnectionStatus();

        Optional optionalMock = mock(Optional.class);
        Optional optionalBindingMountMock = mock(Optional.class);
        NodeId nodeId = new NodeId("nodeId1");
        doReturn(nodeId).when(dataObjectMock).getNodeId();
        doReturn(optionalMock).when(domMountPointServiceMock).getMountPoint((YangInstanceIdentifier)notNull());
        doReturn(optionalBindingMountMock).when(mountPointServiceMock).getMountPoint(any(InstanceIdentifier.class));
        doReturn(true).when(optionalMock).isPresent();
        doReturn(true).when(optionalBindingMountMock).isPresent();

        DOMMountPoint domMountPointMock = mock(DOMMountPoint.class);
        MountPoint mountPointMock = mock(MountPoint.class);
        doReturn(domMountPointMock).when(optionalMock).get();
        doReturn(mountPointMock).when(optionalBindingMountMock).get();

        RpcConsumerRegistry rpcConsumerRegistryMock = mock(RpcConsumerRegistry.class);
        Optional<BindingService> onlyOptionalMock = (Optional<BindingService>) mock(Optional.class);
        NotificationsService notificationsServiceMock = mock(NotificationsService.class);

        doReturn(onlyOptionalMock).when(mountPointMock).getService(RpcConsumerRegistry.class);
        doReturn(rpcConsumerRegistryMock).when(onlyOptionalMock).get();
        doReturn(notificationsServiceMock).when(rpcConsumerRegistryMock).getRpcService(NotificationsService.class);
    }

    @Test
    public void isEventSourceTest() {
        Node nodeMock = mock(Node.class);
        NetconfNode netconfNodeMock = mock(NetconfNode.class);
        AvailableCapabilities availableCapabilitiesMock = mock(AvailableCapabilities.class);
        doReturn(netconfNodeMock).when(nodeMock).getAugmentation(NetconfNode.class);
        doReturn(availableCapabilitiesMock).when(netconfNodeMock).getAvailableCapabilities();
        List<String> availableCapabilityList = new ArrayList<>();
        availableCapabilityList.add("(urn:ietf:params:xml:ns:netconf:notification_availableCapabilityString1");
        doReturn(availableCapabilityList).when(availableCapabilitiesMock).getAvailableCapability();
        assertTrue("Method has not been run correctly.", netconfEventSourceManager.isEventSource(nodeMock));
    }

    @Test
    public void isNotEventSourceTest() {
        Node nodeMock = mock(Node.class);
        NetconfNode netconfNodeMock = mock(NetconfNode.class);
        AvailableCapabilities availableCapabilitiesMock = mock(AvailableCapabilities.class);
        doReturn(netconfNodeMock).when(nodeMock).getAugmentation(NetconfNode.class);
        doReturn(availableCapabilitiesMock).when(netconfNodeMock).getAvailableCapabilities();
        List<String> availableCapabilityList = new ArrayList<>();
        availableCapabilityList.add("availableCapabilityString1");
        doReturn(availableCapabilityList).when(availableCapabilitiesMock).getAvailableCapability();
        assertFalse("Method has not been run correctly.", netconfEventSourceManager.isEventSource(nodeMock));
    }

    @Test
    public void closeTest() {
        netconfEventSourceManager.close();
        verify(listenerRegistrationMock, times(1)).close();
    }
}