mirror of
https://github.com/ae-utbm/sith.git
synced 2025-07-11 20:39:23 +00:00
WIP
This commit is contained in:
@ -3,8 +3,8 @@ from model_bakery import baker
|
||||
|
||||
from core.baker_recipes import old_subscriber_user, subscriber_user
|
||||
from core.models import User
|
||||
from sas.baker_recipes import picture_recipe
|
||||
from sas.models import Picture
|
||||
from sas.baker_recipes import album_recipe, picture_recipe
|
||||
from sas.models import Album, Picture
|
||||
|
||||
|
||||
class TestPictureQuerySet(TestCase):
|
||||
@ -44,3 +44,22 @@ class TestPictureQuerySet(TestCase):
|
||||
user.pictures.create(picture=self.pictures[1]) # moderated
|
||||
pictures = list(Picture.objects.viewable_by(user))
|
||||
assert pictures == [self.pictures[1]]
|
||||
|
||||
|
||||
class TestDeleteAlbum(TestCase):
|
||||
def setUp(cls):
|
||||
cls.album: Album = album_recipe.make()
|
||||
cls.album_pictures = picture_recipe.make(parent=cls.album, _quantity=5)
|
||||
cls.sub_album = album_recipe.make(parent=cls.album)
|
||||
cls.sub_album_pictures = picture_recipe.make(parent=cls.sub_album, _quantity=5)
|
||||
|
||||
def test_delete(self):
|
||||
album_ids = [self.album.id, self.sub_album.id]
|
||||
picture_ids = [
|
||||
*[p.id for p in self.album_pictures],
|
||||
*[p.id for p in self.sub_album_pictures],
|
||||
]
|
||||
self.album.delete()
|
||||
# assert not p.exists()
|
||||
assert not Album.objects.filter(id__in=album_ids).exists()
|
||||
assert not Picture.objects.filter(id__in=picture_ids).exists()
|
||||
|
Reference in New Issue
Block a user