transaction_on_commit_hook_test_mixin.py

from unittest import mock
from django.test import TestCase
from django.db import transaction


class TransactionCommitHooksMixin:
    """
    Fake transaction commit to run delayed on_commit functions
    https://medium.com/gitux/speed-up-django-transaction-hooks-tests-6de4a558ef96
    """

    def run_commit_hooks(self: TestCase):
        for db_name in reversed(self._databases_names()):
            with mock.patch('django.db.backends.base.base.BaseDatabaseWrapper.validate_no_atomic_block',
                            lambda a: None):
                transaction.get_connection(using=db_name).run_and_clear_commit_hooks()