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 = {})
13
13
yield self if block_given?
14
14
end
15
15
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
+
16
25
def prepare_socket ( server )
17
26
socket = server [ :socket ]
18
27
encryption = server [ :encryption ]
@@ -28,7 +37,7 @@ def open_connection(server)
28
37
errors = [ ]
29
38
hosts . each do |host , port |
30
39
begin
31
- prepare_socket ( server . merge ( socket : TCPSocket . new ( host , port ) ) )
40
+ prepare_socket ( server . merge ( socket : socket_class . new ( host , port ) ) )
32
41
return
33
42
rescue Net ::LDAP ::Error , SocketError , SystemCallError ,
34
43
OpenSSL ::SSL ::SSLError => e
You can’t perform that action at this time.
0 commit comments