Convert PowerMgmtImpl of olm into a Component
[transportpce.git] / olm / src / test / java / org / opendaylight / transportpce / olm / power / PowerMgmtTest.java
index 7f51916f8d9e4d6618fa150bb3fd973025ace670..9c6a562b1f3db301f29eafbf5c3570cffc1ee67a 100644 (file)
 
 package org.opendaylight.transportpce.olm.power;
 
+import static org.junit.jupiter.api.Assertions.assertEquals;
+import static org.mockito.ArgumentMatchers.eq;
+import static org.mockito.Mockito.times;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.verifyNoInteractions;
+
+import java.math.BigDecimal;
+import java.util.HashMap;
 import java.util.List;
-import org.junit.Assert;
-import org.junit.Before;
-import org.junit.Test;
+import java.util.Map;
+import java.util.Optional;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.mockito.MockedStatic;
 import org.mockito.Mockito;
-import org.opendaylight.mdsal.binding.api.MountPoint;
-import org.opendaylight.mdsal.binding.api.MountPointService;
-import org.opendaylight.transportpce.common.StringConstants;
+import org.opendaylight.mdsal.binding.api.DataBroker;
 import org.opendaylight.transportpce.common.crossconnect.CrossConnect;
 import org.opendaylight.transportpce.common.crossconnect.CrossConnectImpl;
-import org.opendaylight.transportpce.common.crossconnect.CrossConnectImpl121;
-import org.opendaylight.transportpce.common.crossconnect.CrossConnectImpl221;
 import org.opendaylight.transportpce.common.device.DeviceTransactionManager;
-import org.opendaylight.transportpce.common.device.DeviceTransactionManagerImpl;
-import org.opendaylight.transportpce.common.mapping.MappingUtils;
-import org.opendaylight.transportpce.common.mapping.MappingUtilsImpl;
 import org.opendaylight.transportpce.common.mapping.PortMapping;
-import org.opendaylight.transportpce.common.mapping.PortMappingImpl;
-import org.opendaylight.transportpce.common.mapping.PortMappingVersion121;
-import org.opendaylight.transportpce.common.mapping.PortMappingVersion221;
+import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfaceException;
 import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfaces;
-import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl;
-import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl121;
-import org.opendaylight.transportpce.common.openroadminterfaces.OpenRoadmInterfacesImpl221;
-import org.opendaylight.transportpce.olm.stub.MountPointServiceStub;
-import org.opendaylight.transportpce.olm.stub.MountPointStub;
 import org.opendaylight.transportpce.olm.util.OlmPowerServiceRpcImplUtil;
-import org.opendaylight.transportpce.olm.util.TransactionUtils;
-import org.opendaylight.transportpce.test.AbstractTest;
-import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev170418.ServicePowerSetupInput;
-import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev170418.ServicePowerTurndownInput;
-import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.network.rev180226.NodeId;
-
-public class PowerMgmtTest extends AbstractTest {
+import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.ServicePowerSetupInput;
+import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.olm.rev210618.ServicePowerTurndownInput;
+import org.opendaylight.yang.gen.v1.http.org.opendaylight.transportpce.portmapping.rev220922.OpenroadmNodeVersion;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.types.rev161014.OpticalControlMode;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.common.types.rev161014.RatioDB;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev170206.interfaces.grp.Interface;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.device.rev170206.interfaces.grp.InterfaceBuilder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.optical.channel.interfaces.rev161014.Interface1Builder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.optical.channel.interfaces.rev161014.och.container.OchBuilder;
+import org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev161014.ots.container.OtsBuilder;
+import org.opendaylight.yangtools.yang.common.Decimal64;
 
-    private MountPoint mountPoint;
-    private MountPointService mountPointService;
-    private DeviceTransactionManager deviceTransactionManager;
-    private CrossConnect crossConnect;
+public class PowerMgmtTest {
     private OpenRoadmInterfaces openRoadmInterfaces;
+    private CrossConnect crossConnect;
+    private DeviceTransactionManager deviceTransactionManager;
     private PortMapping portMapping;
     private PowerMgmt powerMgmt;
-    private CrossConnectImpl121 crossConnectImpl121;
-    private CrossConnectImpl221 crossConnectImpl22;
-    private MappingUtils mappingUtils;
-    private OpenRoadmInterfacesImpl121 openRoadmInterfacesImpl121;
-    private OpenRoadmInterfacesImpl221 openRoadmInterfacesImpl22;
-    private PortMappingVersion221 portMappingVersion22;
-    private PortMappingVersion121 portMappingVersion121;
 
-    @Before
-    public void setUp() {
-        this.mountPoint = new MountPointStub(this.getDataBroker());
-        this.mountPointService = new MountPointServiceStub(mountPoint);
-        this.mappingUtils = new MappingUtilsImpl(getDataBroker());
-        this.mappingUtils = Mockito.spy(new MappingUtilsImpl(getDataBroker()));
-        Mockito.doReturn(StringConstants.OPENROADM_DEVICE_VERSION_1_2_1).when(mappingUtils)
-                .getOpenRoadmVersion(Mockito.anyString());
-        this.deviceTransactionManager = new DeviceTransactionManagerImpl(mountPointService, 3000);
-        this.crossConnectImpl121 = new CrossConnectImpl121(deviceTransactionManager);
-        this.crossConnectImpl22 = new CrossConnectImpl221(deviceTransactionManager);
-        this.crossConnect = new CrossConnectImpl(deviceTransactionManager, this.mappingUtils, this.crossConnectImpl121,
-                this.crossConnectImpl22);
-        this.openRoadmInterfacesImpl121 = new OpenRoadmInterfacesImpl121(deviceTransactionManager);
-        this.openRoadmInterfacesImpl22 = new OpenRoadmInterfacesImpl221(deviceTransactionManager);
-        this.openRoadmInterfaces = new OpenRoadmInterfacesImpl((this.deviceTransactionManager),
-                this.mappingUtils,this.openRoadmInterfacesImpl121,this.openRoadmInterfacesImpl22);
-        this.openRoadmInterfaces = Mockito.spy(this.openRoadmInterfaces);
-        this.portMappingVersion22 =
-                new PortMappingVersion221(getDataBroker(), deviceTransactionManager, this.openRoadmInterfaces);
-        this.portMappingVersion121 =
-                new PortMappingVersion121(getDataBroker(), deviceTransactionManager, this.openRoadmInterfaces);
-        this.portMapping = new PortMappingImpl(getDataBroker(), this.portMappingVersion22, this.portMappingVersion121);
-        this.portMapping = Mockito.spy(this.portMapping);
-        this.powerMgmt = new PowerMgmtImpl(this.getDataBroker(), this.openRoadmInterfaces, this.crossConnect,
-                this.deviceTransactionManager);
+    @BeforeEach
+    void setUp() {
+        Mockito.mock(DataBroker.class);
+        this.openRoadmInterfaces = Mockito.mock(OpenRoadmInterfaces.class);
+        this.crossConnect = Mockito.mock((CrossConnectImpl.class));
+        this.deviceTransactionManager = Mockito.mock(DeviceTransactionManager.class);
+        this.portMapping = Mockito.mock(PortMapping.class);
+        this.powerMgmt = new PowerMgmtImpl(this.openRoadmInterfaces, this.crossConnect,
+                this.deviceTransactionManager, this.portMapping, 1000, 1000);
     }
 
     @Test
-    public void testSetPower() {
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
+    void testSetPowerWhenMappingReturnNull() {
+        Mockito.when(this.portMapping.getNode(Mockito.anyString())).thenReturn(null);
+        boolean output = this.powerMgmt.setPower(OlmPowerServiceRpcImplUtil.getServicePowerSetupInput());
+        assertEquals(false, output);
     }
 
     @Test
-    public void testSetPower2() {
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput2();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
-    }
+    void testSetPowerForTransponderAEnd() throws OpenRoadmInterfaceException {
+        Mockito.when(this.portMapping.getNode("xpdr-A"))
+            .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeTpdr("xpdr-A", OpenroadmNodeVersion._121,
+                    List.of("network-A")));
+        Mockito.when(this.portMapping.getNode("roadm-A"))
+            .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeRdm("roadm-A", OpenroadmNodeVersion._121,
+                    List.of("srg1-A", "deg2-A")));
+        Interface interfOch = new InterfaceBuilder()
+                .setName("interface name")
+                .addAugmentation(new Interface1Builder()
+                        .setOch(new OchBuilder().build())
+                        .build())
+                .build();
+        Mockito.when(this.openRoadmInterfaces.getInterface(Mockito.matches("xpdr-A"), Mockito.anyString()))
+            .thenReturn(Optional.of(interfOch));
+        Interface interfOts = new InterfaceBuilder()
+                .setName("interface name")
+                .addAugmentation(
+                        new org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev161014
+                                .Interface1Builder()
+                                .setOts(new OtsBuilder()
+                                        .setSpanLossTransmit(new RatioDB(Decimal64.valueOf("6")))
+                                        .build())
+                                .build())
+                .build();
+        Mockito.when(this.openRoadmInterfaces
+                        .getInterface(Mockito.matches("roadm-A"), Mockito.anyString()))
+                .thenReturn(Optional.of(interfOts));
+        MockedStatic<PowerMgmtVersion121> pmv121 = Mockito.mockStatic(PowerMgmtVersion121.class);
+        pmv121.when(() -> PowerMgmtVersion121.setTransponderPower(Mockito.anyString(), Mockito.anyString(),
+                    Mockito.any(), Mockito.any(), Mockito.any()))
+                .thenReturn(true);
+        Map<String, Double> powerRangeMap = new HashMap<>();
+        powerRangeMap.put("MaxTx", 0.1);
+        powerRangeMap.put("MinTx", -5.1);
+        pmv121.when(() -> PowerMgmtVersion121.getXponderPowerRange(Mockito.anyString(), Mockito.anyString(),
+                    Mockito.anyString(), Mockito.any()))
+                .thenReturn(powerRangeMap);
+        Mockito.when(this.crossConnect.setPowerLevel(Mockito.anyString(), Mockito.anyString(), Mockito.any(),
+                        Mockito.anyString()))
+                .thenReturn(true);
 
-    @Test
-    public void testSetPower3() {
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
+        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInputForTransponder();
+        boolean result = this.powerMgmt.setPower(input);
+        assertEquals(true, result);
     }
 
     @Test
-    public void testSetPower4() {
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput4();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
-    }
+    void testSetPowerForTransponderZEnd() throws OpenRoadmInterfaceException {
+        Mockito.when(this.portMapping.getNode("xpdr-C"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeTpdr("xpdr-C", OpenroadmNodeVersion._121,
+                        List.of("client-C")));
 
-    @Test
-    public void testPowerTurnDown() {
-        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput();
-        boolean output = this.powerMgmt.powerTurnDown(input);
-        Assert.assertEquals(true, output);
+        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil
+                .getServicePowerSetupInputForOneNode("xpdr-C", "network-C", "client-C");
+        boolean result = this.powerMgmt.setPower(input);
+        assertEquals(true, result);
     }
 
     @Test
-    public void testPowerTurnDown2() {
-        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput2();
-        boolean output = this.powerMgmt.powerTurnDown(input);
-        Assert.assertEquals(false, output);
-    }
+    void testSetPowerForRoadmAEnd() throws OpenRoadmInterfaceException {
+        Mockito.when(this.portMapping.getNode("roadm-A"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeRdm("roadm-A", OpenroadmNodeVersion._121,
+                        List.of("srg1-A", "deg2-A")));
+        Mockito.when(this.deviceTransactionManager.getDataFromDevice(Mockito.anyString(), Mockito.any(),
+                        Mockito.any(), Mockito.anyLong(), Mockito.any()))
+                .thenReturn(Optional.empty());
+        Interface interfOts = new InterfaceBuilder()
+                .setName("interface name")
+                .addAugmentation(
+                        new org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev161014
+                            .Interface1Builder()
+                    .setOts(new OtsBuilder()
+                            .setSpanLossTransmit(new RatioDB(Decimal64.valueOf("6")))
+                            .build())
+                    .build())
+                .build();
+        Mockito.when(this.openRoadmInterfaces.getInterface(Mockito.anyString(), Mockito.anyString()))
+                .thenReturn(Optional.of(interfOts));
+        Mockito.when(this.crossConnect.setPowerLevel(Mockito.anyString(), Mockito.anyString(), Mockito.any(),
+                        Mockito.anyString()))
+                .thenReturn(true);
 
-    @Test
-    public void testPowerTurnDown3() {
-        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput3();
-        boolean output = this.powerMgmt.powerTurnDown(input);
-        Assert.assertEquals(true, output);
+        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil
+                .getServicePowerSetupInputForOneNode("roadm-A", "srg1-A", "deg2-A");
+        boolean result = this.powerMgmt.setPower(input);
+        assertEquals(true, result);
+        verify(this.crossConnect, times(1)).setPowerLevel(Mockito.matches("roadm-A"),
+                Mockito.matches(OpticalControlMode.Power.getName()), eq(Decimal64.valueOf("-3.00")),
+                Mockito.matches("srg1-A-deg2-A-761:768"));
+        verify(this.crossConnect, times(1)).setPowerLevel(Mockito.matches("roadm-A"),
+                Mockito.matches(OpticalControlMode.GainLoss.getName()), eq(Decimal64.valueOf("-3.00")),
+                Mockito.matches("srg1-A-deg2-A-761:768"));
     }
 
     @Test
-    public void testPowerTurnDown4() {
-        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput4();
-        boolean output = this.powerMgmt.powerTurnDown(input);
-        Assert.assertEquals(false, output);
-    }
+    void testSetPowerForRoadmZEnd() throws OpenRoadmInterfaceException {
+        Mockito.when(this.portMapping.getNode("roadm-C"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeRdm("roadm-C", OpenroadmNodeVersion._121,
+                        List.of("deg1-C", "srg1-C")));
+        Mockito.when(this.deviceTransactionManager.getDataFromDevice(Mockito.anyString(), Mockito.any(),
+                        Mockito.any(), Mockito.anyLong(), Mockito.any()))
+                .thenReturn(Optional.empty());
+        Mockito.when(this.crossConnect.setPowerLevel(Mockito.anyString(), Mockito.anyString(), Mockito.any(),
+                        Mockito.anyString()))
+                .thenReturn(true);
 
-    @Test
-    public void testSetPowerPresentNodes() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction(nodeId.getValue(), this.getDataBroker(), null);
-            Thread.sleep(1000);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
+        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil
+                .getServicePowerSetupInputForOneNode("roadm-C", "deg1-C", "srg1-C");
+        boolean result = this.powerMgmt.setPower(input);
+        assertEquals(true, result);
+        verify(this.crossConnect, times(1)).setPowerLevel(Mockito.matches("roadm-C"),
+                Mockito.matches(OpticalControlMode.Power.getName()), Mockito.isNull(),
+                Mockito.matches("deg1-C-srg1-C-761:768"));
     }
 
     @Test
-    public void testSetPowerPresentNodes2() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction2(nodeId.getValue(), this.getDataBroker(), null);
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
-    }
+    void testSetPowerForTransponderWhenNoTransponderPort() throws OpenRoadmInterfaceException {
+        Mockito.when(this.portMapping.getNode("xpdr-A"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeTpdr("xpdr-A", OpenroadmNodeVersion._121,
+                        List.of("network-A")));
+        Mockito.when(this.portMapping.getNode("roadm-A"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeRdm("roadm-A", OpenroadmNodeVersion._121,
+                        List.of("srg1-A", "deg2-A")));
+        Interface interfOch = new InterfaceBuilder()
+                .setName("interface name")
+                .addAugmentation(new Interface1Builder()
+                        .setOch(new OchBuilder().build())
+                        .build())
+                .build();
+        Mockito.when(this.openRoadmInterfaces.getInterface(Mockito.matches("xpdr-A"), Mockito.anyString()))
+                .thenReturn(Optional.of(interfOch));
+        Interface interfOts = new InterfaceBuilder()
+                .setName("interface name")
+                .addAugmentation(
+                        new org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev161014
+                                .Interface1Builder()
+                                .setOts(new OtsBuilder()
+                                        .setSpanLossTransmit(new RatioDB(Decimal64.valueOf("6")))
+                                        .build())
+                                .build())
+                .build();
+        Mockito.when(this.openRoadmInterfaces
+                        .getInterface(Mockito.matches("roadm-A"), Mockito.anyString()))
+                .thenReturn(Optional.of(interfOts));
+        try (MockedStatic<PowerMgmtVersion121> pmv121 = Mockito.mockStatic(PowerMgmtVersion121.class)) {
+            pmv121.when(() -> PowerMgmtVersion121.setTransponderPower(Mockito.anyString(), Mockito.anyString(),
+                            Mockito.any(), Mockito.any(), Mockito.any()))
+                    .thenReturn(true);
+            pmv121.when(() -> PowerMgmtVersion121.getXponderPowerRange(Mockito.anyString(), Mockito.anyString(),
+                            Mockito.anyString(), Mockito.any()))
+                    .thenReturn(new HashMap<>());
 
-    @Test
-    public void testSetPowerPresentNodes3() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction3(nodeId.getValue(), this.getDataBroker(), null);
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
-    }
-    /*
-    @Test
-    public void testSetPowerPresentNodes31() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction3(nodeId.getValue(), this.getDataBroker(), "deg");
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(false, output);
-    }*/
+            Mockito.when(this.crossConnect.setPowerLevel(Mockito.anyString(), Mockito.anyString(), Mockito.any(),
+                            Mockito.anyString()))
+                    .thenReturn(true);
 
-    @Test
-    public void testSetPowerPresentNodes312() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction3(nodeId.getValue(), this.getDataBroker(), "deg");
-            Thread.sleep(500);
+
+            ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInputForTransponder();
+            boolean result = this.powerMgmt.setPower(input);
+            assertEquals(true, result);
+            pmv121.verify(() -> PowerMgmtVersion121.setTransponderPower(Mockito.matches("xpdr-A"),
+                    Mockito.anyString(), eq(new BigDecimal("-5")), Mockito.any(), Mockito.any()));
+            verify(this.crossConnect, times(1)).setPowerLevel(Mockito.matches("roadm-A"),
+                    Mockito.matches(OpticalControlMode.GainLoss.getName()), eq(Decimal64.valueOf("-3.00")),
+                    Mockito.matches("srg1-A-deg2-A-761:768"));
         }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput4();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
     }
 
     @Test
-    public void testSetPowerPresentNodes32() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction3(nodeId.getValue(), this.getDataBroker(), null);
-            Thread.sleep(500);
+    void testSetPowerForTransponderAEndWithRoadmPort() throws OpenRoadmInterfaceException {
+        Mockito.when(this.portMapping.getNode("xpdr-A"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeTpdr("xpdr-A", OpenroadmNodeVersion._121,
+                        List.of("network-A")));
+        Mockito.when(this.portMapping.getNode("roadm-A"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeRdm("roadm-A", OpenroadmNodeVersion._121,
+                        List.of("srg1-A", "deg2-A")));
+        Interface interfOch = new InterfaceBuilder()
+                .setName("interface name")
+                .addAugmentation(new Interface1Builder()
+                        .setOch(new OchBuilder().build())
+                        .build())
+                .build();
+        Mockito.when(this.openRoadmInterfaces.getInterface(Mockito.matches("xpdr-A"), Mockito.anyString()))
+                .thenReturn(Optional.of(interfOch));
+        Interface interfOts = new InterfaceBuilder()
+                .setName("interface name")
+                .addAugmentation(
+                        new org.opendaylight.yang.gen.v1.http.org.openroadm.optical.transport.interfaces.rev161014
+                                .Interface1Builder()
+                                .setOts(new OtsBuilder()
+                                        .setSpanLossTransmit(new RatioDB(Decimal64.valueOf("6")))
+                                        .build())
+                                .build())
+                .build();
+        Mockito.when(this.openRoadmInterfaces
+                        .getInterface(Mockito.matches("roadm-A"), Mockito.anyString()))
+                .thenReturn(Optional.of(interfOts));
+        try (MockedStatic<PowerMgmtVersion121> pmv121 = Mockito.mockStatic(PowerMgmtVersion121.class)) {
+
+            pmv121.when(() -> PowerMgmtVersion121.setTransponderPower(Mockito.anyString(), Mockito.anyString(),
+                            Mockito.any(), Mockito.any(), Mockito.any()))
+                    .thenReturn(true);
+            Map<String, Double> powerRangeMapTpdrTx = new HashMap<>();
+            powerRangeMapTpdrTx.put("MaxTx", 0.1);
+            powerRangeMapTpdrTx.put("MinTx", -5.1);
+            pmv121.when(() -> PowerMgmtVersion121.getXponderPowerRange(Mockito.anyString(), Mockito.anyString(),
+                            Mockito.anyString(), Mockito.any()))
+                    .thenReturn(powerRangeMapTpdrTx);
+            Map<String, Double> powerRangeMapSrgRx = new HashMap<>();
+            powerRangeMapSrgRx.put("MaxRx", -4.2);
+            powerRangeMapSrgRx.put("MinRx", -22.2);
+            pmv121.when(() -> PowerMgmtVersion121.getSRGRxPowerRange(Mockito.anyString(), Mockito.anyString(),
+                            Mockito.any(), Mockito.anyString(), Mockito.anyString()))
+                    .thenReturn(powerRangeMapSrgRx);
+            Mockito.when(this.crossConnect.setPowerLevel(Mockito.anyString(), Mockito.anyString(), Mockito.any(),
+                            Mockito.anyString()))
+                    .thenReturn(true);
+
+            ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInputForTransponder();
+            boolean result = this.powerMgmt.setPower(input);
+            assertEquals(true, result);
+            pmv121.verify(() -> PowerMgmtVersion121.setTransponderPower(Mockito.matches("xpdr-A"),
+                    Mockito.anyString(), eq(new BigDecimal("-4.20000000000000017763568394002504646778106689453125")),
+                    Mockito.any(), Mockito.any()));
         }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
     }
 
     @Test
-    public void testSetPowerPresentNodes4() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction(nodeId.getValue(), this.getDataBroker(), "network");
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput2();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
+    void testSetPowerWithoutNode() {
+        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInputWthoutNode();
+        boolean result = this.powerMgmt.setPower(input);
+        assertEquals(false, result);
+        verifyNoInteractions(this.crossConnect);
     }
 
-    /*
     @Test
-    public void testSetPowerPresentNodes41() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction(nodeId.getValue(), this.getDataBroker(), null);
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput2();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(false, output);
-    }*/
+    void testSetPowerForBadNodeType() throws OpenRoadmInterfaceException {
+        Mockito.when(this.portMapping.getNode("ila node"))
+                .thenReturn(OlmPowerServiceRpcImplUtil.getMappingNodeIla());
 
-    @Test
-    public void testSetPowerPresentNodes42() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction(nodeId.getValue(), this.getDataBroker(), "deg");
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
+        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil
+                .getServicePowerSetupInputForOneNode("ila node", "rx-port", "tx-port");
+        boolean result = this.powerMgmt.setPower(input);
+        assertEquals(true, result);
+        verifyNoInteractions(this.crossConnect);
+        verifyNoInteractions(this.openRoadmInterfaces);
     }
 
+
     @Test
-    public void testSetPowerPresentNodes422() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction(nodeId.getValue(), this.getDataBroker(), "deg");
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput4();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
+    void testPowerTurnDownWhenSuccess() {
+        Mockito.when(this.crossConnect.setPowerLevel(Mockito.anyString(), Mockito.anyString(), Mockito.any(),
+                        Mockito.anyString()))
+                .thenReturn(true);
+        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput();
+        boolean result = this.powerMgmt.powerTurnDown(input);
+        assertEquals(true, result);
+        verify(this.crossConnect, times(1)).setPowerLevel(Mockito.matches("roadm-C"),
+                Mockito.matches(OpticalControlMode.Off.getName()), Mockito.isNull(), Mockito.anyString());
+        verify(this.crossConnect, times(1)).setPowerLevel(Mockito.matches("roadm-A"),
+                Mockito.matches(OpticalControlMode.Power.getName()), eq(Decimal64.valueOf("-60")),
+                Mockito.anyString());
+        verify(this.crossConnect, times(1)).setPowerLevel(Mockito.matches("roadm-A"),
+                Mockito.matches(OpticalControlMode.Off.getName()), Mockito.isNull(), Mockito.anyString());
     }
 
     @Test
-    public void testSetPowerPresentNodes43() throws InterruptedException {
-        List<NodeId> nodes = TransactionUtils.getNodeIds();
-        for (NodeId nodeId : nodes) {
-            TransactionUtils.writeNodeTransaction(nodeId.getValue(), this.getDataBroker(), null);
-            Thread.sleep(500);
-        }
-        ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
-        boolean output = this.powerMgmt.setPower(input);
-        Assert.assertEquals(true, output);
+    void testPowerTurnDownWhenFailure() {
+        Mockito.when(this.crossConnect.setPowerLevel(Mockito.anyString(), Mockito.anyString(), Mockito.any(),
+                        Mockito.anyString()))
+                .thenReturn(false);
+        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput();
+        boolean result = this.powerMgmt.powerTurnDown(input);
+        assertEquals(false, result);
+        verify(this.crossConnect, times(2))
+                .setPowerLevel(Mockito.anyString(), Mockito.any(), Mockito.any(), Mockito.anyString());
     }
-
 }