[rb] Allow driver path to be set using ENV variables

rb/lib/selenium/webdriver/common/service.rb

@@ -69,6 +69,7 @@ def driver_path=(path)
       def initialize(path: nil, port: nil, log: nil, args: nil)
         port ||= self.class::DEFAULT_PORT
         args ||= []
+        path ||= env_path
 
         @executable_path = path
         @host = Platform.localhost
@@ -87,16 +88,25 @@ def initialize(path: nil, port: nil, log: nil, args: nil)
       end
 
       def launch
-        @executable_path ||= begin
-          default_options = WebDriver.const_get("#{self.class.name&.split('::')&.[](2)}::Options").new
-          DriverFinder.new(default_options, self).driver_path
-        end
+        @executable_path ||= env_path || find_driver_path
         ServiceManager.new(self).tap(&:start)
       end
 
       def shutdown_supported
         self.class::SHUTDOWN_SUPPORTED
       end
+
+      def find_driver_path
+        default_options = WebDriver.const_get("#{self.class.name&.split('::')&.[](2)}::Options").new
+        DriverFinder.new(default_options, self).driver_path
+      end
+
+      def env_path
+        class_name = self.class.name&.split('::')&.[](2)&.downcase
+        driver_name = class_name == 'firefox' ? 'gecko' : class_name
+        parsed_driver = "SE_#{driver_name&.upcase}DRIVER"
+        ENV.fetch(parsed_driver, nil)
+      end
     end # Service
   end # WebDriver
 end # Selenium

rb/lib/selenium/webdriver/edge/service.rb

@@ -24,7 +24,6 @@ class Service < WebDriver::Service
         DEFAULT_PORT = 9515
         EXECUTABLE = 'msedgedriver'
         SHUTDOWN_SUPPORTED = true
-
         def log
           return @log unless @log.is_a? String
 

rb/lib/selenium/webdriver/safari/service.rb

@@ -24,7 +24,6 @@ class Service < WebDriver::Service
         DEFAULT_PORT = 7050
         EXECUTABLE = 'safaridriver'
         SHUTDOWN_SUPPORTED = false
-
         def initialize(path: nil, port: nil, log: nil, args: nil)
           raise Error::WebDriverError, 'Safari Service does not support setting log output' if log
 

rb/sig/lib/selenium/webdriver/common/service.rbs

@@ -47,6 +47,8 @@ module Selenium
 
       attr_accessor args: untyped
 
+      def env_path: -> String
+
       alias extra_args args
 
       def initialize: (?path: untyped?, ?port: untyped?, ?log: untyped?, ?args: untyped?) -> void

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

@@ -37,6 +37,19 @@ module Chrome
         it 'can be started outside driver' do
           expect(service_manager.uri).to be_a(URI)
         end
+
+        it 'uses the path from the environment' do
+          ENV['SE_CHROMEDRIVER'] = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'updates the path after setting the environment variable' do
+          ENV['SE_CHROMEDRIVER'] = '/foo/bar'
+          service.executable_path = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
       end
     end # Chrome
   end # WebDriver

rb/spec/integration/selenium/webdriver/edge/service_spec.rb

@@ -37,6 +37,19 @@ module Edge
         it 'can be started outside driver' do
           expect(service_manager.uri).to be_a(URI)
         end
+
+        it 'uses the path from the environment' do
+          ENV['SE_EDGEDRIVER'] = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'updates the path after setting the environment variable' do
+          ENV['SE_EDGEDRIVER'] = '/foo/bar'
+          service.executable_path = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
       end
     end # Edge
   end # WebDriver

rb/spec/integration/selenium/webdriver/firefox/service_spec.rb

@@ -37,6 +37,19 @@ module Firefox
         it 'can be started outside driver' do
           expect(service_manager.uri).to be_a(URI)
         end
+
+        it 'uses the path from the environment' do
+          ENV['SE_GECKODRIVER'] = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'updates the path after setting the environment variable' do
+          ENV['SE_GECKODRIVER'] = '/foo/bar'
+          service.executable_path = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
       end
     end # Firefox
   end # WebDriver

rb/spec/integration/selenium/webdriver/ie/service_spec.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+# 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.
+
+require_relative '../spec_helper'
+
+module Selenium
+  module WebDriver
+    module IE
+      describe Service, exclusive: [{bidi: false, reason: 'Not yet implemented with BiDi'}, {browser: :ie}] do
+        let(:service) { described_class.new }
+        let(:service_manager) { service.launch }
+
+        after { service_manager.stop }
+
+        it 'auto uses iedriver' do
+          service.executable_path = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'can be started outside driver' do
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'uses the path from the environment' do
+          ENV['SE_IEIDRIVER'] = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'updates the path after setting the environment variable' do
+          ENV['SE_IEIDRIVER'] = '/foo/bar'
+          service.executable_path = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+      end
+    end # IE
+  end # WebDriver
+end # Selenium

rb/spec/integration/selenium/webdriver/safari/service_spec.rb

@@ -0,0 +1,56 @@
+# frozen_string_literal: true
+
+# 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.
+
+require_relative '../spec_helper'
+
+module Selenium
+  module WebDriver
+    module Safari
+      describe Service, exclusive: [{bidi: false, reason: 'Not yet implemented with BiDi'}, {browser: :safari}] do
+        let(:service) { described_class.new }
+        let(:service_manager) { service.launch }
+
+        after { service_manager.stop }
+
+        it 'auto uses safaridriver' do
+          service.executable_path = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'can be started outside driver' do
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'uses the path from the environment' do
+          ENV['SE_SAFARIDRIVER'] = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+
+        it 'updates the path after setting the environment variable' do
+          ENV['SE_SAFARIDRIVER'] = '/foo/bar'
+          service.executable_path = DriverFinder.new(Options.new, described_class.new).driver_path
+
+          expect(service_manager.uri).to be_a(URI)
+        end
+      end
+    end # Safari
+  end # WebDriver
+end # Selenium