@Test public void testListReservationsEmptyQueue() { ReservationListRequest request = new ReservationListRequestPBImpl(); request.setQueue(""); Plan plan = null; try { plan = rrValidator.validateReservationListRequest(rSystem, request); Assert.fail(); } catch (YarnException e) { Assert.assertNull(plan); String message = e.getMessage(); Assert.assertTrue(message.equals( "The queue is not specified. Please try again with a valid " + "reservable queue.")); LOG.info(message); } }
@Test public void testListReservationsNullPlan() { ReservationListRequest request = new ReservationListRequestPBImpl(); request.setQueue(ReservationSystemTestUtil.reservationQ); when(rSystem.getPlan(ReservationSystemTestUtil.reservationQ)).thenReturn (null); Plan plan = null; try { plan = rrValidator.validateReservationListRequest(rSystem, request); Assert.fail(); } catch (YarnException e) { Assert.assertNull(plan); String message = e.getMessage(); Assert.assertTrue(message.equals( "The specified queue: " + ReservationSystemTestUtil.reservationQ + " is not managed by reservation system." + " Please try again with a valid reservable queue." )); LOG.info(message); } }
@Test public void testListReservationsInvalidTimeInterval() { ReservationListRequest request = new ReservationListRequestPBImpl(); request.setQueue(ReservationSystemTestUtil.reservationQ); request.setEndTime(1000); request.setStartTime(2000); when(rSystem.getPlan(ReservationSystemTestUtil.reservationQ)).thenReturn (this.plan); Plan plan = null; try { plan = rrValidator.validateReservationListRequest(rSystem, request); Assert.fail(); } catch (YarnException e) { Assert.assertNull(plan); String message = e.getMessage(); Assert.assertTrue(message.equals("The specified end time must be " + "greater than the specified start time.")); LOG.info(message); } }
@Test public void testListReservationsNormal() { ReservationListRequest request = new ReservationListRequestPBImpl(); request.setQueue(ReservationSystemTestUtil.reservationQ); request.setEndTime(1000); request.setStartTime(0); when(rSystem.getPlan(ReservationSystemTestUtil.reservationQ)).thenReturn (this.plan); Plan plan = null; try { plan = rrValidator.validateReservationListRequest(rSystem, request); } catch (YarnException e) { Assert.fail(e.getMessage()); } Assert.assertNotNull(plan); }
@Test public void testListReservationsInvalidTimeIntervalDefaults() { ReservationListRequest request = new ReservationListRequestPBImpl(); request.setQueue(ReservationSystemTestUtil.reservationQ); // Negative time gets converted to default values for Start Time and End // Time which are 0 and Long.MAX_VALUE respectively. request.setEndTime(-2); request.setStartTime(-1); when(rSystem.getPlan(ReservationSystemTestUtil.reservationQ)).thenReturn (this.plan); Plan plan = null; try { plan = rrValidator.validateReservationListRequest(rSystem, request); } catch (YarnException e) { Assert.fail(e.getMessage()); } Assert.assertNotNull(plan); }
recordFactory.newRecordInstance(ReservationListResponse.class); Plan plan = rValidator.validateReservationListRequest( reservationSystem, requestInfo); boolean includeResourceAllocations = requestInfo