OpenSlides/openslides/utils/test.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
    openslides.utils.test
    ~~~~~~~~~~~~~~~~~~~~~

    Unit test class.

    :copyright: 2011-2013 by OpenSlides team, see AUTHORS.
    :license: GNU GPL, see LICENSE for more details.
"""


from django.test import TestCase as _TestCase

from openslides.config.api import config
from openslides.core.signals import post_database_setup


class TestCase(_TestCase):
    """
    Overwrites Django's TestCase class to call the post_database_setup
    signal after the preparation of every test. Also refreshs the config cache.
    """
    def _pre_setup(self, *args, **kwargs):
        return_value = super(TestCase, self)._pre_setup(*args, **kwargs)
        post_database_setup.send(sender=self)
        return return_value

    def _post_teardown(self, *args, **kwargs):
        return_value = super(TestCase, self)._post_teardown(*args, **kwargs)
        # Resets the config object by deleting the cache
        try:
            del config._cache
        except AttributeError:
            # The cache has only to be deleted if it exists.
            pass
        return return_value