2 * Copyright (c) 2015 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.controller.messagebus.app.impl;
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;
19 import java.util.HashMap;
20 import java.util.HashSet;
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;
53 import com.google.common.base.Optional;
54 import com.google.common.util.concurrent.CheckedFuture;
56 public class NetconfEventSourceTest {
58 NetconfEventSource netconfEventSource;
59 DOMMountPoint domMountPointMock;
60 JoinTopicInput joinTopicInputMock;
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);
69 RpcConsumerRegistry rpcConsumerRegistryMock = mock(RpcConsumerRegistry.class);
70 Optional<BindingService> onlyOptionalMock = (Optional<BindingService>) mock(Optional.class);
71 NotificationsService notificationsServiceMock = mock(NotificationsService.class);
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);
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();
98 public void joinTopicTest() throws Exception{
99 joinTopicTestHelper();
100 assertNotNull("JoinTopic return value has not been created correctly.", netconfEventSource.joinTopic(joinTopicInputMock));
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();
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);
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();
123 Optional<DOMNotificationService> domNotificationServiceOptionalMock = (Optional<DOMNotificationService>) mock(Optional.class);
124 doReturn(domNotificationServiceOptionalMock).when(domMountPointMock).getService(DOMNotificationService.class);
125 doReturn(true).when(domNotificationServiceOptionalMock).isPresent();
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));
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));