KaneAI - World’s First E2E Software Testing Agent.Book a Demo Join TestMu Conference

How to use _wrap_generator_with_readonly method in autotest

Best Python code snippet using autotest_python

model_logic.py

Source:model_logic.py

...17            finally:18                readonly_connection.connection().unset_django_connection()19        wrapper_method.__name__ = method.__name__20        return wrapper_method21def _wrap_generator_with_readonly(generator):22    """23    We have to wrap generators specially.  Assume it performs24    the query on the first call to next().25    """26    def wrapper_generator(*args, **kwargs):27        generator_obj = generator(*args, **kwargs)28        readonly_connection.connection().set_django_connection()29        try:30            first_value = generator_obj.next()31        finally:32            readonly_connection.connection().unset_django_connection()33        yield first_value34        while True:35            yield generator_obj.next()36    wrapper_generator.__name__ = generator.__name__37    return wrapper_generator38def _make_queryset_readonly(queryset):39    """40    Wrap all methods that do database queries with a readonly connection.41    """42    db_query_methods = ['count', 'get', 'get_or_create', 'latest', 'in_bulk',43                        'delete']44    for method_name in db_query_methods:45        method = getattr(queryset, method_name)46        wrapped_method = _wrap_with_readonly(method)47        setattr(queryset, method_name, wrapped_method)48    queryset.iterator = _wrap_generator_with_readonly(queryset.iterator)49class ReadonlyQuerySet(dbmodels.query.QuerySet):50    """51    QuerySet object that performs all database queries with the read-only52    connection.53    """54    def __init__(self, model=None):55        super(ReadonlyQuerySet, self).__init__(model)56        _make_queryset_readonly(self)57    def values(self, *fields):58        return self._clone(klass=ReadonlyValuesQuerySet, _fields=fields)59class ReadonlyValuesQuerySet(dbmodels.query.ValuesQuerySet):60    def __init__(self, model=None):61        super(ReadonlyValuesQuerySet, self).__init__(model)62        _make_queryset_readonly(self)...

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.