rudra 1.0.14 → 1.0.15

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: f065383cc3c1a776dccc5708a77f627383aaa63f4198b087adad977271f713d6
-  data.tar.gz: 9fc4e3d993f3d3a1b4fb53d57b146df87fe8fa5e561308b726a3c8c46dd5ac1d
+  metadata.gz: 18f80bbed760fbf34d622b60e647507bab5ddfea167a6d144430dae84ed35a98
+  data.tar.gz: '04965c6ea61ea288861f22c6fabbb61240f384faf99b9a9402e51273d201caa4'
 SHA512:
-  metadata.gz: 9f80dd1cc85e183b9ac7ac2be6831a2e1ac56be27ab884133fcbfc4b0f87e03a23d4d9443bd933cfef69292c881d968b2033d3af8cd5a72c909ba8474afcce03
-  data.tar.gz: ad1f215b1ea9c9f6d16e387f95f1a5f9c087845c2db3239d0f64ea719e94241805578fe1a3589b69dafc0207fcc10b6589677259ecb87939752ab0fd0d5c9a52
+  metadata.gz: 55a6791c2d89ecb33b3833ac41ef6f1514b482ac13525070aa246166bc5d530c7dc23bf09d353630f4205ab5959dc8202a49cee271e96554e485e19a5ca24f6e
+  data.tar.gz: 1e9d5d7c2fd2cfa29aa3baa80145647e3088096a3af625f5e49d7baf931d2a72082d73abfeb7e12e44995a8d29607d08693e0533bbffdb5a34995e1182b7ee44

data/lib/rudra.rb

@@ -32,7 +32,8 @@ class Rudra
   ].freeze
 
   attr_reader :browser, :driver, :install_dir, :locale,
-              :headless, :log_prefix, :timeout, :verbose
+              :headless, :screen_dir, :log_prefix,
+              :timeout, :verbose
 
   # Initialize an instance of Rudra
   # @param [Hash] options the options to initialize Rudra
@@ -42,14 +43,16 @@ class Rudra
   #   directory of WebDrivers
   # @option options [Symbol] :locale (:en) the browser locale
   # @option options [Boolean] :headless (false) headless mode
+  # @option options [String] :screen_dir ('.//') the location of screenshots
   # @option options [String] :log_prefix (' - ') log prefix
   # @option options [Integer] :timeout (30) implicit_wait timeout
   # @option options [Boolean] :verbose (true) verbose mode
   def initialize(options = {})
     self.browser = options.fetch(:browser, :chrome)
     self.install_dir = options.fetch(:install_dir, './webdrivers/')
-    self.locale = options.fetch(:locale, :en)
+    self.locale = options.fetch(:locale, :enscreens)
     self.headless = options.fetch(:headless, false)
+    self.screen_dir = options.fetch(:screen_dir, './screens/')
     self.log_prefix = options.fetch(:log_prefix, ' - ')
     self.verbose = options.fetch(:verbose, true)
     self.main_label = caller_locations(2, 1).first.label
@@ -271,11 +274,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
 
@@ -1098,6 +1105,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
@@ -1179,7 +1190,7 @@ class Rudra
   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(', ')
 

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: rudra
 version: !ruby/object:Gem::Version
-  version: 1.0.14
+  version: 1.0.15
 platform: ruby
 authors:
 - Aaron Chen
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2020-06-01 00:00:00.000000000 Z
+date: 2020-06-02 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: yard