[rb] UNSTABLE, Adding driver#send_devtools_command for sending comman…

…ds to Chrome DevTools debugger.

For some reason specs for headless mode are not passing.

Refer:
bayandin/chromedriver@9d18be0

rb/lib/selenium/webdriver/chrome/bridge.rb

@@ -22,7 +22,8 @@ module Bridge
 
         COMMANDS = {
           get_network_conditions: [:get, '/session/:session_id/chromium/network_conditions'.freeze],
-          set_network_conditions: [:post, '/session/:session_id/chromium/network_conditions'.freeze]
+          set_network_conditions: [:post, '/session/:session_id/chromium/network_conditions'.freeze],
+          send_devtools_command: [:post, '/session/:session_id/chromium/send_command'.freeze]
         }.freeze
 
         def commands(command)
@@ -37,6 +38,10 @@ def network_conditions=(conditions)
           execute :set_network_conditions, {}, {network_conditions: conditions}
         end
 
+        def send_devtools_command(command, params)
+          execute :send_devtools_command, {}, {cmd: command, params: params}
+        end
+
       end # Bridge
     end # Chrome
   end # WebDriver

rb/lib/selenium/webdriver/chrome/driver.rb

@@ -26,6 +26,7 @@ module Chrome
 
       class Driver < WebDriver::Driver
         include DriverExtensions::HasNetworkConditions
+        include DriverExtensions::CommunicatesToDevTools
         include DriverExtensions::HasTouchScreen
         include DriverExtensions::HasWebStorage
         include DriverExtensions::TakesScreenshot

rb/lib/selenium/webdriver/common.rb

@@ -58,6 +58,7 @@
 require 'selenium/webdriver/common/driver_extensions/has_network_connection'
 require 'selenium/webdriver/common/driver_extensions/uploads_files'
 require 'selenium/webdriver/common/driver_extensions/has_addons'
+require 'selenium/webdriver/common/driver_extensions/communicates_to_dev_tools'
 require 'selenium/webdriver/common/interactions/interactions'
 require 'selenium/webdriver/common/interactions/input_device'
 require 'selenium/webdriver/common/interactions/interaction'

rb/lib/selenium/webdriver/common/driver_extensions/communicates_to_dev_tools.rb

@@ -0,0 +1,55 @@
+# 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 DriverExtensions
+      module CommunicatesToDevTools
+
+        #
+        # Send commands to the Chrome DevTools Debugger.
+        # Refer: https://chromedevtools.github.io/devtools-protocol/
+        #
+        # @example
+        #   driver.send_devtools_command('Page.navigate', {url: 'http://www.google.com'})
+        #
+        # @param [String] command The command to send to DevTools.
+        # @param [Hash] parameters The paramters of the respective command.
+        #
+
+        def send_devtools_command(command, parameters)
+          @bridge.send_devtools_command(command, parameters)
+        end
+
+        #
+        # Chrome's headless mode has Downloads disabled by default.
+        # Use this method to enable it and set a download directory
+        #
+        # @example
+        #   driver.download_path='/some/dir/which/exists/'
+        #
+        # @param [String] path The path to set as the download directory
+        #
+
+        def download_path=(path)
+          @bridge.send_devtools_command('Page.setDownloadBehavior', behavior: 'allow', downloadPath: path)
+        end
+
+      end # CommunicatesToDevTools
+    end # DriverExtensions
+  end # WebDriver
+end # Selenium

rb/spec/integration/selenium/webdriver/chrome/driver_spec.rb

@@ -43,6 +43,57 @@ module Chrome
             'upload_throughput' => 789
           )
         end
+
+        context "DevTools communication" do
+          it 'can send commands to DevTools' do
+            driver.send_devtools_command('Page.navigate', url: url_for('blank.html'))
+            expect(driver.title).to eq('blank')
+            driver.send_devtools_command('Page.navigate', url: url_for('colorPage.html'))
+            expect(driver.title).to eq('Color Page')
+          end
+
+          describe('#set_download_path') do
+            let(:download_path) { Dir.mktmpdir }
+            let(:file_to_download) { "blank.html" }
+            let(:final_path_to_downloaded_file) { download_path + "/" + file_to_download }
+            let(:page_to_download_from) do
+              'data:text/html,'\
+              '<!DOCTYPE html>'\
+                '<div>'\
+                  '<a download="" href="' + url_for(file_to_download) + '">Go!</a>'\
+                '</div>'\
+              '</html>'
+            end
+
+            before(:each) do
+              File.delete(final_path_to_downloaded_file) if File.exist?(final_path_to_downloaded_file)
+            end
+
+            it 'can download files in headless mode' do
+              options = Selenium::WebDriver::Chrome::Options.new
+              options.headless!
+              local_driver = WebDriver::Driver.for(:chrome, options: options)
+              local_driver.download_path = download_path
+              local_driver.get(page_to_download_from)
+
+              local_driver.find_element(css: 'a').click
+
+              sleep 3
+              local_driver.quit
+              expect(File).to exist(final_path_to_downloaded_file)
+            end
+
+            it 'can download files in non-headless mode' do
+              driver.download_path = download_path
+              driver.get(page_to_download_from)
+
+              driver.find_element(css: 'a').click
+
+              sleep 3
+              expect(File).to exist(final_path_to_downloaded_file)
+            end
+          end
+        end
       end
     end # Chrome
   end # WebDriver