2 * Copyright (c) 2016 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.netconf.messagebus.eventsources.netconf;
10 import static org.mockito.Matchers.any;
11 import static org.mockito.Matchers.eq;
12 import static org.mockito.Matchers.notNull;
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;
18 import com.google.common.base.Optional;
19 import com.google.common.util.concurrent.CheckedFuture;
20 import com.google.common.util.concurrent.Futures;
21 import java.util.HashMap;
23 import org.junit.Before;
24 import org.junit.BeforeClass;
25 import org.junit.Test;
26 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
27 import org.opendaylight.controller.md.sal.binding.api.MountPointService;
28 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataBroker;
29 import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
30 import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
31 import org.opendaylight.controller.md.sal.common.api.data.ReadFailedException;
32 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
33 import org.opendaylight.controller.md.sal.dom.api.DOMDataReadOnlyTransaction;
34 import org.opendaylight.controller.md.sal.dom.api.DOMMountPoint;
35 import org.opendaylight.controller.md.sal.dom.api.DOMMountPointService;
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.messagebus.spi.EventSource;
40 import org.opendaylight.controller.messagebus.spi.EventSourceRegistry;
41 import org.opendaylight.controller.sal.binding.api.RpcProviderRegistry;
42 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.Netconf;
43 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.netconf.Streams;
44 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeConnectionStatus.ConnectionStatus;
45 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
46 import org.opendaylight.yangtools.concepts.ListenerRegistration;
47 import org.opendaylight.yangtools.yang.binding.DataObject;
48 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
49 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
50 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
52 public class NetconfEventSourceManagerTest {
54 NetconfEventSourceManager netconfEventSourceManager;
55 ListenerRegistration listenerRegistrationMock;
56 DOMMountPointService domMountPointServiceMock;
57 MountPointService mountPointServiceMock;
58 EventSourceRegistry eventSourceTopologyMock;
59 AsyncDataChangeEvent asyncDataChangeEventMock;
60 RpcProviderRegistry rpcProviderRegistryMock;
61 EventSourceRegistry eventSourceRegistry;
64 public static void initTestClass() throws IllegalAccessException, InstantiationException {
68 public void setUp() throws Exception {
69 final DataBroker dataBrokerMock = mock(DataBroker.class);
70 final DOMNotificationPublishService domNotificationPublishServiceMock =
71 mock(DOMNotificationPublishService.class);
72 domMountPointServiceMock = mock(DOMMountPointService.class);
73 eventSourceTopologyMock = mock(EventSourceRegistry.class);
74 rpcProviderRegistryMock = mock(RpcProviderRegistry.class);
75 eventSourceRegistry = mock(EventSourceRegistry.class);
77 listenerRegistrationMock = mock(ListenerRegistration.class);
78 doReturn(listenerRegistrationMock).when(dataBrokerMock).registerDataChangeListener(eq(LogicalDatastoreType
79 .OPERATIONAL), any(InstanceIdentifier.class), any(NetconfEventSourceManager.class), eq(
80 AsyncDataBroker.DataChangeScope.SUBTREE));
82 DOMMountPoint domMountPointMock = mock(DOMMountPoint.class);
83 Optional<DOMMountPoint> optionalDomMountServiceMock = Optional.of(domMountPointMock);
84 doReturn(optionalDomMountServiceMock).when(domMountPointServiceMock).getMountPoint((YangInstanceIdentifier)
86 DOMDataBroker mpDataBroker = mock(DOMDataBroker.class);
87 doReturn(Optional.of(mpDataBroker)).when(domMountPointMock).getService(DOMDataBroker.class);
88 doReturn(Optional.of(mock(DOMRpcService.class))).when(domMountPointMock).getService(DOMRpcService.class);
89 doReturn(Optional.of(mock(DOMNotificationService.class))).when(domMountPointMock)
90 .getService(DOMNotificationService.class);
92 DOMDataReadOnlyTransaction rtx = mock(DOMDataReadOnlyTransaction.class);
93 doReturn(rtx).when(mpDataBroker).newReadOnlyTransaction();
94 CheckedFuture<Optional<NormalizedNode<?, ?>>, ReadFailedException> checkFeature = Futures
95 .immediateCheckedFuture(Optional.of(NetconfTestUtils.getStreamsNode("stream-1")));
97 YangInstanceIdentifier pathStream = YangInstanceIdentifier.builder().node(Netconf.QNAME).node(Streams.QNAME)
99 doReturn(checkFeature).when(rtx).read(LogicalDatastoreType.OPERATIONAL, pathStream);
101 netconfEventSourceManager = new NetconfEventSourceManager(dataBrokerMock,
102 domNotificationPublishServiceMock, domMountPointServiceMock, eventSourceRegistry);
103 netconfEventSourceManager.setStreamMap(new HashMap<>());
107 public void onDataChangedCreateEventSourceTestByCreateEntry() throws Exception {
108 onDataChangedTestHelper(true, false, true, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
109 netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
110 verify(eventSourceRegistry, times(1)).registerEventSource(any(EventSource.class));
114 public void onDataChangedCreateEventSourceTestByUpdateEntry() throws Exception {
115 onDataChangedTestHelper(false, true, true, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
116 netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
117 verify(eventSourceRegistry, times(1)).registerEventSource(any(EventSource.class));
121 public void onDataChangedCreateEventSourceTestNotNeconf() throws Exception {
122 onDataChangedTestHelper(false, true, false, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
123 netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
124 verify(eventSourceRegistry, times(0)).registerEventSource(any(EventSource.class));
128 public void onDataChangedCreateEventSourceTestNotNotificationCapability() throws Exception {
129 onDataChangedTestHelper(true, false, true, "bad-prefix");
130 netconfEventSourceManager.onDataChanged(asyncDataChangeEventMock);
131 verify(eventSourceRegistry, times(0)).registerEventSource(any(EventSource.class));
134 private void onDataChangedTestHelper(boolean create, boolean update, boolean isNetconf, String
135 notificationCapabilityPrefix) throws Exception {
136 asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
137 Map<InstanceIdentifier, DataObject> mapCreate = new HashMap<>();
138 Map<InstanceIdentifier, DataObject> mapUpdate = new HashMap<>();
141 String nodeId = "Node01";
142 doReturn(mapCreate).when(asyncDataChangeEventMock).getCreatedData();
143 doReturn(mapUpdate).when(asyncDataChangeEventMock).getUpdatedData();
146 node01 = NetconfTestUtils
147 .getNetconfNode(nodeId, "node01.test.local", ConnectionStatus.Connected,
148 notificationCapabilityPrefix);
151 node01 = NetconfTestUtils.getNode(nodeId);
155 mapCreate.put(NetconfTestUtils.getInstanceIdentifier(node01), node01);
158 mapUpdate.put(NetconfTestUtils.getInstanceIdentifier(node01), node01);