Add test for @seanbell's utility

Currently failing
This commit is contained in:
Matthew Tretter 2013-02-10 15:54:17 -05:00
parent de550b71df
commit 7bf3e4e7a3
2 changed files with 12 additions and 1 deletions

View file

@ -54,3 +54,7 @@ class AbstractImageModel(models.Model):
class ConcreteImageModel(AbstractImageModel):
pass
class ConcreteImageModelSubclass(ConcreteImageModel):
pass

View file

@ -1,8 +1,10 @@
from django.core.files import File
from imagekit.signals import source_created
from imagekit.specs.sourcegroups import ImageFieldSourceGroup
from imagekit.utils import get_nonabstract_descendants
from nose.tools import eq_
from . models import AbstractImageModel, ConcreteImageModel
from . models import (AbstractImageModel, ConcreteImageModel,
ConcreteImageModelSubclass)
from .utils import get_image_file
@ -20,3 +22,8 @@ def test_source_created_signal():
instance.original_image.save('test_source_created_signal.jpg', img)
eq_(count[0], 1)
def test_nonabstract_descendants_generator():
descendants = list(get_nonabstract_descendants(AbstractImageModel))
eq_(descendants, [ConcreteImageModel, ConcreteImageModelSubclass])