Add replacement version of aiohappyeyeballs.start_connection

Author: bmerry

README.md

@@ -111,7 +111,14 @@ def event_loop():
     loop.close()
 ```
 
-### Integration with pytest-aiohttp
+### Integration with aiohappyeyeballs
+
+Unfortunately `aiohappyeyeballs.start_connection` doesn't work out of the box
+because it relies on creating a real socket. I've created a replacement
+`async_solipsism.aiohappyeyeballs_start_connection` that can be used to
+replace it. See the aiohttp section below for example code.
+
+### Integration with aiohttp
 
 A little extra work is required to work with aiohttp's test utilities, but it
 is possible. The example below requires at least aiohttp 3.8.0.
@@ -127,6 +134,12 @@ def event_loop_policy():
     return async_solipsism.EventLoopPolicy()
 
 
+@pytest.fixture(autouse=True)
+def mock_start_connection(monkeypatch):
+    monkeypatch.setattr("aiohappyeyeballs.start_connection",
+                        async_solipsism.aiohappyeyeballs_start_connection)
+
+
 def socket_factory(host, port, family):
     return async_solipsism.ListenSocket((host, port))
 
@@ -174,6 +187,10 @@ Calling functions that are not supported will generally raise
 
 ## Changelog
 
+### 0.7
+
+- Add a replacement function for `aiohappyeyeballs.start_connection`.
+
 ### 0.6
 
 - Drop support for Python 3.6 and 3.7, which have reached end of life.

pyproject.toml

@@ -11,7 +11,7 @@ authors = [
 ]
 requires-python = ">=3.8"
 classifiers = [
-    "Development Status :: 3 - Alpha",
+    "Development Status :: 5 - Production/Stable",
     "Framework :: AsyncIO",
     "License :: OSI Approved :: GNU General Public License v3 or later (GPLv3+)",
     "Topic :: Software Development :: Testing :: Mocking",

src/async_solipsism/__init__.py

@@ -19,4 +19,4 @@
 from .socket import *      # noqa
 from .loop import *        # noqa
 
-__version__ = '0.6'
+__version__ = '0.7'

src/async_solipsism/loop.py

@@ -1,4 +1,4 @@
-# Copyright 2020, 2022 Bruce Merry
+# Copyright 2020, 2022, 2024 Bruce Merry
 #
 # This file is part of async-solipsism.
 #
@@ -24,7 +24,7 @@
 from .exceptions import SolipsismError
 
 
-__all__ = ('EventLoop', 'EventLoopPolicy', 'stream_pairs')
+__all__ = ('EventLoop', 'EventLoopPolicy', 'aiohappyeyeballs_start_connection', 'stream_pairs')
 
 
 class EventLoop(asyncio.selector_events.BaseSelectorEventLoop):
@@ -133,6 +133,27 @@ async def create_server(
             **kwargs
         )
 
+    async def aiohappyeyeballs_start_connection(self,
+                                                addr_infos,
+                                                *,
+                                                local_addr_infos=None,
+                                                happy_eyeballs_delay=None,
+                                                interleave=None):
+        for addr_info in addr_infos:
+            addr = addr_info[4]
+            addr = _socket._normalise_ipv6_sockaddr(addr)
+            if addr in self.__listening_sockets:
+                listener = self.__listening_sockets[addr]
+                if local_addr_infos:
+                    local_addr_info = local_addr_infos[0]
+                else:
+                    port = self.__next_port
+                    self.__next_port += 1
+                    local_addr_info = ('::1', port, 0, 0)
+                sock = await listener.make_connection(local_addr_info)
+                return sock
+        raise ConnectionRefusedError('No socket listening on requested addresses')
+
     async def connect_read_pipe(self, protocol_factory, pipe):
         raise SolipsismError("connect_read_pipe is not supported")
 
@@ -207,3 +228,17 @@ async def stream_pairs(capacity=None):
     streams1 = await asyncio.open_connection(sock=sock1)
     streams2 = await asyncio.open_connection(sock=sock2)
     return streams1, streams2
+
+
+async def aiohappyeyeballs_start_connection(addr_infos,
+                                            *,
+                                            local_addr_infos=None,
+                                            happy_eyeballs_delay=None,
+                                            interleave=None,
+                                            loop=None):
+    if loop is None:
+        loop = asyncio.get_running_loop()
+    return await loop.aiohappyeyeballs_start_connection(addr_infos,
+                                                        local_addr_infos=local_addr_infos,
+                                                        happy_eyeballs_delay=happy_eyeballs_delay,
+                                                        interleave=interleave)

src/async_solipsism/socket.py

@@ -120,7 +120,7 @@ def gettimeout(self):
 
     def setblocking(self, flag):
         if flag:
-            raise SolipsismError('Socket only support non-blocking operation')
+            raise SolipsismError('Socket only supports non-blocking operation')
 
     def setsockopt(self, level, optname, value, optlen=None):
         key = (level, optname)