Add travel booking and place import scripts

tests/test_build_place_yaml.py

@@ -0,0 +1,101 @@
+"""Tests for agenda.build_place_yaml."""
+
+from pathlib import Path
+
+import yaml
+
+from agenda import build_place_yaml
+
+
+def test_upsert_station_adds_new_station(tmp_path: Path) -> None:
+    """Station upsert should add a new station to stations.yaml."""
+    path = tmp_path / "stations.yaml"
+    path.write_text("""- name: London St Pancras
+  latitude: 51.531921
+  longitude: -0.126361
+  country: gb
+  wikidata: Q720102
+  routes: {}
+""")
+
+    replaced = build_place_yaml.upsert_station(
+        tmp_path,
+        {
+            "name": "Paris Gare du Nord",
+            "latitude": 48.8809,
+            "longitude": 2.3553,
+            "country": "fr",
+            "wikidata": "Q624511",
+            "routes": {},
+        },
+    )
+
+    stations = yaml.safe_load(path.read_text())
+    assert replaced is False
+    assert [station["name"] for station in stations] == [
+        "London St Pancras",
+        "Paris Gare du Nord",
+    ]
+    assert "\n\n- name: Paris Gare du Nord\n" in path.read_text()
+
+
+def test_upsert_station_replaces_existing_station(tmp_path: Path) -> None:
+    """Station upsert should replace an existing station with the same name."""
+    path = tmp_path / "stations.yaml"
+    path.write_text("""- name: Paris Gare du Nord
+  latitude: 0
+  longitude: 0
+  country: fr
+  wikidata: Q624511
+  routes: {}
+""")
+
+    replaced = build_place_yaml.upsert_station(
+        tmp_path,
+        {
+            "name": "Paris Gare du Nord",
+            "latitude": 48.8809,
+            "longitude": 2.3553,
+            "country": "fr",
+            "wikidata": "Q624511",
+            "routes": {},
+        },
+    )
+
+    stations = yaml.safe_load(path.read_text())
+    assert replaced is True
+    assert len(stations) == 1
+    assert stations[0]["latitude"] == 48.8809
+    assert "\n\n- name:" not in path.read_text()
+
+
+def test_upsert_airport_adds_mapping_entry(tmp_path: Path) -> None:
+    """Airport upsert should add a new IATA-keyed airport entry."""
+    path = tmp_path / "airports.yaml"
+    path.write_text("""LHR:
+  iata: LHR
+  name: Heathrow Airport
+  city: London
+  country: gb
+  latitude: 51.47
+  longitude: -0.4543
+  qid: Q8691
+""")
+
+    replaced = build_place_yaml.upsert_airport(
+        tmp_path,
+        {
+            "iata": "ORY",
+            "name": "Paris Orly Airport",
+            "city": "Paris Orly Airport",
+            "country": "fr",
+            "latitude": 48.723333,
+            "longitude": 2.379444,
+            "qid": "Q193353",
+        },
+    )
+
+    airports = yaml.safe_load(path.read_text())
+    assert replaced is False
+    assert list(airports) == ["LHR", "ORY"]
+    assert airports["ORY"]["country"] == "fr"