103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
"""Tests for trip map coordinate assembly."""
|
|
|
|
from datetime import date
|
|
|
|
import agenda.trip
|
|
from agenda.types import Trip
|
|
from web_view import app
|
|
|
|
|
|
def test_add_coordinates_for_unbooked_flights_adds_missing_airports() -> None:
|
|
"""Unbooked routes should contribute missing airport pins."""
|
|
routes = [
|
|
{
|
|
"type": "unbooked_flight",
|
|
"key": "LHR_Paris_fr",
|
|
"from_iata": "LHR",
|
|
"to_iata": "CDG",
|
|
"from": (51.47, -0.45),
|
|
"to": (49.01, 2.55),
|
|
}
|
|
]
|
|
coordinates = [
|
|
{
|
|
"name": "Heathrow Airport",
|
|
"type": "airport",
|
|
"latitude": 51.47,
|
|
"longitude": -0.45,
|
|
}
|
|
]
|
|
airports = {
|
|
"LHR": {
|
|
"name": "Heathrow Airport",
|
|
"latitude": 51.47,
|
|
"longitude": -0.45,
|
|
},
|
|
"CDG": {
|
|
"name": "Paris Charles de Gaulle Airport",
|
|
"latitude": 49.01,
|
|
"longitude": 2.55,
|
|
},
|
|
}
|
|
|
|
with app.app_context():
|
|
original_parse_yaml = agenda.trip.travel.parse_yaml
|
|
try:
|
|
agenda.trip.travel.parse_yaml = lambda _name, _data_dir: airports
|
|
agenda.trip.add_coordinates_for_unbooked_flights(
|
|
routes, coordinates, app.config["PERSONAL_DATA"]
|
|
)
|
|
finally:
|
|
agenda.trip.travel.parse_yaml = original_parse_yaml
|
|
|
|
airport_names = {
|
|
coord["name"] for coord in coordinates if coord["type"] == "airport"
|
|
}
|
|
assert airport_names == {"Heathrow Airport", "Paris Charles de Gaulle Airport"}
|
|
|
|
|
|
def test_get_coordinates_and_routes_adds_unbooked_flight_airports() -> None:
|
|
"""Trip list map data should include pins for unbooked flights."""
|
|
trips = [Trip(start=date(2026, 7, 20))]
|
|
unbooked_routes = [
|
|
{
|
|
"type": "unbooked_flight",
|
|
"key": "LHR_Paris_fr",
|
|
"from_iata": "LHR",
|
|
"to_iata": "CDG",
|
|
"from": (51.47, -0.45),
|
|
"to": (49.01, 2.55),
|
|
}
|
|
]
|
|
airports = {
|
|
"LHR": {
|
|
"name": "Heathrow Airport",
|
|
"latitude": 51.47,
|
|
"longitude": -0.45,
|
|
},
|
|
"CDG": {
|
|
"name": "Paris Charles de Gaulle Airport",
|
|
"latitude": 49.01,
|
|
"longitude": 2.55,
|
|
},
|
|
}
|
|
|
|
with app.app_context():
|
|
original_collect_trip_coordinates = agenda.trip.collect_trip_coordinates
|
|
original_get_trip_routes = agenda.trip.get_trip_routes
|
|
original_parse_yaml = agenda.trip.travel.parse_yaml
|
|
try:
|
|
agenda.trip.collect_trip_coordinates = lambda _trip: []
|
|
agenda.trip.get_trip_routes = lambda _trip, _data_dir: unbooked_routes
|
|
agenda.trip.travel.parse_yaml = lambda _name, _data_dir: airports
|
|
|
|
coordinates, _routes = agenda.trip.get_coordinates_and_routes(trips)
|
|
finally:
|
|
agenda.trip.collect_trip_coordinates = original_collect_trip_coordinates
|
|
agenda.trip.get_trip_routes = original_get_trip_routes
|
|
agenda.trip.travel.parse_yaml = original_parse_yaml
|
|
|
|
airport_names = {
|
|
coord["name"] for coord in coordinates if coord["type"] == "airport"
|
|
}
|
|
assert airport_names == {"Heathrow Airport", "Paris Charles de Gaulle Airport"}
|