From 7bf3e4e7a3f99b405c4f653b8a16ff0b18ebcb01 Mon Sep 17 00:00:00 2001 From: Matthew Tretter Date: Sun, 10 Feb 2013 15:54:17 -0500 Subject: [PATCH] Add test for @seanbell's utility Currently failing --- tests/models.py | 4 ++++ tests/test_abstract_models.py | 9 ++++++++- 2 files changed, 12 insertions(+), 1 deletion(-) diff --git a/tests/models.py b/tests/models.py index 61265f0..7520a5c 100644 --- a/tests/models.py +++ b/tests/models.py @@ -54,3 +54,7 @@ class AbstractImageModel(models.Model): class ConcreteImageModel(AbstractImageModel): pass + + +class ConcreteImageModelSubclass(ConcreteImageModel): + pass diff --git a/tests/test_abstract_models.py b/tests/test_abstract_models.py index c144136..a3ec774 100644 --- a/tests/test_abstract_models.py +++ b/tests/test_abstract_models.py @@ -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])