def test_save_flight_when_plane_exists(self, database_session):
# Given
identifier_model = IdentifierModel()
identifier_model.code = 'FDY-198'
database_session.add(identifier_model)
plane_model = PlaneModel()
plane_model.places = 145
plane_model.identifier = identifier_model
database_session.add(plane_model)
flight = Flight(identifier=Identifier('2VG5'),
status='ongoing',
start_time=datetime.now(),
duration=937,
position=None)
plane = Plane(identifier=PlaneIdentifier('FDY-198'),
number_of_places=23)
flight_datasource = FlightDatasource(database_session)
# When
flight_datasource.save_new_flight(flight, plane)
# Then
assert database_session.query(FlightModel).count() == 1
saved_flight = database_session.query(FlightModel).first()
assert saved_flight.plane.identifier.code == 'FDY-198'
def test_return_the_found_flight(self, database_session):
# Given
flight_model = FlightModel()
flight_model.identifier = '3558'
flight_model.status = 'ongoing'
flight_model.start_time = datetime.now()
flight_model.duration = 257
flight_model.position = "40.714,-74.006"
database_session.add(flight_model)
flight_datasource = FlightDatasource(database_session)
# When
flight = flight_datasource.get_one_flight(Identifier('3558'))
# Then
assert flight
assert isinstance(flight, Flight)
assert flight.status == 'ongoing'
assert flight.duration == 257
assert isinstance(flight.position, Position)
assert flight.position.latitude == 40.714
assert flight.position.longitude == -74.006
assert isinstance(flight.identifier, Identifier)
assert flight.identifier.value == '3558'
assert flight.start_time == datetime.now()
def test_attach_a_new_position_for_a_flight(self, database_session):
# Given
position = Position(latitude=40.714, longitude=-74.006)
flight_model = FlightModel()
flight_model.identifier = '2VG5'
flight_model.status = 'ongoing'
flight_model.start_time = datetime.now()
flight_model.duration = 257
database_session.add(flight_model)
flight = Flight(identifier=Identifier('2VG5'),
status='ongoing',
start_time=datetime.now(),
duration=937,
position=None)
flight_datasource = FlightDatasource(database_session)
# When
flight_datasource.save_new_position(flight, position)
# Then
flight = database_session.query(FlightModel) \
.filter(FlightModel.identifier == '2VG5').first()
assert flight is not None
assert flight.position == "40.714,-74.006"
def test_return_an_empty_list_when_no_flight_registered(
self, database_session):
# Given
flight_datasource = FlightDatasource(database_session)
# When
flights = flight_datasource.get_all_flights()
# Then
assert len(flights) == 0
def test_return_none_when_the_flight_does_not_exist(
self, database_session):
# Given
flight_model = FlightModel()
flight_model.identifier = 3558
flight_model.status = 'ongoing'
flight_model.start_time = datetime.now()
flight_model.duration = 257
database_session.add(flight_model)
flight_datasource = FlightDatasource(database_session)
# When
flight = flight_datasource.get_one_flight(Identifier('2456'))
# Then
assert flight is None
def get_one_flight(flight_id):
flight_datasource = FlightDatasource(session=db.session)
flight = get_flight_details(repository=flight_datasource,
identifier=Identifier(flight_id))
if isinstance(flight, UndeclaredFlight):
return {}, 404
return jsonify(FlightSerializer.to_json(flight))
def test_return_(self, database_session):
# Given
flight_model = FlightModel()
flight_model.identifier = '5835'
flight_model.status = 'ongoing'
flight_model.start_time = datetime.now()
flight_model.duration = 257
flight_model.position = None
database_session.add(flight_model)
flight_datasource = FlightDatasource(database_session)
# When
flights = flight_datasource.get_all_flights()
# Then
assert len(flights) == 1
flight = flights[0]
assert flight.position is None
def add_a_plane(flight_id):
flight_repository = FlightDatasource(session=db.session)
payload = request.json
result = add_new_position_for_a_flight(flight_repository=flight_repository,
position=Position(
latitude=payload['latitude'],
longitude=payload['longitude']),
identifier=Identifier(flight_id))
if isinstance(result, UndeclaredFlight):
return {}, 404
return {}, 200
def post_a_new_plane(plane_id):
plane_datasource = PlaneDatasource(session=db.session)
flight_repository = FlightDatasource(session=db.session)
payload = request.json
flight = FlightSerializer.from_json(payload)
result = declare_new_flight(flight=flight,
plane_repository=plane_datasource,
flight_repository=flight_repository,
plane_identifier=PlaneIdentifier(plane_id))
if isinstance(result, UnregisteredPlane):
return {}, 400
return {}, 200
def get_all_flights():
flight_datasource = FlightDatasource(session=db.session)
flights = list_all_flights(repository=flight_datasource)
return jsonify(FlightSerializer.to_json(flights))