Test your AI Agents with the all-new Agent to Agent Testing Platform.Learn More

How to use _save_active_wrapper method in pytest-django

Best Python code snippet using pytest-django_python

plugin.py

Source:plugin.py

...597        # reference to the real implementation.598        if self._real_ensure_connection is None:599            self._real_ensure_connection = BaseDatabaseWrapper.ensure_connection600        return BaseDatabaseWrapper601    def _save_active_wrapper(self):602        return self._history.append(self._dj_db_wrapper.ensure_connection)603    def _blocking_wrapper(*args, **kwargs):604        __tracebackhide__ = True605        __tracebackhide__  # Silence pyflakes606        raise RuntimeError(607            "Database access not allowed, "608            'use the "django_db" mark, or the '609            '"db" or "transactional_db" fixtures to enable it.'610        )611    def unblock(self):612        """Enable access to the Django database."""613        self._save_active_wrapper()614        self._dj_db_wrapper.ensure_connection = self._real_ensure_connection615        return _DatabaseBlockerContextManager(self)616    def block(self):617        """Disable access to the Django database."""618        self._save_active_wrapper()619        self._dj_db_wrapper.ensure_connection = self._blocking_wrapper620        return _DatabaseBlockerContextManager(self)621    def restore(self):622        self._dj_db_wrapper.ensure_connection = self._history.pop()623_blocking_manager = _DatabaseBlocker()624def validate_django_db(marker):625    """Validate the django_db marker.626    It checks the signature and creates the ``transaction`` and627    ``reset_sequences`` attributes on the marker which will have the628    correct values.629    A sequence reset is only allowed when combined with a transaction.630    """631    def apifun(transaction=False, reset_sequences=False):632        return transaction, reset_sequences...

Automation Testing Tutorials

Learn to execute automation testing from scratch with LambdaTest Learning Hub. Right from setting up the prerequisites to run your first automation test, to following best practices and diving deeper into advanced test scenarios. LambdaTest Learning Hubs compile a list of step-by-step guides to help you be proficient with different test automation frameworks i.e. Selenium, Cypress, TestNG etc.