runtests.py

#!/usr/bin/env python


import logging
import os
import sys

import django
from django.test.runner import DiscoverRunner

os.environ["DJANGO_SETTINGS_MODULE"] = "test_project.test_project.settings.test"


def setup_celery():
    try:
        from celery import Celery
    except ImportError:
        return
    app = Celery("test")
    app.config_from_object("django.conf:settings", namespace="CELERY")
    app.autodiscover_tasks()


def log_warnings():
    logger = logging.getLogger("py.warnings")
    handler = logging.StreamHandler()
    logger.addHandler(handler)


def run_tests(reverse=False):
    sys.stdout.write(
        "\nRunning spirit test suite, using settings %(settings)r\n\n"
        % {"settings": os.environ["DJANGO_SETTINGS_MODULE"]}
    )
    return DiscoverRunner(reverse=reverse).run_tests([])


def start():
    django.setup()
    log_warnings()
    setup_celery()
    if run_tests() or run_tests(reverse=True):
        sys.exit(1)


if __name__ == "__main__":
    start()