@pytest.mark.asyncio
async def test_storage_setup():
    """Test that storage setup creates necessary directories and database."""
    # Ensure clean directory
    if os.path.exists(TEST_DB_DIR):
        shutil.rmtree(TEST_DB_DIR)

    storage = LocalStorage(TEST_DB_URI)
    await storage.setup()

    # Check that dirs exist
    assert os.path.exists(os.path.join(TEST_DB_DIR, "configs"))
    assert os.path.exists(os.path.join(TEST_DB_DIR, "data"))
    assert os.path.exists(os.path.join(TEST_DB_DIR, "metadata.db"))

    await storage.close()
    shutil.rmtree(TEST_DB_DIR)

class FileSystemDataHandler:
    """Manages interactions with data/config files on the local filesystem."""

    def __init__(self, data_base_path: str):
        """Args:
        data_base_path: The root directory where configs/, data/, etc. will live.
        """
        self.data_base_path = data_base_path
        self.config_path = os.path.join(self.data_base_path, CONFIGS_DIR)
        self.record_data_path = os.path.join(self.data_base_path, DATA_DIR)
        self.assoc_path = os.path.join(self.data_base_path, ASSOCIATIONS_DIR)
        # TODO: Consider locks if implementing JSONL appends for associations

    async def ensure_base_dirs(self):
        """Ensure all top-level data directories exist."""
        logger.debug("Ensuring base data directories exist...")
        await self._ensure_dir(self.config_path + os.sep)  # Trailing sep ensures dir
        await self._ensure_dir(self.record_data_path + os.sep)
        # TODO: Add associations directory back later
        # await self._ensure_dir(self.assoc_path + os.sep)

    async def _ensure_dir(self, path: str):
        """Asynchronously ensures a directory exists."""
        dir_path = os.path.dirname(path)
        if not await aio_os.path.isdir(dir_path):
            try:
                await aio_os.makedirs(dir_path, exist_ok=True)
                logger.debug(f"Created directory: {dir_path}")
            except Exception as e:
                if not await aio_os.path.isdir(dir_path):
                    # Race condition
                    logger.warning(f"Directory creation failed for {dir_path} due to a race condition.")
                    raise

# Test database location - can be overridden with env var
TEST_DB_DIR = os.environ.get("STARFISH_TEST_DB_DIR", "/tmp/starfish_test_db")
TEST_DB_URI = f"file://{TEST_DB_DIR}"

# Test mode - 'basic' (quick test) or 'full' (comprehensive)
TEST_MODE = os.environ.get("STARFISH_TEST_MODE", "basic")