public SchedulerInfo(ResourceManager rm)
{
// JAXB needs this
ResourceScheduler rs = rm.GetResourceScheduler();
if (rs is CapacityScheduler)
{
this.schedulerName = "Capacity Scheduler";
}
else
{
if (rs is FairScheduler)
{
this.schedulerName = "Fair Scheduler";
}
else
{
if (rs is FifoScheduler)
{
this.schedulerName = "Fifo Scheduler";
}
}
}
this.minAllocResource = new ResourceInfo(rs.GetMinimumResourceCapability());
this.maxAllocResource = new ResourceInfo(rs.GetMaximumResourceCapability());
this.schedulingResourceTypes = rs.GetSchedulingResourceTypes();
}
private static ResourceScheduler MockResourceScheduler()
{
ResourceScheduler scheduler = Org.Mockito.Mockito.Mock <ResourceScheduler>();
Org.Mockito.Mockito.When(scheduler.GetMinimumResourceCapability()).ThenReturn(Resources
.CreateResource(YarnConfiguration.DefaultRmSchedulerMinimumAllocationMb));
Org.Mockito.Mockito.When(scheduler.GetMaximumResourceCapability()).ThenReturn(Resources
.CreateResource(YarnConfiguration.DefaultRmSchedulerMaximumAllocationMb));
ResourceCalculator rc = new DefaultResourceCalculator();
Org.Mockito.Mockito.When(scheduler.GetResourceCalculator()).ThenReturn(rc);
return(scheduler);
}
