How to use

getServices

method

in

org.apache.hadoop.yarn.server.resourcemanager.ResourceManager$RMActiveServices

@Test
public void testPolicyInitializeAfterSchedulerInitialized() {
 @SuppressWarnings("resource")
 MockRM rm = new MockRM(conf);
 rm.init(conf);
 
 // ProportionalCapacityPreemptionPolicy should be initialized after
 // CapacityScheduler initialized. We will 
 // 1) find SchedulingMonitor from RMActiveService's service list, 
 // 2) check if ResourceCalculator in policy is null or not. 
 // If it's not null, we can come to a conclusion that policy initialized
 // after scheduler got initialized
 for (Service service : rm.getRMActiveService().getServices()) {
  if (service instanceof SchedulingMonitor) {
   ProportionalCapacityPreemptionPolicy policy =
     (ProportionalCapacityPreemptionPolicy) ((SchedulingMonitor) service)
       .getSchedulingEditPolicy();
   assertNotNull(policy.getResourceCalculator());
   return;
  }
 }
 
 fail("Failed to find SchedulingMonitor service, please check what happened");
}

SchedulingEditPolicy getSchedulingEditPolicy(MockRM rm) {
 ResourceManager.RMActiveServices activeServices = rm.getRMActiveService();
 SchedulingMonitor mon = null;
 for (Service service : activeServices.getServices()) {
  if (service instanceof SchedulingMonitor) {
   mon = (SchedulingMonitor) service;
   break;
  }
 }
 if (mon != null) {
  return mon.getSchedulingEditPolicy();
 }
 return null;
}