From e06f706102fda3d552198988904052c7dd9307c4 Mon Sep 17 00:00:00 2001 From: Ernesto Rico Schmidt <ernesto@helsinki.at> Date: Wed, 30 Oct 2024 17:57:47 -0400 Subject: [PATCH] test: add factory and fixture for Playlist --- conftest.py | 7 +++++++ program/tests/factories.py | 6 ++++++ 2 files changed, 13 insertions(+) diff --git a/conftest.py b/conftest.py index a640db78..cfad3adc 100644 --- a/conftest.py +++ b/conftest.py @@ -14,6 +14,7 @@ from program.models import ( License, LinkType, MusicFocus, + Playlist, Profile, RadioSettings, RRule, @@ -34,6 +35,7 @@ from program.tests.factories import ( LinkTypeFactory, MusicFocusFactory, OwnerFactory, + PlaylistFactory, ProfileFactory, RadioSettingsFactory, RRuleFactory, @@ -301,3 +303,8 @@ def topic() -> Topic: @pytest.fixture def owner() -> User: return OwnerFactory() + + +@pytest.fixture +def default_playlist(show) -> Playlist: + return PlaylistFactory(description="default playlist", show=show) diff --git a/program/tests/factories.py b/program/tests/factories.py index c4761815..066eb278 100644 --- a/program/tests/factories.py +++ b/program/tests/factories.py @@ -16,6 +16,7 @@ from program.models import ( LinkType, MusicFocus, Note, + Playlist, Profile, RadioSettings, RRule, @@ -167,3 +168,8 @@ class CBAFactory(DjangoModelFactory): username = Sequence(lambda n: "username_%d" % n) user_token = Sequence(lambda n: "user_token_%d" % n) + + +class PlaylistFactory(DjangoModelFactory): + class Meta: + model = Playlist -- GitLab