package org.opendaylight.transportpce.olm.power;
+
import java.util.HashMap;
import java.util.Map;
import java.util.Optional;
this.openRoadmInterfaces = Mockito.spy(this.openRoadmInterfaces);
this.portMapping = Mockito.spy(this.portMapping);
this.powerMgmt = new PowerMgmtImpl(getDataBroker(), this.openRoadmInterfaces, this.crossConnect,
- this.deviceTransactionManager);
+ this.deviceTransactionManager, this.portMapping);
}
@Test
public void testSetPowerMockingUtil() {
Mockito.mockStatic(OlmUtils.class);
- Mockito.when(OlmUtils.getNode(Mockito.anyString(), ArgumentMatchers.eq(getDataBroker())))
- .thenReturn(Optional.of(getXpdrNodesFromNodesBuilderDeg()));
+ Mockito.when(this.portMapping.getNode(Mockito.anyString()))
+ .thenReturn(getXpdrNodesFromNodesBuilderDeg());
ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput2();
boolean output = this.powerMgmt.setPower(input);
Assert.assertEquals(false, output);
public void testSetPowerMockingUtilNetwokType() throws OpenRoadmInterfaceException {
Mockito.mockStatic(OlmUtils.class);
Mockito.mockStatic(PowerMgmtVersion121.class);
- Mockito.when(OlmUtils.getNode(Mockito.anyString(), ArgumentMatchers.eq(getDataBroker())))
- .thenReturn(Optional.of(getXpdrNodesFromNodesBuilderNetwork()));
+ Mockito.when(this.portMapping.getNode(Mockito.anyString()))
+ .thenReturn(getXpdrNodesFromNodesBuilderNetwork());
Map<String, Double> txPowerRangeMap = new HashMap<>();
Mockito.when(PowerMgmtVersion121.getXponderPowerRange(
Mockito.anyString(), Mockito.anyString(), Mockito.anyString(),
public void testSetPowerMockingUtilNetwokTypeMoreThanOneNode() throws OpenRoadmInterfaceException {
Mockito.mockStatic(OlmUtils.class);
Mockito.mockStatic(PowerMgmtVersion121.class);
- Mockito.when(OlmUtils.getNode(Mockito.anyString(), ArgumentMatchers.eq(getDataBroker())))
- .thenReturn(Optional.of(getXpdrNodesFromNodesBuilderNetwork()));
+ Mockito.when(this.portMapping.getNode(Mockito.anyString()))
+ .thenReturn(getXpdrNodesFromNodesBuilderNetwork());
Map<String, Double> txPowerRangeMap = new HashMap<>();
Mockito.when(PowerMgmtVersion121
.getXponderPowerRange(Mockito.anyString(), Mockito.anyString(), Mockito.anyString(),
@Test
public void testSetPowerXpdrNodes() {
Mockito.mockStatic(OlmUtils.class);
- Mockito.when(OlmUtils.getNode(Mockito.anyString(), ArgumentMatchers.eq(getDataBroker())))
- .thenReturn(Optional.of(getXpdrNodesFromNodesBuilderDeg()));
+ Mockito.when(this.portMapping.getNode(Mockito.anyString()))
+ .thenReturn(getXpdrNodesFromNodesBuilderDeg());
ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
boolean output = this.powerMgmt.setPower(input);
Assert.assertEquals(true, output);
@Test
public void testSetPowerRdmNodesReturnInterfaceEmpty() throws OpenRoadmInterfaceException {
Mockito.mockStatic(OlmUtils.class);
- Mockito.when(OlmUtils.getNode(Mockito.anyString(), ArgumentMatchers.eq(getDataBroker())))
- .thenReturn(Optional.of(getRdmNodesFromNodesBuilder()));
+ Mockito.when(this.portMapping.getNode(Mockito.anyString()))
+ .thenReturn(getRdmNodesFromNodesBuilder());
ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
OpenRoadmInterfacesImpl121 openRoadmInterfacesImpl121Spy = Mockito.mock(OpenRoadmInterfacesImpl121.class);
Mockito.when(openRoadmInterfacesImpl121Spy.getInterface(Mockito.anyString(), Mockito.anyString()))
@Test
public void testSetPowerRdmNodesThrowsException() throws OpenRoadmInterfaceException {
Mockito.mockStatic(OlmUtils.class);
- Mockito.when(OlmUtils.getNode(Mockito.anyString(), ArgumentMatchers.eq(getDataBroker())))
- .thenReturn(Optional.of(getRdmNodesFromNodesBuilder()));
+ Mockito.when(this.portMapping.getNode(Mockito.anyString()))
+ .thenReturn(getRdmNodesFromNodesBuilder());
ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
OpenRoadmInterfacesImpl121 openRoadmInterfacesImpl121Spy = Mockito.mock(OpenRoadmInterfacesImpl121.class);
Mockito.when(openRoadmInterfacesImpl121Spy.getInterface(Mockito.anyString(), Mockito.anyString()))
@Test
public void testSetPowerRdmNodesReturnInterface() throws OpenRoadmInterfaceException {
Mockito.mockStatic(OlmUtils.class);
- Mockito.when(OlmUtils.getNode(Mockito.anyString(), ArgumentMatchers.eq(getDataBroker())))
- .thenReturn(Optional.of(getRdmNodesFromNodesBuilder()));
+ Mockito.when(this.portMapping.getNode(Mockito.anyString()))
+ .thenReturn(getRdmNodesFromNodesBuilder());
ServicePowerSetupInput input = OlmPowerServiceRpcImplUtil.getServicePowerSetupInput3();
OpenRoadmInterfacesImpl121 openRoadmInterfacesImpl121Spy = Mockito.mock(OpenRoadmInterfacesImpl121.class);
this.openRoadmInterfacesImpl710);
openRoadmInterfacesSpy = Mockito.spy(openRoadmInterfacesSpy);
return new PowerMgmtImpl(getDataBroker(), openRoadmInterfacesSpy, crossConnectMock,
- this.deviceTransactionManager);
+ this.deviceTransactionManager, this.portMapping);
}
private Nodes getXpdrNodesFromNodesBuilderDeg() {