import static org.junit.Assert.assertNotNull;
import static org.mockito.Matchers.any;
+import static org.mockito.Matchers.eq;
+import static org.mockito.Mockito.doNothing;
import static org.mockito.Mockito.doReturn;
import static org.mockito.Mockito.mock;
import static org.mockito.Mockito.times;
import org.junit.Before;
import org.junit.BeforeClass;
import org.junit.Test;
+import org.opendaylight.controller.md.sal.binding.api.DataBroker;
+import org.opendaylight.controller.md.sal.binding.api.ReadOnlyTransaction;
+import org.opendaylight.controller.md.sal.binding.api.WriteTransaction;
import org.opendaylight.controller.md.sal.common.api.data.AsyncDataChangeEvent;
+import org.opendaylight.controller.md.sal.common.api.data.LogicalDatastoreType;
import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventaggregator.rev141202.NotificationPattern;
import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.EventSourceService;
import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.messagebus.eventsource.rev141202.JoinTopicInput;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.NodeId;
+import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
+import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
+import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
import org.opendaylight.yangtools.yang.binding.DataObject;
import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
+import com.google.common.util.concurrent.CheckedFuture;
+
public class EventSourceTopicTest {
EventSourceTopic eventSourceTopic;
- org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node dataObjectMock;
+ Node dataObjectNodeMock;
NodeId nodeIdMock;
+ DataBroker dataBrokerMock;
EventSourceService eventSourceServiceMock;
+ EventSourceTopology eventSourceTopologyMock;
@BeforeClass
public static void initTestClass() throws IllegalAccessException, InstantiationException {
public void setUp() throws Exception {
NotificationPattern notificationPattern = new NotificationPattern("value1");
eventSourceServiceMock = mock(EventSourceService.class);
- eventSourceTopic = new EventSourceTopic(notificationPattern, "nodeIdPattern1", eventSourceServiceMock);
+ eventSourceTopologyMock = mock(EventSourceTopology.class);
+ dataBrokerMock = mock(DataBroker.class);
+ doReturn(eventSourceServiceMock).when(eventSourceTopologyMock).getEventSourceService();
+ doReturn(dataBrokerMock).when(eventSourceTopologyMock).getDataBroker();
+
+ WriteTransaction writeTransactionMock = mock(WriteTransaction.class);
+ doReturn(writeTransactionMock).when(dataBrokerMock).newWriteOnlyTransaction();
+ doNothing().when(writeTransactionMock).put(any(LogicalDatastoreType.class), any(InstanceIdentifier.class), any(DataObject.class),eq(true));
+ CheckedFuture checkedFutureWriteMock = mock(CheckedFuture.class);
+ doReturn(checkedFutureWriteMock).when(writeTransactionMock).submit();
+
+ ReadOnlyTransaction readOnlyTransactionMock = mock(ReadOnlyTransaction.class);
+ doReturn(readOnlyTransactionMock).when(dataBrokerMock).newReadOnlyTransaction();
+ CheckedFuture checkedFutureReadMock = mock(CheckedFuture.class);
+ doReturn(checkedFutureReadMock).when(readOnlyTransactionMock).read(LogicalDatastoreType.OPERATIONAL, EventSourceTopology.EVENT_SOURCE_TOPOLOGY_PATH);
+ eventSourceTopic = EventSourceTopic.create(notificationPattern, "nodeIdPattern1", eventSourceTopologyMock);
}
@Test
AsyncDataChangeEvent asyncDataChangeEventMock = mock(AsyncDataChangeEvent.class);
onDataChangedTestHelper(asyncDataChangeEventMock);
eventSourceTopic.onDataChanged(asyncDataChangeEventMock);
- verify(dataObjectMock, times(1)).getId();
- verify(nodeIdMock, times(1)).getValue();
+ verify(dataObjectNodeMock, times(2)).getNodeId();
+ verify(nodeIdMock, times(2)).getValue();
}
private void onDataChangedTestHelper(AsyncDataChangeEvent asyncDataChangeEventMock){
Map<InstanceIdentifier<?>, DataObject> map = new HashMap<>();
InstanceIdentifier instanceIdentifierMock = mock(InstanceIdentifier.class);
- dataObjectMock = mock(org.opendaylight.yang.gen.v1.urn.opendaylight.inventory.rev130819.nodes.Node.class);
- map.put(instanceIdentifierMock, dataObjectMock);
+ dataObjectNodeMock = mock(Node.class);
+ doReturn(getNodeKey("testNodeId01")).when(dataObjectNodeMock).getKey();
+ map.put(instanceIdentifierMock, dataObjectNodeMock);
doReturn(map).when(asyncDataChangeEventMock).getUpdatedData();
-
+ doReturn(map).when(asyncDataChangeEventMock).getCreatedData();
nodeIdMock = mock(NodeId.class);
- doReturn(nodeIdMock).when(dataObjectMock).getId();
- doReturn("0").when(nodeIdMock).getValue();
+ doReturn(nodeIdMock).when(dataObjectNodeMock).getNodeId();
+ doReturn("nodeIdPattern1").when(nodeIdMock).getValue();
}
@Test
verify(eventSourceServiceMock, times(1)).joinTopic(any(JoinTopicInput.class));
}
-}
+ public NodeKey getNodeKey(String nodeId){
+ return new NodeKey(new NodeId(nodeId));
+ }
+}
\ No newline at end of file