rudra 1.0.14 → 1.1.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f065383cc3c1a776dccc5708a77f627383aaa63f4198b087adad977271f713d6
-  data.tar.gz: 9fc4e3d993f3d3a1b4fb53d57b146df87fe8fa5e561308b726a3c8c46dd5ac1d
+  metadata.gz: ff3f69221a8e01fb0c0bf5a517c11547872f475b4cdc7aa551a26f81929c8373
+  data.tar.gz: e039db5f85576af890d9b58f8deed4f7f30cf09ea474d8b2226c6574cad9c115
 SHA512:
-  metadata.gz: 9f80dd1cc85e183b9ac7ac2be6831a2e1ac56be27ab884133fcbfc4b0f87e03a23d4d9443bd933cfef69292c881d968b2033d3af8cd5a72c909ba8474afcce03
-  data.tar.gz: ad1f215b1ea9c9f6d16e387f95f1a5f9c087845c2db3239d0f64ea719e94241805578fe1a3589b69dafc0207fcc10b6589677259ecb87939752ab0fd0d5c9a52
+  metadata.gz: a632f5ee68fa9417a78e86642246c47d2b4480292ca4816471b8f948458d7c6444150ccb7eb1a65fe8d7167f25dca32e1271f97cd725dbc86643362d763d4361
+  data.tar.gz: 9edc12b61b22302ef0d80698a682d859a124acaa7c3d121d87353f83c84fef948f9fa93971ce85074c4b7faf52b27bf9ebef4bdedcb14429a8ef7222b73854f3

data/lib/rudra.rb

@@ -1,10 +1,5 @@
 require 'selenium-webdriver'
-require 'webdrivers/chromedriver'
-require 'webdrivers/geckodriver'
-require 'webdrivers/iedriver'
-
-# Selenium::WebDriver::Chrome::Service.driver_path = './webdrivers/chromedriver'
-# Selenium::WebDriver::Firefox::Service.driver_path = './webdrivers/geckodriver'
+require 'webdrivers'
 
 # Selenium IDE-like WebDriver based upon Ruby binding
 # @author Aaron Chen
@@ -13,8 +8,13 @@ require 'webdrivers/iedriver'
 #   of the chosen browser
 # @attr_reader [String] install_dir The install directory of WebDrivers
 # @attr_reader [String] locale The browser locale
+# @attr_reader [Boolean] headless Headless mode for Google Chrome
+# @attr_reader [String] window_size Chrome window size when headless
+# @attr_reader [String] screen_dir The screenshot directory of save_screenshot
+# @attr_reader [String] log_prefix Prefix for logging descriptions and methods
 # @attr_reader [Integer] timeout The driver timeout
-# @attr_reader [Boolean] verbose Verbose mode
+# @attr_reader [Boolean] verbose Turn on/off Verbose mode
+# @attr_reader [Boolean] silent Turn off Turn on/off descriptions
 class Rudra
   # Supported Browsers
   BROWSERS = %i[chrome firefox ie safari].freeze
@@ -28,11 +28,13 @@ class Rudra
   # Attributes
   ATTRIBUTES = %i[
     browser driver install_dir locale
-    headless log_prefix timeout verbose
+    headless window_size screen_dir
+    log_prefix timeout verbose silent
   ].freeze
 
   attr_reader :browser, :driver, :install_dir, :locale,
-              :headless, :log_prefix, :timeout, :verbose
+              :headless, :window_size, :screen_dir,
+              :log_prefix, :timeout, :verbose, :silent
 
   # Initialize an instance of Rudra
   # @param [Hash] options the options to initialize Rudra
@@ -42,16 +44,22 @@ class Rudra
   #   directory of WebDrivers
   # @option options [Symbol] :locale (:en) the browser locale
   # @option options [Boolean] :headless (false) headless mode
-  # @option options [String] :log_prefix (' - ') log prefix
+  # @option options [String] :window_size ('1280,720') window size when headless
+  # @option options [String] :screen_dir ('./screens/') the location of screenshots
+  # @option options [String] :log_prefix (' - ') prefix for logging descriptions and methods
   # @option options [Integer] :timeout (30) implicit_wait timeout
-  # @option options [Boolean] :verbose (true) verbose mode
+  # @option options [Boolean] :verbose (false) Turn on/off verbose mode
+  # @option options [Boolean] :silent (false) Turn on/off descriptions
   def initialize(options = {})
     self.browser = options.fetch(:browser, :chrome)
     self.install_dir = options.fetch(:install_dir, './webdrivers/')
     self.locale = options.fetch(:locale, :en)
     self.headless = options.fetch(:headless, false)
+    self.window_size = options.fetch(:window_size, '1280,720')
+    self.screen_dir = options.fetch(:screen_dir, './screens/')
     self.log_prefix = options.fetch(:log_prefix, ' - ')
-    self.verbose = options.fetch(:verbose, true)
+    self.verbose = options.fetch(:verbose, false)
+    self.silent = options.fetch(:silent, false)
     self.main_label = caller_locations(2, 1).first.label
 
     initialize_driver
@@ -97,6 +105,7 @@ class Rudra
   end
 
   # Send keys to an alert
+  # @param [String] keys keystrokes to send
   def alert_send_keys(keys)
     switch_to_alert.send_keys(keys)
   end
@@ -168,7 +177,7 @@ class Rudra
 
     element ||= driver.find_element(how, what)
 
-    abort("Failed to find element: #{locator}") unless element
+    raise Selenium::WebDriver::Error::NoSuchElementError, "Failed to find element: #{locator}" unless element
 
     wait_for { element.displayed? }
 
@@ -180,8 +189,11 @@ class Rudra
   # @return [Array<Selenium::WebDriver::Element>] the elements found
   def find_elements(locator)
     how, what = parse_locator(locator)
-    driver.find_elements(how, what) ||
-      abort("Failed to find elements: #{locator}")
+    elements = driver.find_elements(how, what)
+
+    raise Selenium::WebDriver::Error::NoSuchElementError, "Failed to find elements: #{locator}" if elements.empty?
+
+    elements
   end
 
   # Move forward a single entry in the browser's history
@@ -201,7 +213,7 @@ class Rudra
 
   # Maximize the current window
   def maximize
-    driver.manage.window.maximize
+    driver.manage.window.maximize unless headless
   end
 
   # Maximize the current window to the size of the screen
@@ -259,6 +271,12 @@ class Rudra
     driver.page_source
   end
 
+  # Print description in the console
+  # @param [String] description description to show
+  def puts(description)
+    $stdout.puts "#{log_prefix}#{description.chomp}" unless silent
+  end
+
   # Refresh the current pagef
   def refresh
     driver.navigate.refresh
@@ -271,11 +289,15 @@ class Rudra
     driver.manage.window.resize_to(width, height)
   end
 
-  # Save a PNG screenshot to the given path
-  # @param [String] png_path the path of PNG screenshot
-  def save_screenshot(png_path)
+  # Save a PNG screenshot to file
+  # @param [String] filename the filename of PNG screenshot
+  def save_screenshot(filename)
+    mkdir(@screen_dir) unless Dir.exist?(@screen_dir)
     driver.save_screenshot(
-      png_path.end_with?('.png') ? png_path : "#{png_path}.png"
+      File.join(
+        @screen_dir,
+        filename.end_with?('.png') ? filename : "#{filename}.png"
+      )
     )
   end
 
@@ -291,6 +313,7 @@ class Rudra
   end
 
   # Switch to the frame with the given id
+  # @param [String] id the frame id
   def switch_to_frame(id)
     driver.switch_to.frame(id)
   end
@@ -326,6 +349,48 @@ class Rudra
     wait_for { find_element(locator).enabled? }
   end
 
+  # Wait until the element, identified by locator, is found in frame
+  # @param [String] frame_id the frame id
+  # @param [String] locator the locator to identify the element
+  def wait_for_element_found_in_frame(frame_id, locator)
+    switch_to_frame frame_id
+
+    how, what = parse_locator(locator)
+
+    wait_for do
+      begin
+        driver.find_element(how, what)
+      rescue Selenium::WebDriver::Error::NoSuchWindowError
+        false
+      end
+    end
+  end
+
+  # Wait (in seconds) until the element is not displayed
+  # @param [String, Selenium::WebDriver::Element] locator the locator to
+  #   identify the element or Selenium::WebDriver::Element
+  # @param [Integer] seconds seconds before timed out
+  def wait_for_not_visible(locator, seconds = 2)
+    how, what = parse_locator(locator)
+
+    begin
+      wait_for(seconds) do
+        begin
+          elements = driver.find_elements(how, what)
+          elements.empty? || elements.map(&:displayed?).none?
+        rescue Selenium::WebDriver::Error::NoSuchElementError
+          true
+        rescue Selenium::WebDriver::Error::StaleElementReferenceError
+          false
+        end
+      end
+    rescue Selenium::WebDriver::Error::TimeoutError
+      true
+    rescue Net::ReadTimeout
+      true
+    end
+  end
+
   # Wait until the title of the page including the given string
   # @param [String] string the string to compare
   def wait_for_title(string)
@@ -374,7 +439,7 @@ class Rudra
   # @param [String] attribute the name of the attribute
   # @return [String, nil] attribute value
   def attribute(locator, attribute)
-    find_element(locator).property(attribute)
+    find_element(locator).attribute(attribute)
   end
 
   # If the element, identified by locator, has the given attribute
@@ -409,7 +474,13 @@ class Rudra
   # @param [String, Selenium::WebDriver::Element] locator the locator to
   #   identify the element or Selenium::WebDriver::Element
   def click(locator)
-    find_element(locator).click
+    wait_for do
+      begin
+        find_element(locator).click.nil?
+      rescue Selenium::WebDriver::Error::ElementClickInterceptedError
+        false
+      end
+    end
   end
 
   # Click the given element, identified by locator, with an offset
@@ -722,6 +793,38 @@ class Rudra
     ), find_element(locator), event)
   end
 
+  # Wait until the element, identified by locator, attribute has value
+  # @param [String, Selenium::WebDriver::Element] locator the locator to identify the element
+  # @param [String] attribute the name of the attribute
+  # @param [String] value the value of the attribute
+  def wait_for_attribute_to_include(locator, attribute, value)
+    how, what = parse_locator(locator)
+
+    wait_for do
+      begin
+        driver.find_element(how, what)&.attribute(attribute)&.downcase&.include?(value.downcase)
+      rescue Selenium::WebDriver::Error::StaleElementReferenceError
+        false
+      end
+    end
+  end
+
+  # Wait until the element, identified by locator, excluding string in text
+  # @param [String, Selenium::WebDriver::Element] locator the locator to
+  #   identify the element or Selenium::WebDriver::Element
+  # @param [String] string the string to exclude
+  def wait_for_text_to_exclude(locator, string)
+    wait_for { text(locator).exclude?(string) }
+  end
+
+  # Wait until the element, identified by locator, including string in text
+  # @param [String, Selenium::WebDriver::Element] locator the locator to
+  #   identify the element or Selenium::WebDriver::Element
+  # @param [String] string the string to compare
+  def wait_for_text_to_include(locator, string)
+    wait_for { text(locator).include?(string) }
+  end
+
   #
   # Tool Functions
   #
@@ -1059,7 +1162,7 @@ class Rudra
   end
 
   (instance_methods - superclass.instance_methods).map do |method_name|
-    next if private_method_defined?(method_name) || ATTRIBUTES.include?(method_name)
+    next if private_method_defined?(method_name) || ATTRIBUTES.include?(method_name) || method_name == :puts
 
     original_method = instance_method(method_name)
 
@@ -1071,7 +1174,8 @@ class Rudra
 
   private
 
-  attr_writer :main_label
+  attr_accessor :main_label
+  attr_writer :silent, :window_size
 
   def browser=(brw)
     unless BROWSERS.include?(brw)
@@ -1098,6 +1202,10 @@ class Rudra
     @headless = true?(mode)
   end
 
+  def screen_dir=(path)
+    @screen_dir = File.join(path, @locale.to_s)
+  end
+
   def log_prefix=(prefix)
     @log_prefix = prefix.chomp
   end
@@ -1120,6 +1228,8 @@ class Rudra
   def initialize_driver
     @driver = if browser == :chrome
                 Selenium::WebDriver.for(:chrome, options: chrome_options)
+              # elsif browser == :edge
+              #   Selenium::WebDriver.for(:edge, options: edge_options)
               elsif browser == :firefox
                 Selenium::WebDriver.for(:firefox, options: firefox_options)
               elsif browser == :ie
@@ -1132,7 +1242,10 @@ class Rudra
   def chrome_options
     options = Selenium::WebDriver::Chrome::Options.new
     options.add_argument('--disable-notifications')
-    options.add_argument('--headless') if headless
+    if headless
+      options.add_argument('--headless')
+      options.add_argument("--window-size=#{window_size}")
+    end
     options.add_option(
       'excludeSwitches',
       %w[enable-automation enable-logging]
@@ -1141,6 +1254,10 @@ class Rudra
     options
   end
 
+  # def edge_options
+  #   Selenium::WebDriver::Edge::Options.new
+  # end
+
   def firefox_options
     options = Selenium::WebDriver::Firefox::Options.new
     options.add_preference('intl.accept_languages', locale)
@@ -1173,17 +1290,17 @@ class Rudra
             how.to_sym
           end
 
-    abort("Cannot parse locator: #{locator}") unless HOWS.include?(how)
+    raise Selenium::WebDriver::Error::InvalidSelectorError, "Cannot parse locator: #{locator}" unless HOWS.include?(how)
 
     [how, what]
   end
 
   def log(method_name, *args)
-    return unless @verbose && caller_locations(2, 1).first.label == main_label
+    return unless verbose && caller_locations(2, 1).first.label == main_label
 
     arguments = args.map(&:to_s).join(', ')
 
-    puts @log_prefix + (
+    puts log_prefix + (
       arguments.empty? ? method_name.to_s : "#{method_name}(#{arguments})"
     )
   end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rudra
 version: !ruby/object:Gem::Version
-  version: 1.0.14
+  version: 1.1.1
 platform: ruby
 authors:
 - Aaron Chen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-06-01 00:00:00.000000000 Z
+date: 2020-06-12 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yard
@@ -24,6 +24,26 @@ dependencies:
     - - "~>"
       - !ruby/object:Gem::Version
         version: 0.9.25
+- !ruby/object:Gem::Dependency
+  name: rspec
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.9.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.9'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 3.9.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '3.9'
 - !ruby/object:Gem::Dependency
   name: selenium-webdriver
   requirement: !ruby/object:Gem::Requirement