import os

import pytest
from asgiref.sync import async_to_sync
from django.test import TestCase, TransactionTestCase
from pytest_django.django_compat import is_django_unittest
from pytest_django.plugin import validate_django_db

from openslides.utils.cache import element_cache


# Set an environment variable to stop the startup command
os.environ["NO_STARTUP"] = "1"


def pytest_collection_modifyitems(items):
    """
    Helper until https://github.com/pytest-dev/pytest-django/issues/214 is fixed.
    """

    def get_marker_transaction(test):
        marker = test.get_closest_marker("django_db")
        if marker:
            validate_django_db(marker)
            return marker.kwargs["transaction"]

        return None

    def has_fixture(test, fixture):
        funcargnames = getattr(test, "funcargnames", None)
        return funcargnames and fixture in funcargnames

    def weight_test_case(test):
        """
        Key function for ordering test cases like the Django test runner.
        """
        is_test_case_subclass = test.cls and issubclass(test.cls, TestCase)
        is_transaction_test_case_subclass = test.cls and issubclass(
            test.cls, TransactionTestCase
        )

        if is_test_case_subclass or get_marker_transaction(test) is False:
            return 0
        elif has_fixture(test, "db"):
            return 0

        if is_transaction_test_case_subclass or get_marker_transaction(test) is True:
            return 1
        elif has_fixture(test, "transactional_db"):
            return 1

        return 0

    items.sort(key=weight_test_case)


@pytest.fixture(autouse=True)
def constants(request):
    """
    Resets the constants on every test.

    Uses fake constants, if the db is not in use.
    """
    from openslides.utils.constants import set_constants, get_constants_from_apps

    if "django_db" in request.node.keywords or is_django_unittest(request):
        # When the db is created, use the original constants
        set_constants(get_constants_from_apps())
    else:
        # Else: Use fake constants
        set_constants({"constant1": "value1", "constant2": "value2"})


@pytest.fixture(autouse=True)
def reset_cache(request):
    """
    Resetts the cache for every test
    """
    if "django_db" in request.node.keywords or is_django_unittest(request):
        # When the db is created, use the original cachables
        async_to_sync(element_cache.cache_provider.clear_cache)()
        element_cache.ensure_cache(reset=True)

    # Set constant start_time
    element_cache.start_time = 1