From eb1afa4245e32f6bf806f8a84817d136d9ca7fa6 Mon Sep 17 00:00:00 2001
From: Bascht74 <66269310+Bascht74@users.noreply.github.com>
Date: Sat, 10 Oct 2026 20:15:30 +0200
Subject: [PATCH] [tests] Close leftover server connections in the tcp_uart
 bridge and TCP client link tests (#20215)

Co-authored-by: J. Nick Koston <nick@koston.org>
---
 tests/integration/test_socket_tcp_client_link.py | 6 ++++++
 tests/integration/test_tcp_uart_bridge.py        | 6 ++++++
 2 files changed, 12 insertions(+)

diff --git a/tests/integration/test_socket_tcp_client_link.py b/tests/integration/test_socket_tcp_client_link.py
index 16b5f544f7..89a6ff4234 100644
--- a/tests/integration/test_socket_tcp_client_link.py
+++ b/tests/integration/test_socket_tcp_client_link.py
@@ -35,6 +35,7 @@ async def test_socket_tcp_client_link(
     link_down = asyncio.Event()
     second_link_up = asyncio.Event()
     link_up_count = 0
+    writers: list[asyncio.StreamWriter] = []
 
     def on_log_line(line: str) -> None:
         nonlocal link_up_count
@@ -48,6 +49,7 @@ async def test_socket_tcp_client_link(
     async def handle(
         reader: asyncio.StreamReader, writer: asyncio.StreamWriter
     ) -> None:
+        writers.append(writer)
         if not echo_done.is_set():
             writer.write(PAYLOAD)
             await writer.drain()
@@ -98,5 +100,9 @@ async def test_socket_tcp_client_link(
                 "Second session echo wrong; stale bytes from the first session?"
             )
     finally:
+        # On Python 3.12 wait_closed() waits for every connection the handlers
+        # left open; nothing else closes them once the device has exited.
+        for writer in writers:
+            writer.close()
         server.close()
         await server.wait_closed()
diff --git a/tests/integration/test_tcp_uart_bridge.py b/tests/integration/test_tcp_uart_bridge.py
index fb22a26b55..70befc9049 100644
--- a/tests/integration/test_tcp_uart_bridge.py
+++ b/tests/integration/test_tcp_uart_bridge.py
@@ -41,12 +41,14 @@ async def test_tcp_uart_bridge(
     service_byte_ok = asyncio.Event()
     reconnect_echo_ok = asyncio.Event()
     sessions = 0
+    writers: list[asyncio.StreamWriter] = []
 
     async def handle(
         reader: asyncio.StreamReader, writer: asyncio.StreamWriter
     ) -> None:
         nonlocal sessions
         sessions += 1
+        writers.append(writer)
         if sessions == 1:
             writer.write(GREETING)
             await writer.drain()
@@ -98,5 +100,9 @@ async def test_tcp_uart_bridge(
                 reconnect_echo_ok, 15.0, "Bridge did not reconnect and echo again"
             )
     finally:
+        # On Python 3.12 wait_closed() waits for every connection the handlers
+        # left open; nothing else closes them once the device has exited.
+        for writer in writers:
+            writer.close()
         server.close()
         await server.wait_closed()