How to use _wrap_generator_with_readonly method in autotest

Best Python code snippet using autotest_python Github


Full Screen

...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 = finally:32 readonly_connection.connection().unset_django_connection()33 yield first_value34 while True:35 yield 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)...

Full Screen

Full Screen

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.

LambdaTest Learning Hubs:


You could also refer to video tutorials over LambdaTest YouTube channel to get step by step demonstration from industry experts.

Run autotest automation tests on LambdaTest cloud grid

Perform automation testing on 3000+ real desktop and mobile devices online.

Try LambdaTest Now !!

Get 100 minutes of automation test minutes FREE!!

Next-Gen App & Browser Testing Cloud

Was this article helpful?