def test_cots_train_trip_removal(mock_navitia_fixture):
"""
test the import of cots_train_6113_trip_removal.json
"""
input_train_trip_removed = get_fixture_data(
'cots_train_6113_trip_removal.json')
with app.app_context():
rt_update = model.RealTimeUpdate(input_train_trip_removed,
connector='cots',
contributor='realtime.cots')
trip_updates = KirinModelBuilder(dumb_nav_wrapper()).build(rt_update)
rt_update.trip_updates = trip_updates
db.session.add(rt_update)
db.session.commit()
assert len(trip_updates) == 1
trip_up = trip_updates[0]
assert trip_up.vj.navitia_trip_id == 'trip:OCETGV-87686006-87751008-2:25768'
assert trip_up.vj_id == trip_up.vj.id
assert trip_up.status == 'delete'
# full trip removal : no stop_time to precise
assert len(trip_up.stop_time_updates) == 0
# verify trip_update effect:
assert trip_up.effect == 'NO_SERVICE'
def test_gtfs_model_builder(basic_gtfs_rt_data):
"""
test the model builder with a simple gtfs-rt
we have realtime data on only 2 stops, so the model builder should only have 2 stops (even if the VJ
have 4 stops)
"""
with app.app_context():
data = ''
rt_update = RealTimeUpdate(data, connector='gtfs-rt')
trip_updates = gtfs_rt.KirinModelBuilder(dumb_nav_wrapper()).build(rt_update, basic_gtfs_rt_data)
# we associate the trip_update manually for sqlalchemy to make the links
rt_update.trip_updates = trip_updates
db.session.add(rt_update)
db.session.commit()
assert len(trip_updates) == 1
assert len(trip_updates[0].stop_time_updates) == 2
second_stop = trip_updates[0].stop_time_updates[0]
assert second_stop.stop_id == 'StopR2'
assert second_stop.arrival_status == 'update'
assert second_stop.arrival_delay == timedelta(minutes=1)
assert second_stop.departure_delay is None
assert second_stop.departure_status == 'none'
assert second_stop.message is None
fourth_stop = trip_updates[0].stop_time_updates[1]
assert fourth_stop.stop_id == 'StopR4'
assert fourth_stop.arrival_status == 'update'
assert fourth_stop.arrival_delay == timedelta(minutes=3)
assert fourth_stop.departure_delay is None
assert fourth_stop.departure_status == 'none'
assert fourth_stop.message is None
def test_cots_train_delayed(mock_navitia_fixture):
"""
test the import of cots_train_96231_delayed.json
"""
input_train_delayed = get_fixture_data('cots_train_96231_delayed.json')
with app.app_context():
rt_update = model.RealTimeUpdate(input_train_delayed,
connector='cots',
contributor='realtime.cots')
trip_updates = KirinModelBuilder(dumb_nav_wrapper()).build(rt_update)
# we associate the trip_update manually for sqlalchemy to make the links
rt_update.trip_updates = trip_updates
db.session.add(rt_update)
db.session.commit()
assert len(trip_updates) == 1
trip_up = trip_updates[0]
assert trip_up.vj.navitia_trip_id == 'trip:OCETrainTER-87212027-85000109-3:11859'
assert trip_up.vj_id == trip_up.vj.id
assert trip_up.status == 'update'
assert trip_up.effect == 'SIGNIFICANT_DELAYS'
# 5 stop times must have been created
assert len(trip_up.stop_time_updates) == 6
# first impacted stop time should be 'gare de Sélestat'
st = trip_up.stop_time_updates[1]
assert st.id
assert st.stop_id == 'stop_point:OCE:SP:TrainTER-87214056'
# the COTS data has no listeHoraireProjeteArrivee, so the status is 'none'
assert st.arrival is None # not computed yet
assert st.arrival_delay is None
assert st.arrival_status == 'none'
assert st.departure is None
assert st.departure_delay == timedelta(minutes=15)
assert st.departure_status == 'update'
assert st.message == 'Affluence exceptionnelle de voyageurs'
# second impacted should be 'gare de Colmar'
st = trip_up.stop_time_updates[2]
assert st.id
assert st.stop_id == 'stop_point:OCE:SP:TrainTER-87182014'
assert st.arrival is None
assert st.arrival_delay == timedelta(minutes=15)
assert st.arrival_status == 'update'
assert st.departure is None
assert st.departure_delay == timedelta(minutes=15)
assert st.departure_status == 'update'
assert st.message == 'Affluence exceptionnelle de voyageurs'
# last should be 'gare de Basel-SBB'
st = trip_up.stop_time_updates[-1]
assert st.id
assert st.stop_id == 'stop_point:OCE:SP:TrainTER-85000109'
assert st.arrival is None
assert st.arrival_delay == timedelta(minutes=15)
assert st.arrival_status == 'update'
# no departure since it's the last (thus the departure will be before the arrival)
assert st.departure is None
assert st.departure_delay is None
assert st.departure_status == 'none'
assert st.message == 'Affluence exceptionnelle de voyageurs'
def test_get_action_on_trip_add(mock_navitia_fixture):
"""
Test the function _get_action_on_trip with different type of flux cots
returns:
1. Fist trip add(AJOUTEE)-> FIRST_TIME_ADDED
2. Add followed by update (PERTURBEE) -> PREVIOUSLY_ADDED
3. Delete followed by add -> FIRST_TIME_ADDED
"""
with app.app_context():
# Test for the first add: should be FIRST_TIME_ADDED
input_trip_add = get_fixture_data('cots_train_151515_added_trip.json')
json_data = json.loads(input_trip_add)
dict_version = model_maker.get_value(json_data, 'nouvelleVersion')
train_numbers = model_maker.get_value(dict_version, 'numeroCourse')
pdps = model_maker._retrieve_interesting_pdp(
model_maker.get_value(dict_version, 'listePointDeParcours'))
action_on_trip = model_maker._get_action_on_trip(
train_numbers, dict_version, pdps)
assert action_on_trip == ActionOnTrip.FIRST_TIME_ADDED.name
# Test for add followed by update should be PREVIOUSLY_ADDED
rt_update = model.RealTimeUpdate(input_trip_add,
connector='cots',
contributor='realtime.cots')
trip_updates = KirinModelBuilder(dumb_nav_wrapper()).build(rt_update)
_, log_dict = handle(rt_update,
trip_updates,
'realtime.cots',
is_new_complete=True)
input_update_added_trip = get_fixture_data(
'cots_train_151515_added_trip_with_delay.json')
json_data = json.loads(input_update_added_trip)
dict_version = model_maker.get_value(json_data, 'nouvelleVersion')
train_numbers = model_maker.get_value(dict_version, 'numeroCourse')
pdps = model_maker._retrieve_interesting_pdp(
model_maker.get_value(dict_version, 'listePointDeParcours'))
action_on_trip = model_maker._get_action_on_trip(
train_numbers, dict_version, pdps)
assert action_on_trip == ActionOnTrip.PREVIOUSLY_ADDED.name
# Clean database for further test
tables = [str(table) for table in db.metadata.sorted_tables]
db.session.execute('TRUNCATE {} CASCADE;'.format(', '.join(tables)))
db.session.commit()
# Delete the recently added trip followed by add: should be FIRST_TIME_ADDED
rt_update = model.RealTimeUpdate(input_trip_add,
connector='cots',
contributor='realtime.cots')
trip_updates = KirinModelBuilder(dumb_nav_wrapper()).build(rt_update)
_, log_dict = handle(rt_update,
trip_updates,
'realtime.cots',
is_new_complete=True)
input_trip_delete = get_fixture_data(
'cots_train_151515_deleted_trip_with_delay_and_stop_time_added.json'
)
rt_update = model.RealTimeUpdate(input_trip_delete,
connector='cots',
contributor='realtime.cots')
trip_updates = KirinModelBuilder(dumb_nav_wrapper()).build(rt_update)
_, log_dict = handle(rt_update,
trip_updates,
'realtime.cots',
is_new_complete=True)
input_added_trip = get_fixture_data(
'cots_train_151515_added_trip.json')
json_data = json.loads(input_added_trip)
dict_version = model_maker.get_value(json_data, 'nouvelleVersion')
train_numbers = model_maker.get_value(dict_version, 'numeroCourse')
pdps = model_maker._retrieve_interesting_pdp(
model_maker.get_value(dict_version, 'listePointDeParcours'))
action_on_trip = model_maker._get_action_on_trip(
train_numbers, dict_version, pdps)
assert action_on_trip == ActionOnTrip.FIRST_TIME_ADDED.name