Use Async::Socket if is present

bblimke · May 22, 2024 · 54519b2 · 54519b2
1 parent 83b6ec7
commit 54519b2
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/lib/webmock/http_lib_adapters/async_http_client_adapter.rb b/lib/webmock/http_lib_adapters/async_http_client_adapter.rb
@@ -152,11 +152,15 @@ def inspect
 
         private
 
+        def socket_class
+          @_socket_class ||= Gem::Dependency.new("async-io").matching_specs.any? ? Async::IO::Socket : Socket
+        end
+
         def create_connected_sockets
           pair = begin
-            Socket.pair(Socket::AF_UNIX, Socket::SOCK_STREAM)
+            socket_class.pair(Socket::AF_UNIX, Socket::SOCK_STREAM)
           rescue Errno::EAFNOSUPPORT
-            Socket.pair(Socket::AF_INET, Socket::SOCK_STREAM)
+            socket_class.pair(Socket::AF_INET, Socket::SOCK_STREAM)
           end
           pair.tap do |sockets|
             sockets.each do |socket|