File tree Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Expand file tree Collapse file tree 1 file changed +10
-1
lines changed Original file line number Diff line number Diff line change @@ -13,6 +13,15 @@ def initialize(server = {})
1313 yield self if block_given?
1414 end
1515
16+ # Allows tests to parameterize what socket class to use
17+ def socket_class
18+ @socket_class || TCPSocket
19+ end
20+
21+ def socket_class = ( socket_class )
22+ @socket_class = socket_class
23+ end
24+
1625 def prepare_socket ( server )
1726 socket = server [ :socket ]
1827 encryption = server [ :encryption ]
@@ -28,7 +37,7 @@ def open_connection(server)
2837 errors = [ ]
2938 hosts . each do |host , port |
3039 begin
31- prepare_socket ( server . merge ( socket : TCPSocket . new ( host , port ) ) )
40+ prepare_socket ( server . merge ( socket : socket_class . new ( host , port ) ) )
3241 return
3342 rescue Net ::LDAP ::Error , SocketError , SystemCallError ,
3443 OpenSSL ::SSL ::SSLError => e
You can’t perform that action at this time.
0 commit comments