from __future__ import annotations

from unittest.mock import AsyncMock, MagicMock, patch

import pytest

from yuxi.channel.startup import get_container, init_channel, register_channel_middleware, shutdown_channel


class TestInitChannel:
    @pytest.mark.asyncio
    async def test_init_channel_sets_global_container(self, tmp_path):
        config_file = tmp_path / "channel_config.yaml"
        config_file.write_text("{}")

        mock_container = MagicMock()
        mock_container.shutdown = AsyncMock()

        with patch("yuxi.channel.startup.ChannelContainerFactory.create", new_callable=AsyncMock, return_value=mock_container):
            container = await init_channel(
                redis_url="redis://localhost:6379/0",
                config_yaml_path=str(config_file),
            )

        assert container is mock_container
        assert get_container() is mock_container

        await shutdown_channel()


class TestShutdownChannel:
    @pytest.mark.asyncio
    async def test_shutdown_channel_calls_shutdown(self):
        mock_container = MagicMock()
        mock_container.shutdown = AsyncMock()

        with patch("yuxi.channel.startup.ChannelContainerFactory.create", new_callable=AsyncMock, return_value=mock_container):
            await init_channel(redis_url="redis://localhost:6379/0")

        await shutdown_channel()

        mock_container.shutdown.assert_awaited_once()
        assert get_container() is None

    @pytest.mark.asyncio
    async def test_shutdown_channel_with_provided_container(self):
        mock_container = MagicMock()
        mock_container.shutdown = AsyncMock()

        await shutdown_channel(mock_container)

        mock_container.shutdown.assert_awaited_once()

    @pytest.mark.asyncio
    async def test_shutdown_channel_handles_none(self):
        await shutdown_channel(None)


class TestGetContainer:
    def test_get_container_returns_none_initially(self):
        assert get_container() is None


class TestRegisterChannelMiddleware:
    def test_register_channel_middleware_sets_state(self):
        mock_container = MagicMock()

        with patch("yuxi.channel.startup._container", mock_container):
            app = MagicMock()
            register_channel_middleware(app)

            assert app.state.channel is mock_container

    def test_register_channel_middleware_skips_when_no_container(self):
        with patch("yuxi.channel.startup._container", None):
            app = MagicMock()
            register_channel_middleware(app)

            assert not hasattr(app.state, "channel")