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.ArgumentMatchers.any;
11 import static org.mockito.ArgumentMatchers.notNull;
12 import static org.mockito.Mockito.doReturn;
13 import static org.mockito.Mockito.mock;
14 import static org.mockito.Mockito.times;
15 import static org.mockito.Mockito.verify;
17 import com.google.common.util.concurrent.FluentFuture;
18 import java.util.Collections;
19 import java.util.HashMap;
20 import java.util.Optional;
21 import org.junit.Before;
22 import org.junit.Test;
23 import org.opendaylight.controller.messagebus.spi.EventSource;
24 import org.opendaylight.controller.messagebus.spi.EventSourceRegistry;
25 import org.opendaylight.mdsal.binding.api.DataBroker;
26 import org.opendaylight.mdsal.binding.api.DataObjectModification;
27 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
28 import org.opendaylight.mdsal.binding.api.DataTreeModification;
29 import org.opendaylight.mdsal.binding.api.MountPointService;
30 import org.opendaylight.mdsal.binding.api.RpcProviderService;
31 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
32 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
33 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction;
34 import org.opendaylight.mdsal.dom.api.DOMMountPoint;
35 import org.opendaylight.mdsal.dom.api.DOMMountPointService;
36 import org.opendaylight.mdsal.dom.api.DOMNotificationPublishService;
37 import org.opendaylight.mdsal.dom.api.DOMNotificationService;
38 import org.opendaylight.mdsal.dom.api.DOMRpcService;
39 import org.opendaylight.mdsal.dom.api.DOMSchemaService;
40 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.Netconf;
41 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.netmod.notification.rev080714.netconf.Streams;
42 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeConnectionStatus.ConnectionStatus;
43 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
44 import org.opendaylight.yangtools.concepts.ListenerRegistration;
45 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
46 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
47 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
49 public class NetconfEventSourceManagerTest extends AbstractCodecTest {
51 private NetconfEventSourceManager netconfEventSourceManager;
52 private ListenerRegistration<?> listenerRegistrationMock;
53 private DOMMountPointService domMountPointServiceMock;
54 private MountPointService mountPointServiceMock;
55 private EventSourceRegistry eventSourceTopologyMock;
56 private DataTreeModification<Node> dataTreeModificationMock;
57 private RpcProviderService rpcProviderRegistryMock;
58 private EventSourceRegistry eventSourceRegistry;
60 @SuppressWarnings("unchecked")
62 public void setUp() throws Exception {
63 final DataBroker dataBrokerMock = mock(DataBroker.class);
64 final DOMNotificationPublishService domNotificationPublishServiceMock =
65 mock(DOMNotificationPublishService.class);
66 domMountPointServiceMock = mock(DOMMountPointService.class);
67 eventSourceTopologyMock = mock(EventSourceRegistry.class);
68 rpcProviderRegistryMock = mock(RpcProviderService.class);
69 eventSourceRegistry = mock(EventSourceRegistry.class);
71 listenerRegistrationMock = mock(ListenerRegistration.class);
72 doReturn(listenerRegistrationMock).when(dataBrokerMock).registerDataTreeChangeListener(
73 any(DataTreeIdentifier.class), any(NetconfEventSourceManager.class));
75 DOMMountPoint domMountPointMock = mock(DOMMountPoint.class);
76 Optional<DOMMountPoint> optionalDomMountServiceMock = Optional.of(domMountPointMock);
77 doReturn(optionalDomMountServiceMock).when(domMountPointServiceMock).getMountPoint((YangInstanceIdentifier)
79 DOMDataBroker mpDataBroker = mock(DOMDataBroker.class);
80 doReturn(Optional.of(mpDataBroker)).when(domMountPointMock).getService(DOMDataBroker.class);
81 doReturn(Optional.of(mock(DOMRpcService.class))).when(domMountPointMock).getService(DOMRpcService.class);
82 doReturn(Optional.of(mock(DOMNotificationService.class))).when(domMountPointMock)
83 .getService(DOMNotificationService.class);
84 doReturn(Optional.of(mock(DOMSchemaService.class))).when(domMountPointMock).getService(DOMSchemaService.class);
86 DOMDataTreeReadTransaction rtx = mock(DOMDataTreeReadTransaction.class);
87 doReturn(rtx).when(mpDataBroker).newReadOnlyTransaction();
88 final FluentFuture<Optional<NormalizedNode<?, ?>>> readStreamFuture =
89 FluentFutures.immediateFluentFuture(Optional.of(NetconfTestUtils.getStreamsNode("stream-1")));
91 YangInstanceIdentifier pathStream = YangInstanceIdentifier.builder().node(Netconf.QNAME).node(Streams.QNAME)
93 doReturn(readStreamFuture).when(rtx).read(LogicalDatastoreType.OPERATIONAL, pathStream);
95 netconfEventSourceManager = new NetconfEventSourceManager(dataBrokerMock, SERIALIZER,
96 domNotificationPublishServiceMock, domMountPointServiceMock, eventSourceRegistry);
97 netconfEventSourceManager.setStreamMap(new HashMap<>());
101 public void onDataChangedCreateEventSourceTestByCreateEntry() throws Exception {
102 onDataChangedTestHelper(true, false, true, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
103 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
104 verify(eventSourceRegistry, times(1)).registerEventSource(any(EventSource.class));
108 public void onDataChangedCreateEventSourceTestByUpdateEntry() throws Exception {
109 onDataChangedTestHelper(false, true, true, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
110 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
111 verify(eventSourceRegistry, times(1)).registerEventSource(any(EventSource.class));
115 public void onDataChangedCreateEventSourceTestNotNeconf() throws Exception {
116 onDataChangedTestHelper(false, true, false, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
117 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
118 verify(eventSourceRegistry, times(0)).registerEventSource(any(EventSource.class));
122 public void onDataChangedCreateEventSourceTestNotNotificationCapability() throws Exception {
123 onDataChangedTestHelper(true, false, true, "bad-prefix");
124 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
125 verify(eventSourceRegistry, times(0)).registerEventSource(any(EventSource.class));
128 @SuppressWarnings("unchecked")
129 private void onDataChangedTestHelper(final boolean create, final boolean update, final boolean isNetconf,
130 final String notificationCapabilityPrefix) throws Exception {
131 dataTreeModificationMock = mock(DataTreeModification.class);
132 DataObjectModification<Node> mockModification = mock(DataObjectModification.class);
133 doReturn(create ? DataObjectModification.ModificationType.WRITE :
134 DataObjectModification.ModificationType.SUBTREE_MODIFIED).when(mockModification).getModificationType();
135 doReturn(mockModification).when(dataTreeModificationMock).getRootNode();
138 String nodeId = "Node01";
140 node01 = NetconfTestUtils
141 .getNetconfNode(nodeId, "node01.test.local", ConnectionStatus.Connected,
142 notificationCapabilityPrefix);
145 node01 = NetconfTestUtils.getNode(nodeId);
148 doReturn(node01).when(mockModification).getDataAfter();
150 doReturn(DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL,
151 NetconfTestUtils.getInstanceIdentifier(node01))).when(dataTreeModificationMock).getRootPath();