From ecb9420e8e5640d3bbb1159013b65d285859094e Mon Sep 17 00:00:00 2001 From: Martin Pitt Date: Tue, 23 Jul 2024 19:59:11 +0200 Subject: [PATCH] screenshot on failure --- bidi.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/bidi.py b/bidi.py index 80b2f1a..39c8ac6 100755 --- a/bidi.py +++ b/bidi.py @@ -1,6 +1,7 @@ #!/usr/bin/python3 import asyncio +import base64 import json import logging import socket @@ -436,7 +437,12 @@ async def main(): # await d.wait_page_load() print("\n\nSTEP: super-user-indicator") - await d.wait("#super-user-indicator") + try: + await d.wait("#super-user-indicator") + except ValueError: + s = await d.bidi("browsingContext.captureScreenshot", context=d.top_context, origin="document") + Path("screenshot.png").write_bytes(base64.b64decode(s["data"])) + raise # FIXME: wait for text helper for _ in range(5): t = await d.text("#super-user-indicator")