@Test
public void testGetStorageUnitIds()
{

    // Create a storage unit availability DTOs.
    StorageUnitAvailabilityDto storageUnitAvailabilityDto1 = new StorageUnitAvailabilityDto();
    storageUnitAvailabilityDto1.setStorageUnitId(INTEGER_VALUE);
    StorageUnitAvailabilityDto storageUnitAvailabilityDto2 = new StorageUnitAvailabilityDto();
    storageUnitAvailabilityDto2.setStorageUnitId(INTEGER_VALUE_2);
    // Get a list of storage unit ids.
    List<Integer> result = storageUnitHelper.getStorageUnitIds(Arrays.asList(storageUnitAvailabilityDto1, storageUnitAvailabilityDto2));
    // Validate the returned object.
    assertEquals(Arrays.asList(INTEGER_VALUE, INTEGER_VALUE_2), result);
    // Get a list of storage unit ids when the list of entities is empty.
    result = storageUnitHelper.getStorageUnitIds(new ArrayList<>());
    // Validate the returned object.
    assertNotNull(result);
    assertEquals(0, result.size());
}