@BeforeClass(groups = { "TestClusterHealService" })
public static void setUp() throws Exception {
service = new ClusterHealService();
// mock cluster entity manager
IClusterEntityManager entityMgr = Mockito.mock(IClusterEntityManager.class);
// mock getDisks
List<DiskEntity> disks = new ArrayList<DiskEntity>();
for (int i = 0; i < 3; i++) {
DiskEntity disk = new DiskEntity(DATA_DISK_NAME_PREFIX + i);
disk.setVmdkPath(LOCAL_DS_MOID_PREFIX + i + "/" + disk.getName());
disk.setDatastoreName(LOCAL_DS_NAME_PREFIX + i);
disk.setDatastoreMoId(LOCAL_DS_MOID_PREFIX + i);
disk.setSizeInMB(20 * 1024);
disk.setDiskType(DiskType.SYSTEM_DISK.type);
disks.add(disk);
}
Mockito.when(entityMgr.getDisks("bj-worker-1")).thenReturn(disks);
// mock findByName(String, String, String)
NodeEntity node = new NodeEntity();
node.setVmName(NODE_1_NAME);
node.setHostName(HOST_NAME);
Mockito.when(
entityMgr.findByName(CLUSTER_NAME, NODE_GROUP_NAME, NODE_1_NAME))
.thenReturn(node);
service.setClusterEntityMgr(entityMgr);
// mock cluster config manager
ClusterConfigManager configMgr = Mockito.mock(ClusterConfigManager.class);