Source code
Revision control
Copy as Markdown
Other Tools
Test Info:
- This WPT test may be referenced by the following Test IDs:
- /webdriver/bidi/browser/close/pending_download.py - WPT Dashboard Interop Dashboard
import pytest
from webdriver.bidi.modules.script import ContextTarget
pytestmark = pytest.mark.asyncio
DOWNLOAD_WILL_BEGIN = "browsingContext.downloadWillBegin"
async def test_close_with_pending_download(
new_session, add_browser_capabilities, inline, event_loop, wait_for_future_safe
):
bidi_session = await new_session(
capabilities={"alwaysMatch": add_browser_capabilities({})}
)
new_tab = await bidi_session.browsing_context.create(type_hint="tab")
url = inline(
"""<a id="download_link" href="/_mozilla/webdriver/support/assets/big.png" download>download</a>"""
)
await bidi_session.browsing_context.navigate(
context=new_tab["context"], url=url, wait="complete"
)
await bidi_session.session.subscribe(events=[DOWNLOAD_WILL_BEGIN])
on_download_will_begin_future = event_loop.create_future()
async def on_event(_, data):
remove_listener()
on_download_will_begin_future.set_result(data)
remove_listener = bidi_session.add_event_listener(DOWNLOAD_WILL_BEGIN, on_event)
result = await bidi_session.browsing_context.locate_nodes(
context=new_tab["context"], locator={"type": "css", "value": "#download_link"}
)
await bidi_session.script.call_function(
arguments=[result["nodes"][0]],
function_declaration="(link) => link.click()",
target=ContextTarget(new_tab["context"]),
await_promise=True,
user_activation=True,
)
await wait_for_future_safe(on_download_will_begin_future)
await bidi_session.browser.close()
# Wait for the browser to actually close.
bidi_session.current_browser.wait()
assert bidi_session.current_browser.is_running is False