Ejemplo n.º 1
0
  @Before
  public void setup() throws IOException {
    FairScheduler scheduler = new FairScheduler();
    Configuration conf = createConfiguration();
    // All tests assume only one assignment per node update
    conf.set(FairSchedulerConfiguration.ASSIGN_MULTIPLE, "false");
    ResourceManager resourceManager = new ResourceManager();
    resourceManager.init(conf);
    ((AsyncDispatcher) resourceManager.getRMContext().getDispatcher()).start();
    scheduler.reinitialize(conf, resourceManager.getRMContext());

    String queueName = "root.queue1";
    QueueManager mockMgr = mock(QueueManager.class);
    when(mockMgr.getMaxResources(queueName)).thenReturn(maxResource);
    when(mockMgr.getMinResources(queueName)).thenReturn(Resources.none());

    schedulable = new FSLeafQueue(queueName, mockMgr, scheduler, null);
  }
Ejemplo n.º 2
0
 private org.apache.hadoop.yarn.server.resourcemanager.NodeManager registerNode(
     String hostName, int containerManagerPort, int nmHttpPort, String rackName, int memory)
     throws IOException {
   return new org.apache.hadoop.yarn.server.resourcemanager.NodeManager(
       hostName,
       containerManagerPort,
       nmHttpPort,
       rackName,
       memory,
       resourceManager.getResourceTrackerService(),
       resourceManager.getRMContext());
 }