Ensure virtual display cleanup runs on browser close failure

This commit is contained in:
joaoneves0 2025-04-14 14:23:38 -03:00
parent 95cc4489d0
commit fd1bb2f878

View file

@ -302,7 +302,11 @@ async def async_attach_vd(
_close = browser.close _close = browser.close
async def new_close(*args: Any, **kwargs: Any): async def new_close(*args: Any, **kwargs: Any):
try:
await _close(*args, **kwargs) await _close(*args, **kwargs)
except Exception:
raise
finally:
if virtual_display: if virtual_display:
virtual_display.kill() virtual_display.kill()
@ -324,7 +328,11 @@ def sync_attach_vd(
_close = browser.close _close = browser.close
def new_close(*args: Any, **kwargs: Any): def new_close(*args: Any, **kwargs: Any):
try:
_close(*args, **kwargs) _close(*args, **kwargs)
except Exception:
raise
finally:
if virtual_display: if virtual_display:
virtual_display.kill() virtual_display.kill()