lib/capybara/headless_chrome/driver.rb in capybara-headless_chrome-0.2.1 vs lib/capybara/headless_chrome/driver.rb in capybara-headless_chrome-0.3.0

@@ -4,25 +4,25 @@
 require "capybara/headless_chrome/downloads"
 
 module Capybara
   module HeadlessChrome
     class Driver < Capybara::Selenium::Driver
-      def initialize app
-        super(app, browser: :chrome, desired_capabilities: chrome_capabilities)
+      def initialize app, args: []
+        super(app, browser: :chrome, desired_capabilities: chrome_capabilities(args))
         configure_downloads
         fix_whitespace
       end
 
       def downloads
         @downloads ||= Downloads.new
       end
 
       private
 
-      def chrome_capabilities
+      def chrome_capabilities args
         ::Selenium::WebDriver::Remote::Capabilities.chrome(
           chromeOptions: {
-            args: chrome_arguments,
+            args: chrome_arguments + args,
             prefs: chrome_preferences
           }
         )
       end