public virtual void SetUp()
{
scheduler = new FairScheduler();
Configuration conf = new YarnConfiguration();
conf.SetClass(YarnConfiguration.RmScheduler, typeof(FairScheduler), typeof(ResourceScheduler
));
conf.Set("yarn.scheduler.fair.event-log-enabled", "true");
// All tests assume only one assignment per node update
conf.Set(FairSchedulerConfiguration.AssignMultiple, "false");
resourceManager = new ResourceManager();
resourceManager.Init(conf);
((AsyncDispatcher)resourceManager.GetRMContext().GetDispatcher()).Start();
scheduler.Init(conf);
scheduler.Start();
scheduler.Reinitialize(conf, resourceManager.GetRMContext());
}
/// <exception cref="System.IO.IOException"/>
public static FairScheduler SetupFairScheduler(ReservationSystemTestUtil testUtil
, RMContext rmContext, Configuration conf, int numContainers)
{
FairScheduler scheduler = new FairScheduler();
scheduler.SetRMContext(rmContext);
Org.Mockito.Mockito.When(rmContext.GetScheduler()).ThenReturn(scheduler);
scheduler.Init(conf);
scheduler.Start();
scheduler.Reinitialize(conf, rmContext);
Resource resource = ReservationSystemTestUtil.CalculateClusterResource(numContainers
);
RMNode node1 = MockNodes.NewNodeInfo(1, resource, 1, "127.0.0.1");
NodeAddedSchedulerEvent nodeEvent1 = new NodeAddedSchedulerEvent(node1);
scheduler.Handle(nodeEvent1);
return(scheduler);
}
public virtual void TestFairReservationSystemReinitialize()
{
ReservationSystemTestUtil.SetupFSAllocationFile(AllocFile);
ReservationSystemTestUtil testUtil = new ReservationSystemTestUtil();
// Setup
RMContext mockRMContext = ReservationSystemTestUtil.CreateRMContext(conf);
scheduler = ReservationSystemTestUtil.SetupFairScheduler(testUtil, mockRMContext,
conf, 10);
FairReservationSystem reservationSystem = new FairReservationSystem();
reservationSystem.SetRMContext(mockRMContext);
try
{
reservationSystem.Reinitialize(scheduler.GetConf(), mockRMContext);
}
catch (YarnException e)
{
NUnit.Framework.Assert.Fail(e.Message);
}
// Assert queue in original config
string planQNam = testUtil.GetFullReservationQueueName();
ReservationSystemTestUtil.ValidateReservationQueue(reservationSystem, planQNam);
// Dynamically add a plan
ReservationSystemTestUtil.UpdateFSAllocationFile(AllocFile);
scheduler.Reinitialize(conf, mockRMContext);
try
{
reservationSystem.Reinitialize(conf, mockRMContext);
}
catch (YarnException e)
{
NUnit.Framework.Assert.Fail(e.Message);
}
string newQueue = "root.reservation";
ReservationSystemTestUtil.ValidateNewReservationQueue(reservationSystem, newQueue
);
}
