ritsudo 0.1.3 → 0.1.4

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: aecf9ff6730444d597a135b519014e951821d0da74e6ac4d4aa012301f93c5ee
-  data.tar.gz: 18d826bea44d1320ce672ac443bff4dc1381d7f6955972f687fa120d9c3f7f3d
+  metadata.gz: ee2a6abdb855e2cdce4490dd6f35fcf71960388b4612e5ebb95508966567c2e5
+  data.tar.gz: b0cc172d1ebf2642a2ec1489f9906d8a0d4e2d204f1504579fb8a61ebad13bf9
 SHA512:
-  metadata.gz: 426554618585d45b1ee152e24945a035f310104f657007cc70e01ea0cd056fdae54218fc263360aa5eadb2be8893b6c0baa75b0436733a1b889f0e3741989ad8
-  data.tar.gz: d028a1734186e28220b718908b1650b1000c849e111c942469348b63d433066c3f8dd90ff56373d09eb7da95234c266a6469894ed499e56ef329919a6ce0532f
+  metadata.gz: 366102b22ebca7f46296a93fb343d9c03cdd7299c22088bc7b0a764b7b6f8ec32545e64d0b445865b065fa5451fe35790074f39ff3e4e4cf62cdb3821ba157a3
+  data.tar.gz: 43a5a6a3efa8981bbdebdf5fda142beb82f19a45e2e46cf14c1b1c4490ae58e2652610e530ea600f328f367df9277d1e415f95b47dd7884da2cd10632cce6d76

data/README.md

@@ -24,7 +24,7 @@ require "ritsudo"
 ua = "Mozilla/5.0 (iPhone; CPU iPhone OS 12_0 like Mac OS X) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/12.0 Mobile/15E148 Safari/604.1"
 collector = Ritsudo::Collector.new(match: /example\.com/)
 benchmark = Ritsudo::Benchmark.new(collector: collector)
-benchmark.do("http:/www.example.com", driver_options: { user_agent: ua })
+benchmark.do("https:/www.example.com", driver_options: { user_agent: ua })
 benchmark.collector.documents.display
 benchmark.collector.xhrs.display
 ```
@@ -43,20 +43,20 @@ Ritsudo requests 5 times: https://example.com
   +------------------+--------+------+-----+-------+
 
 [Document]
-  +--------------------------+--------+--------+--------+-------+
-  | url                      | avg    | max    | min    | count |
-  +--------------------------+--------+--------+--------+-------+
-  | http://www.example.com/  | 597.56 | 858.15 | 433.51 | 5     |
-  +--------------------------+--------+--------+--------+-------+
+  +---------------------------+--------+--------+--------+-------+
+  | url                       | avg    | max    | min    | count |
+  +---------------------------+--------+--------+--------+-------+
+  | https://www.example.com/  | 597.56 | 858.15 | 433.51 | 5     |
+  +---------------------------+--------+--------+--------+-------+
 
 [XHR]
+  +-------------------------------------------+---------+---------+---------+-------+-------------+
+  | url                                       | avg     | max     | min     | count | uncompleted |
+  +-------------------------------------------+---------+---------+---------+-------+-------------+
+  | https://www.example.com/xhr_request       | 300.39  | 500.39  | 250.39  | 4     | 1           |
   +------------------------------------------+---------+---------+---------+-------+-------------+
-  | url                                      | avg     | max     | min     | count | uncompleted |
-  +------------------------------------------+---------+---------+---------+-------+-------------+
-  | http://www.example.com/xhr_request       | 300.39  | 500.39  | 250.39  | 4     | 1           |
-  +------------------------------------------+---------+---------+---------+-------+-------------+
-  | http://www.example.com/slow_xhr_request  | -       | -       | -       | -     | 5           |
-  +------------------------------------------+---------+---------+---------+-------+-------------+
+  | https://www.example.com/slow_xhr_request  | -       | -       | -       | -     | 5           |
+  +-------------------------------------------+---------+---------+---------+-------+-------------+
 
 [Script]
   +--------------------------------+-------+-------+-------+-------+-------------+
@@ -65,6 +65,16 @@ Ritsudo requests 5 times: https://example.com
   | https//example.com/example.js  | 25.88 | 43.15 | 14.96 | 5     | 0           |
   +--------------------------------+-------+-------+-------+-------+-------------+
 ```
+
+### Set Cookies
+```
+ritsudo -a https://example.com/page -m "example.com" -C "hoge=fuga"
+```
+
+#### Problem
+Headless chrome doesn't support to set cookies before access.
+So, Ritsudo access root path(e.g https://example.com") before benchmark.
+
 ## License
 
 The gem is available as open source under the terms of the [MIT License](https://opensource.org/licenses/MIT).

data/lib/ritsudo/benchmark.rb

@@ -5,15 +5,20 @@ module Ritsudo
       @collector = collector
     end
 
-    def do(url, count: 5, wait: 1, sub_process_timeout: 3, 
-                driver_options: {
-                   timeout: 5,
-                   wait_time: 1
-                })
+    def do(url, count: 5, wait: 1, sub_process_timeout: 3, driver_options: {})
+      if driver_options[:cookies]
+        puts <<-EOS 
+WARNING: headless chrome can not set cookie before access. Ritsudo accesses root path before requested path for set cookies.
+
+        EOS
+      end
+
+      default_driver_options = { timeout: 5, wait_time: 1 }
       puts "Ritsudo requests #{count} times: #{url}"
       count.times do
         print(".")
-        collect(url, wait: wait, sub_process_timeout: sub_process_timeout, driver_options: driver_options)
+        collect(url, wait: wait, sub_process_timeout: sub_process_timeout,
+                driver_options: default_driver_options.merge(driver_options))
       end
       puts ""
     end

data/lib/ritsudo/cli.rb

@@ -5,6 +5,7 @@ module Ritsudo
     default_command :benchmark
     option :url, required: true, aliases: ['-a']
     option :count, default: 5, type: :numeric, aliases: ['-c']
+    option :cookies, type: :string, aliases: ['-C']
     option :sub_process_timeout, default: 5, type: :numeric, aliases: ['-s']
     option :timeout, default: 10, type: :numeric, aliases: ['-t']
     option :wait_time, default: 1, type: :numeric, aliases: ['-w']
@@ -21,7 +22,8 @@ module Ritsudo
                    driver_options: {
                      timeout: options[:timeout],
                      wait_time: options[:wait_time],
-                     user_agent: options[:ua]
+                     user_agent: options[:ua],
+                     cookies: options[:cookies]
                    }
                   )
       benchmark.collector.report

data/lib/ritsudo/driver.rb

@@ -1,10 +1,11 @@
 require 'selenium-webdriver'
 require 'forwardable'
+require 'webrick/cookie'
 
 module Ritsudo
   class Driver
     extend Forwardable
-    def_delegators :@driver, :get, :get_log, :manage
+    def_delegators :@driver, :get_log, :manage
 
     def driver
       @driver
@@ -13,6 +14,7 @@ module Ritsudo
     def initialize(logger_level: :warn, timeout: 60, wait_time: 5,
                    logger_output: "./ritsudo.selenium.log",
                    user_agent: nil,
+                   cookies: nil,
                    args: ['--headless',
                           '--window-size=1920,1080',
                           '--ignore-certificate-errors',
@@ -24,17 +26,38 @@ module Ritsudo
         args << "--user-agent=#{user_agent}"
       end
 
+      if cookies
+        @cookie_templates = WEBrick::Cookie.parse(cookies).map do |cookie|
+          { name: cookie.name, value: cookie.value, path: '/' }
+        end
+      end
+
       Selenium::WebDriver.logger.output = logger_output
       Selenium::WebDriver.logger.level = logger_level
       client = Selenium::WebDriver::Remote::Http::Default.new.tap { |c| c.read_timeout = timeout }
-      @driver = Selenium::WebDriver.for(:chrome, options: options(args),
-                                        desired_capabilities: caps,
-                                        http_client: client,).tap do |d|
+      driver_options = { options: options(args), desired_capabilities: caps, http_client: client }
+      @driver = Selenium::WebDriver.for(:chrome, driver_options).tap do |d|
         d.manage.timeouts.implicit_wait = timeout
       end
       #@wait = Selenium::WebDriver::Wait.new(timeout: wait_time)
     end
 
+    def get(url)
+      if @cookie_templates
+        u = URI.parse(url)
+        base_url = u.to_s.sub(u.request_uri, '/')
+        @driver.get(base_url)
+        @driver.execute_script("return window.stop");
+        sleep(1)
+        @driver.manage.logs.get('performance')
+        @cookie_templates&.each do |cookie|
+          manage.add_cookie(cookie.merge(domain: URI.parse(url).host))
+        end
+      end
+      puts @driver.manage.logs.get('performance').inspect
+      @driver.get(url)
+    end
+
     def caps
       Selenium::WebDriver::Remote::Capabilities.chrome(
         loggingPrefs: { performance: 'ALL' },

data/lib/ritsudo/version.rb

@@ -1,3 +1,3 @@
 module Ritsudo
-  VERSION = "0.1.3"
+  VERSION = "0.1.4"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ritsudo
 version: !ruby/object:Gem::Version
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors:
 - shingo morita