From d5068b240c4151bfbf31c2d3354edbac9c3ff8f4 Mon Sep 17 00:00:00 2001
From: Cyber MacGeddon <cybermaggedon@gmail.com>
Date: Tue, 6 Jan 2026 22:01:42 +0000
Subject: [PATCH] Fix tracemalloc async warnings

---
 tests/conftest.py                             | 13 ++++++++++
 .../test_socket_graceful_shutdown.py          | 24 +++++++++++++++----
 2 files changed, 33 insertions(+), 4 deletions(-)

diff --git a/tests/conftest.py b/tests/conftest.py
index e1b98cd5..ccf00ddb 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -5,8 +5,21 @@ This conftest.py applies to all test directories.
 """
 
 import pytest
+import asyncio
+import tracemalloc
+import warnings
 from unittest.mock import MagicMock
 
+# Enable tracemalloc immediately at import time
+tracemalloc.start()
+
+# Enable asyncio debug mode
+asyncio.set_event_loop_policy(asyncio.DefaultEventLoopPolicy())
+
+# Make warnings verbose
+warnings.simplefilter("always", ResourceWarning)
+warnings.simplefilter("always", RuntimeWarning)
+
 
 @pytest.fixture(scope="session", autouse=True)
 def mock_loki_handler(session_mocker=None):
diff --git a/tests/unit/test_gateway/test_socket_graceful_shutdown.py b/tests/unit/test_gateway/test_socket_graceful_shutdown.py
index 50195272..1a63227d 100644
--- a/tests/unit/test_gateway/test_socket_graceful_shutdown.py
+++ b/tests/unit/test_gateway/test_socket_graceful_shutdown.py
@@ -132,6 +132,8 @@ async def test_handle_normal_flow():
 
             # Create proper mock tasks that look like asyncio.Task objects
             def create_task_mock(coro):
+                # Consume the coroutine to avoid "was never awaited" warning
+                coro.close()
                 task = AsyncMock()
                 task.done = MagicMock(return_value=True)
                 task.cancelled = MagicMock(return_value=False)
@@ -187,6 +189,8 @@ async def test_handle_exception_group_cleanup():
 
             # Create proper mock tasks that look like asyncio.Task objects
             def create_task_mock(coro):
+                # Consume the coroutine to avoid "was never awaited" warning
+                coro.close()
                 task = AsyncMock()
                 task.done = MagicMock(return_value=True)
                 task.cancelled = MagicMock(return_value=False)
@@ -195,8 +199,12 @@ async def test_handle_exception_group_cleanup():
             mock_tg.create_task = MagicMock(side_effect=create_task_mock)
             mock_task_group.return_value = mock_tg
 
-            with patch('trustgraph.gateway.endpoint.socket.asyncio.wait_for') as mock_wait_for:
-                mock_wait_for.return_value = None
+            with patch('trustgraph.gateway.endpoint.socket.asyncio.wait_for', new_callable=AsyncMock) as mock_wait_for:
+                # Make wait_for consume the coroutine passed to it
+                async def wait_for_side_effect(coro, timeout=None):
+                    coro.close()  # Consume the coroutine
+                    return None
+                mock_wait_for.side_effect = wait_for_side_effect
 
                 result = await socket_endpoint.handle(request)
 
@@ -246,6 +254,8 @@ async def test_handle_dispatcher_cleanup_timeout():
 
             # Create proper mock tasks that look like asyncio.Task objects
             def create_task_mock(coro):
+                # Consume the coroutine to avoid "was never awaited" warning
+                coro.close()
                 task = AsyncMock()
                 task.done = MagicMock(return_value=True)
                 task.cancelled = MagicMock(return_value=False)
@@ -255,8 +265,12 @@ async def test_handle_dispatcher_cleanup_timeout():
             mock_task_group.return_value = mock_tg
 
             # Mock asyncio.wait_for to raise TimeoutError
-            with patch('trustgraph.gateway.endpoint.socket.asyncio.wait_for') as mock_wait_for:
-                mock_wait_for.side_effect = asyncio.TimeoutError("Cleanup timeout")
+            with patch('trustgraph.gateway.endpoint.socket.asyncio.wait_for', new_callable=AsyncMock) as mock_wait_for:
+                # Make wait_for consume the coroutine before raising
+                async def wait_for_timeout(coro, timeout=None):
+                    coro.close()  # Consume the coroutine
+                    raise asyncio.TimeoutError("Cleanup timeout")
+                mock_wait_for.side_effect = wait_for_timeout
 
                 result = await socket_endpoint.handle(request)
 
@@ -341,6 +355,8 @@ async def test_handle_websocket_already_closed():
 
             # Create proper mock tasks that look like asyncio.Task objects
             def create_task_mock(coro):
+                # Consume the coroutine to avoid "was never awaited" warning
+                coro.close()
                 task = AsyncMock()
                 task.done = MagicMock(return_value=True)
                 task.cancelled = MagicMock(return_value=False)