public void Test_AddCity_AddsCityToFlight()
{
DateTime departing_time= new DateTime(2016,06,05);
DateTime returning_time = new DateTime(2016, 1, 1);
Flight testFlight = new Flight("E201","On time",departing_time,returning_time);
testFlight.Save();
City departCity = new City("Seattle","WA","USA");
City arrivalCity = new City("New York","WA","USA");
departCity.Save();
arrivalCity.Save();
//Act
testFlight.AddCity(departCity,arrivalCity);
List<City> result = testFlight.GetCities();
List<City> testList = new List<City> {departCity,arrivalCity};
//Assert
Assert.Equal(testList, result);
}
public void Test_Delete_DeletesFlightAssociationsFromDatabase()
{
//Arrange
DateTime departing_time= new DateTime(2016,06,05);
DateTime returning_time = new DateTime(2016, 1, 1);
Flight testFlight = new Flight("E201","On time",departing_time,returning_time);
testFlight.Save();
City departCity = new City("Seattle","WA","USA");
City arrivalCity = new City("New York","WA","USA");
departCity.Save();
arrivalCity.Save();
//Act
testFlight.AddCity(departCity,arrivalCity);
testFlight.Delete();
List<Flight> resultCityFlights = departCity.GetFlightsbyDeparting();
List<Flight> testCityFlights = new List<Flight> {};
//Assert
Assert.Equal(testCityFlights, resultCityFlights);
}
