public virtual void TestDeleteReservationNoID()
{
ReservationDeleteRequest request = new ReservationDeleteRequestPBImpl();
Plan plan = null;
try
{
plan = rrValidator.ValidateReservationDeleteRequest(rSystem, request);
NUnit.Framework.Assert.Fail();
}
catch (YarnException e)
{
NUnit.Framework.Assert.IsNull(plan);
string message = e.Message;
NUnit.Framework.Assert.IsTrue(message.StartsWith("Missing reservation id. Please try again by specifying a reservation id."
));
Log.Info(message);
}
}
/// <exception cref="Com.Google.Protobuf.ServiceException"/>
public virtual YarnServiceProtos.ReservationDeleteResponseProto DeleteReservation
(RpcController controller, YarnServiceProtos.ReservationDeleteRequestProto requestProto
)
{
ReservationDeleteRequestPBImpl request = new ReservationDeleteRequestPBImpl(requestProto
);
try
{
ReservationDeleteResponse response = real.DeleteReservation(request);
return(((ReservationDeleteResponsePBImpl)response).GetProto());
}
catch (YarnException e)
{
throw new ServiceException(e);
}
catch (IOException e)
{
throw new ServiceException(e);
}
}
public virtual void TestDeleteReservationInvalidPlan()
{
ReservationDeleteRequest request = new ReservationDeleteRequestPBImpl();
ReservationId reservationID = ReservationSystemTestUtil.GetNewReservationId();
request.SetReservationId(reservationID);
Org.Mockito.Mockito.When(rSystem.GetPlan(PlanName)).ThenReturn(null);
Plan plan = null;
try
{
plan = rrValidator.ValidateReservationDeleteRequest(rSystem, request);
NUnit.Framework.Assert.Fail();
}
catch (YarnException e)
{
NUnit.Framework.Assert.IsNull(plan);
string message = e.Message;
NUnit.Framework.Assert.IsTrue(message.EndsWith(" is not associated with any valid plan. Please try again with a valid reservation."
));
Log.Info(message);
}
}
public virtual void TestDeleteReservationDoesnotExist()
{
ReservationDeleteRequest request = new ReservationDeleteRequestPBImpl();
ReservationId rId = ReservationSystemTestUtil.GetNewReservationId();
request.SetReservationId(rId);
Org.Mockito.Mockito.When(rSystem.GetQueueForReservation(rId)).ThenReturn(null);
Plan plan = null;
try
{
plan = rrValidator.ValidateReservationDeleteRequest(rSystem, request);
NUnit.Framework.Assert.Fail();
}
catch (YarnException e)
{
NUnit.Framework.Assert.IsNull(plan);
string message = e.Message;
NUnit.Framework.Assert.IsTrue(message.Equals(MessageFormat.Format("The specified reservation with ID: {0} is unknown. Please try again with a valid reservation."
, rId)));
Log.Info(message);
}
}
public virtual void TestDeleteReservationNormal()
{
ReservationDeleteRequest request = new ReservationDeleteRequestPBImpl();
ReservationId reservationID = ReservationSystemTestUtil.GetNewReservationId();
request.SetReservationId(reservationID);
ReservationAllocation reservation = Org.Mockito.Mockito.Mock <ReservationAllocation
>();
Org.Mockito.Mockito.When(plan.GetReservationById(reservationID)).ThenReturn(reservation
);
Plan plan = null;
try
{
plan = rrValidator.ValidateReservationDeleteRequest(rSystem, request);
}
catch (YarnException e)
{
NUnit.Framework.Assert.Fail(e.Message);
}
NUnit.Framework.Assert.IsNotNull(plan);
}
