63 lines
2.1 KiB
Python
63 lines
2.1 KiB
Python
from .. import models
|
|
from .factories import GalleryFactory, PhotoFactory
|
|
from .helpers import PhotologueBaseTest
|
|
|
|
|
|
class GalleryTest(PhotologueBaseTest):
|
|
|
|
def setUp(self):
|
|
"""Create a test gallery with 2 photos."""
|
|
super().setUp()
|
|
self.test_gallery = GalleryFactory()
|
|
self.pl2 = PhotoFactory()
|
|
self.test_gallery.photos.add(self.pl)
|
|
self.test_gallery.photos.add(self.pl2)
|
|
|
|
def tearDown(self):
|
|
super().tearDown()
|
|
self.pl2.delete()
|
|
|
|
def test_public(self):
|
|
"""Method 'public' should only return photos flagged as public."""
|
|
self.assertEqual(self.test_gallery.public().count(), 2)
|
|
self.pl.is_public = False
|
|
self.pl.save()
|
|
self.assertEqual(self.test_gallery.public().count(), 1)
|
|
|
|
def test_photo_count(self):
|
|
"""Method 'photo_count' should return the count of the photos in this
|
|
gallery."""
|
|
self.assertEqual(self.test_gallery.photo_count(), 2)
|
|
self.pl.is_public = False
|
|
self.pl.save()
|
|
self.assertEqual(self.test_gallery.photo_count(), 1)
|
|
|
|
# Method takes an optional 'public' kwarg.
|
|
self.assertEqual(self.test_gallery.photo_count(public=False), 2)
|
|
|
|
def test_sample(self):
|
|
"""Method 'sample' should return a random queryset of photos from the
|
|
gallery."""
|
|
|
|
# By default we return all photos from the gallery (but ordered at random).
|
|
_current_sample_size = models.SAMPLE_SIZE
|
|
models.SAMPLE_SIZE = 5
|
|
self.assertEqual(len(self.test_gallery.sample()), 2)
|
|
|
|
# We can state how many photos we want.
|
|
self.assertEqual(len(self.test_gallery.sample(count=1)), 1)
|
|
|
|
# If only one photo is public then the sample cannot have more than one
|
|
# photo.
|
|
self.pl.is_public = False
|
|
self.pl.save()
|
|
self.assertEqual(len(self.test_gallery.sample(count=2)), 1)
|
|
|
|
self.pl.is_public = True
|
|
self.pl.save()
|
|
|
|
# We can limit the number of photos by changing settings.
|
|
models.SAMPLE_SIZE = 1
|
|
self.assertEqual(len(self.test_gallery.sample()), 1)
|
|
|
|
models.SAMPLE_SIZE = _current_sample_size
|