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 org.junit.Before;
12 import org.junit.BeforeClass;
13 import org.junit.Test;
14 import org.opendaylight.controller.config.yang.messagebus.app.impl.NamespaceToStream;
15 import org.opendaylight.controller.md.sal.binding.api.BindingService;
16 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
17 import org.opendaylight.controller.md.sal.binding.api.MountPoint;
18 import org.opendaylight.controller.md.sal.binding.api.MountPointService;
19 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
20 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
21 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
22 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
23 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
24 import org.opendaylight.controller.md.sal.dom.api.DOMNotificationPublishService;
25 import org.opendaylight.controller.sal.binding.api.RpcConsumerRegistry;
26 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netconf.notification._1._0.rev080714.NotificationsService;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeFields;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.fields.AvailableCapabilities;
30 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
31 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
32 import org.opendaylight.yangtools.concepts.ListenerRegistration;
33 import org.opendaylight.yangtools.yang.binding.DataObject;
34 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
35 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
37 import java.util.ArrayList;
38 import java.util.HashMap;
39 import java.util.List;
42 import static org.junit.Assert.assertFalse;
43 import static org.junit.Assert.assertNotNull;
44 import static org.junit.Assert.assertTrue;
45 import static org.mockito.Matchers.any;
46 import static org.mockito.Matchers.eq;
47 import static org.mockito.Mockito.doReturn;
48 import static org.mockito.Mockito.mock;
49 import static org.mockito.Mockito.verify;
50 import static org.mockito.Mockito.times;
51 import static org.mockito.Mockito.notNull;
53 public class NetconfEventSourceManagerTest {
55 NetconfEventSourceManager netconfEventSourceManager;
56 ListenerRegistration listenerRegistrationMock;
57 DOMMountPointService domMountPointServiceMock;
58 MountPointService mountPointServiceMock;
59 EventSourceTopology eventSourceTopologyMock;
60 AsyncDataChangeEvent asyncDataChangeEventMock;
63 public static void initTestClass() throws IllegalAccessException, InstantiationException {
67 public void setUp() throws Exception {
68 DataBroker dataBrokerMock = mock(DataBroker.class);
69 DOMNotificationPublishService domNotificationPublishServiceMock = mock(DOMNotificationPublishService.class);
70 domMountPointServiceMock = mock(DOMMountPointService.class);
71 mountPointServiceMock = mock(MountPointService.class);
72 eventSourceTopologyMock = mock(EventSourceTopology.class);
73 List<NamespaceToStream> namespaceToStreamList = new ArrayList<>();
75 listenerRegistrationMock = mock(ListenerRegistration.class);
76 doReturn(listenerRegistrationMock).when(dataBrokerMock).registerDataChangeListener(eq(LogicalDatastoreType.OPERATIONAL), any(InstanceIdentifier.class), any(NetconfEventSourceManager.class), eq(AsyncDataBroker.DataChangeScope.SUBTREE));
78 netconfEventSourceManager = new NetconfEventSourceManager(dataBrokerMock, domNotificationPublishServiceMock, domMountPointServiceMock,
79 mountPointServiceMock, eventSourceTopologyMock, namespaceToStreamList);
83 public void constructorTest() {
84 assertNotNull("Instance has not been created correctly.", netconfEventSourceManager);
88 public void onDataChangedTest() {
89 AsyncDataChangeEvent asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
90 Map<InstanceIdentifier, DataObject> map = new HashMap<>();
91 InstanceIdentifier instanceIdentifierMock = mock(InstanceIdentifier.class);
92 Node dataObjectMock = mock(Node.class);
93 map.put(instanceIdentifierMock, dataObjectMock);
94 doReturn(map).when(asyncDataChangeEventMock).getCreatedData();
95 doReturn(map).when(asyncDataChangeEventMock).getUpdatedData();
96 netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
97 verify(dataObjectMock, times(2)).getAugmentation(NetconfNode.class);
101 public void onDataChangedCreateEventSourceTest() {
102 onDataChangedCreateEventSourceTestHelper();
103 netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
104 verify(eventSourceTopologyMock, times(1)).register(any(Node.class), any(NetconfEventSource.class));
107 private void onDataChangedCreateEventSourceTestHelper(){
108 asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
109 Map<InstanceIdentifier, DataObject> map = new HashMap<>();
110 InstanceIdentifier instanceIdentifierMock = mock(InstanceIdentifier.class);
111 Node dataObjectMock = mock(Node.class);
112 map.put(instanceIdentifierMock, dataObjectMock);
113 doReturn(map).when(asyncDataChangeEventMock).getCreatedData();
114 doReturn(map).when(asyncDataChangeEventMock).getUpdatedData();
116 NetconfNode netconfNodeMock = mock(NetconfNode.class);
117 AvailableCapabilities availableCapabilitiesMock = mock(AvailableCapabilities.class);
118 doReturn(netconfNodeMock).when(dataObjectMock).getAugmentation(NetconfNode.class);
119 doReturn(availableCapabilitiesMock).when(netconfNodeMock).getAvailableCapabilities();
120 List<String> availableCapabilityList = new ArrayList<>();
121 availableCapabilityList.add("(urn:ietf:params:xml:ns:netconf:notification_availableCapabilityString1");
122 doReturn(availableCapabilityList).when(availableCapabilitiesMock).getAvailableCapability();
124 doReturn(NetconfNodeFields.ConnectionStatus.Connected).when(netconfNodeMock).getConnectionStatus();
126 Optional optionalMock = mock(Optional.class);
127 Optional optionalBindingMountMock = mock(Optional.class);
128 NodeId nodeId = new NodeId("nodeId1");
129 doReturn(nodeId).when(dataObjectMock).getNodeId();
130 doReturn(optionalMock).when(domMountPointServiceMock).getMountPoint((YangInstanceIdentifier)notNull());
131 doReturn(optionalBindingMountMock).when(mountPointServiceMock).getMountPoint(any(InstanceIdentifier.class));
132 doReturn(true).when(optionalMock).isPresent();
133 doReturn(true).when(optionalBindingMountMock).isPresent();
135 DOMMountPoint domMountPointMock = mock(DOMMountPoint.class);
136 MountPoint mountPointMock = mock(MountPoint.class);
137 doReturn(domMountPointMock).when(optionalMock).get();
138 doReturn(mountPointMock).when(optionalBindingMountMock).get();
140 RpcConsumerRegistry rpcConsumerRegistryMock = mock(RpcConsumerRegistry.class);
141 Optional<BindingService> onlyOptionalMock = (Optional<BindingService>) mock(Optional.class);
142 NotificationsService notificationsServiceMock = mock(NotificationsService.class);
144 doReturn(onlyOptionalMock).when(mountPointMock).getService(RpcConsumerRegistry.class);
145 doReturn(rpcConsumerRegistryMock).when(onlyOptionalMock).get();
146 doReturn(notificationsServiceMock).when(rpcConsumerRegistryMock).getRpcService(NotificationsService.class);
150 public void isEventSourceTest() {
151 Node nodeMock = mock(Node.class);
152 NetconfNode netconfNodeMock = mock(NetconfNode.class);
153 AvailableCapabilities availableCapabilitiesMock = mock(AvailableCapabilities.class);
154 doReturn(netconfNodeMock).when(nodeMock).getAugmentation(NetconfNode.class);
155 doReturn(availableCapabilitiesMock).when(netconfNodeMock).getAvailableCapabilities();
156 List<String> availableCapabilityList = new ArrayList<>();
157 availableCapabilityList.add("(urn:ietf:params:xml:ns:netconf:notification_availableCapabilityString1");
158 doReturn(availableCapabilityList).when(availableCapabilitiesMock).getAvailableCapability();
159 assertTrue("Method has not been run correctly.", netconfEventSourceManager.isEventSource(nodeMock));
163 public void isNotEventSourceTest() {
164 Node nodeMock = mock(Node.class);
165 NetconfNode netconfNodeMock = mock(NetconfNode.class);
166 AvailableCapabilities availableCapabilitiesMock = mock(AvailableCapabilities.class);
167 doReturn(netconfNodeMock).when(nodeMock).getAugmentation(NetconfNode.class);
168 doReturn(availableCapabilitiesMock).when(netconfNodeMock).getAvailableCapabilities();
169 List<String> availableCapabilityList = new ArrayList<>();
170 availableCapabilityList.add("availableCapabilityString1");
171 doReturn(availableCapabilityList).when(availableCapabilitiesMock).getAvailableCapability();
172 assertFalse("Method has not been run correctly.", netconfEventSourceManager.isEventSource(nodeMock));
176 public void closeTest() {
177 netconfEventSourceManager.close();
178 verify(listenerRegistrationMock, times(1)).close();