appium_lib_core 1.7.2 → 1.8.0

data/lib/appium_lib_core/common/device/ime_actions.rb

@@ -0,0 +1,29 @@
+module Appium
+  module Core
+    class Base
+      module Device
+        module ImeActions
+          def ime_activate(ime_name)
+            execute :ime_activate_engine, {}, engine: ime_name
+          end
+
+          def ime_available_engines
+            execute :ime_get_available_engines
+          end
+
+          def ime_active_engine
+            execute :ime_get_active_engine
+          end
+
+          def ime_activated
+            execute :ime_is_activated
+          end
+
+          def ime_deactivate
+            execute :ime_deactivate, {}
+          end
+        end # module ImeActions
+      end # module Device
+    end # class Base
+  end # module Core
+end # module Appium

data/lib/appium_lib_core/common/device/keyboard.rb

@@ -0,0 +1,22 @@
+module Appium
+  module Core
+    class Base
+      module Device
+        module Keyboard
+          def hide_keyboard(close_key = nil, strategy = nil)
+            option = {}
+
+            option[:key] = close_key || 'Done'        # default to Done key.
+            option[:strategy] = strategy || :pressKey # default to pressKey
+
+            execute :hide_keyboard, {}, option
+          end
+
+          def is_keyboard_shown # rubocop:disable Naming/PredicateName
+            execute :is_keyboard_shown
+          end
+        end # module Keyboard
+      end # module Device
+    end # class Base
+  end # module Core
+end # module Appium

data/lib/appium_lib_core/common/device/keyevent.rb

@@ -0,0 +1,38 @@
+module Appium
+  module Core
+    class Base
+      module Device
+        module KeyEvent
+          # Only for Selendroid
+          def keyevent(key, metastate = nil)
+            args             = { keycode: key }
+            args[:metastate] = metastate if metastate
+            execute :keyevent, {}, args
+          end
+
+          def press_keycode(key, metastate: [], flags: [])
+            raise ArgumentError, 'flags should be Array' unless flags.is_a? Array
+            raise ArgumentError, 'metastates should be Array' unless metastate.is_a? Array
+
+            args             = { keycode: key }
+            args[:metastate] = metastate.reduce(0) { |acc, meta| acc | meta } unless metastate.empty?
+            args[:flags]     = flags.reduce(0) { |acc, flag| acc | flag } unless flags.empty?
+
+            execute :press_keycode, {}, args
+          end
+
+          def long_press_keycode(key, metastate: [], flags: [])
+            raise ArgumentError, 'flags should be Array' unless flags.is_a? Array
+            raise ArgumentError, 'metastates should be Array' unless metastate.is_a? Array
+
+            args             = { keycode: key }
+            args[:metastate] = metastate.reduce(0) { |acc, meta| acc | meta } unless metastate.empty?
+            args[:flags]     = flags.reduce(0) { |acc, flag| acc | flag } unless flags.empty?
+
+            execute :long_press_keycode, {}, args
+          end
+        end # module KeyEvent
+      end # module Device
+    end # class Base
+  end # module Core
+end # module Appium

data/lib/appium_lib_core/common/device/screen_record.rb

@@ -0,0 +1,54 @@
+module Appium
+  module Core
+    class Base
+      module Device
+        class ScreenRecord
+          #
+          # @api private
+          #
+
+          attr_reader :upload_option
+
+          METHOD = %w(POST PUT).freeze
+
+          def initialize(remote_path: nil, user: nil, pass: nil, method: 'PUT', force_restart: nil)
+            @upload_option = if remote_path.nil?
+                               {}
+                             else
+                               raise 'method should be POST or PUT' unless METHOD.member?(method.to_s.upcase)
+
+                               option = {}
+                               option[:remotePath] = remote_path
+                               option[:user] = user unless user.nil?
+                               option[:pass] = pass unless pass.nil?
+                               option[:method] = method
+                               option
+                             end
+
+            return if force_restart.nil?
+
+            raise 'force_restart should be true or false' unless [true, false].member?(force_restart)
+            @upload_option[:forceRestart] = force_restart
+          end
+
+          module Command
+            def stop_recording_screen(remote_path: nil, user: nil, pass: nil, method: 'PUT')
+              option = ::Appium::Core::Base::Device::ScreenRecord.new(
+                remote_path: remote_path, user: user, pass: pass, method: method
+              ).upload_option
+
+              params = option.empty? ? {} : { options: option }
+
+              execute(:stop_recording_screen, {}, params)
+            end
+
+            def stop_and_save_recording_screen(file_path)
+              base64data = execute(:stop_recording_screen, {}, {})
+              File.open(file_path, 'wb') { |f| f << Base64.decode64(base64data) }
+            end
+          end # module Command
+        end # class ScreenRecord
+      end # module Device
+    end # class Base
+  end # module Core
+end # module Appium

data/lib/appium_lib_core/common/device/setting.rb

@@ -0,0 +1,17 @@
+module Appium
+  module Core
+    class Base
+      module Device
+        module Setting
+          def get_settings
+            execute :get_settings, {}
+          end
+
+          def update_settings(settings)
+            execute :update_settings, {}, settings: settings
+          end
+        end # module Setting
+      end # module Device
+    end # class Base
+  end # module Core
+end # module Appium

data/lib/appium_lib_core/common/device/touch_actions.rb

@@ -0,0 +1,21 @@
+require_relative '../touch_action/touch_actions'
+require_relative '../touch_action/multi_touch'
+
+module Appium
+  module Core
+    class Base
+      module Device
+        module TouchActions
+          def touch_actions(actions)
+            actions = { actions: [actions].flatten }
+            execute :touch_actions, {}, actions
+          end
+
+          def multi_touch(actions)
+            execute :multi_touch, {}, actions: actions
+          end
+        end # module TouchActions
+      end # module Device
+    end # class Base
+  end # module Core
+end # module Appium

data/lib/appium_lib_core/common/device/value.rb

@@ -0,0 +1,19 @@
+module Appium
+  module Core
+    class Base
+      module Device
+        module Value
+          def set_immediate_value(element, *value)
+            keys = ::Selenium::WebDriver::Keys.encode(value)
+            execute :set_immediate_value, { id: element.ref }, value: Array(keys)
+          end
+
+          def replace_value(element, *value)
+            keys = ::Selenium::WebDriver::Keys.encode(value)
+            execute :replace_value, { id: element.ref }, value: Array(keys)
+          end
+        end # module Value
+      end # module Device
+    end # class Base
+  end # module Core
+end # module Appium

data/lib/appium_lib_core/device.rb

@@ -1,23 +1,5 @@
-require_relative 'common/touch_action/touch_actions'
-require_relative 'common/touch_action/multi_touch'
-
 require_relative 'element/image'
 
-require_relative 'device/screen_record'
-require_relative 'device/app_state'
-require_relative 'device/clipboard_content_type'
-require_relative 'device/image_comparison'
-require_relative 'device/app_management'
-require_relative 'device/keyboard'
-require_relative 'device/file_management'
-require_relative 'device/touch_actions'
-require_relative 'device/device_lock'
-require_relative 'device/ime_actions'
-require_relative 'device/context'
-require_relative 'device/keyevent'
-require_relative 'device/setting'
-require_relative 'device/value'
-
 require 'base64'
 
 module Appium
@@ -25,541 +7,38 @@ module Appium
     module Device
       extend Forwardable
 
-      # rubocop:disable Metrics/LineLength
-
-      ####
-      ## No argument
-      ####
-
-      # @!method launch_app
-      # Start the simulator and application configured with desired capabilities
-      #
-      # @example
-      #
-      #   @driver.launch_app
-      #
-
-      # @!method close_app
-      # Close an app on device
-      #
-      # @example
-      #
-      #   @driver.close_app
-      #
-
-      # @!method reset
-      # Reset the device, relaunching the application.
-      #
-      # @example
-      #
-      #   @driver.reset
-      #
-
-      # @!method shake
-      # Cause the device to shake
-      #
-      # @example
-      #
-      #   @driver.shake
-      #
-
-      # @!method unlock
-      # Unlock the device
-      #
-      # @example
-      #
-      #   @driver.unlock
-      #
-
-      # @!method device_locked?
-      # Check current device status is weather locked or not
-      #
-      # @example
-      #
-      #   @driver.device_locked?
-      #
-
-      # @!method device_time
-      #   Get the time on the device
-      #
-      # @param [String] format The set of format specifiers. Read https://momentjs.com/docs/ to get the full list of supported
-      #                        datetime format specifiers. The default format is `YYYY-MM-DDTHH:mm:ssZ`,
-      #                        which complies to ISO-8601
-      # @return [String] Formatted datetime string or the raw command output if formatting fails
-      #
-      # @example
-      #
-      #   @driver.device_time #=> "2018-06-12T11:13:31+02:00"
-      #   @driver.device_time "YYYY-MM-DD" #=> "2018-06-12"
-      #
-
-      ####
-      ## With arguments
-      ####
-
-      # @!method install_app(path, replace: nil, timeout: nil, allow_test_packages: nil, use_sdcard: nil, grant_permissions: nil)
-      # Install the given app onto the device
-      #
-      # @param [String] path The absolute local path or remote http URL to an .ipa or .apk file, or a .zip containing one of these.
-      # @param [Boolean] replace: Only for Android. Whether to reinstall/upgrade the package if it is already present on the device under test. `true` by default
-      # @param [Integer] timeout: Only for Android. How much time to wait for the installation to complete. 60000ms by default.
-      # @param [Boolean] allow_test_packages: Only for Android. Whether to allow installation of packages marked as test in the manifest. `false` by default
-      # @param [Boolean] use_sdcard: Only for Android. Whether to use the SD card to install the app. `false` by default
-      # @param [Boolean] grant_permissions:  Only for Android. whether to automatically grant application permissions on Android 6+ after the installation completes. `false` by default
-      #
-      # @example
-      #
-      #   @driver.install_app("/path/to/test.apk")
-      #   @driver.install_app("/path/to/test.apk", replace: true, timeout: 20000, allow_test_packages: true, use_sdcard: false, grant_permissions: false)
-      #
-
-      # @!method remove_app(app_id, keep_data: nil, timeout: nil)
-      #
-      # @param [Strong] app_id BundleId for iOS or package name for Android
-      # @param [Boolean] keep_data: Only for Android. Whether to keep application data and caches after it is uninstalled. `false` by default
-      # @param [Integer] timeout: Only for Android. How much time to wait for the uninstall to complete. 20000ms by default.
-      #
-      # @example
-      #
-      #   @driver.remove_app("io.appium.bundle")
-      #   @driver.remove_app("io.appium.bundle", keep_data: false, timeout, 10000)
-      #
-
-      # @!method app_installed?(app_id)
-      # Check whether the specified app is installed on the device
-      # @return [Boolean]
-      #
-      # @example
-      #
-      #   @driver.app_installed?("io.appium.bundle")
-      #
-
-      # @!method terminate_app(app_id)
-      # Terminate the specified app.
-      #
-      # @param [Strong] app_id BundleId for iOS or package name for Android
-      # @param [Integer] timeout: Only for Android. How much time to wait for the application termination to complete. 500ms by default.
-      # @return [Boolean]
-      #
-      # @example
-      #
-      #   @driver.terminate_app("io.appium.bundle") # true
-      #   @driver.terminate_app("io.appium.bundle", timeout: 500)
-      #
-
-      # @!method activate_app(app_id)
-      # Activate(Launch) the specified app.
-      # @return [Hash]
-      #
-      # @example
-      #
-      #   @driver.activate_app("io.appium.bundle") #=> {}
-      #
-
-      # Get the status of an existing application on the device.
-      # State:
-      #   :not_installed : The current application state cannot be determined/is unknown
-      #   :not_running : The application is not running
-      #   :running_in_background_suspended : The application is running in the background and is suspended
-      #   :running_in_background : The application is running in the background and is not suspended
-      #   :running_in_foreground : The application is running in the foreground
-      #
-      # For more details: https://developer.apple.com/documentation/xctest/xcuiapplicationstate
-      #
-      # @param [String] bundle_id A target app's bundle id
-      # @return [AppState::STATUS] A number of the state
-      #
-      # @example
-      #
-      #      @driver.app_state("io.appium.bundle") #=> :not_running
-      #
-
-      # @!method app_strings(language = nil)
-      # Return the hash of all localization strings.
-      # @return [Hash]
-      #
-      # @example
-      #
-      #   @driver.app_strings #=> "TransitionsTitle"=>"Transitions", "WebTitle"=>"Web"
-      #
-
-      # @!method lock(duration = nil)
-      # Lock the device
-      # @return [String]
-      #
-      # @example
-      #
-      #   @driver.lock    #=> Lock the device
-      #   @driver.lock(5) #=> Lock the device in 5 sec and unlock the device after 5 sec.
-      #                   #   Block other commands during locking the device.
-      #
-
-      # @!method background_app(duration = 0)
-      # Backgrounds the app for a set number of seconds.
-      # This is a blocking application
-      # @param [Integer] duration How many seconds to background the app for.
-      # @return [String]
-      #
-      # @example
-      #
-      #   @driver.background_app
-      #   @driver.background_app(5)
-      #   @driver.background_app(-1) #=> the app never come back. https://github.com/appium/appium/issues/7741
-      #
-
-      # @!method hide_keyboard(close_key = nil, strategy = nil)
-      # Hide the onscreen keyboard
-      # @param [String] close_key The name of the key which closes the keyboard.
-      #   Defaults to 'Done' for iOS(except for XCUITest).
-      # @param [Symbol] strategy The symbol of the strategy which closes the keyboard.
-      #   XCUITest ignore this argument.
-      #   Default for iOS is `:pressKey`. Default for Android is `:tapOutside`.
-      #
-      # @example
-      #
-      #   @driver.hide_keyboard # Close a keyboard with the 'Done' button
-      #   @driver.hide_keyboard('Finished') # Close a keyboard with the 'Finished' button
-      #   @driver.hide_keyboard(nil, :tapOutside) # Close a keyboard with tapping out side of keyboard
-      #
-
-      # @!method is_keyboard_shown
-      # Get whether keyboard is displayed or not.
-      # @return [Boolean] Return true if keyboard is shown. Return false if keyboard is hidden.
-      #
-      # @example
-      #   @driver.is_keyboard_shown # false
-      #
-
-      # @!method keyevent(key, metastate = nil)
-      # Send keyevent on the device.(Only for Selendroid)
-      # http://developer.android.com/reference/android/view/KeyEvent.html
-      # @param [integer] key The key to press.
-      # @param [String] metastate The state the metakeys should be in when pressing the key.
-      #
-      # @example
-      #
-      #   @driver.keyevent 82
-      #
-
-      # @!method press_keycode(key, metastate: [], flags: [])
-      # Press keycode on the device.
-      # http://developer.android.com/reference/android/view/KeyEvent.html
-      # @param [Integer] key The key to press. The values which have `KEYCODE_` prefix in http://developer.android.com/reference/android/view/KeyEvent.html
-      #                      e.g.: KEYCODE_HOME is `3` or `0x00000003`
-      # @param [[Integer]] metastate: The state the metakeys should be in when pressing the key. Default is empty Array.
-      #                            Metastate have `META_` prefix in https://developer.android.com/reference/android/view/KeyEvent.html
-      #                            e.g.: META_SHIFT_ON is `1` or `0x00000001`
-      # @param [[Integer]] flags: Native Android flag value. Several flags can be combined into a single key event.
-      #                           Default is empty Array.  Can set multiple flags as Array.
-      #                           Flags have `FLAG_` prefix in http://developer.android.com/reference/android/view/KeyEvent.html
-      #                           e.g.: FLAG_CANCELED is `32` or `0x00000020`
-      #
-      # @example
-      #
-      #   @driver.press_keycode 66
-      #   @driver.press_keycode 66, flags: [0x02]
-      #   @driver.press_keycode 66, metastate: [1], flags: [32]
-      #
-
-      # @!method long_press_keycode(key, metastate: [], flags: [])
-      # Long press keycode on the device.
-      # http://developer.android.com/reference/android/view/KeyEvent.html
-      # @param [Integer] key The key to long press. The values which have `KEYCODE_` prefix in http://developer.android.com/reference/android/view/KeyEvent.html
-      #                      e.g.: KEYCODE_HOME is `3` or `0x00000003`
-      # @param [[Integer]] metastate: The state the metakeys should be in when pressing the key. Default is empty Array.
-      #                            Metastate have `META_` prefix in https://developer.android.com/reference/android/view/KeyEvent.html
-      #                            e.g.: META_SHIFT_ON is `1` or `0x00000001`
-      # @param [[Integer]] flags: Native Android flag value. Several flags can be combined into a single key event.
-      #                           Default is empty Array. Can set multiple flags as Array.
-      #                           Flags have `FLAG_` prefix in http://developer.android.com/reference/android/view/KeyEvent.html
-      #                           e.g.: FLAG_CANCELED is `32` or `0x00000020`
-      #
-      # @example
-      #
-      #   @driver.long_press_keycode 66
-      #   @driver.long_press_keycode 66, flags: [0x20, 0x2000]
-      #   @driver.long_press_keycode 66, metastate: [1], flags: [32, 8192]
-      #
-
-      # @!method push_file(path, filedata)
-      # Place a file in a specific location on the device.
-      # On iOS, the server should have ifuse libraries installed and configured properly for this feature to work on real devices.
-      # On Android, the application under test should be built with debuggable flag enabled in order to get access to
-      # its container on the internal file system.
-      #
-      # @see https://github.com/libimobiledevice/ifuse iFuse GitHub page6
-      # @see https://github.com/osxfuse/osxfuse/wiki/FAQ osxFuse FAQ
-      # @see https://developer.android.com/studio/debug/ 'Debug Your App' developer article
-      #
-      # @param [String] path The absolute path on the device to store data at.
-      #                      If the path starts with application id prefix, then the file will be pushed to
-      #                      the root of the corresponding application container.
-
-      # @param [String] path Either an absolute path OR, for iOS devices, a path relative to the app, as described.
-      #                      If the path starts with application id prefix, then the file will be pushed to the root of
-      #                      the corresponding application container.
-      # @param [String] filedata Raw file data to be sent to the device. Converted to base64 in the method.
-      #
-      # @example
-      #
-      #   @driver.push_file "/file/to/path", "data"
-      #
-
-      # @!method pull_file(path)
-      # Pull a file from the simulator/device.
-      # On iOS the server should have ifuse
-      # libraries installed and configured properly for this feature to work on real devices.
-      # On Android the application under test should be built with debuggable flag enabled in order to get access
-      # to its container on the internal file system.
-      #
-      # @see https://github.com/libimobiledevice/ifuse iFuse GitHub page6
-      # @see https://github.com/osxfuse/osxfuse/wiki/FAQ osxFuse FAQ
-      # @see https://developer.android.com/studio/debug/ 'Debug Your App' developer article
-      #
-      # @param [String] path Either an absolute path OR, for iOS devices, a path relative to the app, as described.
-      #                      If the path starts with application id prefix, then the file will be pulled from the root
-      #                      of the corresponding application container.
-      #                      Otherwise the root folder is considered as / on Android and on iOS it is a media folder root
-      #                      (real devices only).
-      # @return [Base64-decoded] Base64 decoded data
-      #
-      # @example
-      #
-      #   @driver.pull_file '/local/data/some/path'     #=> Get the file at that path
-      #   @driver.pull_file 'Shenanigans.app/some/file' #=> Get 'some/file' from the install location of Shenanigans.app
-      #
-
-      # @!method pull_folder(path)
-      # Pull a folder content from the simulator/device.
-      # On iOS the server should have ifuse libraries installed and configured properly for this feature to work on real devices.
-      # On Android the application under test should be built with debuggable flag enabled in order to get access to its container
-      # on the internal file system.
-      #
-      # @see https://github.com/libimobiledevice/ifuse iFuse GitHub page6
-      # @see https://github.com/osxfuse/osxfuse/wiki/FAQ osxFuse FAQ
-      # @see https://developer.android.com/studio/debug/ 'Debug Your App' developer article
-      #
-      # @param [String] path Absolute path to the folder.
-      #                      If the path starts with <em>@applicationId/</em> prefix, then the folder will be pulled
-      #                      from the root of the corresponding application container.
-      #                      Otherwise the root folder is considered as / on Android and on iOS it is a media folder root
-      #                      (real devices only).
-      #
-      # @return [Base64-decoded] Base64 decoded data which is zip archived
-      #
-      # @example
-      #
-      #   @driver.pull_folder '/data/local/tmp' #=> Get the folder at that path
-      #
-
-      # @!method get_settings
-      # Get appium Settings for current test session
-      #
-      # @example
-      #
-      #   @driver.pull_folder '/data/local/tmp' #=> Get the folder at that path
-      #
-
-      # @since 1.3.4
-      # @!method save_viewport_screenshot
-      # Save screenshot except for status bar while `@driver.save_screenshot` save entire screen.
-      #
-      # @example
-      #
-      #   @driver.save_viewport_screenshot 'path/to/save.png' #=> Get the File instance of viewport_screenshot
-      #
-
-      # @!method update_settings(settings)
-      # Update Appium Settings for current test session
-      # @param [Hash] settings Settings to update, keys are settings, values to value to set each setting to
-      #
-      # @example
-      #
-      #   @driver.update_settings('allowInvisibleElements': true)
-      #
-
-      # @!method set_immediate_value(element, *value)
-      #   Set the value to element directly
-      #
-      # @example
-      #
-      #   set_immediate_value element, 'hello'
-      #
-
-      # @!method replace_value(element, *value)
-      #   Replace the value to element directly
-      #
-      # @example
-      #
-      #   replace_value element, 'hello'
-      #
-
-      # @!method ime_activate(ime_name)
-      # Android only. Make an engine that is available active.
-      # @param [String] ime_name The IME owning the activity [required]
-      #
-      # @example
-      #
-      #   ime_activate engine: 'com.android.inputmethod.latin/.LatinIME'
-      #
-
-      # @!method ime_available_engines
-      # Android only. List all available input engines on the machine.
-      #
-      # @example
-      #
-      #   ime_available_engines #=> Get the list of IME installed in the target device
-      #
-
-      # @!method ime_active_engine
-      # Android only. Get the name of the active IME engine.
-      #
-      # @example
-      #
-      #   ime_active_engine #=> Get the current active IME such as 'com.android.inputmethod.latin/.LatinIME'
-      #
-
-      # @!method ime_activated
-      #   Android only. Indicates whether IME input is active at the moment (not if it is available).
-      #
-      # @example
-      #
-      #   ime_activated #=> True if IME is activated
-      #
-
-      # @!method ime_deactivate
-      # Android only. De-activates the currently-active IME engine.
-      #
-      # @example
-      #
-      #   ime_deactivate #=> Deactivate current IME engine
-      #
-
-      # @!method set_context(context)
-      # Change the context to the given context.
-      # @param [String] context The context to change to
-      #
-      # @example
-      #
-      #   @driver.set_context "NATIVE_APP"
-      #
-
-      # @!method current_context
-      # @return [String] The context currently being used.
-      #
-      # @example
-      #
-      #   @driver.current_context
-      #
-
-      # @!method available_contexts
-      # @return [Array<String>] All usable contexts, as an array of strings.
-      #
-      # @example
-      #
-      #   @driver.available_contexts
-      #
-
-      # Perform a block within the given context, then switch back to the starting context.
-      # @param [String] context The context to switch to for the duration of the block.
-      #
-      # @example
-      #
-      #   result = @driver.within_context('NATIVE_APP') do
-      #     @driver.find_element :tag, "button"
-      #   end # The result of `find_element :tag, "button"`
-      #
-
-      # Change to the default context. This is equivalent to `set_context nil`.
-      #
-      # @example
-      #
-      #   @driver.switch_to_default_context
-      #
-
-      # @!method stop_recording_screen(remote_path: nil, user: nil, pass: nil, method: 'PUT')
-      # @param [String] remote_path: The path to the remote location, where the resulting video should be uploaded.
-      #                             The following protocols are supported: http/https, ftp.
-      #                             Null or empty string value (the default setting) means the content of resulting
-      #                             file should be encoded as Base64 and passed as the endpoint response value.
-      #                             An exception will be thrown if the generated media file is too big to
-      #                             fit into the available process memory.
-      # @param [String] user: The name of the user for the remote authentication.
-      # @param [String] pass: The password for the remote authentication.
-      # @param [String] method: The http multipart upload method name. The 'PUT' one is used by default.
-      #
-      # @example
-      #
-      #    @driver.stop_recording_screen
-      #    @driver.stop_recording_screen remote_path: 'https://example.com', user: 'example', pass: 'pass', method: 'POST'
-      #
-
-      # @!method stop_and_save_recording_screen(file_path)
-      # @param [String] file_path The path to save video decoded from base64 from Appium server.
-      #
-      # @example
-      #
-      #    @driver.stop_and_save_recording_screen 'example.mp4'
-      #
-
-      # rubocop:enable Metrics/LineLength
-
-      ####
-      ## class << self
-      ####
-
       class << self
         def extended(_mod)
           extend_webdriver_with_forwardable
 
-          add_endpoint_method(:shake) do
-            def shake
-              execute :shake
-            end
-          end
-
-          add_endpoint_method(:device_time) do
-            def device_time(format = nil)
-              arg = {}
-              arg[:format] = format unless format.nil?
-              execute :device_time, {}, arg
-            end
-          end
-
-          add_endpoint_method(:save_viewport_screenshot) do
-            def save_viewport_screenshot(png_path)
-              extension = File.extname(png_path).downcase
-              if extension != '.png'
-                ::Appium::Logger.warn 'name used for saved screenshot does not match file type. '\
-                                  'It should end with .png extension'
-              end
-              viewport_screenshot_encode64 = execute_script('mobile: viewportScreenshot')
-              File.open(png_path, 'wb') { |f| f << viewport_screenshot_encode64.unpack('m')[0] }
-            end
-          end
-
-          Value.add_methods
-          Setting.add_methods
-          KeyEvent.add_methods
-          Context.add_methods
-          ImeActions.add_methods
-          DeviceLock.add_methods
-          TouchActions.add_methods
-          FileManagement.add_methods
-          Keyboard.add_methods
-          AppManagement.add_methods
-          ScreenRecord.add_methods
-          ImageComparison.add_methods
-          AppState.add_methods
-
           # Compatibility for appium_lib
           # TODO: Will remove
-          delegate_from_appium_driver :take_element_screenshot
+          [
+            :take_element_screenshot, :save_viewport_screenshot,
+            :lock, :device_locked?, :unlock,
+            :hide_keyboard, :is_keyboard_shown,
+            :ime_activate, :ime_available_engines, :ime_active_engine, :ime_activated, :ime_deactivate,
+            :get_settings, :update_settings,
+            :within_context, :switch_to_default_context, :current_context, :available_contexts, :set_context,
+            :set_immediate_value, :replace_value,
+            :push_file, :pull_file, :pull_folder,
+            :keyevent, :press_keycode, :long_press_keycode,
+            :match_images_features, :find_image_occurrence, :get_images_similarity, :compare_images,
+            :launch_app, :close_app, :reset, :app_strings, :background_app,
+            :install_app, :remove_app, :app_installed?, :activate_app, :terminate_app,
+            :app_state,
+            :stop_recording_screen, :stop_and_save_recording_screen,
+            :shake, :device_time,
+            :touch_actions, :multi_touch
+          ].each do |key|
+            delegate_from_appium_driver key
+          end
         end
 
         # def extended
 
         # @private
+        # Define method in Bridges
         def add_endpoint_method(method)
           block_given? ? create_bridge_command(method, &Proc.new) : create_bridge_command(method)
 
@@ -575,19 +54,17 @@ module Appium
           end
         end
 
-        # For ruby_lib compatibility
-        # @private
+        private
+
         def delegate_from_appium_driver(method, delegation_target = :driver)
           def_delegator delegation_target, method
         end
 
-        # @private
         def delegate_driver_method(method)
           return if ::Appium::Core::Base::Driver.method_defined? method
           ::Appium::Core::Base::Driver.class_eval { def_delegator :@bridge, method }
         end
 
-        # @private
         def create_bridge_command(method)
           ::Appium::Core::Base::Bridge::MJSONWP.class_eval do
             block_given? ? class_eval(&Proc.new) : define_method(method) { execute method }
@@ -597,6 +74,6 @@ module Appium
           end
         end
       end # class << self
-    end # module Device
+    end
   end # module Core
 end # module Appium