opendaylight/md-sal/messagebus-impl/src/test/java/org/opendaylight/controller/messagebus/app/impl/NetconfEventSourceTest.java

   1 /*
   2  * Copyright (c) 2015 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.controller.messagebus.app.impl;
   9
  10 import com.google.common.base.Optional;
  11 import com.google.common.util.concurrent.CheckedFuture;
  12 import org.junit.Before;
  13 import org.junit.BeforeClass;
  14 import org.junit.Test;
  15 import org.opendaylight.controller.md.sal.binding.api.BindingService;
  16 import org.opendaylight.controller.md.sal.binding.api.MountPoint;
  17 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  18 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
  19 import org.opendaylight.controller.md.sal.dom.api.DOMNotification;
  20 import org.opendaylight.controller.md.sal.dom.api.DOMNotificationPublishService;
  21 import org.opendaylight.controller.md.sal.dom.api.DOMNotificationService;
  22 import org.opendaylight.controller.md.sal.dom.api.DOMService;
  23 import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
  24 import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
  25 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.NotificationPattern;
  26 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicInput;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.NotificationsService;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.inventory.rev140108.NetconfNode;
  33 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  34 import org.opendaylight.yangtools.yang.binding.DataObject;
  35 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  36 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  37 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
  38 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  39 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  40
  41 import org.opendaylight.yangtools.yang.common.QName;
  42
  43 import java.lang.reflect.Field;
  44 import java.net.URI;
  45 import java.util.HashMap;
  46 import java.util.HashSet;
  47 import java.util.Map;
  48 import java.util.Set;
  49 import java.util.List;
  50
  51 import static org.junit.Assert.assertEquals;
  52 import static org.junit.Assert.assertNotNull;
  53 import static org.mockito.Matchers.any;
  54 import static org.mockito.Mockito.doReturn;
  55 import static org.mockito.Mockito.mock;
  56 import static org.mockito.Mockito.verify;
  57 import static org.mockito.Mockito.times;
  58
  59 public class NetconfEventSourceTest {
  60
  61     NetconfEventSource netconfEventSource;
  62     DOMMountPoint domMountPointMock;
  63     JoinTopicInput joinTopicInputMock;
  64
  65     @BeforeClass
  66     public static void initTestClass() throws IllegalAccessException, InstantiationException {
  67     }
  68
  69     @Before
  70     public void setUp() throws Exception {
  71         Map<String, String> streamMap = new HashMap<>();
  72         streamMap.put("string1", "string2");
  73         domMountPointMock = mock(DOMMountPoint.class);
  74         DOMNotificationPublishService domNotificationPublishServiceMock = mock(DOMNotificationPublishService.class);
  75         MountPoint mountPointMock = mock(MountPoint.class);
  76
  77         RpcConsumerRegistry rpcConsumerRegistryMock = mock(RpcConsumerRegistry.class);
  78         Optional<BindingService> onlyOptionalMock = (Optional<BindingService>) mock(Optional.class);
  79         NotificationsService notificationsServiceMock = mock(NotificationsService.class);
  80
  81         doReturn(onlyOptionalMock).when(mountPointMock).getService(RpcConsumerRegistry.class);
  82         doReturn(rpcConsumerRegistryMock).when(onlyOptionalMock).get();
  83         doReturn(notificationsServiceMock).when(rpcConsumerRegistryMock).getRpcService(NotificationsService.class);
  84         netconfEventSource = new NetconfEventSource("nodeId1", streamMap, domMountPointMock, domNotificationPublishServiceMock, mountPointMock);
  85     }
  86
  87     @Test
  88     public void constructorTest() {
  89         assertNotNull("Instance has not been created correctly.", netconfEventSource);
  90     }
  91
  92     @Test
  93     public void joinTopicTest() throws Exception{
  94         joinTopicTestHelper();
  95         assertNotNull("JoinTopic return value has not been created correctly.", netconfEventSource.joinTopic(joinTopicInputMock));
  96     }
  97
  98     private void joinTopicTestHelper() throws Exception{
  99         joinTopicInputMock = mock(JoinTopicInput.class);
 100         NotificationPattern notificationPatternMock = mock(NotificationPattern.class);
 101         doReturn(notificationPatternMock).when(joinTopicInputMock).getNotificationPattern();
 102         doReturn("regexString1").when(notificationPatternMock).getValue();
 103
 104         SchemaContext schemaContextMock = mock(SchemaContext.class);
 105         doReturn(schemaContextMock).when(domMountPointMock).getSchemaContext();
 106         Set<NotificationDefinition> notificationDefinitionSet = new HashSet<>();
 107         NotificationDefinition notificationDefinitionMock = mock(NotificationDefinition.class);
 108         notificationDefinitionSet.add(notificationDefinitionMock);
 109
 110         URI uri = new URI("uriStr1");
 111         QName qName = new QName(uri, "localName1");
 112         org.opendaylight.yangtools.yang.model.api.SchemaPath schemaPath = SchemaPath.create(true, qName);
 113         doReturn(notificationDefinitionSet).when(schemaContextMock).getNotifications();
 114         doReturn(schemaPath).when(notificationDefinitionMock).getPath();
 115
 116         Optional<DOMNotificationService> domNotificationServiceOptionalMock = (Optional<DOMNotificationService>) mock(Optional.class);
 117         doReturn(domNotificationServiceOptionalMock).when(domMountPointMock).getService(DOMNotificationService.class);
 118         doReturn(true).when(domNotificationServiceOptionalMock).isPresent();
 119
 120         DOMNotificationService domNotificationServiceMock = mock(DOMNotificationService.class);
 121         doReturn(domNotificationServiceMock).when(domNotificationServiceOptionalMock).get();
 122         ListenerRegistration listenerRegistrationMock = mock(ListenerRegistration.class);
 123         doReturn(listenerRegistrationMock).when(domNotificationServiceMock).registerNotificationListener(any(NetconfEventSource.class), any(List.class));
 124     }
 125
 126     @Test (expected=NullPointerException.class)
 127     public void onNotificationTest() {
 128         DOMNotification domNotificationMock = mock(DOMNotification.class);
 129         ContainerNode containerNodeMock = mock(ContainerNode.class);
 130         SchemaContext schemaContextMock = mock(SchemaContext.class);
 131         SchemaPath schemaPathMock = mock(SchemaPath.class);
 132         doReturn(schemaContextMock).when(domMountPointMock).getSchemaContext();
 133         doReturn(schemaPathMock).when(domNotificationMock).getType();
 134         doReturn(containerNodeMock).when(domNotificationMock).getBody();
 135         netconfEventSource.onNotification(domNotificationMock);
 136     }
 137
 138     @Test
 139     public void onDataChangedTest() {
 140         InstanceIdentifier brmIdent = InstanceIdentifier.create(Nodes.class)
 141                 .child(Node.class, new NodeKey(new NodeId("brm"))).augmentation(NetconfNode.class);
 142         AsyncDataChangeEvent asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
 143         NetconfNode dataObjectMock = mock(NetconfNode.class);
 144         Map<InstanceIdentifier, DataObject> dataChangeMap = new HashMap<>();
 145         dataChangeMap.put(brmIdent, dataObjectMock);
 146         doReturn(dataChangeMap).when(asyncDataChangeEventMock).getOriginalData();
 147         doReturn(dataChangeMap).when(asyncDataChangeEventMock).getUpdatedData();
 148
 149         netconfEventSource.onDataChanged(asyncDataChangeEventMock);
 150         verify(dataObjectMock, times(2)).isConnected();
 151     }
 152
 153     @Test
 154     public void onDataChangedResubscribeTest() throws Exception{
 155         InstanceIdentifier brmIdent = InstanceIdentifier.create(Nodes.class)
 156                 .child(Node.class, new NodeKey(new NodeId("brm"))).augmentation(NetconfNode.class);
 157         AsyncDataChangeEvent asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
 158         NetconfNode dataObjectMock = mock(NetconfNode.class);
 159         Map<InstanceIdentifier, DataObject> dataChangeMap = new HashMap<>();
 160         dataChangeMap.put(brmIdent, dataObjectMock);
 161         doReturn(dataChangeMap).when(asyncDataChangeEventMock).getUpdatedData();
 162         doReturn(true).when(dataObjectMock).isConnected();
 163
 164         Set<String> localSet = getActiveStreams();
 165         localSet.add("activeStream1");
 166
 167         Optional<DOMService> optionalMock = (Optional<DOMService>) mock(Optional.class);
 168         doReturn(optionalMock).when(domMountPointMock).getService(DOMRpcService.class);
 169         DOMRpcService domRpcServiceMock = mock(DOMRpcService.class);
 170         doReturn(domRpcServiceMock).when(optionalMock).get();
 171         CheckedFuture checkedFutureMock = mock(CheckedFuture.class);
 172         doReturn(checkedFutureMock).when(domRpcServiceMock).invokeRpc(any(SchemaPath.class), any(ContainerNode.class));
 173
 174         netconfEventSource.onDataChanged(asyncDataChangeEventMock);
 175         verify(dataObjectMock, times(1)).isConnected();
 176         assertEquals("Size of set has not been set correctly.", 1, getActiveStreams().size());
 177     }
 178
 179     private Set getActiveStreams() throws Exception{
 180         Field nesField = NetconfEventSource.class.getDeclaredField("activeStreams");
 181         nesField.setAccessible(true);
 182         return (Set) nesField.get(netconfEventSource);
 183     }
 184
 185 }