networkmodel/src/test/java/org/opendaylight/transportpce/networkmodel/NetConfTopologyListenerTest.java

   1 /*
   2  * Copyright © 2020 Orange.  All rights reserved.
   3  *
   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
   7  */
   8
   9 package org.opendaylight.transportpce.networkmodel;
  10
  11 import static org.mockito.Mockito.anyString;
  12 import static org.mockito.Mockito.mock;
  13 import static org.mockito.Mockito.never;
  14 import static org.mockito.Mockito.times;
  15 import static org.mockito.Mockito.verify;
  16 import static org.mockito.Mockito.when;
  17 import static org.opendaylight.transportpce.common.StringConstants.OPENROADM_DEVICE_VERSION_2_2_1;
  18
  19 import java.util.ArrayList;
  20 import java.util.Collection;
  21 import java.util.HashSet;
  22 import java.util.List;
  23 import java.util.Map;
  24 import org.junit.Test;
  25 import org.junit.runner.RunWith;
  26 import org.mockito.Mock;
  27 import org.mockito.junit.MockitoJUnitRunner;
  28 import org.opendaylight.mdsal.binding.api.DataBroker;
  29 import org.opendaylight.mdsal.binding.api.DataObjectModification;
  30 import org.opendaylight.mdsal.binding.api.DataTreeModification;
  31 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
  32 import org.opendaylight.transportpce.common.mapping.PortMapping;
  33 import org.opendaylight.transportpce.networkmodel.dto.NodeRegistration;
  34 import org.opendaylight.transportpce.networkmodel.service.NetworkModelService;
  35 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Host;
  36 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
  37 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
  38 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
  39 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
  40 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeBuilder;
  41 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeConnectionStatus;
  42 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.AvailableCapabilities;
  43 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.AvailableCapabilitiesBuilder;
  44 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.available.capabilities.AvailableCapability;
  45 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.available.capabilities.AvailableCapabilityBuilder;
  46 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPasswordBuilder;
  47 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
  48 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
  49 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
  50 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeKey;
  51 import org.opendaylight.yangtools.yang.common.Uint16;
  52 import org.opendaylight.yangtools.yang.common.Uint32;
  53
  54 @RunWith(MockitoJUnitRunner.StrictStubs.class)
  55 public class NetConfTopologyListenerTest {
  56
  57     @Mock
  58     private NetworkModelService networkModelService;
  59     @Mock
  60     private DataBroker dataBroker;
  61     @Mock
  62     private DeviceTransactionManager deviceTransactionManager;
  63     @Mock
  64     private PortMapping portMapping;
  65     @Mock
  66     private Map<String, NodeRegistration> registrations;
  67
  68     @Test
  69     public void testOnDataTreeChangedWhenDeleteNode() {
  70         @SuppressWarnings("unchecked") final DataObjectModification<Node> node = mock(DataObjectModification.class);
  71         final Collection<DataTreeModification<Node>> changes = new HashSet<>();
  72         @SuppressWarnings("unchecked") final DataTreeModification<Node> ch = mock(DataTreeModification.class);
  73         final NodeRegistration nodeRegistration = mock(NodeRegistration.class);
  74         changes.add(ch);
  75         when(ch.getRootNode()).thenReturn(node);
  76
  77         final Node netconfNode = getNetconfNode("netconfNode1", NetconfNodeConnectionStatus.ConnectionStatus.Connecting,
  78             OPENROADM_DEVICE_VERSION_2_2_1);
  79         when(node.getModificationType()).thenReturn(DataObjectModification.ModificationType.DELETE);
  80         when(node.getDataBefore()).thenReturn(netconfNode);
  81         when(registrations.remove(anyString())).thenReturn(nodeRegistration);
  82
  83         NetConfTopologyListener listener = new NetConfTopologyListener(networkModelService, dataBroker,
  84             deviceTransactionManager, portMapping, registrations);
  85         listener.onDataTreeChanged(changes);
  86         verify(ch, times(1)).getRootNode();
  87         verify(node, times(1)).getModificationType();
  88         verify(node, times(3)).getDataBefore();
  89         verify(networkModelService, times(1)).deleteOpenRoadmnode(anyString());
  90         verify(nodeRegistration, times(1)).unregisterListeners();
  91     }
  92
  93     @Test
  94     public void testOnDataTreeChangedWhenAddNode() {
  95         @SuppressWarnings("unchecked") final DataObjectModification<Node> node = mock(DataObjectModification.class);
  96         final Collection<DataTreeModification<Node>> changes = new HashSet<>();
  97         @SuppressWarnings("unchecked") final DataTreeModification<Node> ch = mock(DataTreeModification.class);
  98         changes.add(ch);
  99         when(ch.getRootNode()).thenReturn(node);
 100
 101         final Node netconfNodeBefore = getNetconfNode("netconfNode1",
 102             NetconfNodeConnectionStatus.ConnectionStatus.Connecting, OPENROADM_DEVICE_VERSION_2_2_1);
 103         final Node netconfNodeAfter = getNetconfNode("netconfNode1",
 104             NetconfNodeConnectionStatus.ConnectionStatus.Connected, OPENROADM_DEVICE_VERSION_2_2_1);
 105         when(node.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
 106         when(node.getDataBefore()).thenReturn(netconfNodeBefore);
 107         when(node.getDataAfter()).thenReturn(netconfNodeAfter);
 108
 109         NetConfTopologyListener listener = new NetConfTopologyListener(networkModelService, dataBroker,
 110             deviceTransactionManager, portMapping);
 111         listener.onDataTreeChanged(changes);
 112         verify(ch, times(1)).getRootNode();
 113         verify(node, times(1)).getModificationType();
 114         verify(node, times(3)).getDataBefore();
 115         verify(node, times(1)).getDataAfter();
 116         verify(networkModelService, times(1)).createOpenRoadmNode(anyString(), anyString());
 117     }
 118
 119     @Test
 120     public void testOnDataTreeChangedWhenDisconnectingNode() {
 121         @SuppressWarnings("unchecked") final DataObjectModification<Node> node = mock(DataObjectModification.class);
 122         final Collection<DataTreeModification<Node>> changes = new HashSet<>();
 123         @SuppressWarnings("unchecked") final DataTreeModification<Node> ch = mock(DataTreeModification.class);
 124         changes.add(ch);
 125         when(ch.getRootNode()).thenReturn(node);
 126
 127         final Node netconfNodeBefore = getNetconfNode("netconfNode1",
 128             NetconfNodeConnectionStatus.ConnectionStatus.Connected, OPENROADM_DEVICE_VERSION_2_2_1);
 129         final Node netconfNodeAfter = getNetconfNode("netconfNode1",
 130             NetconfNodeConnectionStatus.ConnectionStatus.Connecting, OPENROADM_DEVICE_VERSION_2_2_1);
 131         when(node.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
 132         when(node.getDataBefore()).thenReturn(netconfNodeBefore);
 133         when(node.getDataAfter()).thenReturn(netconfNodeAfter);
 134
 135         NetConfTopologyListener listener = new NetConfTopologyListener(networkModelService, dataBroker,
 136             deviceTransactionManager, portMapping);
 137         listener.onDataTreeChanged(changes);
 138         verify(ch, times(1)).getRootNode();
 139         verify(node, times(1)).getModificationType();
 140         verify(node, times(3)).getDataBefore();
 141         verify(node, times(1)).getDataAfter();
 142         verify(networkModelService, never()).createOpenRoadmNode(anyString(), anyString());
 143         verify(networkModelService, never()).deleteOpenRoadmnode(anyString());
 144     }
 145
 146     @Test
 147     public void testOnDataTreeChangedWhenShouldNeverHappen() {
 148         @SuppressWarnings("unchecked") final DataObjectModification<Node> node = mock(DataObjectModification.class);
 149         final Collection<DataTreeModification<Node>> changes = new HashSet<>();
 150         @SuppressWarnings("unchecked") final DataTreeModification<Node> ch = mock(DataTreeModification.class);
 151         changes.add(ch);
 152         when(ch.getRootNode()).thenReturn(node);
 153
 154         final Node netconfNodeBefore = getNetconfNode("netconfNode1",
 155             NetconfNodeConnectionStatus.ConnectionStatus.Connected, OPENROADM_DEVICE_VERSION_2_2_1);
 156         when(node.getModificationType()).thenReturn(DataObjectModification.ModificationType.SUBTREE_MODIFIED);
 157         when(node.getDataBefore()).thenReturn(netconfNodeBefore);
 158
 159         NetConfTopologyListener listener = new NetConfTopologyListener(networkModelService, dataBroker,
 160             deviceTransactionManager, portMapping);
 161         listener.onDataTreeChanged(changes);
 162         verify(ch, times(1)).getRootNode();
 163         verify(node, times(2)).getModificationType();
 164         verify(node, times(3)).getDataBefore();
 165         verify(node, never()).getDataAfter();
 166         verify(networkModelService, never()).createOpenRoadmNode(anyString(), anyString());
 167         verify(networkModelService, never()).deleteOpenRoadmnode(anyString());
 168     }
 169
 170     private Node getNetconfNode(final String nodeId, final NetconfNodeConnectionStatus.ConnectionStatus cs,
 171         final String openRoadmVersion) {
 172         final List<AvailableCapability> avCapList = new ArrayList<>();
 173         avCapList.add(new AvailableCapabilityBuilder()
 174             .setCapabilityOrigin(AvailableCapability.CapabilityOrigin.UserDefined)
 175             .setCapability(openRoadmVersion)
 176             .build());
 177         final AvailableCapabilities avCaps = new AvailableCapabilitiesBuilder().setAvailableCapability(avCapList)
 178             .build();
 179         final NetconfNode netconfNode = new NetconfNodeBuilder()
 180             .setConnectionStatus(cs)
 181             .setAvailableCapabilities(avCaps)
 182             .setHost(new Host(new IpAddress(new Ipv4Address("127.0.0.1"))))
 183             .setPort(new PortNumber(Uint16.valueOf(9999)))
 184             .setReconnectOnChangedSchema(true)
 185             .setDefaultRequestTimeoutMillis(Uint32.valueOf(1000))
 186             .setBetweenAttemptsTimeoutMillis(Uint16.valueOf(100))
 187             .setKeepaliveDelay(Uint32.valueOf(1000))
 188             .setTcpOnly(true)
 189             .setCredentials(new LoginPasswordBuilder()
 190                 .setUsername("testuser")
 191                 .setPassword("testpassword")
 192                 .build())
 193             .build();
 194         return new NodeBuilder()
 195             .withKey(new NodeKey(new NodeId(nodeId)))
 196             .setNodeId(new NodeId(nodeId))
 197             .addAugmentation(netconfNode)
 198             .build();
 199     }
 200 }