Fix test race condition

Author: estolfo

spec/mongo/server/connection_spec.rb

@@ -546,9 +546,6 @@
 
     before do
       connection.connect!
-      socket = connection.instance_variable_get(:@socket)
-      max_message_size = connection.send(:max_message_size)
-      allow(Mongo::Protocol::Reply).to receive(:deserialize).with(socket, max_message_size).and_return(reply)
     end
 
     context 'when the ismaster response indicates the auth mechanism is :scram' do
@@ -564,13 +561,19 @@
       context 'when the server auth mechanism is scram', if: scram_sha_1_enabled? do
 
         it 'uses scram' do
+          socket = connection.instance_variable_get(:@socket)
+          max_message_size = connection.send(:max_message_size)
+          allow(Mongo::Protocol::Reply).to receive(:deserialize).with(socket, max_message_size).and_return(reply)
           expect(connection.send(:default_mechanism)).to eq(:scram)
         end
       end
 
       context 'when the server auth mechanism is the default (mongodb_cr)', unless: scram_sha_1_enabled?  do
 
         it 'uses scram' do
+          socket = connection.instance_variable_get(:@socket)
+          max_message_size = connection.send(:max_message_size)
+          allow(Mongo::Protocol::Reply).to receive(:deserialize).with(socket, max_message_size).and_return(reply)
           expect(connection.send(:default_mechanism)).to eq(:scram)
         end
       end
@@ -589,13 +592,19 @@
       context 'when the server auth mechanism is scram', if: scram_sha_1_enabled? do
 
         it 'uses scram' do
+          socket = connection.instance_variable_get(:@socket)
+          max_message_size = connection.send(:max_message_size)
+          allow(Mongo::Protocol::Reply).to receive(:deserialize).with(socket, max_message_size).and_return(reply)
           expect(connection.send(:default_mechanism)).to eq(:scram)
         end
       end
 
       context 'when the server auth mechanism is the default (mongodb_cr)', unless: scram_sha_1_enabled?  do
 
         it 'uses mongodb_cr' do
+          socket = connection.instance_variable_get(:@socket)
+          max_message_size = connection.send(:max_message_size)
+          allow(Mongo::Protocol::Reply).to receive(:deserialize).with(socket, max_message_size).and_return(reply)
           expect(connection.send(:default_mechanism)).to eq(:mongodb_cr)
         end
       end