@Mock
public static VcCluster findVcCluster(final String clusterName) {
if (flag) {
VcCluster vcCluster = Mockito.mock(VcCluster.class);
VcClusterConfig config = Mockito.mock(VcClusterConfig.class);
Mockito.when(vcCluster.getConfig()).thenReturn(config);
Mockito.when(config.getDRSEnabled()).thenReturn(true);
return vcCluster;
} else {
return null;
}
}