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.BeforeClass;
23 import org.junit.Test;
24 import org.opendaylight.controller.messagebus.spi.EventSource;
25 import org.opendaylight.controller.messagebus.spi.EventSourceRegistry;
26 import org.opendaylight.mdsal.binding.api.DataBroker;
27 import org.opendaylight.mdsal.binding.api.DataObjectModification;
28 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
29 import org.opendaylight.mdsal.binding.api.DataTreeModification;
30 import org.opendaylight.mdsal.binding.api.MountPointService;
31 import org.opendaylight.mdsal.binding.api.RpcProviderService;
32 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
33 import org.opendaylight.mdsal.dom.api.DOMDataBroker;
34 import org.opendaylight.mdsal.dom.api.DOMDataTreeReadTransaction;
35 import org.opendaylight.mdsal.dom.api.DOMMountPoint;
36 import org.opendaylight.mdsal.dom.api.DOMMountPointService;
37 import org.opendaylight.mdsal.dom.api.DOMNotificationPublishService;
38 import org.opendaylight.mdsal.dom.api.DOMNotificationService;
39 import org.opendaylight.mdsal.dom.api.DOMRpcService;
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 {
51 NetconfEventSourceManager netconfEventSourceManager;
52 ListenerRegistration listenerRegistrationMock;
53 DOMMountPointService domMountPointServiceMock;
54 MountPointService mountPointServiceMock;
55 EventSourceRegistry eventSourceTopologyMock;
56 DataTreeModification<Node> dataTreeModificationMock;
57 RpcProviderService rpcProviderRegistryMock;
58 EventSourceRegistry eventSourceRegistry;
61 public static void initTestClass() throws IllegalAccessException, InstantiationException {
64 @SuppressWarnings("unchecked")
66 public void setUp() throws Exception {
67 final DataBroker dataBrokerMock = mock(DataBroker.class);
68 final DOMNotificationPublishService domNotificationPublishServiceMock =
69 mock(DOMNotificationPublishService.class);
70 domMountPointServiceMock = mock(DOMMountPointService.class);
71 eventSourceTopologyMock = mock(EventSourceRegistry.class);
72 rpcProviderRegistryMock = mock(RpcProviderService.class);
73 eventSourceRegistry = mock(EventSourceRegistry.class);
75 listenerRegistrationMock = mock(ListenerRegistration.class);
76 doReturn(listenerRegistrationMock).when(dataBrokerMock).registerDataTreeChangeListener(
77 any(DataTreeIdentifier.class), any(NetconfEventSourceManager.class));
79 DOMMountPoint domMountPointMock = mock(DOMMountPoint.class);
80 Optional<DOMMountPoint> optionalDomMountServiceMock = Optional.of(domMountPointMock);
81 doReturn(optionalDomMountServiceMock).when(domMountPointServiceMock).getMountPoint((YangInstanceIdentifier)
83 DOMDataBroker mpDataBroker = mock(DOMDataBroker.class);
84 doReturn(Optional.of(mpDataBroker)).when(domMountPointMock).getService(DOMDataBroker.class);
85 doReturn(Optional.of(mock(DOMRpcService.class))).when(domMountPointMock).getService(DOMRpcService.class);
86 doReturn(Optional.of(mock(DOMNotificationService.class))).when(domMountPointMock)
87 .getService(DOMNotificationService.class);
89 DOMDataTreeReadTransaction rtx = mock(DOMDataTreeReadTransaction.class);
90 doReturn(rtx).when(mpDataBroker).newReadOnlyTransaction();
91 final FluentFuture<Optional<NormalizedNode<?, ?>>> readStreamFuture =
92 FluentFutures.immediateFluentFuture(Optional.of(NetconfTestUtils.getStreamsNode("stream-1")));
94 YangInstanceIdentifier pathStream = YangInstanceIdentifier.builder().node(Netconf.QNAME).node(Streams.QNAME)
96 doReturn(readStreamFuture).when(rtx).read(LogicalDatastoreType.OPERATIONAL, pathStream);
98 netconfEventSourceManager = new NetconfEventSourceManager(dataBrokerMock,
99 domNotificationPublishServiceMock, domMountPointServiceMock, eventSourceRegistry);
100 netconfEventSourceManager.setStreamMap(new HashMap<>());
104 public void onDataChangedCreateEventSourceTestByCreateEntry() throws Exception {
105 onDataChangedTestHelper(true, false, true, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
106 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
107 verify(eventSourceRegistry, times(1)).registerEventSource(any(EventSource.class));
111 public void onDataChangedCreateEventSourceTestByUpdateEntry() throws Exception {
112 onDataChangedTestHelper(false, true, true, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
113 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
114 verify(eventSourceRegistry, times(1)).registerEventSource(any(EventSource.class));
118 public void onDataChangedCreateEventSourceTestNotNeconf() throws Exception {
119 onDataChangedTestHelper(false, true, false, NetconfTestUtils.NOTIFICATION_CAPABILITY_PREFIX);
120 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
121 verify(eventSourceRegistry, times(0)).registerEventSource(any(EventSource.class));
125 public void onDataChangedCreateEventSourceTestNotNotificationCapability() throws Exception {
126 onDataChangedTestHelper(true, false, true, "bad-prefix");
127 netconfEventSourceManager.onDataTreeChanged(Collections.singletonList(dataTreeModificationMock));
128 verify(eventSourceRegistry, times(0)).registerEventSource(any(EventSource.class));
131 @SuppressWarnings("unchecked")
132 private void onDataChangedTestHelper(final boolean create, final boolean update, final boolean isNetconf,
133 final String notificationCapabilityPrefix) throws Exception {
134 dataTreeModificationMock = mock(DataTreeModification.class);
135 DataObjectModification<Node> mockModification = mock(DataObjectModification.class);
136 doReturn(create ? DataObjectModification.ModificationType.WRITE :
137 DataObjectModification.ModificationType.SUBTREE_MODIFIED).when(mockModification).getModificationType();
138 doReturn(mockModification).when(dataTreeModificationMock).getRootNode();
141 String nodeId = "Node01";
143 node01 = NetconfTestUtils
144 .getNetconfNode(nodeId, "node01.test.local", ConnectionStatus.Connected,
145 notificationCapabilityPrefix);
148 node01 = NetconfTestUtils.getNode(nodeId);
151 doReturn(node01).when(mockModification).getDataAfter();
153 doReturn(DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL,
154 NetconfTestUtils.getInstanceIdentifier(node01))).when(dataTreeModificationMock).getRootPath();