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|