fix: test allow_hosts for unix sockets

pytest_socket.py

@@ -130,7 +130,7 @@ def _resolve_allow_hosts(item):
         hosts = mark_restrictions.args[0]
     elif cli_restrictions:
         hosts = cli_restrictions
-    socket_allow_hosts(hosts)
+    socket_allow_hosts(hosts, allow_unix_socket=item.config.__socket_allow_unix_socket)
     return hosts
 
 
@@ -151,7 +151,7 @@ def host_from_connect_args(args):
         return host_from_address(address)
 
 
-def socket_allow_hosts(allowed=None):
+def socket_allow_hosts(allowed=None, allow_unix_socket=False):
     """ disable socket.socket.connect() to disable the Internet. useful in testing.
     """
     if isinstance(allowed, str):
@@ -161,8 +161,17 @@ def socket_allow_hosts(allowed=None):
 
     def guarded_connect(inst, *args):
         host = host_from_connect_args(args)
-        if host and host in allowed:
+        if host in allowed:
             return _true_connect(inst, *args)
+
+        try:
+            is_unix_socket = inst.family == socket.AF_UNIX
+        except AttributeError:
+            # AF_UNIX not supported on Windows https://bugs.python.org/issue33408
+            is_unix_socket = False
+        if allow_unix_socket and is_unix_socket:
+            return _true_connect(inst, *args)
+
         raise SocketConnectBlockedError(allowed, host)
 
     socket.socket.connect = guarded_connect

tests/test_combinations.py

@@ -1,4 +1,6 @@
 """Test module to express odd combinations of configurations."""
+# TODO: Move this to a more common location, like `conftest.py`.
+from test_async import unix_sockets_only
 
 
 def test_parametrize_with_socket_enabled_and_allow_hosts(testdir, httpbin):
@@ -45,3 +47,31 @@ def test_remote_not_allowed_fails():
     result.stdout.fnmatch_lines(
         "*SocketConnectBlockedError: A test tried to use socket.socket.connect() with host*"
     )
+
+
+@unix_sockets_only
+def test_combine_unix_and_allow_hosts(testdir, httpbin):
+    """Test combination of disable, allow-unix and allow-hosts.
+
+    From https://github.com/miketheman/pytest-socket/issues/78
+    """
+    testdir.makepyfile(
+        f"""
+        import socket
+
+        import pytest
+
+
+        def test_unix_connect():
+            sock = socket.socket(socket.AF_UNIX, socket.SOCK_STREAM)
+            with pytest.raises(FileNotFoundError):
+                sock.connect('/tmp/socket.sock')
+
+
+        def test_inet_connect():
+            sock = socket.socket(socket.AF_INET, socket.SOCK_STREAM)
+            sock.connect(('{httpbin.host}', {httpbin.port}))
+        """
+    )
+    result = testdir.runpytest("--disable-socket", "--allow-unix-socket", f"--allow-hosts={httpbin.host}")
+    result.assert_outcomes(passed=2)