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 static org.junit.Assert.assertNotNull;
11 import static org.mockito.ArgumentMatchers.any;
12 import static org.mockito.Mockito.doNothing;
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.util.concurrent.FluentFuture;
19 import java.util.Collections;
20 import org.junit.Before;
21 import org.junit.BeforeClass;
22 import org.junit.Test;
23 import org.opendaylight.mdsal.binding.api.DataBroker;
24 import org.opendaylight.mdsal.binding.api.DataObjectModification;
25 import org.opendaylight.mdsal.binding.api.DataTreeIdentifier;
26 import org.opendaylight.mdsal.binding.api.DataTreeModification;
27 import org.opendaylight.mdsal.binding.api.ReadTransaction;
28 import org.opendaylight.mdsal.binding.api.WriteTransaction;
29 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
30 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.NotificationPattern;
31 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.EventSourceService;
32 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicInput;
33 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicOutputBuilder;
34 import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicStatus;
35 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
36 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
37 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
38 import org.opendaylight.yangtools.yang.binding.DataObject;
39 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
40 import org.opendaylight.yangtools.yang.common.RpcResultBuilder;
42 public class EventSourceTopicTest {
44 EventSourceTopic eventSourceTopic;
45 DataBroker dataBrokerMock;
46 EventSourceService eventSourceServiceMock;
47 EventSourceTopology eventSourceTopologyMock;
50 public static void initTestClass() {
55 final NotificationPattern notificationPattern = new NotificationPattern("value1");
56 eventSourceServiceMock = mock(EventSourceService.class);
57 doReturn(RpcResultBuilder.success(new JoinTopicOutputBuilder().setStatus(JoinTopicStatus.Up).build())
58 .buildFuture()).when(eventSourceServiceMock).joinTopic(any(JoinTopicInput.class));
60 eventSourceTopologyMock = mock(EventSourceTopology.class);
61 dataBrokerMock = mock(DataBroker.class);
62 doReturn(eventSourceServiceMock).when(eventSourceTopologyMock).getEventSourceService();
63 doReturn(dataBrokerMock).when(eventSourceTopologyMock).getDataBroker();
65 WriteTransaction writeTransactionMock = mock(WriteTransaction.class);
66 doReturn(writeTransactionMock).when(dataBrokerMock).newWriteOnlyTransaction();
67 doNothing().when(writeTransactionMock).mergeParentStructurePut(any(LogicalDatastoreType.class),
68 any(InstanceIdentifier.class), any(DataObject.class));
69 FluentFuture checkedFutureWriteMock = mock(FluentFuture.class);
70 doReturn(checkedFutureWriteMock).when(writeTransactionMock).commit();
72 ReadTransaction readOnlyTransactionMock = mock(ReadTransaction.class);
73 doReturn(readOnlyTransactionMock).when(dataBrokerMock).newReadOnlyTransaction();
74 FluentFuture checkedFutureReadMock = mock(FluentFuture.class);
75 doReturn(checkedFutureReadMock).when(readOnlyTransactionMock).read(LogicalDatastoreType.OPERATIONAL,
76 EventSourceTopology.EVENT_SOURCE_TOPOLOGY_PATH);
77 eventSourceTopic = EventSourceTopic.create(notificationPattern, "nodeIdPattern1", eventSourceTopologyMock);
81 public void createModuleTest() {
82 assertNotNull("Instance has not been created correctly.", eventSourceTopic);
86 public void getTopicIdTest() {
87 assertNotNull("Topic has not been created correctly.", eventSourceTopic.getTopicId());
90 @SuppressWarnings("unchecked")
92 public void onDataTreeChangedTest() {
93 InstanceIdentifier<Node> instanceIdentifierMock = mock(InstanceIdentifier.class);
94 DataTreeModification<Node> mockDataTreeModification = mock(DataTreeModification.class);
95 DataObjectModification<Node> mockModification = mock(DataObjectModification.class);
96 doReturn(mockModification).when(mockDataTreeModification).getRootNode();
97 doReturn(DataTreeIdentifier.create(LogicalDatastoreType.OPERATIONAL, instanceIdentifierMock))
98 .when(mockDataTreeModification).getRootPath();
99 doReturn(DataObjectModification.ModificationType.WRITE).when(mockModification).getModificationType();
101 Node dataObjectNodeMock = mock(Node.class);
102 doReturn(getNodeKey("testNodeId01")).when(dataObjectNodeMock).key();
103 NodeId nodeIdMock = mock(NodeId.class);
104 doReturn(nodeIdMock).when(dataObjectNodeMock).getNodeId();
105 doReturn("nodeIdPattern1").when(nodeIdMock).getValue();
107 doReturn(dataObjectNodeMock).when(mockModification).getDataAfter();
109 eventSourceTopic.onDataTreeChanged(Collections.singletonList(mockDataTreeModification));
110 verify(dataObjectNodeMock).getNodeId();
111 verify(nodeIdMock).getValue();
115 public void notifyNodeTest() {
116 InstanceIdentifier instanceIdentifierMock = mock(InstanceIdentifier.class);
117 eventSourceTopic.notifyNode(instanceIdentifierMock);
118 verify(eventSourceServiceMock, times(1)).joinTopic(any(JoinTopicInput.class));
121 public NodeKey getNodeKey(final String nodeId) {
122 return new NodeKey(new NodeId(nodeId));