Test public usage response dispatcher

Author: PrashantBhanage

server/src/test/java/com/cloud/api/ApiResponseHelperTest.java

@@ -34,6 +34,8 @@
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.DisplayName;
 import org.junit.jupiter.api.Test;
+import org.junit.jupiter.params.ParameterizedTest;
+import org.junit.jupiter.params.provider.ValueSource;
 import org.junit.jupiter.api.extension.ExtendWith;
 import org.mockito.InjectMocks;
 import org.mockito.Mock;
@@ -1413,6 +1415,71 @@ public void populateVmDiskBytesWriteUsageResponseTest() throws Exception {
         verify(entityManagerMock).findByIdIncludingRemoved(VolumeVO.class, 28L);
     }
 
+    @ParameterizedTest
+    @ValueSource(ints = {
+            UsageTypes.RUNNING_VM,
+            UsageTypes.IP_ADDRESS,
+            UsageTypes.NETWORK_BYTES_SENT,
+            UsageTypes.VM_DISK_IO_READ,
+            UsageTypes.VOLUME,
+            UsageTypes.TEMPLATE,
+            UsageTypes.SNAPSHOT,
+            UsageTypes.LOAD_BALANCER_POLICY,
+            UsageTypes.PORT_FORWARDING_RULE,
+            UsageTypes.NETWORK_OFFERING,
+            UsageTypes.VPN_USERS,
+            UsageTypes.SECURITY_GROUP,
+            UsageTypes.BACKUP,
+            UsageTypes.VM_SNAPSHOT,
+            UsageTypes.VOLUME_SECONDARY,
+            UsageTypes.VM_SNAPSHOT_ON_PRIMARY,
+            UsageTypes.BUCKET
+    })
+    public void createUsageResponseDispatchesAllUsageTypes(int usageType) {
+        UsageVO usageRecord = mock(UsageVO.class);
+        Account account = mock(Account.class);
+        DomainVO domain = new DomainVO();
+
+        when(usageRecord.getAccountId()).thenReturn(1L);
+        when(usageRecord.getZoneId()).thenReturn(null);
+        when(usageRecord.getDescription()).thenReturn("usage description");
+        when(usageRecord.getUsageDisplay()).thenReturn("usage");
+        when(usageRecord.getUsageType()).thenReturn(usageType);
+        when(usageRecord.getVmInstanceId()).thenReturn(null);
+        when(usageRecord.getVmName()).thenReturn(null);
+        when(usageRecord.getTemplateId()).thenReturn(null);
+        when(usageRecord.getOfferingId()).thenReturn(1L);
+        when(usageRecord.getUsageId()).thenReturn(1L);
+        when(usageRecord.getNetworkId()).thenReturn(null);
+        when(usageRecord.getRawUsage()).thenReturn(null);
+        when(usageRecord.getSize()).thenReturn(null);
+        when(usageRecord.getVirtualSize()).thenReturn(null);
+        when(usageRecord.getType()).thenReturn("UserVm");
+        when(usageRecord.getStartDate()).thenReturn(null);
+        when(usageRecord.getEndDate()).thenReturn(null);
+
+        when(account.getType()).thenReturn(Account.Type.NORMAL);
+        when(account.getId()).thenReturn(1L);
+        when(account.getUuid()).thenReturn("account-uuid");
+        when(account.getAccountName()).thenReturn("account");
+        when(account.getDomainId()).thenReturn(2L);
+        domain.setId(2L);
+        domain.setUuid("domain-uuid");
+        domain.setName("domain");
+        domain.setPath("/domain/");
+
+        try (MockedStatic<ApiDBUtils> ignored = Mockito.mockStatic(ApiDBUtils.class)) {
+            when(ApiDBUtils.findAccountById(1L)).thenReturn(account);
+            when(ApiDBUtils.findDomainById(2L)).thenReturn(domain);
+
+            UsageRecordResponse response = helper.createUsageResponse(usageRecord, null, false);
+
+            assertEquals(usageType, response.getUsageType());
+            assertEquals("account-uuid", response.getAccountId());
+            assertEquals("domain-uuid", response.getDomainId());
+        }
+    }
+
     private Object invokeUsageDetailsHelper(String methodName, Class<?>[] parameterTypes, Object... args) throws Exception {
         Method method = ApiResponseHelper.class.getDeclaredMethod(methodName, parameterTypes);
         method.setAccessible(true);