From f999232b8e6d053e4fd26a1c58668d976830eac8 Mon Sep 17 00:00:00 2001 From: Boris Arzentar Date: Wed, 30 Apr 2025 15:45:15 +0200 Subject: [PATCH] fix: add asyncio mark to neo4j deadlock tests --- .../unit/infrastructure/databases/graph/neo4j_deadlock_test.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/cognee/tests/unit/infrastructure/databases/graph/neo4j_deadlock_test.py b/cognee/tests/unit/infrastructure/databases/graph/neo4j_deadlock_test.py index a1e2d5bae..01043602d 100644 --- a/cognee/tests/unit/infrastructure/databases/graph/neo4j_deadlock_test.py +++ b/cognee/tests/unit/infrastructure/databases/graph/neo4j_deadlock_test.py @@ -5,6 +5,7 @@ from neo4j.exceptions import Neo4jError from cognee.infrastructure.databases.graph.neo4j_driver.deadlock_retry import deadlock_retry +@pytest.mark.asyncio async def test_deadlock_retry_errored(): mock_return = asyncio.Future() mock_return.set_result(True) @@ -18,6 +19,7 @@ async def test_deadlock_retry_errored(): await wrapped_function(self=None) +@pytest.mark.asyncio async def test_deadlock_retry(): mock_return = asyncio.Future() mock_return.set_result(True) @@ -29,6 +31,7 @@ async def test_deadlock_retry(): assert result, "Function should have succeded on second time" +@pytest.mark.asyncio async def test_deadlock_retry_exhaustive(): mock_return = asyncio.Future() mock_return.set_result(True)