Fix random test failure in TestChefApiClient
[appc.git] / appc-adapters / appc-iaas-adapter / appc-iaas-adapter-bundle / src / test / java / org / onap / appc / adapter / iaas / provider / operation / impl / TestVmStatuschecker.java
1 package org.onap.appc.adapter.iaas.provider.operation.impl;
2
3 import org.junit.Assert;
4 import org.junit.Test;
5 import org.onap.appc.Constants;
6 import org.onap.appc.exceptions.APPCException;
7 import static org.mockito.Mockito.verify;
8 import static org.mockito.Mockito.atLeastOnce;
9 import com.att.cdp.zones.model.ModelObject;
10 import com.att.cdp.zones.model.Server;
11 import com.att.cdp.zones.model.Server.Status;
12
13 public class TestVmStatuschecker {
14
15     @Test
16     public void vmStatuscheckerSuspended() {
17         MockGenerator mg = new MockGenerator(Status.SUSPENDED);
18         Server server = mg.getServer();
19         VmStatuschecker rbs = new VmStatuschecker();
20         rbs.setProviderCache(mg.getProviderCacheMap());
21         ModelObject mo = null;
22         try {
23             mo = rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
24         } catch (APPCException e) {
25             Assert.fail("Exception during VmStatuschecker.executeProviderOperation");
26         }
27         verify(mg.getSvcLogicContext(), atLeastOnce()).setAttribute(Constants.STATUS_OF_VM,
28                 "suspended");
29     }
30
31     @Test
32     public void vmStatuscheckerRunning() {
33         MockGenerator mg = new MockGenerator(Status.RUNNING);
34         Server server = mg.getServer();
35         VmStatuschecker rbs = new VmStatuschecker();
36         rbs.setProviderCache(mg.getProviderCacheMap());
37         ModelObject mo = null;
38         try {
39             mo = rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
40         } catch (APPCException e) {
41             Assert.fail("Exception during VmStatuschecker.executeProviderOperation");
42         }
43         verify(mg.getSvcLogicContext(), atLeastOnce()).setAttribute(Constants.STATUS_OF_VM,
44                 "running");
45     }
46
47     @Test
48     public void vmStatuscheckerError() {
49         MockGenerator mg = new MockGenerator(Status.ERROR);
50         Server server = mg.getServer();
51         VmStatuschecker rbs = new VmStatuschecker();
52         rbs.setProviderCache(mg.getProviderCacheMap());
53         ModelObject mo = null;
54         try {
55             mo = rbs.executeProviderOperation(mg.getParams(), mg.getSvcLogicContext());
56         } catch (APPCException e) {
57             Assert.fail("Exception during VmStatuschecker.executeProviderOperation");
58         }
59         verify(mg.getSvcLogicContext(), atLeastOnce()).setAttribute(Constants.STATUS_OF_VM,
60                 "error");
61     }
62 }