diff --git a/program/tests/factories.py b/program/tests/factories.py index 066eb27855c36802b6a3eb20f3b1488fff407142..eb82b75ffef0454dfd5d95ae1e412cff35983eb7 100644 --- a/program/tests/factories.py +++ b/program/tests/factories.py @@ -17,6 +17,7 @@ from program.models import ( MusicFocus, Note, Playlist, + PlaylistEntry, Profile, RadioSettings, RRule, @@ -173,3 +174,13 @@ class CBAFactory(DjangoModelFactory): class PlaylistFactory(DjangoModelFactory): class Meta: model = Playlist + + +class PlaylistEntryFactory(DjangoModelFactory): + class Meta: + model = PlaylistEntry + + duration = 73 + file_id = 42 + line_num = Sequence(lambda n: int(n)) + uri = f"file://0/{file_id}"