fix: pilot-db tests

Author: aldbr

diracx-db/tests/pilots/test_pilot_management.py

@@ -3,7 +3,6 @@
 from datetime import datetime, timezone
 
 import pytest
-import pytest_asyncio
 
 from diracx.core.exceptions import (
     PilotAlreadyAssociatedWithJobError,
@@ -17,7 +16,7 @@
 MAIN_VO = "lhcb"
 
 
-@pytest_asyncio.fixture
+@pytest.fixture
 async def pilot_db():
     agents_db = PilotAgentsDB("sqlite+aiosqlite:///:memory:")
     async with agents_db.engine_context():
@@ -26,7 +25,6 @@ async def pilot_db():
         yield agents_db
 
 
-@pytest.mark.asyncio
 async def test_register_pilots_roundtrip(pilot_db: PilotAgentsDB):
     """Register a pilot and read it back with its defaults."""
     async with pilot_db as db:
@@ -47,7 +45,6 @@ async def test_register_pilots_roundtrip(pilot_db: PilotAgentsDB):
         assert not pilot["AccountingSent"]
 
 
-@pytest.mark.asyncio
 async def test_update_pilot_metadata_heterogeneous(pilot_db: PilotAgentsDB):
     """A single call must support rows that set different field subsets.
 
@@ -78,7 +75,6 @@ async def test_update_pilot_metadata_heterogeneous(pilot_db: PilotAgentsDB):
         assert rows["stamp-2"]["BenchMark"] == 42.0
 
 
-@pytest.mark.asyncio
 async def test_update_pilot_metadata_missing_pilot_raises(pilot_db: PilotAgentsDB):
     async with pilot_db as db:
         with pytest.raises(PilotNotFoundError):
@@ -87,7 +83,6 @@ async def test_update_pilot_metadata_missing_pilot_raises(pilot_db: PilotAgentsD
             )
 
 
-@pytest.mark.asyncio
 async def test_assign_jobs_to_pilot_duplicate_raises(pilot_db: PilotAgentsDB):
     """Second assignment of the same (pilot, job) pair must raise."""
     async with pilot_db as db:
@@ -109,7 +104,6 @@ async def test_assign_jobs_to_pilot_duplicate_raises(pilot_db: PilotAgentsDB):
             )
 
 
-@pytest.mark.asyncio
 async def test_delete_cascades_logs_and_mappings(pilot_db: PilotAgentsDB):
     """Deleting a pilot removes its logs and job associations."""
     async with pilot_db as db:

diracx-db/tests/pilots/test_query.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import pytest
-import pytest_asyncio
 
 from diracx.core.models.pilot import PilotMetadata, PilotStatus
 from diracx.core.models.search import (
@@ -18,7 +17,7 @@
 N = 20
 
 
-@pytest_asyncio.fixture
+@pytest.fixture
 async def pilot_db():
     agents_db = PilotAgentsDB("sqlite+aiosqlite:///:memory:")
     async with agents_db.engine_context():
@@ -27,7 +26,7 @@ async def pilot_db():
         yield agents_db
 
 
-@pytest_asyncio.fixture
+@pytest.fixture
 async def populated_pilot_db(pilot_db):
     """Small, focused dataset: N pilots with alternating statuses."""
     stamps = [f"stamp_{i + 1}" for i in range(N)]
@@ -51,7 +50,6 @@ async def populated_pilot_db(pilot_db):
     yield pilot_db
 
 
-@pytest.mark.asyncio
 async def test_search_smoke(populated_pilot_db):
     """Smoke-test the generic search helpers through the pilots table.
 
@@ -102,7 +100,6 @@ async def test_search_smoke(populated_pilot_db):
         assert [r["PilotID"] for r in result][0] == N
 
 
-@pytest.mark.asyncio
 async def test_pilot_summary_groups_by_status(populated_pilot_db):
     """`pilot_summary` must aggregate by the requested column."""
     async with populated_pilot_db as db: