gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: add asyncio mark to neo4j deadlock tests

Browse source
This commit is contained in:
Boris Arzentar 2025-04-30 15:45:15 +02:00
parent c5d5c5578a
commit f999232b8e
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/tests/unit/infrastructure/databases/graph/neo4j_deadlock_test.py
View file

@ -5,6 +5,7 @@ from neo4j.exceptions import Neo4jError
from cognee.infrastructure.databases.graph.neo4j_driver.deadlock_retry import deadlock_retry from cognee.infrastructure.databases.graph.neo4j_driver.deadlock_retry import deadlock_retry
@pytest.mark.asyncio
async def test_deadlock_retry_errored(): async def test_deadlock_retry_errored():
mock_return = asyncio.Future() mock_return = asyncio.Future()
mock_return.set_result(True) mock_return.set_result(True)
@ -18,6 +19,7 @@ async def test_deadlock_retry_errored():
await wrapped_function(self=None) await wrapped_function(self=None)
@pytest.mark.asyncio
async def test_deadlock_retry(): async def test_deadlock_retry():
mock_return = asyncio.Future() mock_return = asyncio.Future()
mock_return.set_result(True) mock_return.set_result(True)
@ -29,6 +31,7 @@ async def test_deadlock_retry():
assert result, "Function should have succeded on second time" assert result, "Function should have succeded on second time"
@pytest.mark.asyncio
async def test_deadlock_retry_exhaustive(): async def test_deadlock_retry_exhaustive():
mock_return = asyncio.Future() mock_return = asyncio.Future()
mock_return.set_result(True) mock_return.set_result(True)