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 static org.junit.Assert.assertEquals;
  11 import static org.junit.Assert.assertNotNull;
  12 import static org.mockito.Matchers.any;
  13 import static org.mockito.Mockito.doReturn;
  14 import static org.mockito.Mockito.mock;
  15 import static org.mockito.Mockito.times;
  16 import static org.mockito.Mockito.verify;
  17
  18 import java.net.URI;
  19 import java.util.HashMap;
  20 import java.util.HashSet;
  21 import java.util.Map;
  22 import java.util.Set;
  23
  24 import org.junit.Before;
  25 import org.junit.Test;
  26 import org.opendaylight.controller.md.sal.binding.api.BindingService;
  27 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
  28 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
  29 import org.opendaylight.controller.md.sal.dom.api.DOMNotificationPublishService;
  30 import org.opendaylight.controller.md.sal.dom.api.DOMNotificationService;
  31 import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
  32 import org.opendaylight.controller.md.sal.dom.api.DOMService;
  33 import org.opendaylight.controller.messagebus.eventsources.netconf.NetconfEventSource;
  34 import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
  35 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.NotificationPattern;
  36 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.TopicId;
  37 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicInput;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.NotificationsService;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.Nodes;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.NodeKey;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.inventory.rev140108.NetconfNode;
  44 import org.opendaylight.yangtools.concepts.ListenerRegistration;
  45 import org.opendaylight.yangtools.yang.binding.DataObject;
  46 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  47 import org.opendaylight.yangtools.yang.common.QName;
  48 import org.opendaylight.yangtools.yang.data.api.schema.ContainerNode;
  49 import org.opendaylight.yangtools.yang.model.api.NotificationDefinition;
  50 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  51 import org.opendaylight.yangtools.yang.model.api.SchemaPath;
  52
  53 import com.google.common.base.Optional;
  54 import com.google.common.util.concurrent.CheckedFuture;
  55
  56 public class NetconfEventSourceTest {
  57
  58     NetconfEventSource netconfEventSource;
  59     DOMMountPoint domMountPointMock;
  60     JoinTopicInput joinTopicInputMock;
  61
  62     @Before
  63     public void setUp() throws Exception {
  64         Map<String, String> streamMap = new HashMap<>();
  65         streamMap.put("uriStr1", "string2");
  66         domMountPointMock = mock(DOMMountPoint.class);
  67         DOMNotificationPublishService domNotificationPublishServiceMock = mock(DOMNotificationPublishService.class);
  68
  69         RpcConsumerRegistry rpcConsumerRegistryMock = mock(RpcConsumerRegistry.class);
  70         Optional<BindingService> onlyOptionalMock = (Optional<BindingService>) mock(Optional.class);
  71         NotificationsService notificationsServiceMock = mock(NotificationsService.class);
  72
  73         doReturn(notificationsServiceMock).when(rpcConsumerRegistryMock).getRpcService(NotificationsService.class);
  74         org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node node
  75                 = mock(org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node.class);
  76         org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId nodeId
  77                 = new org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId("NodeId1");
  78         doReturn(nodeId).when(node).getNodeId();
  79         netconfEventSource = new NetconfEventSource(node, streamMap, domMountPointMock, domNotificationPublishServiceMock);
  80     }
  81
  82     @Test
  83     public void onDataChangedTest(){
  84         InstanceIdentifier brmIdent = InstanceIdentifier.create(Nodes.class)
  85                 .child(Node.class, new NodeKey(new NodeId("brm"))).augmentation(NetconfNode.class);
  86         AsyncDataChangeEvent asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
  87         NetconfNode dataObjectMock = mock(NetconfNode.class);
  88         Map<InstanceIdentifier, DataObject> dataChangeMap = new HashMap<>();
  89         dataChangeMap.put(brmIdent, dataObjectMock);
  90         doReturn(dataChangeMap).when(asyncDataChangeEventMock).getOriginalData();
  91         doReturn(dataChangeMap).when(asyncDataChangeEventMock).getUpdatedData();
  92         doReturn(true).when(dataObjectMock).isConnected();
  93         netconfEventSource.onDataChanged(asyncDataChangeEventMock);
  94         verify(dataObjectMock, times(2)).isConnected();
  95     }
  96
  97     @Test
  98     public void joinTopicTest() throws Exception{
  99         joinTopicTestHelper();
 100         assertNotNull("JoinTopic return value has not been created correctly.", netconfEventSource.joinTopic(joinTopicInputMock));
 101     }
 102
 103     private void joinTopicTestHelper() throws Exception{
 104         joinTopicInputMock = mock(JoinTopicInput.class);
 105         TopicId topicId = new TopicId("topicID007");
 106         doReturn(topicId).when(joinTopicInputMock).getTopicId();
 107         NotificationPattern notificationPatternMock = mock(NotificationPattern.class);
 108         doReturn(notificationPatternMock).when(joinTopicInputMock).getNotificationPattern();
 109         doReturn("uriStr1").when(notificationPatternMock).getValue();
 110
 111         SchemaContext schemaContextMock = mock(SchemaContext.class);
 112         doReturn(schemaContextMock).when(domMountPointMock).getSchemaContext();
 113         Set<NotificationDefinition> notificationDefinitionSet = new HashSet<>();
 114         NotificationDefinition notificationDefinitionMock = mock(NotificationDefinition.class);
 115         notificationDefinitionSet.add(notificationDefinitionMock);
 116
 117         URI uri = new URI("uriStr1");
 118         QName qName = new QName(uri, "localName1");
 119         org.opendaylight.yangtools.yang.model.api.SchemaPath schemaPath = SchemaPath.create(true, qName);
 120         doReturn(notificationDefinitionSet).when(schemaContextMock).getNotifications();
 121         doReturn(schemaPath).when(notificationDefinitionMock).getPath();
 122
 123         Optional<DOMNotificationService> domNotificationServiceOptionalMock = (Optional<DOMNotificationService>) mock(Optional.class);
 124         doReturn(domNotificationServiceOptionalMock).when(domMountPointMock).getService(DOMNotificationService.class);
 125         doReturn(true).when(domNotificationServiceOptionalMock).isPresent();
 126
 127         DOMNotificationService domNotificationServiceMock = mock(DOMNotificationService.class);
 128         doReturn(domNotificationServiceMock).when(domNotificationServiceOptionalMock).get();
 129         ListenerRegistration<NetconfEventSource> listenerRegistrationMock = (ListenerRegistration<NetconfEventSource>)mock(ListenerRegistration.class);
 130         doReturn(listenerRegistrationMock).when(domNotificationServiceMock).registerNotificationListener(any(NetconfEventSource.class), any(SchemaPath.class));
 131
 132         Optional<DOMService> optionalMock = (Optional<DOMService>) mock(Optional.class);
 133         doReturn(optionalMock).when(domMountPointMock).getService(DOMRpcService.class);
 134         DOMRpcService domRpcServiceMock = mock(DOMRpcService.class);
 135         doReturn(domRpcServiceMock).when(optionalMock).get();
 136         CheckedFuture checkedFutureMock = mock(CheckedFuture.class);
 137         doReturn(checkedFutureMock).when(domRpcServiceMock).invokeRpc(any(SchemaPath.class), any(ContainerNode.class));
 138     }
 139
 140 }