PYTHON-1235 - Fix auto reconnect test under Jython 2.7

behackett · behackett · commit 9e123e3c111b · 2017-02-06T13:38:11.000-08:00
diff --git a/test/test_replica_set_client.py b/test/test_replica_set_client.py
@@ -382,14 +382,19 @@ def test_auto_reconnect_exception_when_read_preference_is_secondary(self):
         def raise_socket_error(*args, **kwargs):
             raise socket.error
 
-        old_sendall = socket.socket.sendall
-        socket.socket.sendall = raise_socket_error
+        # In Jython 2.7 socket.socket is a function, not a class.
+        sock = socket.socket()
+        klass = sock.__class__
+        old_sendall = klass.sendall
+        klass.sendall = raise_socket_error
 
         try:
             cursor = db.test.find(read_preference=ReadPreference.SECONDARY)
             self.assertRaises(AutoReconnect, cursor.next)
         finally:
-            socket.socket.sendall = old_sendall
+            klass.sendall = old_sendall
+            # Silence resource warnings.
+            sock.close()
 
     def test_operations(self):
         c = self._get_client()
