"""Tests for CLI container composition root."""

import pytest

from basic_memory.cli.container import (
    CliContainer,
    get_container,
    set_container,
    get_or_create_container,
)
from basic_memory.runtime import RuntimeMode


class TestCliContainer:
    """Tests for CliContainer."""

    def test_create_from_config(self, app_config):
        """Container can be created from config."""
        container = CliContainer(config=app_config, mode=RuntimeMode.LOCAL)
        assert container.config == app_config
        assert container.mode == RuntimeMode.LOCAL

    def test_is_cloud_mode_when_cloud(self, app_config):
        """is_cloud_mode returns True in cloud mode."""
        container = CliContainer(config=app_config, mode=RuntimeMode.CLOUD)
        assert container.is_cloud_mode is True

    def test_is_cloud_mode_when_local(self, app_config):
        """is_cloud_mode returns False in local mode."""
        container = CliContainer(config=app_config, mode=RuntimeMode.LOCAL)
        assert container.is_cloud_mode is False

    def test_is_cloud_mode_when_test(self, app_config):
        """is_cloud_mode returns False in test mode."""
        container = CliContainer(config=app_config, mode=RuntimeMode.TEST)
        assert container.is_cloud_mode is False


class TestContainerAccessors:
    """Tests for container get/set functions."""

    def test_get_container_raises_when_not_set(self, monkeypatch):
        """get_container raises RuntimeError when container not initialized."""
        import basic_memory.cli.container as container_module

        monkeypatch.setattr(container_module, "_container", None)

        with pytest.raises(RuntimeError, match="CLI container not initialized"):
            get_container()

    def test_set_and_get_container(self, app_config, monkeypatch):
        """set_container allows get_container to return the container."""
        import basic_memory.cli.container as container_module

        container = CliContainer(config=app_config, mode=RuntimeMode.LOCAL)
        monkeypatch.setattr(container_module, "_container", None)

        set_container(container)
        assert get_container() is container


class TestGetOrCreateContainer:
    """Tests for get_or_create_container - unique to CLI container."""

    def test_creates_new_when_none_exists(self, monkeypatch):
        """get_or_create_container creates a new container when none exists."""
        import basic_memory.cli.container as container_module

        monkeypatch.setattr(container_module, "_container", None)

        container = get_or_create_container()
        assert container is not None
        assert isinstance(container, CliContainer)

    def test_returns_existing_when_set(self, app_config, monkeypatch):
        """get_or_create_container returns existing container if already set."""
        import basic_memory.cli.container as container_module

        existing = CliContainer(config=app_config, mode=RuntimeMode.LOCAL)
        monkeypatch.setattr(container_module, "_container", existing)

        result = get_or_create_container()
        assert result is existing

    def test_sets_module_level_container(self, monkeypatch):
        """get_or_create_container sets the module-level container."""
        import basic_memory.cli.container as container_module

        monkeypatch.setattr(container_module, "_container", None)

        container = get_or_create_container()

        # Verify it was set at module level
        assert container_module._container is container
        # Verify get_container now works
        assert get_container() is container