def setUp(self):
self.gallery = Gallery.objects.get(slug='TestGallery')
self.picture = create_picture(self.gallery)
def test_unique_slug_multiple_conflicts(self):
create_picture(self.gallery, "slug").save()
create_picture(self.gallery, "slug-1").save()
create_picture(self.gallery, "slug-2").save()
self.assertEquals("slug-3", _unique_slug("slug"), "The slug should be incremented.")
def test_count_with_pictures(self):
expected = 4
for i in range(expected):
create_picture(self.gallery).save()
actual = self.gallery.picture_count()
self.assertEquals(expected, actual)
