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 ); }
public virtual void TestFairReservationSystemInitialize() { 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); } ReservationSystemTestUtil.ValidateReservationQueue(reservationSystem, testUtil.GetFullReservationQueueName ()); }