diff --git a/conftest.py b/conftest.py index a640db7822fd7ec1bc313b40cc103f213cf33403..cfad3adc62be1645c282f815347f8bed10f88abe 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 c4761815c2f697f5c0c5c49f24403c3575c432a4..066eb27855c36802b6a3eb20f3b1488fff407142 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