use instance method

appium · Oct 18, 2024 · 34dd9cc · 34dd9cc
1 parent 0564815
commit 34dd9cc
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/appium/webdriver/extensions/context.py b/appium/webdriver/extensions/context.py
@@ -58,6 +58,6 @@ def context(self) -> str:
         return self.current_context
 
     def _add_commands(self) -> None:
-        self.command_executor.extra_commands[Command.CONTEXTS] = ('GET', '/session/$sessionId/contexts')
-        self.command_executor.extra_commands[Command.GET_CURRENT_CONTEXT] = ('GET', '/session/$sessionId/context')
-        self.command_executor.extra_commands[Command.SWITCH_TO_CONTEXT] = ('POST', '/session/$sessionId/context')
+        self.command_executor.add_command(Command.CONTEXTS, 'GET', '/session/$sessionId/contexts')
+        self.command_executor.add_command(Command.GET_CURRENT_CONTEXT, 'GET', '/session/$sessionId/context')
+        self.command_executor.add_command(Command.SWITCH_TO_CONTEXT, 'POST', '/session/$sessionId/context')