Snippet

Benutzerdefiniertes TestCase-Mixin für Programmatische SQL-Abfrage-Prüfungen

Durch Erweitern von Djangos `CaptureQueriesContext` lassen sich benutzerdefinierte Test-Mixins erstellen, die Datenbankabfragen erfassen und Tests automatisch fehlschlagen lassen, wenn Grenzwerte überschritten werden.

snippet.py

python

from django.test import TestCase
from django.test.utils import CaptureQueriesContext
from django.db import connection
 
class AssertQueryLimitMixin:
    def assert_max_queries(self, limit: int):
        class QueryContextManager(CaptureQueriesContext):
            def __exit__(self, exc_type, exc_val, exc_tb):
                super().__exit__(exc_type, exc_val, exc_tb)
                query_count = len(self.captured_queries)
                if query_count > limit:
                    raise AssertionError(f'Query limit exceeded: {query_count} > {limit}')
        return QueryContextManager(connection)

django

Erklärung

class AssertQueryLimitMixin:

Definiert ein wiederverwendbares Test-Mixin zur Überprüfung von Datenbankabfrage-Limits.

class QueryContextManager(CaptureQueriesContext):

Erweitert den Kontextmanager von Django zur Erfassung von SQL-Abfragen.

if query_count > limit:

Vergleicht die Anzahl erfasster SQL-Abfragen mit dem Schwellenwert und wirft bei Überschreitung einen AssertionError.

Vorheriges Snippet Nächstes Snippet

Mehr Python-Übungen für Experten →

Benutzerdefiniertes TestCase-Mixin für Programmatische SQL-Abfrage-Prüfungen

Verwandt