fix: ensure cancellation is not swallowed on connect failure and in r…

…esponse handlers (#57)

src/aioharmony/hubconnector_websocket.py

@@ -7,6 +7,7 @@
 import asyncio
 import logging
 import socket
+import sys
 from contextlib import suppress
 from typing import Optional, Union
 from urllib.parse import urlparse
@@ -399,6 +400,12 @@ async def _listener(self, websocket=None) -> None:
 
             except asyncio.CancelledError:
                 _LOGGER.debug("%s: Received STOP for listener", self._ip_address)
+                if (
+                    sys.version_info >= (3, 11)
+                    and (task := asyncio.current_task())
+                    and task.cancelling()
+                ):
+                    raise
                 break
 
             # pylint: disable=broad-except

src/aioharmony/hubconnector_xmpp.py

@@ -7,6 +7,7 @@
 import asyncio
 import json
 import logging
+import sys
 from typing import Optional
 from uuid import uuid4
 
@@ -215,6 +216,12 @@ def remove_handlers():
                 )
                 # Remove the handlers.
                 remove_handlers()
+                if (
+                    sys.version_info >= (3, 11)
+                    and (task := asyncio.current_task())
+                    and task.cancelling()
+                ):
+                    raise
                 return False
             except OSError as exc:
                 _LOGGER.log(

src/aioharmony/responsehandler.py

@@ -8,6 +8,7 @@
 
 import asyncio
 import logging
+import sys
 from datetime import datetime, timedelta, timezone
 from re import Pattern
 from typing import NamedTuple, Optional, Union
@@ -312,6 +313,12 @@ async def _callback_handler(self) -> None:
 
             except asyncio.CancelledError:
                 _LOGGER.debug("%s: Received STOP for callback handler", self._name)
+                if (
+                    sys.version_info >= (3, 11)
+                    and (task := asyncio.current_task())
+                    and task.cancelling()
+                ):
+                    raise
                 break
 
             # Need to catch everything here to prevent an issue in a