// mock a clusterEntityMgr and node group entity
ClusterEntity clusterEntity = Mockito.mock(ClusterEntity.class);
Mockito.when(clusterEntity.getIoShares()).thenReturn(Priority.HIGH);
IClusterEntityManager entityMgr = Mockito.mock(IClusterEntityManager.class);
NodeGroupEntity nodeGroup = Mockito.mock(NodeGroupEntity.class);
Mockito.when(entityMgr.findByName("test", "master"))
.thenReturn(nodeGroup);
Mockito.when(entityMgr.findByName("test")).thenReturn(clusterEntity);
Field field = service.getClass().getDeclaredField("clusterEntityMgr");
field.setAccessible(true);
field.set(service, entityMgr);