conftest.py

import pytest
from django.contrib.auth.models import Permission

from django_sql_dashboard.models import Dashboard


def pytest_collection_modifyitems(items):
    """Add django_db marker with databases to tests that need database access."""
    for item in items:
        fixturenames = getattr(item, "fixturenames", ())
        # Tests using client fixtures or dashboard_db need both databases
        if any(f in fixturenames for f in ("admin_client", "client", "dashboard_db")):
            item.add_marker(pytest.mark.django_db(databases=["default", "dashboard"]))


@pytest.fixture
def dashboard_db(settings):
    settings.DATABASES["dashboard"]["OPTIONS"] = {
        "options": "-c default_transaction_read_only=on -c statement_timeout=100"
    }


@pytest.fixture
def execute_sql_permission():
    return Permission.objects.get(
        content_type__app_label="django_sql_dashboard",
        content_type__model="dashboard",
        codename="execute_sql",
    )


@pytest.fixture
def saved_dashboard(dashboard_db):
    dashboard = Dashboard.objects.create(
        slug="test",
        title="Test dashboard",
        description="This [supports markdown](http://example.com/)",
        view_policy="public",
    )
    dashboard.queries.create(sql="select 11 + 33")
    dashboard.queries.create(sql="select 22 + 55")
    return dashboard