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):
|
|
|
|
name = factory.Faker('name')
|
|
|
|
user = factory.SubFactory(UserFactory)
|
|
|
|
|
|
|
|
class Meta:
|
|
|
|
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:
|
|
|
|
model = 'playlists.PlaylistTrack'
|