From 5b86fd384caef972c21d9bf76984ea6a8d54f09f Mon Sep 17 00:00:00 2001 From: "stone-24tch3r (aider)" <100294019+stone-w4tch3r@users.noreply.github.com> Date: Sun, 22 Sep 2024 22:43:05 +0500 Subject: [PATCH] refactor: Update test class to use IsolatedAsyncioTestCase --- tests/test_trace_id_handler.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/tests/test_trace_id_handler.py b/tests/test_trace_id_handler.py index f6ae8b0..20e04ff 100644 --- a/tests/test_trace_id_handler.py +++ b/tests/test_trace_id_handler.py @@ -1,11 +1,10 @@ import unittest import uuid -import asyncio from unittest.mock import patch from core.trace_id_handler import TraceIdHandler -class TestTraceIdHandler(unittest.TestCase): +class TestTraceIdHandler(unittest.IsolatedAsyncioTestCase): async def test_logging_scope(self): async with TraceIdHandler.logging_scope(): @@ -24,9 +23,9 @@ class TestTraceIdHandler(unittest.TestCase): self.assertEqual(TraceIdHandler.get_current_trace_id(), mock_uuid) self.assertEqual(TraceIdHandler.get_current_trace_id(), mock_uuid) # Should return the same UUID - def test_get_current_trace_id_outside_scope(self): + async def test_get_current_trace_id_outside_scope(self): self.assertIsNone(TraceIdHandler.get_current_trace_id()) if __name__ == '__main__': - asyncio.run(unittest.main()) + unittest.main()