diff options
| -rw-r--r-- | tests/cli/test_fietsctl.py | 26 | ||||
| -rw-r--r-- | tests/testutils.py | 25 |
2 files changed, 49 insertions, 2 deletions
diff --git a/tests/cli/test_fietsctl.py b/tests/cli/test_fietsctl.py index 6e19f21..afc77fe 100644 --- a/tests/cli/test_fietsctl.py +++ b/tests/cli/test_fietsctl.py @@ -182,3 +182,29 @@ def test_user_modify(fietsctl, dbengine, data_manager): assert not user.is_verified assert user.is_admin assert user.email == "ser.davos@seaworth.com" + + +def test_track_list(fietsctl, dbengine, data_manager): + populate(dbengine, data_manager) + + res = fietsctl(["track", "list"]) + assert res.successful() + assert res.has_line("Jon.*Trip around Winterfell") + assert res.has_line("Jon.*Road to Riverrun") + + +def test_track_del(fietsctl, dbengine, data_manager): + ids = populate(dbengine, data_manager) + + res = fietsctl(["track", "del", "-i", str(ids.winterfell)]) + assert not res.successful() + + res = fietsctl(["track", "del", "-i", str(ids.winterfell), "-f"]) + assert res.successful() + + with Session(dbengine) as session: + track = session.get(models.Track, ids.winterfell) + assert track is None + + with pytest.raises(FileNotFoundError): + data_manager.open(ids.winterfell) 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], + ) |