2017-12-24 18:15:21 +00:00
|
|
|
import factory
|
|
|
|
|
|
|
|
from funkwhale_api.factories import registry
|
2018-03-18 20:31:59 +00:00
|
|
|
from funkwhale_api.music.factories import TrackFactory
|
2017-12-24 18:15:21 +00:00
|
|
|
from funkwhale_api.users.factories import UserFactory
|
|
|
|
|
|
|
|
|
|
|
|
@registry.register
|
|
|
|
class PlaylistFactory(factory.django.DjangoModelFactory):
|
2018-06-09 13:36:16 +00:00
|
|
|
name = factory.Faker("name")
|
2017-12-24 18:15:21 +00:00
|
|
|
user = factory.SubFactory(UserFactory)
|
|
|
|
|
|
|
|
class Meta:
|
2018-06-09 13:36:16 +00:00
|
|
|
model = "playlists.Playlist"
|
2018-03-18 20:31:59 +00:00
|
|
|
|
|
|
|
|
|
|
|
@registry.register
|
|
|
|
class PlaylistTrackFactory(factory.django.DjangoModelFactory):
|
|
|
|
playlist = factory.SubFactory(PlaylistFactory)
|
|
|
|
track = factory.SubFactory(TrackFactory)
|
|
|
|
|
|
|
|
class Meta:
|
2018-06-09 13:36:16 +00:00
|
|
|
model = "playlists.PlaylistTrack"
|