257ef1e25ef48cdee71ccefc10027ef8dc621404
[transportpce.git] / olm / src / test / java / org / opendaylight / transportpce / olm / service / OlmPowerServiceImplSpanLossBaseTest.java
1 /*
2  * Copyright © 2018 Orange, Inc. and others.  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.olm.service;
10
11 import java.math.BigDecimal;
12 import java.util.Optional;
13
14 import org.junit.Assert;
15 import org.junit.Before;
16 import org.junit.Test;
17 import org.mockito.Mockito;
18 import org.mockito.MockitoAnnotations;
19 import org.opendaylight.mdsal.binding.api.DataBroker;
20 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
21 import org.opendaylight.transportpce.common.NetworkUtils;
22 import org.opendaylight.transportpce.common.StringConstants;
23 import org.opendaylight.transportpce.common.Timeouts;
24 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
25 import org.opendaylight.transportpce.common.mapping.MappingUtils;
26 import org.opendaylight.transportpce.common.mapping.PortMapping;
27 import org.opendaylight.transportpce.common.mapping.PortMappingImpl;
28 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfaceException;
29 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfaces;
30 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl;
31 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl121;
32 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl221;
33 import org.opendaylight.transportpce.olm.power.PowerMgmt;
34 import org.opendaylight.transportpce.olm.power.PowerMgmtImpl;
35 import org.opendaylight.transportpce.olm.util.OlmPowerServiceRpcImplUtil;
36 import org.opendaylight.transportpce.olm.util.OlmTransactionUtils;
37 import org.opendaylight.transportpce.test.AbstractTest;
38 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev170418.CalculateSpanlossBaseInput;
39 import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev170418.CalculateSpanlossBaseOutput;
40 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.types.rev181019.RatioDB;
41 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.interfaces.grp.Interface;
42 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.interfaces.grp.InterfaceBuilder;
43 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.interfaces.grp.InterfaceKey;
44 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.org.openroadm.device.container.OrgOpenroadmDevice;
45 import org.opendaylight.yang.gen.v1.http.org.openroadm.interfaces.rev170626.OpticalTransport;
46 import org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev181019.Interface1;
47 import org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev181019.Interface1Builder;
48 import org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev181019.ots.container.Ots;
49 import org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev181019.ots.container.OtsBuilder;
50 import org.opendaylight.yang.gen.v1.http.org.openroadm.pm.rev181019.CurrentPmList;
51 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NetworkId;
52 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.Networks;
53 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.Network;
54 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.networks.NetworkKey;
55 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
56 import org.slf4j.Logger;
57 import org.slf4j.LoggerFactory;
58
59 public class OlmPowerServiceImplSpanLossBaseTest extends AbstractTest {
60
61     private static final Logger LOG = LoggerFactory.getLogger(OlmPowerServiceImplSpanLossBaseTest.class);
62     private DeviceTransactionManager deviceTransactionManager;
63     private OpenRoadmInterfaces openRoadmInterfaces;
64     private PortMapping portMapping;
65     private PowerMgmt powerMgmt;
66     private MappingUtils mappingUtils;
67     private OpenRoadmInterfacesImpl221 openRoadmInterfacesImpl221;
68     private OpenRoadmInterfacesImpl121 openRoadmInterfacesImpl121;
69     private OlmPowerService olmPowerService;
70     private DataBroker dataBroker;
71
72     @Before
73     public void setUp() throws OpenRoadmInterfaceException {
74         this.openRoadmInterfacesImpl121 = Mockito.mock(OpenRoadmInterfacesImpl121.class);
75         this.openRoadmInterfacesImpl221 = Mockito.mock(OpenRoadmInterfacesImpl221.class);
76         this.mappingUtils = Mockito.mock(MappingUtils.class);
77         this.portMapping = Mockito.mock(PortMappingImpl.class);
78         this.deviceTransactionManager = Mockito.mock(DeviceTransactionManager.class);
79         this.powerMgmt = Mockito.mock(PowerMgmtImpl.class);
80         this.openRoadmInterfaces = new OpenRoadmInterfacesImpl(this.deviceTransactionManager, this.mappingUtils,
81                 this.openRoadmInterfacesImpl121, this.openRoadmInterfacesImpl221);
82         this.olmPowerService = new OlmPowerServiceImpl(this.getDataBroker(), this.powerMgmt,
83                 this.deviceTransactionManager, this.portMapping, this.mappingUtils, this.openRoadmInterfaces);
84         this.dataBroker = this.getDataBroker();
85         MockitoAnnotations.initMocks(this);
86         Mockito.doReturn(StringConstants.OPENROADM_DEVICE_VERSION_2_2_1).when(this.mappingUtils)
87                 .getOpenRoadmVersion(Mockito.anyString());
88
89         Mockito.when(this.portMapping.getMapping("ROADM-A1", "DEG2-TTP-TXRX"))
90                 .thenReturn(OlmTransactionUtils.getMapping1());
91         Mockito.when(this.portMapping.getMapping("ROADM-C1", "DEG1-TTP-TXRX"))
92                 .thenReturn(OlmTransactionUtils.getMapping2());
93
94         InstanceIdentifier<CurrentPmList> iidCurrentPmList = InstanceIdentifier.create(CurrentPmList.class);
95         Mockito.when(this.deviceTransactionManager.getDataFromDevice("ROADM-A1", LogicalDatastoreType.OPERATIONAL,
96                 iidCurrentPmList, Timeouts.DEVICE_READ_TIMEOUT,
97                 Timeouts.DEVICE_READ_TIMEOUT_UNIT)).thenReturn(OlmTransactionUtils.getCurrentPmListA());
98         Mockito.when(this.deviceTransactionManager.getDataFromDevice("ROADM-C1", LogicalDatastoreType.OPERATIONAL,
99                 iidCurrentPmList, Timeouts.DEVICE_READ_TIMEOUT,
100                 Timeouts.DEVICE_READ_TIMEOUT_UNIT)).thenReturn(OlmTransactionUtils.getCurrentPmListC());
101
102         InstanceIdentifier<Interface> interfacesIIDA = InstanceIdentifier.create(OrgOpenroadmDevice.class)
103                 .child(Interface.class, new InterfaceKey("OTS-DEG2-TTP-TXRX"));
104         InstanceIdentifier<Interface> interfacesIIDC = InstanceIdentifier.create(OrgOpenroadmDevice.class)
105                 .child(Interface.class, new InterfaceKey("OTS-DEG1-TTP-TXRX"));
106         Optional<Interface> interfaceA = Optional.of(new InterfaceBuilder().setName("OTS-DEG2-TTP-TXRX").build());
107         Optional<Interface> interfaceC = Optional.of(new InterfaceBuilder().setName("OTS-DEG1-TTP-TXRX").build());
108         Mockito.when(this.deviceTransactionManager.getDataFromDevice("ROADM-A1", LogicalDatastoreType.CONFIGURATION,
109                 interfacesIIDA, Timeouts.DEVICE_READ_TIMEOUT,
110                 Timeouts.DEVICE_READ_TIMEOUT_UNIT)).thenReturn(interfaceA);
111         Mockito.when(this.deviceTransactionManager.getDataFromDevice("ROADM-C1", LogicalDatastoreType.CONFIGURATION,
112                 interfacesIIDC, Timeouts.DEVICE_READ_TIMEOUT,
113                 Timeouts.DEVICE_READ_TIMEOUT_UNIT)).thenReturn(interfaceC);
114
115         Ots otsValue = new OtsBuilder().setSpanLossTransmit(new RatioDB(new BigDecimal(25)))
116                 .setSpanLossReceive(new RatioDB(new BigDecimal(25))).build();
117         Interface1 ots = new Interface1Builder().setOts(otsValue).build();
118         Interface interA = new InterfaceBuilder().setName("OTS-DEG2-TTP-TXRX").setType(OpticalTransport.class)
119                 .addAugmentation(Interface1.class, ots).build();
120         Interface interC = new InterfaceBuilder().setName("OTS-DEG1-TTP-TXRX").setType(OpticalTransport.class)
121                 .addAugmentation(Interface1.class, ots).build();
122         Optional<Interface> interOptA = Optional.of(interA);
123         Optional<Interface> interOptC = Optional.of(interC);
124         Mockito.when(this.openRoadmInterfacesImpl221.getInterface("ROADM-A1", "OTS-DEG2-TTP-TXRX"))
125                 .thenReturn(interOptA);
126         Mockito.when(this.openRoadmInterfacesImpl221.getInterface("ROADM-C1", "OTS-DEG1-TTP-TXRX"))
127                 .thenReturn(interOptC);
128
129         InterfaceBuilder ifABldr = new InterfaceBuilder();
130         Mockito.doNothing().when(this.openRoadmInterfacesImpl221).postInterface("ROADM-A1", ifABldr);
131         Mockito.doNothing().when(this.openRoadmInterfacesImpl221).postInterface("ROADM-C1", ifABldr);
132
133     }
134
135     @Test
136     public void testCalculateSpanlossBaseLink() {
137         // initialise and store openroadm-topology in datastore
138         NetworkKey overlayTopologyKey = new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID));
139         InstanceIdentifier<Network> ietfNetworkIID = InstanceIdentifier.builder(Networks.class)
140                 .child(Network.class, overlayTopologyKey).build();
141         Network openroadmTopology = OlmTransactionUtils.getNetworkForSpanLoss();
142         OlmTransactionUtils.writeTransaction(this.dataBroker, ietfNetworkIID, openroadmTopology);
143         try {
144             Thread.sleep(1000);
145         } catch (InterruptedException e) {
146             LOG.error("Write transaction failed !", e);
147         }
148
149         CalculateSpanlossBaseInput input = OlmPowerServiceRpcImplUtil.getCalculateSpanlossBaseInputLink();
150         CalculateSpanlossBaseOutput output = this.olmPowerService.calculateSpanlossBase(input);
151
152         Assert.assertEquals("Success", output.getResult());
153         Assert.assertEquals("ROADM-A1-to-ROADM-C1", output.getSpans().get(0).getLinkId().getValue());
154         Assert.assertEquals("15", output.getSpans().get(0).getSpanloss());
155     }
156
157     @Test
158     public void testCalculateSpanlossBaseAll() {
159
160         // initialise and store openroadm-topology in datastore
161         NetworkKey overlayTopologyKey = new NetworkKey(new NetworkId(NetworkUtils.OVERLAY_NETWORK_ID));
162         InstanceIdentifier<Network> ietfNetworkIID = InstanceIdentifier.builder(Networks.class)
163                 .child(Network.class, overlayTopologyKey).build();
164         Network openroadmTopology = OlmTransactionUtils.getNetworkForSpanLoss();
165         OlmTransactionUtils.writeTransaction(this.dataBroker, ietfNetworkIID, openroadmTopology);
166         try {
167             Thread.sleep(1000);
168         } catch (InterruptedException e) {
169             LOG.error("Write transaction failed !", e);
170         }
171
172         CalculateSpanlossBaseInput input = OlmPowerServiceRpcImplUtil.getCalculateSpanlossBaseInputAll();
173         CalculateSpanlossBaseOutput output = this.olmPowerService.calculateSpanlossBase(input);
174
175         Assert.assertEquals("Success", output.getResult());
176         Assert.assertEquals("ROADM-A1-to-ROADM-C1", output.getSpans().get(0).getLinkId().getValue());
177         Assert.assertEquals("15", output.getSpans().get(0).getSpanloss());
178     }
179 }