conftest.py

import pytest
import sys
from django.conf import settings


def pytest_configure():
    settings.configure(
        DATABASES={
            "default": {
                "ENGINE": "django.db.backends.sqlite3",
                # Override Django’s default behaviour of using an in-memory database
                # in tests for SQLite, since that avoids connection.close() working.
                "TEST": {"NAME": "test_db.sqlite3"},
            }
        },
        INSTALLED_APPS=[
            "django.contrib.auth",
            "django.contrib.contenttypes",
            "django.contrib.sessions",
            "django.contrib.admin",
            "channels",
        ],
        SECRET_KEY="Not_a_secret_key",
    )


def pytest_generate_tests(metafunc):
    if "samesite" in metafunc.fixturenames:
        metafunc.parametrize("samesite", ["Strict", "None"], indirect=True)


@pytest.fixture
def samesite(request, settings):
    """Set samesite flag to strict."""
    settings.SESSION_COOKIE_SAMESITE = request.param


@pytest.fixture
def samesite_invalid(settings):
    """Set samesite flag to strict."""
    settings.SESSION_COOKIE_SAMESITE = "Hello"


@pytest.fixture
def root_urlconf(settings):
    """Set ROOT_WEBSOCKET_URLCONF."""
    settings.ROOT_WEBSOCKET_URLCONF = "__src.routings"
    return settings.ROOT_WEBSOCKET_URLCONF


@pytest.fixture(autouse=True)
def mock_modules():
    """Save original modules for each test and clear a cache"""
    original_modules = sys.modules.copy()
    yield
    sys.modules = original_modules
    from django.urls.base import _get_cached_resolver
    _get_cached_resolver.cache_clear()