Change a test behavior for unix_connect.get_socket

i2y · vasily-v-ryabov · commit 1127b311b3dc · 2022-12-25T21:28:24.000+03:00
diff --git a/test/test_unix_connect.py b/test/test_unix_connect.py
@@ -72,8 +72,8 @@ def _get_socket(socket_type, raises, *params):
         def path_exists(returns, path):
             calls.append(('os.path.exists', path))
             return returns
-        def fcntl(*args):
-            calls.append(('fcntl',) + args)
+        def ensure_not_inheritable(*args):
+            calls.append(('ensure_not_inheritable',) + args)
         for params, allow_unix, unix_addr_exists, allow_tcp, expect_connection_error, expected_calls in (
             # Successful explicit TCP socket connection.
             (('tcp/host:6', None, 'host', 6), False, False, True, False, [
@@ -141,17 +141,15 @@ def fcntl(*args):
                           partial(_get_socket, 'tcp', not allow_tcp)), \
                     patch('os.path.exists',
                           partial(path_exists, unix_addr_exists)), \
-                    patch('fcntl.fcntl', fcntl):
+                    patch('Xlib.support.unix_connect._ensure_not_inheritable', ensure_not_inheritable):
                 del calls[:]
                 if expect_connection_error:
                     with self.assertRaises(DisplayConnectionError):
                         unix_connect.get_socket(*params)
                 else:
                     s = unix_connect.get_socket(*params)
                     self.assertIsInstance(s, FakeSocket)
-                    expected_calls.append(('fcntl', 42,
-                                           unix_connect.F_SETFD,
-                                           unix_connect.FD_CLOEXEC))
+                    expected_calls.append(('ensure_not_inheritable', s))
                 self.assertEqual(calls, expected_calls)
 
 
