|
| 1 | +module Appium |
| 2 | + module Android |
| 3 | + module Espresso |
| 4 | + module Element |
| 5 | + # Find the first button that contains value or by index. |
| 6 | + # @param value [String, Integer] the value to exactly match. |
| 7 | + # If int then the button at that index is returned. |
| 8 | + # @return [Button] |
| 9 | + def button(value) |
| 10 | + # Don't use ele_index because that only works on one element type. |
| 11 | + # Android needs to combine button and image button to match iOS. |
| 12 | + if value.is_a? Numeric |
| 13 | + index = value |
| 14 | + raise "#{index} is not a valid index. Must be >= 1" if index <= 0 |
| 15 | + |
| 16 | + # zero index |
| 17 | + _button_visible_selectors_xpath(index: index - 1) |
| 18 | + end |
| 19 | + |
| 20 | + i = find_elements :xpath, _button_contains_string_xpath(Button, value) |
| 21 | + e = find_elements :xpath, _button_contains_string_xpath(ImageButton, value) |
| 22 | + |
| 23 | + raise_no_such_element_if_empty(i + e) |
| 24 | + |
| 25 | + (i + e)[0] |
| 26 | + end |
| 27 | + |
| 28 | + # Find all buttons containing value. |
| 29 | + # If value is omitted, all buttons are returned. |
| 30 | + # @param value [String] the value to search for |
| 31 | + # @return [Array<Button>] |
| 32 | + def buttons(value = false) |
| 33 | + return _button_visible_selectors_xpath unless value |
| 34 | + |
| 35 | + i = find_elements :xpath, _button_contains_string_xpath(Button, value) |
| 36 | + e = find_elements :xpath, _button_contains_string_xpath(ImageButton, value) |
| 37 | + i + e |
| 38 | + end |
| 39 | + |
| 40 | + # Find the first button. |
| 41 | + # @return [Button] |
| 42 | + def first_button |
| 43 | + _button_visible_selectors_xpath(button_index: 0, image_button_index: 0) |
| 44 | + end |
| 45 | + |
| 46 | + # Find the last button. |
| 47 | + # @return [Button] |
| 48 | + def last_button |
| 49 | + # uiautomator index doesn't support last |
| 50 | + # and it's 0 indexed |
| 51 | + button_index = tags(::Appium::Android::Button).length |
| 52 | + button_index -= 1 if button_index > 0 |
| 53 | + image_button_index = tags(::Appium::Android::ImageButton).length |
| 54 | + image_button_index -= 1 if image_button_index > 0 |
| 55 | + |
| 56 | + _button_visible_selectors_xpath(button_index: button_index, |
| 57 | + image_button_index: image_button_index) |
| 58 | + end |
| 59 | + |
| 60 | + # Find the first button that exactly matches value. |
| 61 | + # @param value [String] the value to match exactly |
| 62 | + # @return [Button] |
| 63 | + def button_exact(value) |
| 64 | + i = find_elements :xpath, _button_exact_string_xpath(Button, value) |
| 65 | + e = find_elements :xpath, _button_exact_string_xpath(ImageButton, value) |
| 66 | + |
| 67 | + raise_no_such_element_if_empty(i + e) |
| 68 | + |
| 69 | + (i + e)[0] |
| 70 | + end |
| 71 | + |
| 72 | + # Find all buttons that exactly match value. |
| 73 | + # @param value [String] the value to match exactly |
| 74 | + # @return [Array<Button>] |
| 75 | + def buttons_exact(value) |
| 76 | + i = find_elements :xpath, _button_exact_string_xpath(Button, value) |
| 77 | + e = find_elements :xpath, _button_exact_string_xpath(ImageButton, value) |
| 78 | + i + e |
| 79 | + end |
| 80 | + |
| 81 | + private |
| 82 | + |
| 83 | + # @private |
| 84 | + def raise_no_such_element_if_empty(elements) |
| 85 | + raise _no_such_element if elements.empty? |
| 86 | + |
| 87 | + elements.first |
| 88 | + end |
| 89 | + |
| 90 | + def _button_visible_selectors_xpath(opts = {}) |
| 91 | + button_index = opts.fetch :button_index, false |
| 92 | + image_button_index = opts.fetch :image_button_index, false |
| 93 | + |
| 94 | + index = opts.fetch :index, false |
| 95 | + |
| 96 | + b = find_elements :xpath, "//#{Button}" |
| 97 | + i = find_elements :xpath, "//#{ImageButton}" |
| 98 | + |
| 99 | + if index |
| 100 | + raise_no_such_element_if_empty(b + i) |
| 101 | + (b + i)[index] |
| 102 | + elsif button_index && image_button_index |
| 103 | + raise_no_such_element_if_empty(b + i) |
| 104 | + b_index = button_index + image_button_index |
| 105 | + (b + i)[b_index] |
| 106 | + else |
| 107 | + b + i |
| 108 | + end |
| 109 | + end |
| 110 | + |
| 111 | + def _button_exact_string_xpath(class_name, value) |
| 112 | + r_id = resource_id(value, " or @resource-id='#{value}'") |
| 113 | + "//#{class_name}[@text='#{value}' or @content-desc='#{value}'#{r_id}]" |
| 114 | + end |
| 115 | + |
| 116 | + def _button_contains_string_xpath(class_name, value) |
| 117 | + r_id = resource_id(value, " or @resource-id='#{value}'") |
| 118 | + "//#{class_name}[contains(translate(@text,'#{value.upcase}', '#{value}'), '#{value}')" \ |
| 119 | + " or contains(translate(@content-desc,'#{value.upcase}', '#{value}'), '#{value}')#{r_id}]" |
| 120 | + end |
| 121 | + end # module Element |
| 122 | + end # module Espresso |
| 123 | + end # module Android |
| 124 | +end # module Appium |
0 commit comments