JezK
Edit File: search_context.rb
# frozen_string_literal: true # Licensed to the Software Freedom Conservancy (SFC) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information # regarding copyright ownership. The SFC licenses this file # to you under the Apache License, Version 2.0 (the # "License"); you may not use this file except in compliance # with the License. You may obtain a copy of the License at # # http://www.apache.org/licenses/LICENSE-2.0 # # Unless required by applicable law or agreed to in writing, # software distributed under the License is distributed on an # "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY # KIND, either express or implied. See the License for the # specific language governing permissions and limitations # under the License. module Selenium module WebDriver module SearchContext # @api private FINDERS = { class: 'class name', class_name: 'class name', css: 'css selector', id: 'id', link: 'link text', link_text: 'link text', name: 'name', partial_link_text: 'partial link text', relative: 'relative', tag_name: 'tag name', xpath: 'xpath' }.freeze # # Find the first element matching the given arguments # # When using Element#find_element with :xpath, be aware that webdriver # follows standard conventions: a search prefixed with "//" will search # the entire document, not just the children of this current node. Use # ".//" to limit your search to the children of the receiving Element. # # @overload find_element(how, what) # @param [Symbol, String] how The method to find the element by # @param [String] what The locator to use # @overload find_element(opts) # @param [Hash] opts Find options # @option opts [Symbol] :how Key named after the method to find the element by, containing the locator # @return [Element] # # @raise [Error::NoSuchElementError] if the element doesn't exist # def find_element(*args) how, what = extract_args(args) by = FINDERS[how.to_sym] raise ArgumentError, "cannot find element by #{how.inspect}" unless by bridge.find_element_by by, what, ref end # # Find all elements matching the given arguments # # @see SearchContext#find_element # def find_elements(*args) how, what = extract_args(args) by = FINDERS[how.to_sym] raise ArgumentError, "cannot find elements by #{how.inspect}" unless by bridge.find_elements_by by, what, ref end private def extract_args(args) case args.size when 2 args when 1 arg = args.first unless arg.respond_to?(:shift) raise ArgumentError, "expected #{arg.inspect}:#{arg.class} to respond to #shift" end # this will be a single-entry hash, so use #shift over #first or #[] arr = arg.dup.shift raise ArgumentError, "expected #{arr.inspect} to have 2 elements" unless arr.size == 2 arr else raise ArgumentError, "wrong number of arguments (#{args.size} for 2)" end end end # SearchContext end # WebDriver end # Selenium