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