mirror of
https://github.com/python/cpython.git
synced 2025-07-28 21:55:21 +00:00
[3.13] gh-122133: Rework pure Python socketpair tests to avoid use of importlib.reload. (GH-122493) (#122504)
gh-122133: Rework pure Python socketpair tests to avoid use of importlib.reload. (GH-122493)
(cherry picked from commit f071f01b7b
)
Co-authored-by: Russell Keith-Magee <russell@keith-magee.com>
Co-authored-by: Gregory P. Smith <greg@krypto.org>
This commit is contained in:
parent
4a6365c5f4
commit
c21a36112a
2 changed files with 64 additions and 77 deletions
|
@ -4861,7 +4861,6 @@ class BasicSocketPairTest(SocketPairTest):
|
|||
|
||||
|
||||
class PurePythonSocketPairTest(SocketPairTest):
|
||||
|
||||
# Explicitly use socketpair AF_INET or AF_INET6 to ensure that is the
|
||||
# code path we're using regardless platform is the pure python one where
|
||||
# `_socket.socketpair` does not exist. (AF_INET does not work with
|
||||
|
@ -4876,28 +4875,21 @@ class PurePythonSocketPairTest(SocketPairTest):
|
|||
# Local imports in this class make for easy security fix backporting.
|
||||
|
||||
def setUp(self):
|
||||
import _socket
|
||||
self._orig_sp = getattr(_socket, 'socketpair', None)
|
||||
if self._orig_sp is not None:
|
||||
if hasattr(_socket, "socketpair"):
|
||||
self._orig_sp = socket.socketpair
|
||||
# This forces the version using the non-OS provided socketpair
|
||||
# emulation via an AF_INET socket in Lib/socket.py.
|
||||
del _socket.socketpair
|
||||
import importlib
|
||||
global socket
|
||||
socket = importlib.reload(socket)
|
||||
socket.socketpair = socket._fallback_socketpair
|
||||
else:
|
||||
pass # This platform already uses the non-OS provided version.
|
||||
# This platform already uses the non-OS provided version.
|
||||
self._orig_sp = None
|
||||
super().setUp()
|
||||
|
||||
def tearDown(self):
|
||||
super().tearDown()
|
||||
import _socket
|
||||
if self._orig_sp is not None:
|
||||
# Restore the default socket.socketpair definition.
|
||||
_socket.socketpair = self._orig_sp
|
||||
import importlib
|
||||
global socket
|
||||
socket = importlib.reload(socket)
|
||||
socket.socketpair = self._orig_sp
|
||||
|
||||
def test_recv(self):
|
||||
msg = self.serv.recv(1024)
|
||||
|
|
Loading…
Add table
Add a link
Reference in a new issue