rudra 1.1.0 → 1.1.5

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: b59cfa6497e4560d80369dc076df45638ec043abe069b5f6665e5749aefdef87
-  data.tar.gz: 51e97a957330ffce97411d9f3ad40e01a4be439cc677306d5507ae0136577beb
+  metadata.gz: 27e45015364b99148e3e92162d6069446d7b711745d660089661d14ce1da5113
+  data.tar.gz: c9fb052da4d604b1247440f1ce660b31c0cbeb67d20cad5c0c90bc481a5805cd
 SHA512:
-  metadata.gz: f7984dccb05ea056c716dd533c5d4662e83ea8711c62a2fc3f02f95d0d7a795914d16c2ceacd2ef2cf1eff26374ae9c8ed795a14797186f696356f3c02c687bf
-  data.tar.gz: 4bfb07ee8abe3ce212c85ca19ab052fb134b47c42e225e7160ea6463795e425e22a594998d2e89365709a3d2b1495a2c5cbbb0894b64bd1bfc073a190519ef8f
+  metadata.gz: a8213217762733881e6249e7e5f23c380ff88ea6b4378a2b495792f8838327dee6f6de1ed4c940f0742b09da9c7357a69352652047da038deb2c81baaaed335c
+  data.tar.gz: 56d3540043140a8c1c2ed1b00e1fd16e4ec7521956670a9a158455b5881b49628de0c30fc48f5f0b0bb182850462efb742954b3e0409ad47758ebaaac43b4716

data/lib/rudra.rb

@@ -1,5 +1,9 @@
 require 'selenium-webdriver'
 require 'webdrivers'
+require 'zip'
+require 'base64'
+require 'json'
+require 'stringio'
 
 # Selenium IDE-like WebDriver based upon Ruby binding
 # @author Aaron Chen
@@ -15,6 +19,8 @@ require 'webdrivers'
 # @attr_reader [Integer] timeout The driver timeout
 # @attr_reader [Boolean] verbose Turn on/off Verbose mode
 # @attr_reader [Boolean] silent Turn off Turn on/off descriptions
+# @attr_reader [String] auth_username username for Basic Access Authentication Extension (Chrome only)
+# @attr_reader [String] auth_password password for Basic Access Authentication Extension (Chrome only)
 class Rudra
   # Supported Browsers
   BROWSERS = %i[chrome firefox ie safari].freeze
@@ -30,11 +36,13 @@ class Rudra
     browser driver install_dir locale
     headless window_size screen_dir
     log_prefix timeout verbose silent
+    auth_username auth_password
   ].freeze
 
   attr_reader :browser, :driver, :install_dir, :locale,
               :headless, :window_size, :screen_dir,
-              :log_prefix, :timeout, :verbose, :silent
+              :log_prefix, :timeout, :verbose, :silent,
+              :auth_username, :auth_password
 
   # Initialize an instance of Rudra
   # @param [Hash] options the options to initialize Rudra
@@ -50,6 +58,8 @@ class Rudra
   # @option options [Integer] :timeout (30) implicit_wait timeout
   # @option options [Boolean] :verbose (false) Turn on/off verbose mode
   # @option options [Boolean] :silent (false) Turn on/off descriptions
+  # @option options [String] :auth_username ('') username for Basic Access Authentication extension
+  # @option options [String] :auth_password ('') password for Basic Access Authentication extension
   def initialize(options = {})
     self.browser = options.fetch(:browser, :chrome)
     self.install_dir = options.fetch(:install_dir, './webdrivers/')
@@ -60,6 +70,8 @@ class Rudra
     self.log_prefix = options.fetch(:log_prefix, ' - ')
     self.verbose = options.fetch(:verbose, false)
     self.silent = options.fetch(:silent, false)
+    self.auth_username = options.fetch(:auth_username, '')
+    self.auth_password = options.fetch(:auth_password, '')
     self.main_label = caller_locations(2, 1).first.label
 
     initialize_driver
@@ -149,6 +161,26 @@ class Rudra
     driver.manage.delete_cookie(name)
   end
 
+  # Check if an element is found
+  # @param [String, Selenium::WebDriver::Element] locator the locator to
+  #   identify the element or Selenium::WebDriver::Element
+  # @param [Integer] seconds seconds before timed out
+  def element_found?(locator, seconds = 1)
+    how, what = parse_locator(locator)
+
+    implicit_wait(seconds)
+
+    begin
+      wait_for(seconds) { driver.find_element(how, what).displayed? }
+    rescue Selenium::WebDriver::Error::TimeoutError
+      false
+    rescue Net::ReadTimeout
+      false
+    ensure
+      implicit_wait(timeout)
+    end
+  end
+
   # Execute the given JavaScript
   # @param [String] script JavaScript source to execute
   # @param [Selenium::WebDriver::Element, Integer, Float, Boolean, NilClass,
@@ -177,7 +209,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? }
 
@@ -189,8 +221,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
@@ -289,13 +324,16 @@ class Rudra
   # 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(
-      File.join(
-        @screen_dir,
-        filename.end_with?('.png') ? filename : "#{filename}.png"
-      )
+    file = File.join(
+      @screen_dir,
+      sanitize(filename.end_with?('.png') ? filename : "#{filename}.png")
     )
+
+    dir = File.dirname(file)
+
+    mkdir(dir) unless Dir.exist?(dir)
+
+    driver.save_screenshot(file)
   end
 
   # Switch to the currently active modal dialog
@@ -310,6 +348,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
@@ -345,6 +384,50 @@ class Rudra
     wait_for { find_element(locator).enabled? }
   end
 
+  # Switch to a frame and wait until the element, identified by locator, is found
+  # @param [String] frame_id the frame id
+  # @param [String] locator the locator to identify the element
+  def switch_to_frame_and_wait_for_element_found(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 = 3)
+    how, what = parse_locator(locator)
+
+    implicit_wait(seconds)
+
+    begin
+      wait_for(seconds) do
+        begin
+          elements = driver.find_elements(how, what)
+          elements.empty? || elements.map(&:displayed?).none?
+        rescue Selenium::WebDriver::Error::StaleElementReferenceError
+          false
+        end
+      end
+    rescue Selenium::WebDriver::Error::TimeoutError
+      true
+    rescue Net::ReadTimeout
+      true
+    ensure
+      implicit_wait(timeout)
+    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)
@@ -393,7 +476,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
@@ -428,7 +511,14 @@ 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
+        element = find_element(locator)
+        element.enabled? && element.click.nil?
+      rescue Selenium::WebDriver::Error::ElementClickInterceptedError
+        false
+      end
+    end
   end
 
   # Click the given element, identified by locator, with an offset
@@ -741,6 +831,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
   #
@@ -1091,7 +1213,7 @@ class Rudra
   private
 
   attr_accessor :main_label
-  attr_writer :silent, :window_size
+  attr_writer :silent, :window_size, :auth_username, :auth_password
 
   def browser=(brw)
     unless BROWSERS.include?(brw)
@@ -1162,6 +1284,14 @@ class Rudra
       options.add_argument('--headless')
       options.add_argument("--window-size=#{window_size}")
     end
+    if auth_username
+      if headless
+        $stdout.puts('Basic Access Authentication Extension cannot be installed while headless')
+      else
+        encoded = chrome_basic_auth_extension(auth_username, auth_password)
+        options.add_encoded_extension(encoded)
+      end
+    end
     options.add_option(
       'excludeSwitches',
       %w[enable-automation enable-logging]
@@ -1206,7 +1336,7 @@ 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
@@ -1221,9 +1351,61 @@ class Rudra
     )
   end
 
+  def sanitize(filename)
+    invalid_characters = ['/', '\\', '?', '%', '*', ':', '|', '"', '<', '>']
+    invalid_characters.each do |character|
+      filename.gsub!(character, '')
+    end
+    filename
+  end
+
   def random_id(length = 8)
     charset = [(0..9), ('a'..'z')].flat_map(&:to_a)
     id = Array.new(length) { charset.sample }.join
     "rudra_#{id}"
   end
+
+  def chrome_basic_auth_extension(username, password)
+    manifest = {
+      "manifest_version": 2,
+      "name": 'Rudra Basic Access Authentication Extension',
+      "version": '1.0.0',
+      "permissions": ['*://*/*', 'webRequest', 'webRequestBlocking'],
+      "background": {
+        "scripts": ['background.js']
+      }
+    }
+
+    background = <<~JAVASCRIPT
+      var username = '#{username}';
+      var password = '#{password}';
+
+      chrome.webRequest.onAuthRequired.addListener(
+        function handler(details) {
+          if (username == null) {
+            return { cancel: true };
+          }
+
+          var authCredentials = { username: username, password: username };
+          username = password = null;
+
+          return { authCredentials: authCredentials };
+        },
+        { urls: ['<all_urls>'] },
+        ['blocking']
+      );
+    JAVASCRIPT
+
+    stringio = Zip::OutputStream.write_buffer do |zos|
+      zos.put_next_entry('manifest.json')
+      zos.write manifest.to_json
+      zos.put_next_entry('background.js')
+      zos.write background
+    end
+    # File.open('basic_auth.crx', 'wb') do |f|
+    #   f << stringio.string
+    # end
+
+    Base64.strict_encode64(stringio.string)
+  end
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rudra
 version: !ruby/object:Gem::Version
-  version: 1.1.0
+  version: 1.1.5
 platform: ruby
 authors:
 - Aaron Chen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-06-08 00:00:00.000000000 Z
+date: 2020-06-30 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yard
@@ -24,6 +24,46 @@ 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: rubyzip
+  requirement: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.3.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
+  type: :runtime
+  prerelease: false
+  version_requirements: !ruby/object:Gem::Requirement
+    requirements:
+    - - ">="
+      - !ruby/object:Gem::Version
+        version: 2.3.0
+    - - "~>"
+      - !ruby/object:Gem::Version
+        version: '2.3'
 - !ruby/object:Gem::Dependency
   name: selenium-webdriver
   requirement: !ruby/object:Gem::Requirement