Source code for pytoolbox.django.test.runner.mixins

"""
Mix-ins for building your own test runners.
"""

import tempfile

from django.conf import settings

__all__ = ['CeleryInMemoryMixin', 'FastPasswordHasherMixin', 'TemporarySendfileRootMixin']


[docs]class CeleryInMemoryMixin(object):

[docs]    def setup_test_environment(self):
        super().setup_test_environment()
        settings.BROKER_BACKEND = 'memory'
        settings.CELERY_EAGER_PROPAGATES_EXCEPTIONS = True
        settings.CELERY_ALWAYS_EAGER = True


[docs]class FastPasswordHasherMixin(object):

[docs]    def setup_test_environment(self):
        super().setup_test_environment()
        settings.PASSWORD_HASHERS = ('django.contrib.auth.hashers.MD5PasswordHasher', )


[docs]class TemporarySendfileRootMixin(object):

[docs]    def setup_test_environment(self):
        super().setup_test_environment()
        settings.SENDFILE_ROOT = tempfile.mkdtemp()