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

How to use decorate_callable method in freezegun

Best Python code snippet using freezegun

decorator.py

Source:decorator.py

...43                    "_call_func",44                )45            ):46                continue47            setattr(klass, attr, self.decorate_callable(attr_value))48        return klass49    @abstractmethod50    def decorate_callable(self, func: Callable[..., T]) -> Callable[..., T]:51        pass52    def __call__(self, func: Callable[..., T]) -> Callable[..., T]:53        if isinstance(func, type):54            return self.decorate_class(func)55        return self.decorate_callable(func)56    def _call_func(self, func: Callable[..., T], *args: Any, **kwargs: Any) -> T:57        try:58            return func(*args, **kwargs)59        except TypeError as e:60            # static functions61            try:62                return func(*args[1:], **kwargs)63            except TypeError:64                # it wasn't that it was a static function...

decorators.py

Source:decorators.py

...15                continue16            attr_value = getattr(klass, attr)17            if not hasattr(attr_value, "__call__"):18                continue19            setattr(klass, attr, decorate_callable(attr_value))20        return klass21    def decorate_callable(test):22        @functools.wraps(test)23        def wrapper(*args, **kw):24            with mock.patch(25                'ecommerce.core.models.SiteConfiguration.course_catalog_api_client',26                mock.PropertyMock(return_value=EdxRestApiClient(27                    settings.COURSE_CATALOG_API_URL,28                    jwt='auth-token'29                ))30            ):31                return test(*args, **kw)32        return wrapper33    if isinstance(test, type):34        return decorate_class(test)35    return decorate_callable(test)36def mock_enterprise_api_client(test):37    """38    Custom decorator for mocking the property "enterprise_api_client" of39    siteconfiguration to construct a new instance of EdxRestApiClient with a40    dummy jwt value.41    """42    def decorate_class(klass):43        for attr in dir(klass):44            # Decorate only callable unit tests.45            if not attr.startswith('test_'):46                continue47            attr_value = getattr(klass, attr)48            if not hasattr(attr_value, '__call__'):49                continue50            setattr(klass, attr, decorate_callable(attr_value))51        return klass52    def decorate_callable(test):53        @functools.wraps(test)54        def wrapper(*args, **kw):55            with mock.patch(56                'ecommerce.core.models.SiteConfiguration.enterprise_api_client',57                mock.PropertyMock(58                    return_value=EdxRestApiClient(59                        settings.ENTERPRISE_API_URL,60                        jwt='auth-token'61                    )62                )63            ):64                return test(*args, **kw)65        return wrapper66    if isinstance(test, type):67        return decorate_class(test)...

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.