Avoid the multiprocessing forkserver method

Author: hroncok

tests/test_multiprocessing.py

@@ -1,6 +1,5 @@
 import contextlib
 import multiprocessing
-import sys
 
 import pytest
 import redis
@@ -9,9 +8,6 @@
 
 from .conftest import _get_client
 
-if sys.platform == "darwin":
-    multiprocessing.set_start_method("fork", force=True)
-
 
 @contextlib.contextmanager
 def exit_callback(callback, *args):
@@ -22,6 +18,17 @@ def exit_callback(callback, *args):
 
 
 class TestMultiprocessing:
+
+    # On macOS and newly non-macOS POSIX systems (since Python 3.14),
+    # the default method has been changed to forkserver.
+    # The code in this module does not work with it,
+    # hence the explicit change to 'fork'
+    # See https://github.com/python/cpython/issues/125714
+    if multiprocessing.get_start_method() == "forkserver":
+        _mp_context = multiprocessing.get_context(method="fork")
+    else:
+        _mp_context = multiprocessing.get_context()
+
     # Test connection sharing between forks.
     # See issue #1085 for details.
 
@@ -45,7 +52,7 @@ def target(conn):
             assert conn.read_response() == b"PONG"
             conn.disconnect()
 
-        proc = multiprocessing.Process(target=target, args=(conn,))
+        proc = self._mp_context.Process(target=target, args=(conn,))
         proc.start()
         proc.join(3)
         assert proc.exitcode == 0
@@ -75,7 +82,7 @@ def target(conn, ev):
                 conn.send_command("ping")
 
         ev = multiprocessing.Event()
-        proc = multiprocessing.Process(target=target, args=(conn, ev))
+        proc = self._mp_context.Process(target=target, args=(conn, ev))
         proc.start()
 
         conn.disconnect()
@@ -143,7 +150,7 @@ def target(pool):
                     assert conn.send_command("ping") is None
                     assert conn.read_response() == b"PONG"
 
-        proc = multiprocessing.Process(target=target, args=(pool,))
+        proc = self._mp_context.Process(target=target, args=(pool,))
         proc.start()
         proc.join(3)
         assert proc.exitcode == 0
@@ -181,7 +188,7 @@ def target(pool, disconnect_event):
 
         ev = multiprocessing.Event()
 
-        proc = multiprocessing.Process(target=target, args=(pool, ev))
+        proc = self._mp_context.Process(target=target, args=(pool, ev))
         proc.start()
 
         pool.disconnect()
@@ -197,7 +204,7 @@ def target(client):
             assert client.ping() is True
             del client
 
-        proc = multiprocessing.Process(target=target, args=(r,))
+        proc = self._mp_context.Process(target=target, args=(r,))
         proc.start()
         proc.join(3)
         assert proc.exitcode == 0