72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
from __future__ import unicode_literals
|
|
import datetime
|
|
import django
|
|
import os
|
|
import shutil
|
|
import sys
|
|
from django.test.utils import get_runner
|
|
|
|
from base.console_scripts.admin import DJANGO_MAIN_MODULE
|
|
from base.tests.utils import mkdtemp
|
|
|
|
|
|
class DjangoEnvironment(object):
|
|
@staticmethod
|
|
def _install_djangoproject(path):
|
|
cmd = 'django-test-admin setup "{}"'.format(path)
|
|
os.system(cmd)
|
|
|
|
def __init__(self, path=None, remove_after=True):
|
|
self.path = path
|
|
|
|
self._remove_after = remove_after
|
|
self._original_sys_path = None
|
|
self._modified_sys_path = None
|
|
|
|
def __enter__(self):
|
|
if self.path is None:
|
|
prefix = 'testrun-{datetime}-'.format(
|
|
datetime=datetime.datetime.now().strftime('%Y%m%d-%H%M')
|
|
)
|
|
self.path = mkdtemp(prefix=prefix)
|
|
|
|
self._install_djangoproject(self.path)
|
|
|
|
self._original_sys_path = sys.path
|
|
sys.path.append(self.path)
|
|
self._modified_sys_path = sys.path
|
|
|
|
os.environ['DJANGO_SETTINGS_MODULE'] = '{}.settings'.format(DJANGO_MAIN_MODULE)
|
|
django.setup()
|
|
|
|
from django.conf import settings
|
|
self.settings = settings
|
|
|
|
return self
|
|
|
|
def __exit__(self, *args):
|
|
if self._modified_sys_path is not None and self._original_sys_path is not None:
|
|
if sys.path == self._modified_sys_path:
|
|
sys.path = self._original_sys_path
|
|
if self._remove_after:
|
|
shutil.rmtree(self.path)
|
|
|
|
|
|
class TestSuite(object):
|
|
@staticmethod
|
|
def run():
|
|
tests = ['base']
|
|
test_tags = None
|
|
exclude_test_tags = None
|
|
|
|
with DjangoEnvironment() as env:
|
|
test_runner_class = get_runner(env.settings)
|
|
test_runner = test_runner_class(tags=test_tags, exclude_tags=exclude_test_tags)
|
|
failures = test_runner.run_tests(tests)
|
|
|
|
return bool(failures)
|
|
|
|
def __call__(self):
|
|
sys.exit(self.run())
|