feat: Adding getAllSessions (appium#446)

manoj9788 · KazuCocoa · commit 2c2a55ee3b70 · 2019-10-09T15:50:42.000+01:00
* Adding getAllSessions

* adjust per lint

* fix comments
diff --git a/appium/webdriver/mobilecommand.py b/appium/webdriver/mobilecommand.py
@@ -16,6 +16,7 @@
 class MobileCommand(object):
     # Common
     GET_SESSION = 'getSession'
+    GET_ALL_SESSIONS = 'getAllSessions'
 
     GET_LOCATION = 'getLocation'
     SET_LOCATION = 'setLocation'
diff --git a/appium/webdriver/webdriver.py b/appium/webdriver/webdriver.py
@@ -701,6 +701,16 @@ def session(self):
         """
         return self.execute(Command.GET_SESSION)['value']
 
+    @property
+    def all_sessions(self):
+        """ Retrieves all sessions that are open
+        Usage:
+            sessions = driver.allSessions
+        Returns:
+            `dict containing all open sessions`
+        """
+        return self.execute(Command.GET_ALL_SESSIONS)['value']
+
     @property
     def events(self):
         """ Retrieves events information from the current session
diff --git a/test/functional/ios/webdriver_tests.py b/test/functional/ios/webdriver_tests.py
@@ -24,10 +24,21 @@ class WebDriverTests(unittest.TestCase):
     def setUp(self):
         desired_caps = desired_capabilities.get_desired_capabilities('UICatalog.app.zip')
         self.driver = webdriver.Remote('http://localhost:4723/wd/hub', desired_caps)
+        desired_caps['deviceName'] = 'iPhone Xs Max'
+        desired_caps['wdaLocalPort'] = '8102'
 
     def tearDown(self):
         self.driver.quit()
 
+    def testAllSessions(self):
+        desired_caps = desired_capabilities.get_desired_capabilities('UICatalog.app.zip')
+        desired_caps['deviceName'] = 'iPhone Xs Max'
+        desired_caps['wdaLocalPort'] = '8102'
+        self.driver1 = webdriver.Remote('http://localhost:4723/wd/hub', desired_caps)
+        self.assertEqual(2, len(self.driver.all_sessions))
+        if self.driver1:
+            self.driver1.quit()
+
     def test_app_management(self):
         # this only works in Xcode9+
         if float(desired_capabilities.get_desired_capabilities(
diff --git a/test/unit/webdriver/webdriver_test.py b/test/unit/webdriver/webdriver_test.py
@@ -257,6 +257,17 @@ def test_create_session_register_uridirect_no_direct_connect_path(self):
         assert 'http://localhost:4723/wd/hub' == driver.command_executor._url
         assert ['NATIVE_APP', 'CHROMIUM'] == driver.contexts
 
+    @httpretty.activate
+    def test_get_all_sessions(self):
+        driver = ios_w3c_driver()
+        httpretty.register_uri(
+            httpretty.GET,
+            appium_command('/sessions'),
+            body=json.dumps({'value': {'deviceName': 'iPhone Simulator', 'events': {'simStarted': [1234567891]}}})
+        )
+        session = driver.all_sessions
+        assert len(session) != 1
+
     @httpretty.activate
     def test_get_session(self):
         driver = ios_w3c_driver()
