fix context switch tests in unit test

Author: KazuCocoa

.github/workflows/functional-test.yml

@@ -99,7 +99,7 @@ jobs:
             name: func_test_android3
           - target: test/functional/android/finger_print_tests.py test/functional/android/screen_record_tests.py test/functional/android/settings_tests.py test/functional/android/chrome_tests.py
             name: func_test_android4
-          - target: test/functional/android/context_switching_tests.py test/functional/android/remote_fs_tests.py
+          - target: test/functional/android/remote_fs_tests.py
             name: func_test_android5
           - target: test/functional/android/common_tests.py test/functional/android/webelement_tests.py
             name: func_test_android6

test/functional/android/context_switching_tests.py

@@ -14,12 +14,41 @@
 
 import httpretty
 
-from test.unit.helper.test_helper import android_w3c_driver, appium_command
+from test.unit.helper.test_helper import android_w3c_driver, appium_command, get_httpretty_request_body
 
 
 class TestWebDriverContext(object):
+    @httpretty.activate
+    def test_current_contexts(self):
+        driver = android_w3c_driver()
+        httpretty.register_uri(
+            httpretty.GET, appium_command('/session/1234567890/context'), body='{"value": "NATIVE_APP"}'
+        )
+        assert driver.current_context == 'NATIVE_APP'
+
     @httpretty.activate
     def test_get_contexts(self):
         driver = android_w3c_driver()
-        httpretty.register_uri(httpretty.GET, appium_command('/session/1234567890/context'), body='{"value": "NATIVE"}')
-        assert driver.current_context == 'NATIVE'
+        httpretty.register_uri(
+            httpretty.GET, appium_command('/session/1234567890/contexts'), body='{"value": ["NATIVE_APP", "CHROMIUM"]}'
+        )
+
+        assert ['NATIVE_APP', 'CHROMIUM'] == driver.contexts
+
+    @httpretty.activate
+    def test_switch_to_context(self):
+        driver = android_w3c_driver()
+        httpretty.register_uri(httpretty.POST, appium_command('/session/1234567890/context'), body='{"value": null}')
+
+        driver.switch_to.context(None)
+
+        assert {'name': None}, get_httpretty_request_body(httpretty.last_request())
+
+    @httpretty.activate
+    def test_switch_to_context_native_app(self):
+        driver = android_w3c_driver()
+        httpretty.register_uri(httpretty.POST, appium_command('/session/1234567890/context'), body='{"value": null}')
+
+        driver.switch_to.context('NATIVE_APP')
+
+        assert {'name': 'NATIVE_APP'}, get_httpretty_request_body(httpretty.last_request())

test/unit/webdriver/context_test.py

@@ -384,7 +384,7 @@ def test_extention_command_check(self):
         assert {
             'args': [{'component': 'io.appium.android.apis/.accessibility.AccessibilityNodeProviderActivity'}],
             'script': 'mobile: startActivity',
-        }, get_httpretty_request_body(httpretty.last_request())
+        } == get_httpretty_request_body(httpretty.last_request())
 
 
 class SubWebDriver(WebDriver):