From 80d3b2a19e2a96eda8b9671e4cf0fad5eb61e4d6 Mon Sep 17 00:00:00 2001 From: Cliff Hill Date: Tue, 5 Oct 2021 17:10:40 -0400 Subject: [PATCH] Fine tuning tests. Signed-off-by: Cliff Hill --- src/playlist/data/models.py | 4 +++- tests/plex/test_server.py | 4 ++-- 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/src/playlist/data/models.py b/src/playlist/data/models.py index e100260..d6a5116 100644 --- a/src/playlist/data/models.py +++ b/src/playlist/data/models.py @@ -23,7 +23,9 @@ class Track(base.BaseData): metadata={"sa": sqlalchemy.Column(sqlalchemy.Integer, primary_key=True)}, ) plex_id: int = dataclasses.field( - metadata={"sa": sqlalchemy.Column(sqlalchemy.Integer, nullable=False)}, + metadata={ + "sa": sqlalchemy.Column(sqlalchemy.Integer, nullable=False, unique=True), + }, ) track_num: int = dataclasses.field( metadata={"sa": sqlalchemy.Column(sqlalchemy.Integer, nullable=False)}, diff --git a/tests/plex/test_server.py b/tests/plex/test_server.py index 85a85a4..b3d2a4f 100644 --- a/tests/plex/test_server.py +++ b/tests/plex/test_server.py @@ -39,7 +39,7 @@ def test_calc_duration(mocker): # type: ignore [no-untyped-def] @pytest.mark.asyncio -async def test_gen_tracks(mocker): # type: ignore [no-untyped-def] +async def test_gen_tracks(mocker, fake_track): # type: ignore [no-untyped-def] """Test the gen_tracks asynchronous generator.""" mock_downloader = mocker.patch( "playlist.plex.server._downloader", @@ -49,7 +49,7 @@ async def test_gen_tracks(mocker): # type: ignore [no-untyped-def] mock_models = mocker.patch("playlist.plex.server.models") mock_calc_delay = mocker.patch("playlist.plex.server.calc_delay") mock_calc_duration = mocker.patch("playlist.plex.server.calc_duration") - mock_downloader.return_value = (1, ["Not a real track"]) + mock_downloader.return_value = (1, [fake_track]) async def fake_gen_batch_params(*args): # type: ignore for item in [(0, 1), (1, 1)]: