Last active
August 25, 2016 21:18
-
-
Save jsatt/48c745fc5c8edfae76e0d9674b0e3afe to your computer and use it in GitHub Desktop.
Test unmanaged models
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
TEST_RUNNER = b'test_runner.CustomRunner' |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from django_nose import NoseTestSuiteRunner | |
class CustomRunner(NoseTestSuiteRunner): | |
def setup_test_environment(self, *args, **kwargs): | |
from django.db.models.loading import get_models | |
self.unmanaged_models = [m for m in get_models() if not m._meta.managed] | |
for m in self.unmanaged_models: | |
m._meta.managed = True | |
super(CustomRunner, self).setup_test_environment(*args, **kwargs) | |
def teardown_test_environment(self, *args, **kwargs): | |
super(CustomRunner, self).teardown_test_environment(*args, **kwargs) | |
# reset unmanaged models | |
for m in self.unmanaged_models: | |
m._meta.managed = False |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment