Skip to content

Instantly share code, notes, and snippets.

@jsatt
Last active August 25, 2016 21:18
Show Gist options
  • Save jsatt/48c745fc5c8edfae76e0d9674b0e3afe to your computer and use it in GitHub Desktop.
Save jsatt/48c745fc5c8edfae76e0d9674b0e3afe to your computer and use it in GitHub Desktop.
Test unmanaged models
TEST_RUNNER = b'test_runner.CustomRunner'
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