From d9304916ccf0127f714f25c906b02e9d71064674 Mon Sep 17 00:00:00 2001
From: luisramos0 <luisramos0@gmail.com>
Date: Mon, 25 Mar 2019 15:43:28 +0000
Subject: [PATCH] Add chrome driver config to make testing csv downloads with
 headless chrome possible

---
 spec/spec_helper.rb              | 13 ++++++++++++-
 spec/support/downloads_helper.rb |  7 +++++--
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/spec/spec_helper.rb b/spec/spec_helper.rb
index 68a523c4af..103fbd96d6 100644
--- a/spec/spec_helper.rb
+++ b/spec/spec_helper.rb
@@ -46,7 +46,17 @@ Capybara.register_driver :chrome do |app|
   options = Selenium::WebDriver::Chrome::Options.new(
     args: %w[headless disable-gpu no-sandbox window-size=1280,768]
   )
-  Capybara::Selenium::Driver.new(app, browser: :chrome, options: options)
+  driver = Capybara::Selenium::Driver.new(app, browser: :chrome, options: options)
+
+  ### Allow file downloads in headless chrome
+  ### https://bugs.chromium.org/p/chromium/issues/detail?id=696481#c89
+  bridge = driver.browser.send(:bridge)
+  path = '/session/:session_id/chromium/send_command'
+  path[':session_id'] = bridge.session_id
+  bridge.http.call(:post, path, cmd: 'Page.setDownloadBehavior',
+                                params: { behavior: 'allow', downloadPath: DownloadsHelper.path.to_s })
+
+  driver
 end
 
 Capybara.default_max_wait_time = 30
@@ -145,6 +155,7 @@ RSpec.configure do |config|
   config.include ActionView::Helpers::DateHelper
   config.include OpenFoodNetwork::DelayedJobHelper
   config.include OpenFoodNetwork::PerformanceHelper
+  config.include DownloadsHelper
 
   # FactoryBot
   require 'factory_bot_rails'
diff --git a/spec/support/downloads_helper.rb b/spec/support/downloads_helper.rb
index a0980e8900..732bf22769 100644
--- a/spec/support/downloads_helper.rb
+++ b/spec/support/downloads_helper.rb
@@ -1,6 +1,9 @@
 module DownloadsHelper
   TIMEOUT = 10
-  PATH    = Rails.root.join("tmp", "downloads")
+
+  def self.path
+    Rails.root.join("tmp", "downloads")
+  end
 
   def downloaded_filename
     wait_for_download
@@ -19,7 +22,7 @@ module DownloadsHelper
   private
 
   def downloaded_filenames
-    Dir[PATH.join("*")]
+    Dir[DownloadsHelper.path.join("*")]
   end
 
   def wait_for_download