SNOW-3244317: Fix pytest session hang caused by pytest-rerunfailures socket server (#2807)

Author: sfc-gh-fpawlowski

DESCRIPTION.md

@@ -7,7 +7,7 @@ https://docs.snowflake.com/
 Source code is also available at: https://github.com/snowflakedb/snowflake-connector-python
 
 # Release Notes
-- v4.4.0(March 23,2026)
+- v4.4.0(March 24,2026)
   - Bump lower boundary of cryptography to 46.0.5 due to CVE-2026-26007.
   - Added support for Python 3.14.
   - Removed pyOpenSSL upper bound dependency constraint to allow installation of pyOpenSSL 26.0.0+, which includes a fix for GHSA-vp96-hxj8-p424.

test/conftest.py

@@ -2,6 +2,7 @@
 from __future__ import annotations
 
 import os
+import socket
 from contextlib import contextmanager
 from logging import getLogger
 from pathlib import Path
@@ -157,6 +158,27 @@ def pytest_runtest_setup(item) -> None:
             pytest.skip("Skipping WIF test in current environment")
 
 
+def pytest_unconfigure(config) -> None:
+    """Shut down the pytest-rerunfailures socket server to prevent session hangs.
+
+    The ServerStatusDB.run_server thread blocks forever on socket.accept()
+    with no shutdown mechanism. Closing the socket here unblocks it so the
+    interpreter can exit cleanly instead of deadlocking during teardown.
+    See: https://github.com/pytest-dev/pytest-rerunfailures/issues/295
+    """
+    db = getattr(config, "failures_db", None)
+    sock = getattr(db, "sock", None)
+    if sock is not None:
+        try:
+            sock.shutdown(socket.SHUT_RDWR)
+        except OSError:
+            pass
+        try:
+            sock.close()
+        except OSError:
+            pass
+
+
 def get_server_parameter_value(connection, parameter_name: str) -> str | None:
     """Get server parameter value, returns None if parameter doesn't exist."""
     try: