from django.conf import settings
from django.test.simple import DjangoTestSuiteRunner
from django_coverage.coverage_runner import CoverageRunner
from django_jenkins.runner import CITestSuiteRunner


class OurTestRunner(DjangoTestSuiteRunner):
    def build_suite(self, test_labels, *args, **kwargs):
        suite = super(OurTestRunner, self).build_suite(
            test_labels or settings.PROJECT_APPS, *args, **kwargs
        )

        return suite


class OurCoverageRunner(OurTestRunner, CoverageRunner):
    pass


class JenkinsTestRunner(OurTestRunner, CITestSuiteRunner):
    pass