Add NetConfTopologyListenerTest
[transportpce.git] / 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.mock;
12 import static org.mockito.Mockito.times;
13 import static org.mockito.Mockito.verify;
14 import static org.mockito.Mockito.when;
15 import static org.opendaylight.transportpce.common.StringConstants.OPENROADM_DEVICE_VERSION_1_2_1;
16 import static org.opendaylight.transportpce.common.StringConstants.OPENROADM_DEVICE_VERSION_2_2_1;
17
18 import java.util.ArrayList;
19 import java.util.Collection;
20 import java.util.HashSet;
21 import java.util.List;
22 import org.junit.Before;
23 import org.junit.Test;
24 import org.opendaylight.mdsal.binding.api.DataBroker;
25 import org.opendaylight.mdsal.binding.api.DataObjectModification;
26 import org.opendaylight.mdsal.binding.api.DataTreeModification;
27 import org.opendaylight.mdsal.binding.api.MountPoint;
28 import org.opendaylight.mdsal.binding.api.MountPointService;
29 import org.opendaylight.transportpce.common.DataStoreContextImpl;
30 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
31 import org.opendaylight.transportpce.common.device.DeviceTransactionManagerImpl;
32 import org.opendaylight.transportpce.common.mapping.MappingUtils;
33 import org.opendaylight.transportpce.common.mapping.MappingUtilsImpl;
34 import org.opendaylight.transportpce.common.mapping.PortMapping;
35 import org.opendaylight.transportpce.common.mapping.PortMappingImpl;
36 import org.opendaylight.transportpce.common.mapping.PortMappingVersion121;
37 import org.opendaylight.transportpce.common.mapping.PortMappingVersion221;
38 import org.opendaylight.transportpce.common.network.NetworkTransactionImpl;
39 import org.opendaylight.transportpce.common.network.NetworkTransactionService;
40 import org.opendaylight.transportpce.common.network.RequestProcessor;
41 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl;
42 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl121;
43 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl221;
44 import org.opendaylight.transportpce.networkmodel.service.NetworkModelService;
45 import org.opendaylight.transportpce.networkmodel.stub.MountPointServiceStub;
46 import org.opendaylight.transportpce.networkmodel.stub.MountPointStub;
47 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Host;
48 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.IpAddress;
49 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.Ipv4Address;
50 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev130715.PortNumber;
51 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNode;
52 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeBuilder;
53 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.NetconfNodeConnectionStatus;
54 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.AvailableCapabilities;
55 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.AvailableCapabilitiesBuilder;
56 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.available.capabilities.AvailableCapability;
57 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.connection.status.available.capabilities.AvailableCapabilityBuilder;
58 import org.opendaylight.yang.gen.v1.urn.opendaylight.netconf.node.topology.rev150114.netconf.node.credentials.credentials.LoginPasswordBuilder;
59 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.NodeId;
60 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.Node;
61 import org.opendaylight.yang.gen.v1.urn.tbd.params.xml.ns.yang.network.topology.rev131021.network.topology.topology.NodeBuilder;
62 import org.opendaylight.yangtools.yang.common.Uint16;
63 import org.opendaylight.yangtools.yang.common.Uint32;
64
65
66 public class NetConfTopologyListenerTest {
67
68     @Before
69     public void setUp() {
70
71     }
72
73     @Test
74     public void testOnDataTreeChanged() {
75
76         final DataObjectModification<Node> newNode = mock(DataObjectModification.class);
77         when(newNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.WRITE);
78         final Collection<DataTreeModification<Node>> changes = new HashSet<>();
79         final DataTreeModification<Node> ch = mock(DataTreeModification.class);
80         changes.add(ch);
81         when(ch.getRootNode()).thenReturn(newNode);
82
83         DataStoreContextImpl dataStoreContext = new DataStoreContextImpl();
84         DataBroker dataBroker = dataStoreContext.getDataBroker();
85         RequestProcessor requestProcessor = new RequestProcessor(dataBroker);
86         NetworkTransactionService networkTransactionService = new NetworkTransactionImpl(requestProcessor);
87         MountPoint mountPoint = new MountPointStub(dataBroker);
88         MountPointService mountPointService = new MountPointServiceStub(mountPoint);
89         DeviceTransactionManager deviceTransactionManager =
90                 new DeviceTransactionManagerImpl(mountPointService, 3000);
91         R2RLinkDiscovery linkDiskovery = new R2RLinkDiscovery(
92                 dataBroker, deviceTransactionManager, networkTransactionService);
93         OpenRoadmInterfacesImpl121 openRoadmInterfacesImpl121 =
94                 new OpenRoadmInterfacesImpl121(deviceTransactionManager);
95         OpenRoadmInterfacesImpl221 openRoadmInterfacesImpl221 =
96                 new OpenRoadmInterfacesImpl221(deviceTransactionManager);
97         MappingUtils mappingUtils = new MappingUtilsImpl(dataBroker);
98         OpenRoadmInterfacesImpl openRoadmInterfaces =
99                 new OpenRoadmInterfacesImpl(deviceTransactionManager, mappingUtils,
100                 openRoadmInterfacesImpl121, openRoadmInterfacesImpl221);
101         PortMappingVersion121 p1 = new PortMappingVersion121(dataBroker, deviceTransactionManager, openRoadmInterfaces);
102         PortMappingVersion221 p2 = new PortMappingVersion221(dataBroker, deviceTransactionManager, openRoadmInterfaces);
103         PortMapping portMapping = new PortMappingImpl(dataBroker, p2, p1);
104         NetworkModelService networkModelService = mock(NetworkModelService.class);
105
106         //Start Netconf Topology listener and start adding nodes to the Netconf Topology to verify behaviour
107         NetConfTopologyListener listener = new NetConfTopologyListener(networkModelService, dataBroker,
108                 deviceTransactionManager);
109
110         //A new node appears in Netconf Topology, status is Connecting
111         final Node netconfNode = getNetconfNode("test1",
112                 NetconfNodeConnectionStatus.ConnectionStatus.Connecting, OPENROADM_DEVICE_VERSION_2_2_1);
113         when(newNode.getDataAfter()).thenReturn(netconfNode);
114         listener.onDataTreeChanged(changes);
115         verify(ch).getRootNode();
116         verify(newNode, times(3)).getDataAfter();
117         verify(newNode, times(2)).getModificationType();
118
119         //A new node appears in Netconf Topology, status is Connected, version is 121
120         final Node netconfNode2 = getNetconfNode("test2", NetconfNodeConnectionStatus.ConnectionStatus.Connected,
121                 OPENROADM_DEVICE_VERSION_1_2_1);
122         when(newNode.getDataAfter()).thenReturn(netconfNode2);
123         listener.onDataTreeChanged(changes);
124         verify(ch, times(2)).getRootNode();
125         verify(newNode, times(6)).getDataAfter();
126         verify(newNode, times(4)).getModificationType();
127
128         //A new node appears in Netconf Topology, status is Connected, version is 221
129         final Node netconfNode3 = getNetconfNode("test3", NetconfNodeConnectionStatus.ConnectionStatus.Connected,
130                 OPENROADM_DEVICE_VERSION_2_2_1);
131         when(newNode.getDataAfter()).thenReturn(netconfNode3);
132         listener.onDataTreeChanged(changes);
133         verify(ch, times(3)).getRootNode();
134         verify(newNode, times(9)).getDataAfter();
135         verify(newNode, times(6)).getModificationType();
136
137         //A new node is deleted from Netconf Topology, Data Before was empty
138         when(newNode.getModificationType()).thenReturn(DataObjectModification.ModificationType.DELETE);
139         when(newNode.getDataBefore()).thenReturn(netconfNode3);
140         listener.onDataTreeChanged(changes);
141         verify(ch, times(4)).getRootNode();
142         verify(newNode, times(10)).getDataAfter();
143         verify(newNode, times(7)).getModificationType();
144
145     }
146
147     public Node getNetconfNode(final String nodeId, final NetconfNodeConnectionStatus.ConnectionStatus cs,
148                                final String openRoadmVersion) {
149         final List<AvailableCapability> avCapList = new ArrayList<>();
150         avCapList.add(new AvailableCapabilityBuilder()
151                 .setCapabilityOrigin(AvailableCapability.CapabilityOrigin.UserDefined)
152                 .setCapability(openRoadmVersion)
153                 .build());
154         final AvailableCapabilities avCaps =
155                 new AvailableCapabilitiesBuilder().setAvailableCapability(avCapList).build();
156         final NetconfNode netconfNode = new NetconfNodeBuilder()
157                 .setConnectionStatus(cs)
158                 .setAvailableCapabilities(avCaps)
159                 .setHost(new Host(new IpAddress(new Ipv4Address("127.0.0.1"))))
160                 .setPort(new PortNumber(Uint16.valueOf(9999)))
161                 .setReconnectOnChangedSchema(true)
162                 .setDefaultRequestTimeoutMillis(Uint32.valueOf(1000))
163                 .setBetweenAttemptsTimeoutMillis(Uint16.valueOf(100))
164                 .setKeepaliveDelay(Uint32.valueOf(1000))
165                 .setTcpOnly(true)
166                 .setCredentials(new LoginPasswordBuilder()
167                         .setUsername("testuser")
168                         .setPassword("testpassword")
169                         .build())
170                 .build();
171         final NodeBuilder nn = new NodeBuilder().setNodeId(new NodeId(nodeId))
172                 .addAugmentation(NetconfNode.class, netconfNode);
173         return nn.build();
174
175     }
176 }