From b09034c23061e6cc06a10829d3edc251602d6fca Mon Sep 17 00:00:00 2001
From: Julian Descottes <jdescottes@mozilla.com>
Date: Tue, 6 Feb 2024 13:40:11 +0000
Subject: [PATCH] [wdspec] Add test for browser.removeUserContext to check
 browsing contexts are closed

Depends on D200023

Differential Revision: https://phabricator.services.mozilla.com/D200339

bugzilla-url: https://bugzilla.mozilla.org/show_bug.cgi?id=1874918
gecko-commit: 4be6b1980ac25cdf37cea10a1b246151b6646811
gecko-reviewers: webdriver-reviewers, whimboo
---
 .../remove_user_context/user_context.py       | 60 +++++++++++++++++++
 1 file changed, 60 insertions(+)

diff --git a/webdriver/tests/bidi/browser/remove_user_context/user_context.py b/webdriver/tests/bidi/browser/remove_user_context/user_context.py
index 3a5f7f6c05bfb1d..98d6a2e2c8bfc7a 100644
--- a/webdriver/tests/bidi/browser/remove_user_context/user_context.py
+++ b/webdriver/tests/bidi/browser/remove_user_context/user_context.py
@@ -1,5 +1,6 @@
 import pytest
 
+from tests.support.sync import AsyncPoll
 import webdriver.bidi.error as error
 
 from .. import get_user_context_ids
@@ -13,3 +14,62 @@ async def test_remove_context(bidi_session, create_user_context):
     await bidi_session.browser.remove_user_context(user_context=user_context)
     assert user_context not in await get_user_context_ids(bidi_session)
     assert "default" in await get_user_context_ids(bidi_session)
+
+
+@pytest.mark.parametrize("type_hint", ["tab", "window"])
+@pytest.mark.asyncio
+async def test_remove_context_closes_contexts(
+    bidi_session, subscribe_events, wait_for_event, create_user_context, type_hint
+):
+    # Subscribe to all browsing context events
+    await subscribe_events(events=["browsingContext.contextDestroyed"])
+
+    user_context_1 = await create_user_context()
+    user_context_2 = await create_user_context()
+
+    # context 1 and 2 are owned by user context 1
+    context_1 = await bidi_session.browsing_context.create(
+        user_context=user_context_1, type_hint=type_hint
+    )
+    context_2 = await bidi_session.browsing_context.create(
+        user_context=user_context_1, type_hint=type_hint
+    )
+    # context 3 and 4 are owned by user context 2
+    context_3 = await bidi_session.browsing_context.create(
+        user_context=user_context_2, type_hint=type_hint
+    )
+    context_4 = await bidi_session.browsing_context.create(
+        user_context=user_context_2, type_hint=type_hint
+    )
+
+    # Track all received browsingContext.contextDestroyed events in the events array
+    events = []
+
+    async def on_event(method, data):
+        events.append(data)
+
+    remove_listener = bidi_session.add_event_listener("browsingContext.contextDestroyed", on_event)
+
+    # destroy user context 1 and wait for context 1 and 2 to be destroyed
+    await bidi_session.browser.remove_user_context(user_context=user_context_1)
+
+    wait = AsyncPoll(bidi_session, timeout=2)
+    await wait.until(lambda _: len(events) >= 2)
+
+    assert len(events) == 2
+    destroyed_contexts = [event["context"] for event in events]
+    assert context_1["context"] in destroyed_contexts
+    assert context_2["context"] in destroyed_contexts
+
+    # destroy user context 1 and wait for context 3 and 4 to be destroyed
+    await bidi_session.browser.remove_user_context(user_context=user_context_2)
+
+    wait = AsyncPoll(bidi_session, timeout=2)
+    await wait.until(lambda _: len(events) >= 4)
+
+    assert len(events) == 4
+    destroyed_contexts = [event["context"] for event in events]
+    assert context_3["context"] in destroyed_contexts
+    assert context_4["context"] in destroyed_contexts
+
+    remove_listener()