django/tests/invalid_models_tests/base.py

# -*- encoding: utf-8 -*-
from __future__ import unicode_literals

from django.apps import apps
from django.test import SimpleTestCase


class IsolatedModelsTestCase(SimpleTestCase):

    def setUp(self):
        # The unmanaged models need to be removed after the test in order to
        # prevent bad interactions with the flush operation in other tests.
        self._old_models = apps.app_configs['invalid_models_tests'].models.copy()

    def tearDown(self):
        apps.app_configs['invalid_models_tests'].models = self._old_models
        apps.all_models['invalid_models_tests'] = self._old_models
        apps.clear_cache()