FIXUP: expose AddrInfoType

aio-libs · Mar 11, 2025 · 3a87c34 · 3a87c34
1 parent 057ecc9
commit 3a87c34
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 2 deletions.
diff --git a/aiohttp/__init__.py b/aiohttp/__init__.py
@@ -112,6 +112,7 @@
 __all__: Tuple[str, ...] = (
     "hdrs",
     # client
+    "AddrInfoType",
     "BaseConnector",
     "ClientConnectionError",
     "ClientConnectionResetError",
@@ -146,6 +147,7 @@
     "ServerDisconnectedError",
     "ServerFingerprintMismatch",
     "ServerTimeoutError",
+    "SocketFactoryType",
     "SocketTimeoutError",
     "TCPConnector",
     "TooManyRedirects",

diff --git a/aiohttp/connector.py b/aiohttp/connector.py
@@ -96,7 +96,14 @@
 # which first appeared in Python 3.12.7 and 3.13.1
 
 
-__all__ = ("BaseConnector", "TCPConnector", "UnixConnector", "NamedPipeConnector")
+__all__ = (
+    "BaseConnector",
+    "TCPConnector",
+    "UnixConnector",
+    "NamedPipeConnector",
+    "AddrInfoType",
+    "SocketFactoryType",
+)
 
 
 if TYPE_CHECKING:

diff --git a/tests/test_connector.py b/tests/test_connector.py
@@ -26,7 +26,6 @@
 from unittest import mock
 
 import pytest
-from aiohappyeyeballs import AddrInfoType
 from pytest_mock import MockerFixture
 from yarl import URL
 
@@ -44,6 +43,7 @@
 from aiohttp.connector import (
     _SSL_CONTEXT_UNVERIFIED,
     _SSL_CONTEXT_VERIFIED,
+    AddrInfoType,
     Connection,
     TCPConnector,
     _DNSCacheTable,