1 files changed, 109 insertions, 0 deletions
diff --git a/appc-adapters/appc-iaas-adapter/appc-iaas-adapter-bundle/src/test/java/org/onap/appc/adapter/iaas/provider/operation/impl/TestRebuildServer.java b/appc-adapters/appc-iaas-adapter/appc-iaas-adapter-bundle/src/test/java/org/onap/appc/adapter/iaas/provider/operation/impl/TestRebuildServer.java
new file mode 100644
index 000000000..5e16c00ed
--- /dev/null
+++ b/appc-adapters/appc-iaas-adapter/appc-iaas-adapter-bundle/src/test/java/org/onap/appc/adapter/iaas/provider/operation/impl/TestRebuildServer.java
@@ -0,0 +1,109 @@
+package org.onap.appc.adapter.iaas.provider.operation.impl;
+
+import org.junit.Assert;
+import org.junit.Test;
+import static org.mockito.Mockito.verify;
+import static org.mockito.Mockito.inOrder;
+import org.mockito.InOrder;
+import com.att.cdp.zones.model.Server;
+import com.att.cdp.zones.model.Server.Status;
+import com.att.cdp.exceptions.ZoneException;
+import org.onap.appc.adapter.iaas.provider.operation.impl.RebuildServer;
+import org.onap.appc.exceptions.APPCException;
+import org.onap.appc.configuration.Configuration;
+import org.onap.appc.configuration.ConfigurationFactory;
+
+
+public class TestRebuildServer {
+    protected static final Configuration configuration = ConfigurationFactory.getConfiguration();
+
+    @Test
+    public void rebuildServerRunning() throws ZoneException {
+        MockGenerator mg = new MockGenerator(Status.RUNNING);
+        Server server = mg.getServer();
+        RebuildServer rbs = new RebuildServer();
+        rbs.setProviderCache(mg.getProviderCacheMap());
+        rbs.setRebuildSleepTime(0);
+        try {
+            rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
+        } catch (APPCException e) {
+            Assert.fail("Exception during RebuildServer.executeProviderOperation");
+        }
+        InOrder inOrderTest = inOrder(server);
+        inOrderTest.verify(server).stop();
+        inOrderTest.verify(server).rebuild("linuxBase");
+        inOrderTest.verify(server).start();
+
+    }
+
+    @Test
+    public void rebuildServerReady() throws ZoneException {
+        MockGenerator mg = new MockGenerator(Status.READY);
+        Server server = mg.getServer();
+        RebuildServer rbs = new RebuildServer();
+        rbs.setProviderCache(mg.getProviderCacheMap());
+        rbs.setRebuildSleepTime(0);
+        try {
+            rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
+        } catch (APPCException e) {
+            Assert.fail("Exception during RebuildServer.executeProviderOperation");
+        }
+        InOrder inOrderTest = inOrder(server);
+        inOrderTest.verify(server).rebuild("linuxBase");
+        inOrderTest.verify(server).start();
+    }
+
+    @Test
+    public void rebuildServerPause() throws ZoneException {
+        MockGenerator mg = new MockGenerator(Status.PAUSED);
+        Server server = mg.getServer();
+        RebuildServer rbs = new RebuildServer();
+        rbs.setProviderCache(mg.getProviderCacheMap());
+        rbs.setRebuildSleepTime(0);
+        try {
+            rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
+        } catch (APPCException e) {
+            Assert.fail("Exception during RebuildServer.executeProviderOperation");
+        }
+        InOrder inOrderTest = inOrder(server);
+        inOrderTest.verify(server).unpause();
+        inOrderTest.verify(server).stop();
+        inOrderTest.verify(server).rebuild("linuxBase");
+        inOrderTest.verify(server).start();
+    }
+
+    @Test
+    public void rebuildServerError() {
+        MockGenerator mg = new MockGenerator(Status.ERROR);
+        Server server = mg.getServer();
+        RebuildServer rbs = new RebuildServer();
+        rbs.setProviderCache(mg.getProviderCacheMap());
+        try {
+            rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
+        } catch (APPCException e) {
+            Assert.fail("Exception during RebuildServer.executeProviderOperation");
+        }
+        verify(mg.getSvcLogicContext()).setAttribute(org.onap.appc.Constants.ATTRIBUTE_ERROR_CODE,
+                "405");
+    }
+
+    @Test
+    public void rebuildServerSuspended() throws ZoneException {
+        MockGenerator mg = new MockGenerator(Status.SUSPENDED);
+        Server server = mg.getServer();
+        RebuildServer rbs = new RebuildServer();
+        rbs.setProviderCache(mg.getProviderCacheMap());
+        rbs.setRebuildSleepTime(0);
+        try {
+            rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
+        } catch (APPCException e) {
+            Assert.fail("Exception during RebuildServer.executeProviderOperation");
+        }
+        InOrder inOrderTest = inOrder(server);
+        inOrderTest.verify(server).resume();
+        inOrderTest.verify(server).stop();
+        inOrderTest.verify(server).rebuild("linuxBase");
+        inOrderTest.verify(server).start();
+    }
+
+}