fix tests

ydb-platform · Sep 26, 2024 · 2534b70 · 2534b70
1 parent d489a62
commit 2534b70
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/ydb/_topic_common/common.py b/ydb/_topic_common/common.py
@@ -31,10 +31,10 @@ def wrapper(rpc_state, response_pb, driver=None):
 
 if sys.hexversion < 0x03080000:
     def wrap_set_name_for_asyncio_task(task: asyncio.Task, task_name: str) -> asyncio.Task:
-        task.set_name(task_name)
         return task
 else:
     def wrap_set_name_for_asyncio_task(task: asyncio.Task, task_name: str) -> asyncio.Task:
+        task.set_name(task_name)
         return task
 
 

diff --git a/ydb/_topic_common/common_test.py b/ydb/_topic_common/common_test.py
@@ -87,8 +87,7 @@ async def some_async_task():
         asyncio_task = loop.create_task(some_async_task())
         wrap_set_name_for_asyncio_task(asyncio_task, task_name=task_name)
 
-        if sys.hexversion >= 0x03080000:
-            assert asyncio_task.get_name() == task_name
+        assert asyncio_task.get_name() == task_name