common/src/test/java/org/opendaylight/transportpce/common/crossconnect/CrossConnectImpl221Test.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.common.crossconnect;
  10
  11 import static org.mockito.ArgumentMatchers.any;
  12 import static org.mockito.Mockito.mock;
  13
  14 import java.math.BigDecimal;
  15 import java.util.List;
  16 import java.util.Optional;
  17 import java.util.concurrent.CompletableFuture;
  18 import org.junit.Assert;
  19 import org.junit.Before;
  20 import org.junit.Test;
  21 import org.mockito.Mockito;
  22 import org.opendaylight.mdsal.binding.api.DataBroker;
  23 import org.opendaylight.mdsal.binding.api.MountPoint;
  24 import org.opendaylight.mdsal.binding.api.MountPointService;
  25 import org.opendaylight.mdsal.binding.api.ReadWriteTransaction;
  26 import org.opendaylight.mdsal.common.api.LogicalDatastoreType;
  27 import org.opendaylight.transportpce.common.Timeouts;
  28 import org.opendaylight.transportpce.common.device.DeviceTransaction;
  29 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
  30 import org.opendaylight.transportpce.common.device.DeviceTransactionManagerImpl;
  31 import org.opendaylight.transportpce.common.fixedflex.SpectrumInformation;
  32 import org.opendaylight.yang.gen.v1.http.org.openroadm.common.types.rev181019.OpticalControlMode;
  33 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.org.openroadm.device.container.OrgOpenroadmDevice;
  34 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.org.openroadm.device.container.org.openroadm.device.RoadmConnections;
  35 import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev181019.org.openroadm.device.container.org.openroadm.device.RoadmConnectionsKey;
  36 import org.opendaylight.yang.gen.v1.http.org.transportpce.common.types.rev210930.otn.renderer.nodes.Nodes;
  37 import org.opendaylight.yangtools.util.concurrent.FluentFutures;
  38 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  39 import org.opendaylight.yangtools.yang.common.Uint32;
  40
  41 public class CrossConnectImpl221Test {
  42     private CrossConnectImpl221 crossConnectImpl221 = null;
  43     private DeviceTransactionManager deviceTransactionManager = null;
  44
  45     private MountPointService mountPointServiceMock = mock(MountPointService.class);
  46     private MountPoint mountPointMock = mock(MountPoint.class);
  47     private DataBroker dataBrokerMock = mock(DataBroker.class);
  48     private ReadWriteTransaction rwTransactionMock = mock(ReadWriteTransaction.class);
  49
  50     @Before
  51     public void setup() {
  52         deviceTransactionManager = mock(DeviceTransactionManager.class);
  53         crossConnectImpl221 = new CrossConnectImpl221(deviceTransactionManager);
  54
  55         //mock responses for deviceTransactionManager calls
  56         InstanceIdentifier<RoadmConnections> deviceIID =
  57                 InstanceIdentifier
  58                         .create(OrgOpenroadmDevice.class)
  59                         .child(RoadmConnections.class, new RoadmConnectionsKey("1"));
  60
  61         Mockito.when(deviceTransactionManager.getDataFromDevice("deviceId",
  62                 LogicalDatastoreType.CONFIGURATION, deviceIID,
  63                 Timeouts.DEVICE_READ_TIMEOUT, Timeouts.DEVICE_READ_TIMEOUT_UNIT))
  64             .thenReturn(Optional.of(mock(RoadmConnections.class)));
  65     }
  66
  67     @Test
  68     public void getCrossConnectTest() {
  69         Optional<RoadmConnections> res =
  70                 crossConnectImpl221.getCrossConnect("deviceId", "1");
  71         Assert.assertTrue("Optional object should have a value", res.isPresent());
  72     }
  73
  74     @Test
  75     public void postCrossConnectTest() {
  76         Mockito.when(mountPointServiceMock.getMountPoint(any())).thenReturn(Optional.of(mountPointMock));
  77         Mockito.when(mountPointMock.getService(any())).thenReturn(Optional.of(dataBrokerMock));
  78         Mockito.when(dataBrokerMock.newReadWriteTransaction()).thenReturn(rwTransactionMock);
  79         Mockito.when(rwTransactionMock.commit()).thenReturn(FluentFutures.immediateNullFluentFuture());
  80         deviceTransactionManager = new DeviceTransactionManagerImpl(mountPointServiceMock, 3000);
  81         crossConnectImpl221 = new CrossConnectImpl221(deviceTransactionManager);
  82         SpectrumInformation spectrumInformation = new SpectrumInformation();
  83         spectrumInformation.setWaveLength(Uint32.valueOf(1));
  84         spectrumInformation.setLowerSpectralSlotNumber(761);
  85         spectrumInformation.setHigherSpectralSlotNumber(768);
  86         Optional<String> res = crossConnectImpl221.postCrossConnect("deviceId", "srcTp", "destTp", spectrumInformation);
  87         Assert.assertEquals(res.get(), "srcTp-destTp-761:768");
  88     }
  89
  90     @Test(expected = NullPointerException.class)
  91     public void setPowerLevelTest() {
  92         InstanceIdentifier<RoadmConnections> deviceIID =
  93                 InstanceIdentifier.create(OrgOpenroadmDevice.class)
  94                         .child(RoadmConnections.class, new RoadmConnectionsKey("1"));
  95         Mockito.when(deviceTransactionManager.getDataFromDevice("deviceId",
  96                 LogicalDatastoreType.OPERATIONAL, deviceIID,
  97                 Timeouts.DEVICE_READ_TIMEOUT, Timeouts.DEVICE_READ_TIMEOUT_UNIT))
  98             .thenReturn(Optional.of(mock(RoadmConnections.class)));
  99
 100         Mockito.when(deviceTransactionManager.getDeviceTransaction("deviceId"))
 101             .thenReturn(CompletableFuture.completedFuture(Optional.of(mock(DeviceTransaction.class))));
 102         crossConnectImpl221.setPowerLevel("deviceId", OpticalControlMode.GainLoss, new BigDecimal(100), "1");
 103
 104         Assert.assertTrue("set Level should be true", true);
 105     }
 106
 107     @Test(expected = NullPointerException.class)
 108     public void postOtnCrossConnect() {
 109         Nodes nodes = Mockito.mock(Nodes.class);
 110         Mockito.when(nodes.getNodeId()).thenReturn("nodeId");
 111         Mockito.when(deviceTransactionManager.getDeviceTransaction(any()))
 112             .thenReturn(CompletableFuture.completedFuture(Optional.of(mock(DeviceTransaction.class))));
 113         Optional<String> res = crossConnectImpl221.postOtnCrossConnect(List.of("src1", "src2"), nodes);
 114         Assert.assertTrue("Optional value should have a value", res.isPresent());
 115     }
 116 }