r2rpy/py/tests/integration/test_base.py

from typing import Optional

from r2r import R2RException


class BaseTest:
    """Base class for all test classes with common utilities."""

    @staticmethod
    async def cleanup_resource(cleanup_func,
                               resource_id: Optional[str] = None) -> None:
        """Generic cleanup helper that won't fail the test if cleanup fails."""
        if resource_id:
            try:
                await cleanup_func(id=resource_id)
            except R2RException:
                pass