add bidi command session status

Author: navin772

py/selenium/webdriver/common/bidi/session.py

@@ -44,3 +44,21 @@ def session_unsubscribe(*events, browsing_contexts=None):
         cmd_dict["params"]["browsingContexts"] = browsing_contexts
     _ = yield cmd_dict
     return None
+
+
+def session_status():
+    """
+    The session.status command returns information about the remote end's readiness
+    to create new sessions and may include implementation-specific metadata.
+
+    Returns
+    -------
+    dict
+        Dictionary containing the ready state (bool), message (str) and metadata
+    """
+    cmd_dict = {
+        "method": "session.status",
+        "params": {},
+    }
+    result = yield cmd_dict
+    return result

py/selenium/webdriver/remote/webdriver.py

@@ -1269,6 +1269,30 @@ def network(self):
 
         return self._network
 
+    def get_bidi_session_status(self):
+        """
+        Get the session status using WebDriver BiDi.
+        Returns information about whether a remote end is in a state
+        in which it can create new sessions.
+
+        Returns:
+        -------
+        dict
+            Dictionary containing the ready state (bool), message (str) and metadata
+
+        Example:
+        --------
+        >>> status = driver.get_bidi_session_status()
+        >>> print(status["ready"])
+        >>> print(status["message"])
+        """
+        if not self._websocket_connection:
+            self._start_bidi()
+
+        from selenium.webdriver.common.bidi.session import session_status
+
+        return self._websocket_connection.execute(session_status())
+
     def _get_cdp_details(self):
         import json
 

py/test/selenium/webdriver/common/bidi_session_tests.py

@@ -0,0 +1,51 @@
+# Licensed to the Software Freedom Conservancy (SFC) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The SFC licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+import pytest
+
+from selenium.webdriver.common.window import WindowTypes
+
+
+@pytest.mark.xfail_safari
+def test_session_status(driver):
+    result = driver.get_bidi_session_status()
+
+    assert result is not None
+    assert "ready" in result
+    assert "message" in result
+    assert isinstance(result["ready"], bool)
+    assert isinstance(result["message"], str)
+
+
+@pytest.mark.xfail_safari
+def test_session_status_not_closed_with_one_window(driver):
+    # initial session status
+    initial_status = driver.get_bidi_session_status()
+    assert initial_status is not None
+
+    # Open new window and tab
+    driver.switch_to.new_window(WindowTypes.WINDOW)
+    driver.switch_to.new_window(WindowTypes.TAB)
+
+    # Close one window
+    driver.close()
+
+    # Session should still be active
+    status_after_closing = driver.get_bidi_session_status()
+    assert status_after_closing is not None
+    assert "ready" in status_after_closing
+    assert "message" in status_after_closing