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); }