1717package org .apache .cloudstack .storage .allocator ;
1818
1919
20- import static org .mockito .Mockito .when ;
21-
22- import java .util .ArrayList ;
23- import java .util .HashSet ;
24- import java .util .List ;
25- import java .util .Set ;
20+ import com .cloud .deploy .DeploymentPlan ;
21+ import com .cloud .deploy .DeploymentPlanner ;
22+ import com .cloud .storage .Storage ;
23+ import com .cloud .storage .StoragePool ;
24+ import com .cloud .storage .dao .VolumeDao ;
25+ import com .cloud .user .Account ;
26+ import com .cloud .vm .DiskProfile ;
27+ import com .cloud .vm .VirtualMachineProfile ;
2628
2729import org .apache .cloudstack .storage .datastore .db .StoragePoolVO ;
30+
2831import org .junit .After ;
2932import org .junit .Assert ;
3033import org .junit .Before ;
3437import org .mockito .Mockito ;
3538import org .mockito .junit .MockitoJUnitRunner ;
3639
37- import com .cloud .deploy .DeploymentPlan ;
38- import com .cloud .deploy .DeploymentPlanner ;
39- import com .cloud .storage .Storage ;
40- import com .cloud .storage .StoragePool ;
41- import com .cloud .storage .dao .VolumeDao ;
42- import com .cloud .user .Account ;
43- import com .cloud .vm .DiskProfile ;
44- import com .cloud .vm .VirtualMachineProfile ;
40+ import java .util .ArrayList ;
41+ import java .util .HashSet ;
42+ import java .util .List ;
43+ import java .util .Set ;
44+
45+ import static org .mockito .Mockito .when ;
4546
4647@ RunWith (MockitoJUnitRunner .class )
4748public class AbstractStoragePoolAllocatorTest {
@@ -74,25 +75,23 @@ public void tearDown() {
7475
7576 @ Test
7677 public void reorderStoragePoolsBasedOnAlgorithm_random () {
77- allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account );
78+ allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account , "random" );
7879 Mockito .verify (allocator , Mockito .times (0 )).reorderPoolsByCapacity (plan , pools );
7980 Mockito .verify (allocator , Mockito .times (0 )).reorderPoolsByNumberOfVolumes (plan , pools , account );
8081 Mockito .verify (allocator , Mockito .times (1 )).reorderRandomPools (pools );
8182 }
8283
8384 @ Test
8485 public void reorderStoragePoolsBasedOnAlgorithm_userdispersing () {
85- allocator .volumeAllocationAlgorithm = "userdispersing" ;
8686 Mockito .doReturn (pools ).when (allocator ).reorderPoolsByNumberOfVolumes (plan , pools , account );
87- allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account );
87+ allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account , "userdispersing" );
8888 Mockito .verify (allocator , Mockito .times (0 )).reorderPoolsByCapacity (plan , pools );
8989 Mockito .verify (allocator , Mockito .times (1 )).reorderPoolsByNumberOfVolumes (plan , pools , account );
9090 Mockito .verify (allocator , Mockito .times (0 )).reorderRandomPools (pools );
9191 }
9292
9393 @ Test
9494 public void reorderStoragePoolsBasedOnAlgorithm_userdispersing_reorder_check () {
95- allocator .volumeAllocationAlgorithm = "userdispersing" ;
9695 allocator .volumeDao = volumeDao ;
9796
9897 when (plan .getDataCenterId ()).thenReturn (1l );
@@ -104,7 +103,7 @@ public void reorderStoragePoolsBasedOnAlgorithm_userdispersing_reorder_check() {
104103 poolIds .add (9l );
105104 when (volumeDao .listPoolIdsByVolumeCount (1l ,1l ,1l ,1l )).thenReturn (poolIds );
106105
107- List <StoragePool > reorderedPools = allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account );
106+ List <StoragePool > reorderedPools = allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account , "userdispersing" );
108107 Assert .assertEquals (poolIds .size (),reorderedPools .size ());
109108
110109 Mockito .verify (allocator , Mockito .times (0 )).reorderPoolsByCapacity (plan , pools );
@@ -115,9 +114,8 @@ public void reorderStoragePoolsBasedOnAlgorithm_userdispersing_reorder_check() {
115114
116115 @ Test
117116 public void reorderStoragePoolsBasedOnAlgorithm_firstfitleastconsumed () {
118- allocator .volumeAllocationAlgorithm = "firstfitleastconsumed" ;
119117 Mockito .doReturn (pools ).when (allocator ).reorderPoolsByCapacity (plan , pools );
120- allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account );
118+ allocator .reorderStoragePoolsBasedOnAlgorithm (pools , plan , account , "firstfitleastconsumed" );
121119 Mockito .verify (allocator , Mockito .times (1 )).reorderPoolsByCapacity (plan , pools );
122120 Mockito .verify (allocator , Mockito .times (0 )).reorderPoolsByNumberOfVolumes (plan , pools , account );
123121 Mockito .verify (allocator , Mockito .times (0 )).reorderRandomPools (pools );
0 commit comments