TestMu Conference is Back! Future of Testing + AI LambdaTest G2 Announcement

How to use _inject_screenshot_and_page_source_pre_hooks method in Selene

Best Python code snippet using selene_python

config.py

Source:config.py

...210        folder = os.path.dirname(file)211        if not os.path.exists(folder) and folder:212            os.makedirs(folder)213        return file214    def _inject_screenshot_and_page_source_pre_hooks(self, hook):215        def save_and_log_screenshot(error: TimeoutException) -> Exception:216            path = WebHelper(self.driver).save_screenshot(217                self.generate_filename(suffix='.png')218            )219            self.last_screenshot = path220            return TimeoutException(221                error.msg222                + f'''223Screenshot: file://{path}'''224            )225        def save_and_log_page_source(error: TimeoutException) -> Exception:226            filename = (227                self.last_screenshot.replace('.png', '.html')228                if self.last_screenshot229                else self.generate_filename(suffix='.html')230            )231            path = WebHelper(self.driver).save_page_source(filename)232            self.last_page_source = path233            return TimeoutException(234                error.msg235                + f'''236PageSource: file://{path}'''237            )238        return pipe(239            save_and_log_screenshot240            if self.save_screenshot_on_failure241            else None,242            save_and_log_page_source243            if self.save_page_source_on_failure244            else None,245            hook,246        )247    def wait(self, entity):248        hook = self._inject_screenshot_and_page_source_pre_hooks(249            self.hook_wait_failure250        )251        return Wait(entity, at_most=self.timeout, or_fail_with=hook)252    @Config.timeout.setter253    def timeout(self, value: int):254        self._timeout = value255    @Config.base_url.setter256    def base_url(self, value: str):257        self._base_url = value258    @Config.set_value_by_js.setter259    def set_value_by_js(self, value: bool):260        self._set_value_by_js = value261    @Config.type_by_js.setter262    def type_by_js(self, value: bool):...

conftest.py

Source:conftest.py

...46#         return report.step(original_open)(relative_or_absolute_url)47#48#     @monkey.patch_method_in(SharedConfig)  # todo: consider patching Wait explicitly49#     def wait(self, entity):50#         hook = self._inject_screenshot_and_page_source_pre_hooks(51#             self.hook_wait_failure52#         )53#54#         from framework.assist.selene.report.wait import ReportedWait55#         return ReportedWait(56#             entity,57#             at_most=self.timeout,58#             or_fail_with=hook59#         )60@pytest.fixture(autouse=True)61def browser_management():62    yield shared.browser63    if (64        shared.admin_pro_driver is not ......

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.