Plain Old CLR Objects - Entity Framework 6 Recipes - page 287

repository.AddTrain(train);
repository.AddSchedule(schedule);
repository.AddReservation(reservation);
repository.SaveChanges();
}
[TestMethod]
[ExpectedException(typeof(InvalidOperationException))]
public void TestForDuplicateReservation()
{
var repository = new ReservationRepository(_context);
var schedule = repository.GetActiveSchedulesForTrain(1).First();
var reservation = new Reservation { ReservationId = 2,
Schedule = schedule,
Passenger = "Phil Marlowe" };
repository.AddReservation(reservation);
repository.SaveChanges();
}
[TestMethod]
[ExpectedException(typeof(InvalidOperationException))]
public void TestForArrivalDateGreaterThanDepartureDate()
{
var repository = new ReservationRepository(_context);
var schedule = new Schedule { ScheduleId = 2, TrainId = 1,
ArrivalDate = DateTime.Today,
DepartureDate = DateTime.Now,
ArrivesAt = "New York",
LeavesFrom = "Chicago" };
repository.AddSchedule(schedule);
repository.SaveChanges();
}
[TestMethod]
[ExpectedException(typeof(InvalidOperationException))]
public void TestForArrivesAndLeavesFromSameLocation()
{
var repository = new ReservationRepository(_context);
var schedule = new Schedule { ScheduleId = 3, TrainId = 1,
ArrivalDate = DateTime.Now,
DepartureDate = DateTime.Today,
ArrivesAt = "Dallas",
LeavesFrom = "Dallas" };
repository.AddSchedule(schedule);
repository.SaveChanges();
}
}