django-imagekit/tests/test_abstract_models.py
2013-02-10 15:42:39 -05:00

22 lines
735 B
Python

from django.core.files import File
from imagekit.signals import source_created
from imagekit.specs.sourcegroups import ImageFieldSourceGroup
from nose.tools import eq_
from . models import AbstractImageModel, ConcreteImageModel
from .utils import get_image_file
def test_source_created_signal():
source_group = ImageFieldSourceGroup(AbstractImageModel, 'original_image')
count = [0]
def receiver(sender, *args, **kwargs):
if sender is source_group:
count[0] += 1
source_created.connect(receiver, dispatch_uid='test_source_created')
instance = ConcreteImageModel()
img = File(get_image_file())
instance.original_image.save('test_source_created_signal.jpg', img)
eq_(count[0], 1)