Patch httpretty socket for latest urllib3 release

PyGithub · Dec 28, 2024 · 10a7135 · 10a7135
1 parent 3d84a47
commit 10a7135
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/Framework.py b/tests/Framework.py
@@ -62,6 +62,8 @@
 from typing import Optional
 
 import httpretty  # type: ignore
+import urllib3
+from packaging.version import Version
 from requests.structures import CaseInsensitiveDict
 from urllib3.util import Url  # type: ignore
 
@@ -87,6 +89,22 @@
 """
 
 
+# patch httpretty against urllib3>=2.3.0 https://github.com/PyGithub/PyGithub/issues/3101
+if Version(urllib3.__version__) >= Version("2.3.0"):
+    getattr = httpretty.core.fakesock.socket.__getattr__
+
+    def patched_getattr(self, name):
+        def shutdown(how: int):
+            pass
+
+        if name == "shutdown" and not httpretty.core.httpretty.allow_net_connect and not self.truesock:
+            return shutdown
+
+        return getattr(self, name)
+
+    httpretty.core.fakesock.socket.__getattr__ = patched_getattr
+
+
 def readLine(file_):
     line = file_.readline()
     if isinstance(line, bytes):