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()
|