Fix specs

Author: zuchmanski

lib/aerospike/node_validator.rb

@@ -20,12 +20,13 @@
 module Aerospike
   class NodeValidator # :nodoc:
 
-    attr_reader :host, :aliases, :name, :features, :cluster_name, :tls_options, :conn
+    attr_reader :host, :timeout, :aliases, :name, :features, :cluster_name, :tls_options, :conn
 
     def initialize(cluster, host, timeout, cluster_name, tls_options = {})
       @cluster = cluster
       @features = Set.new
       @host = host
+      @timeout = timeout
       @cluster_name = cluster_name
       @tls_options = tls_options
 
@@ -44,7 +45,7 @@ def get_hosts(address)
 
       begin
         conn = Cluster::CreateConnection.(@cluster, Host.new(address, host.port, host.tls_name))
-        conn.timeout = @cluster.connection_timeout
+        conn.timeout = timeout
 
         commands = %w[node features]
         commands << address_command unless is_loopback?(address)

spec/aerospike/node_validator_spec.rb

@@ -21,6 +21,7 @@
     before do
       allow(socket).to receive(:write).and_return(nil)
       allow(socket).to receive(:read).and_return(nil)
+      allow(socket).to receive(:timeout=).and_return(nil)
       expect(::Aerospike::Cluster::CreateConnection).to receive(:call).and_return(socket)
     end
 
@@ -31,6 +32,7 @@
     let(:hosts) { [::Aerospike::Host.new('192.168.1.1', '3000')] }
 
     before do
+      allow(socket).to receive(:timeout=).and_return(nil)
       allow(::Aerospike::Cluster::CreateConnection).to receive(:call).and_return(socket)
 
       expect(::Aerospike::Info).to receive(:request).and_return(
@@ -49,6 +51,7 @@
     let(:hosts) { [::Aerospike::Host.new('my.lb.com', '3000')] }
 
     before do
+      allow(socket).to receive(:timeout=).and_return(nil)
       allow(::Aerospike::Cluster::CreateConnection).to receive(:call).and_return(socket)
       expect(Resolv).to receive(:getaddresses).and_return(['101.1.1.1', '102.1.1.1'])
 
@@ -79,6 +82,7 @@
     let(:policy) { ::Aerospike::ClientPolicy.new(tls: { enable: true }) }
 
     before do
+      allow(socket).to receive(:timeout=).and_return(nil)
       allow(::Aerospike::Cluster::CreateConnection).to receive(:call).and_return(socket)
       expect(Resolv).to receive(:getaddresses).and_return(['101.1.1.1', '102.1.1.1'])