52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
from pathlib import Path
|
|
|
|
from django.apps import apps
|
|
from django.test.runner import DiscoverRunner
|
|
|
|
|
|
class AppDiscoverRunner(DiscoverRunner):
|
|
def load_tests_for_label(self, label, discover_kwargs):
|
|
if label.startswith("apps."):
|
|
try:
|
|
app_config = apps.get_app_config(label.split(".", 1)[1])
|
|
except LookupError:
|
|
return super().load_tests_for_label(label, discover_kwargs)
|
|
|
|
test_dir = Path(app_config.path) / "tests"
|
|
if test_dir.exists():
|
|
repo_root = Path(__file__).resolve().parent.parent
|
|
discover_kwargs = dict(discover_kwargs)
|
|
discover_kwargs["top_level_dir"] = str(repo_root)
|
|
return self.test_loader.discover(
|
|
start_dir=str(test_dir),
|
|
**discover_kwargs,
|
|
)
|
|
|
|
return super().load_tests_for_label(label, discover_kwargs)
|
|
|
|
def build_suite(self, test_labels=None, extra_tests=None, **kwargs):
|
|
if test_labels:
|
|
return super().build_suite(
|
|
test_labels,
|
|
extra_tests=extra_tests,
|
|
**kwargs,
|
|
)
|
|
|
|
suite = self.test_suite()
|
|
repo_root = Path(__file__).resolve().parent.parent
|
|
|
|
for app_config in apps.get_app_configs():
|
|
test_dir = Path(app_config.path) / "tests"
|
|
if app_config.name.startswith("apps.") and test_dir.exists():
|
|
suite.addTests(
|
|
self.test_loader.discover(
|
|
start_dir=str(test_dir),
|
|
pattern=self.pattern,
|
|
top_level_dir=str(repo_root),
|
|
)
|
|
)
|
|
|
|
if extra_tests:
|
|
suite.addTests(extra_tests)
|
|
return suite
|