mirror of
				https://github.com/ae-utbm/sith.git
				synced 2025-10-31 17:13:08 +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