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
9 package org.opendaylight.groupbasedpolicy.renderer.ios_xe_provider.impl.listener;
11 import java.util.Collections;
12 import org.junit.After;
13 import org.junit.Before;
14 import org.junit.Test;
15 import org.junit.runner.RunWith;
16 import org.mockito.Matchers;
17 import org.mockito.Mock;
18 import org.mockito.Mockito;
19 import org.mockito.runners.MockitoJUnitRunner;
20 import org.opendaylight.controller.md.sal.binding.api.DataBroker;
21 import org.opendaylight.controller.md.sal.binding.api.DataObjectModification;
22 import org.opendaylight.controller.md.sal.binding.api.DataTreeChangeListener;
23 import org.opendaylight.controller.md.sal.binding.api.DataTreeIdentifier;
24 import org.opendaylight.controller.md.sal.binding.api.DataTreeModification;
25 import org.opendaylight.groupbasedpolicy.renderer.ios_xe_provider.impl.manager.NodeManager;
26 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NetworkTopology;
27 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
28 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
29 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
30 import org.opendaylight.yangtools.concepts.ListenerRegistration;
33 * Test for {@link IosXeCapableNodeListenerImpl}.
35 @RunWith(MockitoJUnitRunner.class)
36 public class IosXeCapableNodeListenerImplTest {
39 private DataBroker dataBroker;
41 private NodeManager nodeManager;
43 private ListenerRegistration<DataTreeChangeListener<Node>> listenerRegistration;
45 private DataTreeModification<Node> dataTreeModification;
47 private DataObjectModification<Node> rootNode;
49 private IosXeCapableNodeListenerImpl listener;
52 public void setUp() throws Exception {
53 Mockito.when(dataBroker.registerDataTreeChangeListener(
54 Matchers.<DataTreeIdentifier<Node>>any(),
55 Matchers.<DataTreeChangeListener<Node>>any()))
56 .thenReturn(listenerRegistration);
57 listener = new IosXeCapableNodeListenerImpl(dataBroker, nodeManager);
58 Mockito.verify(dataBroker).registerDataTreeChangeListener(
59 Matchers.<DataTreeIdentifier<NetworkTopology>>any(),
60 Matchers.<DataTreeChangeListener<NetworkTopology>>any());
64 public void tearDown() throws Exception {
65 Mockito.verifyNoMoreInteractions(dataBroker, nodeManager, listenerRegistration);
69 public void testOnDataTreeChanged_add() throws Exception {
70 final Node topologyNode = createNetworkTopologyNode("topology-node-id-1");
71 Mockito.when(dataTreeModification.getRootNode()).thenReturn(rootNode);
72 Mockito.when(rootNode.getDataBefore()).thenReturn(topologyNode);
73 Mockito.when(rootNode.getDataAfter()).thenReturn(null);
75 listener.onDataTreeChanged(Collections.singleton(dataTreeModification));
76 Mockito.verify(nodeManager).syncNodes(null, topologyNode);
81 public void testOnDataTreeChanged_update() throws Exception {
82 final Node topologyNodeBefore = createNetworkTopologyNode("topology-node-id-1");
83 final Node topologyNodeAfter = createNetworkTopologyNode("topology-node-id-2");
84 Mockito.when(dataTreeModification.getRootNode()).thenReturn(rootNode);
85 Mockito.when(rootNode.getDataBefore()).thenReturn(topologyNodeBefore);
86 Mockito.when(rootNode.getDataAfter()).thenReturn(topologyNodeAfter);
88 listener.onDataTreeChanged(Collections.singleton(dataTreeModification));
89 Mockito.verify(nodeManager).syncNodes(topologyNodeAfter, topologyNodeBefore);
93 public void testOnDataTreeChanged_remove() throws Exception {
94 final Node topologyNode = createNetworkTopologyNode("topology-node-id-2");
95 Mockito.when(dataTreeModification.getRootNode()).thenReturn(rootNode);
96 Mockito.when(rootNode.getDataBefore()).thenReturn(null);
97 Mockito.when(rootNode.getDataAfter()).thenReturn(topologyNode);
99 listener.onDataTreeChanged(Collections.singleton(dataTreeModification));
100 Mockito.verify(nodeManager).syncNodes(topologyNode, null);
103 private Node createNetworkTopologyNode(final String nodeId) {
104 return new NodeBuilder()
105 .setNodeId(new NodeId(nodeId))
110 public void testClose() throws Exception {
111 Mockito.verify(listenerRegistration, Mockito.never()).close();
113 Mockito.verify(listenerRegistration).close();