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