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)