Add unit test for UsageManagerImpl.configureUsageManagerServicePid()

Author: GabrielBrascher

usage/src/main/java/com/cloud/usage/UsageManagerImpl.java

@@ -283,7 +283,7 @@ public boolean configure(String name, Map<String, Object> params) throws Configu
             throw new ConfigurationException("Unhandled exception configuring UsageManager " + e.toString());
         }
 
-        configureUsageManagerServicePid();
+        _pid = configureUsageManagerServicePid();
         return true;
     }
 
@@ -296,7 +296,7 @@ public boolean configure(String name, Map<String, Object> params) throws Configu
      *  <li>NumberUtils.toInt returns 0 (zero) if the conversion fails (pid is null or "")</li>
      * </ul>
      */
-    protected void configureUsageManagerServicePid() {
+    protected int configureUsageManagerServicePid() {
         File usageServicePid = new File(CLOUDSTACK_USAGE_SERVER_SERVICE_PID_FILE);
         if (!usageServicePid.exists()) {
             throw new CloudRuntimeException(String.format("Cannot find file [%s].", CLOUDSTACK_USAGE_SERVER_SERVICE_PID_FILE));
@@ -307,11 +307,12 @@ protected void configureUsageManagerServicePid() {
         try {
             String pidAsString = FileUtils.readFileToString(usageServicePid, Charset.defaultCharset());
             String pidAsStringWithoutLineSeparator = pidAsString.replace(System.getProperty("line.separator"), "");
-            _pid = NumberUtils.toInt(pidAsStringWithoutLineSeparator);
-            if (_pid == 0) {
+            int pid = NumberUtils.toInt(pidAsStringWithoutLineSeparator);
+            if (pid == 0) {
                 throw new CloudRuntimeException(
                         String.format("Failed to retrieve a valid cloudstack-usage.service pid [file:%s, pid at file:%s]", CLOUDSTACK_USAGE_SERVER_SERVICE_PID_FILE, pidAsString));
             }
+            return pid;
         } catch (IOException e) {
             throw new CloudRuntimeException(e);
         }

usage/src/test/java/com/cloud/usage/UsageManagerImplTest.java

@@ -0,0 +1,68 @@
+package com.cloud.usage;
+
+import java.io.File;
+import java.io.IOException;
+import java.nio.charset.Charset;
+
+import org.apache.commons.io.FileUtils;
+import org.apache.commons.lang3.math.NumberUtils;
+import org.junit.Assert;
+import org.junit.Test;
+import org.junit.runner.RunWith;
+import org.mockito.InOrder;
+import org.mockito.Mockito;
+import org.powermock.api.mockito.PowerMockito;
+import org.powermock.core.classloader.annotations.PrepareForTest;
+import org.powermock.modules.junit4.PowerMockRunner;
+
+import com.cloud.utils.exception.CloudRuntimeException;
+
+@RunWith(PowerMockRunner.class)
+@PrepareForTest({FileUtils.class, UsageManagerImpl.class, NumberUtils.class})
+public class UsageManagerImplTest {
+
+    private static final String CLOUDSTACK_USAGE_SERVER_SERVICE_PID_FILE = "/var/run/cloudstack-usage.service.pid";
+    private UsageManagerImpl usageManagerImpl = new UsageManagerImpl();
+
+    @Test
+    public void configureUsageManagerServicePidTestCompleteExecutionFlow() throws Exception {
+        prepareAndVerifyTestConfigureUsageManagerServicePidTest(true, true, 1, 1234, "1234" + System.getProperty("line.separator"));
+    }
+
+    @Test(expected=CloudRuntimeException.class)
+    public void configureUsageManagerServicePidTestFileDoesNotExist() throws Exception {
+        prepareAndVerifyTestConfigureUsageManagerServicePidTest(false, true, 0, 1234, "1234" + System.getProperty("line.separator"));
+    }
+
+    @Test(expected = CloudRuntimeException.class)
+    public void configureUsageManagerServicePidTestCannotReadFile() throws Exception {
+        prepareAndVerifyTestConfigureUsageManagerServicePidTest(true, false, 1, 1234, "1234" + System.getProperty("line.separator"));
+    }
+
+    @Test(expected = CloudRuntimeException.class)
+    public void configureUsageManagerServicePidTestPidEqualsToZero() throws Exception {
+        prepareAndVerifyTestConfigureUsageManagerServicePidTest(true, true, 1, 0, "abc");
+    }
+
+    private void prepareAndVerifyTestConfigureUsageManagerServicePidTest(boolean fileExists, boolean canReadFile, int timesCanReadIsExecuted, int expectedPid,
+            String readedStringFromFile)
+            throws Exception, IOException {
+        File usageServicePid = Mockito.mock(File.class);
+        PowerMockito.whenNew(File.class).withArguments(CLOUDSTACK_USAGE_SERVER_SERVICE_PID_FILE).thenReturn(usageServicePid);
+
+        Mockito.when(usageServicePid.exists()).thenReturn(fileExists);
+        Mockito.when(usageServicePid.canRead()).thenReturn(canReadFile);
+
+        PowerMockito.mockStatic(FileUtils.class);
+        Mockito.when(FileUtils.readFileToString(usageServicePid, Charset.defaultCharset())).thenReturn(readedStringFromFile);
+
+        int result = usageManagerImpl.configureUsageManagerServicePid();
+
+        InOrder inOrder = Mockito.inOrder(usageServicePid);
+        inOrder.verify(usageServicePid).exists();
+        inOrder.verify(usageServicePid, Mockito.times(timesCanReadIsExecuted)).canRead();
+
+        Assert.assertEquals(expectedPid, result);
+    }
+
+}