32 lines
804 B
Python
32 lines
804 B
Python
|
from __future__ import absolute_import, unicode_literals
|
||
|
|
import os
|
||
|
|
|
||
|
|
test_runner = None
|
||
|
|
old_config = None
|
||
|
|
|
||
|
|
os.environ["DJANGO_SETTINGS_MODULE"] = "tests.settings"
|
||
|
|
|
||
|
|
import django
|
||
|
|
if hasattr(django, "setup"):
|
||
|
|
django.setup()
|
||
|
|
|
||
|
|
def setup():
|
||
|
|
global test_runner
|
||
|
|
global old_config
|
||
|
|
|
||
|
|
# If you want to support Django 1.5 and older, you need
|
||
|
|
# this try-except block.
|
||
|
|
try:
|
||
|
|
from django.test.runner import DiscoverRunner
|
||
|
|
test_runner = DiscoverRunner()
|
||
|
|
except ImportError:
|
||
|
|
from django.test.simple import DjangoTestSuiteRunner
|
||
|
|
test_runner = DjangoTestSuiteRunner()
|
||
|
|
|
||
|
|
test_runner.setup_test_environment()
|
||
|
|
old_config = test_runner.setup_databases()
|
||
|
|
|
||
|
|
def teardown():
|
||
|
|
test_runner.teardown_databases(old_config)
|
||
|
|
test_runner.teardown_test_environment()
|