[rb] fix SeleniumHQ#8525 by limiting concurrent sessions in test to t…

…he number of processors detected
titusfortner · Feb 11, 2021 · fd88246 · fd88246
1 parent 06f5a88
commit fd88246
Show file tree

Hide file tree

Showing 4 changed files with 24 additions and 9 deletions.
diff --git a/rb/.rubocop.yml b/rb/.rubocop.yml
@@ -107,8 +107,7 @@ RSpec/AnyInstance:
   Enabled: false
 
 RSpec/BeforeAfterAll:
-  Exclude:
-    - 'spec/unit/selenium/webdriver/socket_poller_spec.rb'
+  Enabled: false
 
 RSpec/ContextWording:
   Enabled: false

diff --git a/rb/lib/selenium/webdriver/common/platform.rb b/rb/lib/selenium/webdriver/common/platform.rb
@@ -50,6 +50,22 @@ def os
                 end
       end
 
+      def processors
+        case os
+        when :macosx
+          (`which hwprefs` != '' ? `hwprefs thread_count` : `sysctl -n hw.ncpu`).to_i
+        when :linux
+          `cat /proc/cpuinfo | grep processor | wc -l`.to_i
+        when :unix
+          `sysctl -n hw.ncpu`.to_i
+        when :windows
+          require 'win32ole'
+          wmi = WIN32OLE.connect("winmgmts://")
+          cpu = wmi.ExecQuery("select NumberOfCores from Win32_Processor")
+          cpu.to_enum.first.NumberOfCores
+        end
+      end
+
       def ci
         if ENV['TRAVIS']
           :travis

diff --git a/rb/spec/integration/selenium/webdriver/driver_spec.rb b/rb/spec/integration/selenium/webdriver/driver_spec.rb
@@ -22,10 +22,7 @@
 module Selenium
   module WebDriver
     describe Driver do
-      it_behaves_like 'driver that can be started concurrently', exclude: [{browser: %i[safari safari_preview]},
-                                                                           {driver: :remote,
-                                                                            platform: :linux,
-                                                                            reason: 8525}]
+      it_behaves_like 'driver that can be started concurrently', exclude: {browser: %i[safari safari_preview]}
 
       it 'creates default capabilities' do
         reset_driver! do |driver|

diff --git a/rb/spec/integration/selenium/webdriver/spec_support/shared_examples/concurrent_driver.rb b/rb/spec/integration/selenium/webdriver/spec_support/shared_examples/concurrent_driver.rb
@@ -18,6 +18,8 @@
 # under the License.
 
 shared_examples_for 'driver that can be started concurrently' do |guard|
+  before(:all) { quit_driver }
+
   after do
     drivers.each(&:quit)
     threads.select(&:alive?).each(&:kill)
@@ -26,8 +28,9 @@
   let(:drivers) { [] }
   let(:threads) { [] }
 
-  it 'starts 3 drivers sequentially', guard do
-    3.times do
+  it 'starts multiple drivers sequentially', guard do
+    expected_count = [4, WebDriver::Platform.processors].min
+    expected_count.times do
       thread = Thread.new do
         drivers << create_driver!
       end
@@ -36,7 +39,7 @@
     end
 
     expect { threads.each(&:join) }.not_to raise_error
-    expect(drivers.count).to eq(3)
+    expect(drivers.count).to eq(expected_count)
 
     # make any wire call
     expect { drivers.each(&:title) }.not_to raise_error