netconf/messagebus-netconf/src/test/java/org/opendaylight/netconf/messagebus/eventsources/netconf/NetconfEventSourceTest.java

   1 /*
   2  * Copyright (c) 2016 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.netconf.messagebus.eventsources.netconf;
   9
  10 import static org.junit.Assert.assertNotNull;
  11 import static org.mockito.Matchers.any;
  12 import static org.mockito.Mockito.doReturn;
  13 import static org.mockito.Mockito.mock;
  14 import static org.mockito.Mockito.verify;
  15
  16 import com.google.common.base.Optional;
  17 import com.google.common.util.concurrent.CheckedFuture;
  18 import java.net.URI;
  19 import java.util.Collections;
  20 import java.util.HashMap;
  21 import java.util.HashSet;
  22 import java.util.Map;
  23 import java.util.Set;
  24 import org.junit.Assert;
  25 import org.junit.Before;
  26 import org.junit.Test;
  27 import org.mockito.ArgumentCaptor;
  28 import org.opendaylight.controller.md.sal.binding.api.BindingService;
  29 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
  30 import org.opendaylight.controller.md.sal.binding.api.MountPoint;
  31 import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
  32 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
  33 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
  34 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
  35 import org.opendaylight.controller.md.sal.dom.api.DOMNotification;
  36 import org.opendaylight.controller.md.sal.dom.api.DOMNotificationPublishService;
  37 import org.opendaylight.controller.md.sal.dom.api.DOMNotificationService;
  38 import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
  39 import org.opendaylight.controller.md.sal.dom.api.DOMService;
  40 import org.opendaylight.controller.messagebus.app.util.TopicDOMNotification;
  41 import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
  42 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.NotificationPattern;
  43 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.TopicId;
  44 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicInput;
  45 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicInputBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.NotificationsService;
  47 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.Netconf;
  48 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.netconf.Streams;
  49 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeConnectionStatus.ConnectionStatus;
  50 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  51 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  52 import org.opendaylight.yangtools.yang.common.QName;
  53 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  54 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  55 import org.opendaylight.yangtools.yang.data.impl.schema.Builders;
  56 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
  57 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  58 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  59
  60 public class NetconfEventSourceTest {
  61
  62     NetconfEventSource netconfEventSource;
  63     DOMMountPoint domMountPointMock;
  64     MountPoint mountPointMock;
  65     JoinTopicInput joinTopicInputMock;
  66     DOMNotificationPublishService domNotificationPublishServiceMock;
  67     DOMNotification notification;
  68
  69     @Before
  70     public void setUp() throws Exception {
  71         Map<String, String> streamMap = new HashMap<>();
  72         streamMap.put("uriStr1", "string2");
  73         domMountPointMock = mock(DOMMountPoint.class);
  74         mountPointMock = mock(MountPoint.class);
  75         domNotificationPublishServiceMock = mock(DOMNotificationPublishService.class);
  76         RpcConsumerRegistry rpcConsumerRegistryMock = mock(RpcConsumerRegistry.class);
  77         Optional<BindingService> onlyOptionalMock = (Optional<BindingService>) mock(Optional.class);
  78         NotificationsService notificationsServiceMock = mock(NotificationsService.class);
  79         doReturn(notificationsServiceMock).when(rpcConsumerRegistryMock).getRpcService(NotificationsService.class);
  80
  81         final NotificationDefinition notificationDefinitionMock = getNotificationDefinitionMock("urn:cisco:params:xml:ns:yang:messagebus:eventsource", "2014-12-02", "event-source-status");
  82         Set<NotificationDefinition> notifications = Collections.singleton(notificationDefinitionMock);
  83         ContainerNode node = Builders.containerBuilder()
  84                 .withNodeIdentifier(YangInstanceIdentifier.NodeIdentifier.create(QName.create("notification-namespace", "2016-02-17", "name")))
  85                 .build();
  86         notification = mock(DOMNotification.class);
  87         doReturn(node).when(notification).getBody();
  88         doReturn(notificationDefinitionMock.getPath()).when(notification).getType();
  89         SchemaContext schema = mock(SchemaContext.class);
  90         doReturn(notifications).when(schema).getNotifications();
  91         doReturn(schema).when(domMountPointMock).getSchemaContext();
  92         doReturn(Optional.of(mock(DOMNotificationService.class))).when(domMountPointMock).getService(DOMNotificationService.class);
  93
  94         Optional<DataBroker> optionalMpDataBroker = (Optional<DataBroker>) mock(Optional.class);
  95         DataBroker mpDataBroker = mock(DataBroker.class);
  96         doReturn(optionalMpDataBroker).when(mountPointMock).getService(DataBroker.class);
  97         doReturn(true).when(optionalMpDataBroker).isPresent();
  98         doReturn(mpDataBroker).when(optionalMpDataBroker).get();
  99
 100         ReadOnlyTransaction rtx = mock(ReadOnlyTransaction.class);
 101         doReturn(rtx).when(mpDataBroker).newReadOnlyTransaction();
 102         CheckedFuture<Optional<Streams>, ReadFailedException> checkFeature = (CheckedFuture<Optional<Streams>, ReadFailedException>)mock(CheckedFuture.class);
 103         InstanceIdentifier<Streams> pathStream = InstanceIdentifier.builder(Netconf.class).child(Streams.class).build();
 104         doReturn(checkFeature).when(rtx).read(LogicalDatastoreType.OPERATIONAL, pathStream);
 105         Optional<Streams> avStreams = NetconfTestUtils.getAvailableStream("stream01", true);
 106         doReturn(avStreams).when(checkFeature).checkedGet();
 107
 108         netconfEventSource = new NetconfEventSource(
 109                 NetconfTestUtils.getNetconfNode("NodeId1", "node.test.local", ConnectionStatus.Connected,
 110                     NetconfTestUtils.notification_capability_prefix),
 111                 streamMap,
 112                 domMountPointMock,
 113                 mountPointMock ,
 114                 domNotificationPublishServiceMock);
 115
 116     }
 117
 118     @Test
 119     public void joinTopicTest() throws Exception{
 120         joinTopicTestHelper();
 121         assertNotNull("JoinTopic return value has not been created correctly.", netconfEventSource.joinTopic(joinTopicInputMock));
 122     }
 123
 124     @Test
 125     public void testOnNotification() throws Exception {
 126         final JoinTopicInput topic1 = new JoinTopicInputBuilder()
 127                 .setTopicId(TopicId.getDefaultInstance("topic1"))
 128                 .setNotificationPattern(NotificationPattern.getDefaultInstance(".*"))
 129                 .build();
 130         netconfEventSource.joinTopic(topic1);
 131
 132
 133         ArgumentCaptor<DOMNotification> captor = ArgumentCaptor.forClass(DOMNotification.class);
 134         netconfEventSource.onNotification(notification);
 135         verify(domNotificationPublishServiceMock).putNotification(captor.capture());
 136         final TopicDOMNotification value = (TopicDOMNotification) captor.getValue();
 137         final Object actualTopicId = value.getBody().getChild(new YangInstanceIdentifier.NodeIdentifier(QName.create("urn:cisco:params:xml:ns:yang:messagebus:eventaggregator", "2014-12-02", "topic-id"))).get().getValue();
 138         Assert.assertEquals(topic1.getTopicId(), actualTopicId);
 139     }
 140
 141     private void joinTopicTestHelper() throws Exception{
 142         joinTopicInputMock = mock(JoinTopicInput.class);
 143         TopicId topicId = new TopicId("topicID007");
 144         doReturn(topicId).when(joinTopicInputMock).getTopicId();
 145         NotificationPattern notificationPatternMock = mock(NotificationPattern.class);
 146         doReturn(notificationPatternMock).when(joinTopicInputMock).getNotificationPattern();
 147         doReturn("uriStr1").when(notificationPatternMock).getValue();
 148
 149         SchemaContext schemaContextMock = mock(SchemaContext.class);
 150         doReturn(schemaContextMock).when(domMountPointMock).getSchemaContext();
 151         Set<NotificationDefinition> notificationDefinitionSet = new HashSet<>();
 152         NotificationDefinition notificationDefinitionMock = mock(NotificationDefinition.class);
 153         notificationDefinitionSet.add(notificationDefinitionMock);
 154
 155         URI uri = new URI("uriStr1");
 156         QName qName = new QName(uri, "localName1");
 157         org.opendaylight.yangtools.yang.model.api.SchemaPath schemaPath = SchemaPath.create(true, qName);
 158         doReturn(notificationDefinitionSet).when(schemaContextMock).getNotifications();
 159         doReturn(schemaPath).when(notificationDefinitionMock).getPath();
 160
 161         Optional<DOMNotificationService> domNotificationServiceOptionalMock = (Optional<DOMNotificationService>) mock(Optional.class);
 162         doReturn(domNotificationServiceOptionalMock).when(domMountPointMock).getService(DOMNotificationService.class);
 163         doReturn(true).when(domNotificationServiceOptionalMock).isPresent();
 164
 165         DOMNotificationService domNotificationServiceMock = mock(DOMNotificationService.class);
 166         doReturn(domNotificationServiceMock).when(domNotificationServiceOptionalMock).get();
 167         ListenerRegistration<NetconfEventSource> listenerRegistrationMock = (ListenerRegistration<NetconfEventSource>)mock(ListenerRegistration.class);
 168         doReturn(listenerRegistrationMock).when(domNotificationServiceMock).registerNotificationListener(any(NetconfEventSource.class), any(SchemaPath.class));
 169
 170         Optional<DOMService> optionalMock = (Optional<DOMService>) mock(Optional.class);
 171         doReturn(optionalMock).when(domMountPointMock).getService(DOMRpcService.class);
 172         doReturn(true).when(optionalMock).isPresent();
 173         DOMRpcService domRpcServiceMock = mock(DOMRpcService.class);
 174         doReturn(domRpcServiceMock).when(optionalMock).get();
 175         CheckedFuture checkedFutureMock = mock(CheckedFuture.class);
 176         doReturn(checkedFutureMock).when(domRpcServiceMock).invokeRpc(any(SchemaPath.class), any(ContainerNode.class));
 177
 178     }
 179
 180     private NotificationDefinition getNotificationDefinitionMock(String namespace, String revision, String name) {
 181         NotificationDefinition notification = mock(NotificationDefinition.class);
 182         final QName qName = QName.create(namespace, revision, name);
 183         doReturn(qName).when(notification).getQName();
 184         doReturn(SchemaPath.create(true, qName)).when(notification).getPath();
 185         return notification;
 186     }
 187
 188 }