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)