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