add unit test for PowerMgmtImpl

Signed-off-by: Shaaban Ebrahim <shaaban.altanany@gmail.com>
Change-Id: Icb1c1f8f53bdd47b418c14fd959f23bb546b157a

diff --git a/olm/src/test/java/org/opendaylight/transportpce/olm/power/PowerMgmtTest.java b/olm/src/test/java/org/opendaylight/transportpce/olm/power/PowerMgmtTest.java
index d56307df73567c2a279623768f0f80efa53f51ff..d2f2c039ae36ff741535b539a21fad6b1f801739 100644 (file)
--- a/olm/src/test/java/org/opendaylight/transportpce/olm/power/PowerMgmtTest.java
+++ b/olm/src/test/java/org/opendaylight/transportpce/olm/power/PowerMgmtTest.java
@@ -38,6 +38,7 @@ 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 {
@@ -121,6 +122,34 @@ public class PowerMgmtTest extends AbstractTest {
 
     }
 
+    @Test
+    public void testPowerTurnDown() {
+        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput();
+        boolean output = this.powerMgmt.powerTurnDown(input);
+        Assert.assertEquals(true, output);
+    }
+
+    @Test
+    public void testPowerTurnDown2() {
+        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput2();
+        boolean output = this.powerMgmt.powerTurnDown(input);
+        Assert.assertEquals(false, output);
+    }
+
+    @Test
+    public void testPowerTurnDown3() {
+        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput3();
+        boolean output = this.powerMgmt.powerTurnDown(input);
+        Assert.assertEquals(true, output);
+    }
+
+    @Test
+    public void testPowerTurnDown4() {
+        ServicePowerTurndownInput input = OlmPowerServiceRpcImplUtil.getServicePowerTurndownInput4();
+        boolean output = this.powerMgmt.powerTurnDown(input);
+        Assert.assertEquals(false, output);
+    }
+
     @Test
     public void testSetPowerPresentNodes() throws InterruptedException {
         List<NodeId> nodes = TransactionUtils.getNodeIds();

diff --git a/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java b/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java
index f6d783fd44b03ee900af18f90798afbba39a073d..ec9955157bfaecaffd188b68cf23e3e382b679cd 100644 (file)
--- a/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java
+++ b/olm/src/test/java/org/opendaylight/transportpce/olm/util/OlmPowerServiceRpcImplUtil.java
@@ -98,9 +98,51 @@ public final class OlmPowerServiceRpcImplUtil {
         nodes.add(node1);
         nodes.add(node2);
         ServicePowerTurndownInput input = new ServicePowerTurndownInputBuilder()
-            .setNodes(nodes)
-            .setServiceName("service 1")
-            .setWaveNumber(Long.valueOf("1234")).build();
+                .setNodes(nodes)
+                .setServiceName("service 1")
+                .setWaveNumber(Long.valueOf("1234")).build();
+
+        return input;
+    }
+
+    public static ServicePowerTurndownInput getServicePowerTurndownInput2() {
+        Nodes node1 = new NodesBuilder().setDestTp("destdeg").setSrcTp("src").setNodeId("node 1").build();
+        Nodes node2 = new NodesBuilder().setDestTp("destdeg").setSrcTp("src").setNodeId("node 2").build();
+        List<Nodes> nodes = new ArrayList<>();
+        nodes.add(node1);
+        nodes.add(node2);
+        ServicePowerTurndownInput input = new ServicePowerTurndownInputBuilder()
+                .setNodes(nodes)
+                .setServiceName("service 1")
+                .setWaveNumber(Long.valueOf("1234")).build();
+
+        return input;
+    }
+
+    public static ServicePowerTurndownInput getServicePowerTurndownInput3() {
+        Nodes node1 = new NodesBuilder().setDestTp("destsrg").setSrcTp("src").setNodeId("node 1").build();
+        Nodes node2 = new NodesBuilder().setDestTp("destsrg").setSrcTp("src").setNodeId("node 2").build();
+        List<Nodes> nodes = new ArrayList<>();
+        nodes.add(node1);
+        nodes.add(node2);
+        ServicePowerTurndownInput input = new ServicePowerTurndownInputBuilder()
+                .setNodes(nodes)
+                .setServiceName("service 1")
+                .setWaveNumber(Long.valueOf("1234")).build();
+
+        return input;
+    }
+
+    public static ServicePowerTurndownInput getServicePowerTurndownInput4() {
+        Nodes node1 = new NodesBuilder().setDestTp("destdeg").setSrcTp("src").setNodeId("node 1").build();
+        Nodes node2 = new NodesBuilder().setDestTp("destdeg").setSrcTp("src").setNodeId("node 2").build();
+        List<Nodes> nodes = new ArrayList<>();
+        nodes.add(node1);
+        nodes.add(node2);
+        ServicePowerTurndownInput input = new ServicePowerTurndownInputBuilder()
+                .setNodes(nodes)
+                .setServiceName("service 1")
+                .setWaveNumber(Long.valueOf("1234")).build();
 
         return input;
     }