diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py index 253ae7a6119..3bdab1d231a 100644 --- a/tests/unit/conftest.py +++ b/tests/unit/conftest.py @@ -1,16 +1,23 @@ from collections.abc import Iterator +from typing import Final import pytest from pytest_socket import enable_socket, socket_allow_hosts +LOOPBACK_HOSTS: Final = ["127.0.0.1", "::1"] + + +def _allow_loopback_only() -> None: + socket_allow_hosts(LOOPBACK_HOSTS, allow_unix_socket=True) + @pytest.fixture(autouse=True, scope="session") def block_external_sockets() -> Iterator[None]: - socket_allow_hosts(["127.0.0.1", "::1"], allow_unix_socket=True) + _allow_loopback_only() yield enable_socket() @pytest.hookimpl(trylast=True) def pytest_runtest_setup() -> None: - socket_allow_hosts(["127.0.0.1", "::1"], allow_unix_socket=True) + _allow_loopback_only()