Update socket poller test for Ruby 2.4

SeleniumHQ · Apr 26, 2019 · 87d060e · 87d060e
1 parent d3834b5
commit 87d060e
Showing 1 changed file with 10 additions and 8 deletions.
diff --git a/rb/spec/unit/selenium/webdriver/socket_poller_spec.rb b/rb/spec/unit/selenium/webdriver/socket_poller_spec.rb
@@ -23,15 +23,17 @@ module Selenium
   module WebDriver
     describe SocketPoller do
       around do |example|
-        server_thread = Thread.new do
-          server = TCPServer.open(9250)
-          Thread.current.thread_variable_set(:server, server)
-          loop { server.accept.close }
+        begin
+          server_thread = Thread.new do
+            server = TCPServer.open(9250)
+            Thread.current.thread_variable_set(:server, server)
+            loop { server.accept.close }
+          end
+          server_thread.report_on_exception = false
+          example.call
+        ensure
+          server_thread.thread_variable_get(:server).close
         end
-        server_thread.report_on_exception = false
-        example.call
-      ensure
-        server_thread.thread_variable_get(:server).close
       end
 
       def poller(port)