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 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;
41 import org.opendaylight.yangtools.yang.common.QName;
43 import java.lang.reflect.Field;
45 import java.util.HashMap;
46 import java.util.HashSet;
49 import java.util.List;
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;
59 public class NetconfEventSourceTest {
61 NetconfEventSource netconfEventSource;
62 DOMMountPoint domMountPointMock;
63 JoinTopicInput joinTopicInputMock;
66 public static void initTestClass() throws IllegalAccessException, InstantiationException {
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);
77 RpcConsumerRegistry rpcConsumerRegistryMock = mock(RpcConsumerRegistry.class);
78 Optional<BindingService> onlyOptionalMock = (Optional<BindingService>) mock(Optional.class);
79 NotificationsService notificationsServiceMock = mock(NotificationsService.class);
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);
88 public void constructorTest() {
89 assertNotNull("Instance has not been created correctly.", netconfEventSource);
93 public void joinTopicTest() throws Exception{
94 joinTopicTestHelper();
95 assertNotNull("JoinTopic return value has not been created correctly.", netconfEventSource.joinTopic(joinTopicInputMock));
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();
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);
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();
116 Optional<DOMNotificationService> domNotificationServiceOptionalMock = (Optional<DOMNotificationService>) mock(Optional.class);
117 doReturn(domNotificationServiceOptionalMock).when(domMountPointMock).getService(DOMNotificationService.class);
118 doReturn(true).when(domNotificationServiceOptionalMock).isPresent();
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));
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);
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();
149 netconfEventSource.onDataChanged(asyncDataChangeEventMock);
150 verify(dataObjectMock, times(2)).isConnected();
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();
164 Set<String> localSet = getActiveStreams();
165 localSet.add("activeStream1");
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));
174 netconfEventSource.onDataChanged(asyncDataChangeEventMock);
175 verify(dataObjectMock, times(1)).isConnected();
176 assertEquals("Size of set has not been set correctly.", 1, getActiveStreams().size());
179 private Set getActiveStreams() throws Exception{
180 Field nesField = NetconfEventSource.class.getDeclaredField("activeStreams");
181 nesField.setAccessible(true);
182 return (Set) nesField.get(netconfEventSource);