How to use

TestReservationInputValidator

in

org.apache.hadoop.yarn.server.resourcemanager.reservation

private ReservationSubmissionRequest createSimpleReservationSubmissionRequest(
  int numRequests, int numContainers, long arrival, long deadline,
  long duration) {
 return createSimpleReservationSubmissionRequest(numRequests, numContainers,
   arrival, deadline, duration, "0");
}

private ReservationUpdateRequest createSimpleReservationUpdateRequest(
  int numRequests, int numContainers, long arrival, long deadline,
  long duration) {
 return createSimpleReservationUpdateRequest(numRequests, numContainers,
   arrival, deadline, duration, "0");
}

@Test
public void testUpdateReservationInvalidPlan() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(1, 1, 1, 5, 4);
 when(rSystem.getPlan(PLAN_NAME)).thenReturn(null);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert
    .assertTrue(message
      .endsWith(" is not associated with any valid plan. Please try again with a valid reservation."));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationMaxPeriodIndivisibleByRecurrenceExp() {
 long indivisibleRecurrence =
   YarnConfiguration.DEFAULT_RM_RESERVATION_SYSTEM_MAX_PERIODICITY / 2 + 1;
 String recurrenceExp = Long.toString(indivisibleRecurrence);
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 1, 1, 5, 3, recurrenceExp);
 plan = null;
 try {
  plan = rrValidator.validateReservationSubmissionRequest(rSystem, request,
    ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message.startsWith("The maximum periodicity:"));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationInvalidPlan() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(1, 1, 1, 5, 4);
 when(rSystem.getPlan(PLAN_NAME)).thenReturn(null);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert
    .assertTrue(message
      .endsWith(" is not associated with any valid plan. Please try again with a valid reservation."));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationInvalidPlan() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 1, 1, 5, 3);
 when(rSystem.getPlan(PLAN_NAME)).thenReturn(null);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert
    .assertTrue(message
      .endsWith(" is not managed by reservation system. Please try again with a valid reservable queue."));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationInvalidRR() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(0, 0, 1, 5, 3);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("No resources have been specified to reserve"));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationInvalidPlan() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 1, 1, 5, 3);
 when(rSystem.getPlan(PLAN_NAME)).thenReturn(null);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert
    .assertTrue(message
      .endsWith(" is not managed by reservation system. Please try again with a valid reservable queue."));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationDoesnotExist() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(1, 1, 1, 5, 4);
 ReservationId rId = request.getReservationId();
 when(rSystem.getQueueForReservation(rId)).thenReturn(null);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert
    .assertTrue(message.equals(MessageFormat
      .format(
        "The specified reservation with ID: {0} is unknown. Please try again with a valid reservation.",
        rId)));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationExceedsGangSize() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 1, 1, 5, 4);
 Resource resource = Resource.newInstance(512, 1);
 when(plan.getTotalCapacity()).thenReturn(resource);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message.startsWith(
    "The size of the largest gang in the reservation definition"));
  Assert.assertTrue(message.contains(
    "exceed the capacity available "));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationEmptyRR() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(1, 0, 1, 5, 3);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("No resources have been specified to reserve"));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationInvalidDeadline() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 1, 1, 0, 3);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("The specified deadline: 0 is the past"));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationInvalidDeadline() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(1, 1, 1, 0, 3);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("The specified deadline: 0 is the past"));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationEmptyRR() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 0, 1, 5, 3);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("No resources have been specified to reserve"));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationInvalidRR() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(0, 0, 1, 5, 3);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("No resources have been specified to reserve"));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationInvalidDuration() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 1, 1, 3, 4);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message.startsWith("The time difference"));
  Assert
    .assertTrue(message
      .contains("must  be greater or equal to the minimum resource duration"));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationDoesnotExist() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(1, 1, 1, 5, 4);
 ReservationId rId = request.getReservationId();
 when(rSystem.getQueueForReservation(rId)).thenReturn(null);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert
    .assertTrue(message.equals(MessageFormat
      .format(
        "The specified reservation with ID: {0} is unknown. Please try again with a valid reservation.",
        rId)));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationExceedsGangSize() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(1, 1, 1, 5, 4);
 Resource resource = Resource.newInstance(512, 1);
 when(plan.getTotalCapacity()).thenReturn(resource);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert
    .assertTrue(message
      .startsWith("The size of the largest gang in the reservation refinition"));
  Assert.assertTrue(message.contains("exceed the capacity available "));
  LOG.info(message);
 }
}

@Test
public void testUpdateReservationInvalidDeadline() {
 ReservationUpdateRequest request =
   createSimpleReservationUpdateRequest(1, 1, 1, 0, 3);
 Plan plan = null;
 try {
  plan = rrValidator.validateReservationUpdateRequest(rSystem, request);
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("The specified deadline: 0 is the past"));
  LOG.info(message);
 }
}

@Test
public void testSubmitReservationInvalidRR() {
 ReservationSubmissionRequest request =
   createSimpleReservationSubmissionRequest(0, 0, 1, 5, 3);
 Plan plan = null;
 try {
  plan =
    rrValidator.validateReservationSubmissionRequest(rSystem, request,
      ReservationSystemTestUtil.getNewReservationId());
  Assert.fail();
 } catch (YarnException e) {
  Assert.assertNull(plan);
  String message = e.getMessage();
  Assert.assertTrue(message
    .startsWith("No resources have been specified to reserve"));
  LOG.info(message);
 }
}