add tests for fietsctl track list/delcli-tests

1 files changed, 23 insertions, 2 deletions

diff --git a/tests/testutils.py b/tests/testutils.py
index 9db3673..9a62062 100644
--- a/tests/testutils.py
+++ b/tests/testutils.py
@@ -17,6 +17,8 @@ class PopulationIds(NamedTuple):
     """Collection of database IDs that :func:`populate` returns."""
     jon: int
     davos: int
+    winterfell: int
+    riverrun: int
 
 
 def load_test_asset(filename: str) -> bytes:
@@ -65,7 +67,18 @@ def extract_and_upload(page: Page, filename: str, tmp_path: Path):
 
 
 def populate(dbengine: Engine, data_manager: DataManager) -> PopulationIds:
-    """Populates the database and data directory with some test data."""
+    """Populates the database and data directory with some test data.
+
+    This adds:
+
+    Jon (jon.snow@nw.org, admin, password: ygritte)
+      Trip around Winterfell (2 images)
+      Road to Riverrun
+
+    Davos (davos@seaworth.com, password: 123456)
+
+    :return: An object carrying the database IDs of the added objects.
+    """
     user_ids = []
     with Session(dbengine) as session:
         user = models.User(name="Jon", email="jon.snow@nw.org", is_verified=True, is_admin=True)
@@ -87,6 +100,7 @@ def populate(dbengine: Engine, data_manager: DataManager) -> PopulationIds:
     for user_id in user_ids:
         data_manager.initialize_user(user_id)
 
+    track_ids = []
     # First track for Jon:
 
     gpx_data = load_gpx_asset("Teasi_1.gpx.gz")
@@ -110,6 +124,7 @@ def populate(dbengine: Engine, data_manager: DataManager) -> PopulationIds:
         session.flush()
 
         assert track.id is not None
+        track_ids.append(track.id)
         manager = data_manager.initialize(track.id)
         manager.compress_backup(gpx_data)
 
@@ -152,6 +167,7 @@ def populate(dbengine: Engine, data_manager: DataManager) -> PopulationIds:
         session.flush()
 
         assert track.id is not None
+        track_ids.append(track.id)
         manager = data_manager.initialize(track.id)
         manager.compress_backup(gpx_data)
 
@@ -161,4 +177,9 @@ def populate(dbengine: Engine, data_manager: DataManager) -> PopulationIds:
 
         session.commit()
 
-    return PopulationIds(jon=user_ids[0], davos=user_ids[1])
+    return PopulationIds(
+        jon=user_ids[0],
+        davos=user_ids[1],
+        winterfell=track_ids[0],
+        riverrun=track_ids[1],
+    )