class TestFlight(unittest.TestCase):
def setUp(self):
self.flight = None
self.airport = Airport()
def test_flight_constructor(self):
self.flight = Flight(start_time=Date(29, 11, 2016, hour='12:20'),
end_time=Date(29, 11, 2016, hour='15:30'),
passengers=100,
max_passengers=120,
from_dest="Sofia",
to_dest="London",
terminal=Terminal(2, 30),
declined=False)
def test_terminal_constructor(self):
self.terminal = Terminal(number=1, max_flights=20)
def test_passenger_constructor(self):
self.passenger = Passenger(first_name="Rositsa",
last_name="Zlateva",
flight=Flight(),
age=22)
def test_reservation_constructor(self):
self.reservation = Reservation(flight=Flight(),
passenger=Passenger(),
accepted=True)
def test_get_flights_for(self):
self.flight = constants.flights[2]
self.assertEqual(
self.airport.get_flights_for(Date(29, 11, 2016, hour='12:20'))[0],
self.flight)
def test_get_flight_before(self):
self.assertEqual(
self.airport.get_flights_before(Date(29, 11, 2016, hour='15:29')),
[constants.flights[2]])
def test_get_flight_from(self):
self.assertEqual(constants.flights[0],
self.airport.get_flight_from('Sofia')[0])
def test_get_flight_to(self):
self.assertEqual(constants.flights[0],
self.airport.get_flight_to('Sofia')[0])
