import os
from unittest.mock import patch

import pytest

from app.infrastructure.database.connection import init_db


class TestInitDB:
    @pytest.mark.asyncio
    async def test_init_db_skipped_when_skip_env_set(self) -> None:
        with patch.dict(os.environ, {"SKIP_INIT_DB": "1"}):
            await init_db()

    @pytest.mark.asyncio
    async def test_init_db_runs_when_no_env(self) -> None:
        with patch.dict(os.environ, {}, clear=True):
            await init_db()