API¶

@pytest.fixture(autouse=True)
def pico_isolation(request, monkeypatch):
    """Tests must not depend on which pico plugins happen to be installed
    in the venv — auto-discovery is opt-in via the pico_auto_plugins marker."""
    if request.node.get_closest_marker(AUTO_PLUGINS_MARKER) is None:
        monkeypatch.setenv("PICO_BOOT_AUTO_PLUGINS", "false")

@pytest.fixture
def make_container(request):
    """Factory for pico containers with automatic shutdown on teardown.

    The ``pico_module`` ini option (usually the package under test) is
    prepended to the modules list, so ``make_container()`` boots it alone.

    Usage::

        def test_service(make_container):
            container = make_container("my_pkg", config={"my": {"key": "value"}})
            svc = container.get(MyService)
    """
    created = []
    base = request.config.getini("pico_module")

    def _make(*modules, config=None, boot=False):
        from pico_ioc import DictSource, configuration

        if config is None or isinstance(config, dict):
            config = configuration(DictSource(config or {}))
        if boot:
            from pico_boot import init
        else:
            from pico_ioc import init
        container = init(modules=([base] if base else []) + list(modules), config=config)
        created.append(container)
        return container

    yield _make
    for container in reversed(created):
        container.shutdown()

@pytest.fixture
def make_client():
    """FastAPI TestClient over a container's app; lifespan runs, closed on teardown.

    The client exit triggers the pico-fastapi lifespan shutdown AND the
    container fixture shuts down again: requires pico-ioc >= 2.3.3
    (idempotent shutdown).

    Usage::

        def test_endpoint(make_container, make_client):
            container = make_container("pico_fastapi", config={"fastapi": {"title": "t"}})
            client = make_client(container)
            assert client.get("/api/ping").status_code == 200
    """
    import contextlib

    with contextlib.ExitStack() as stack:

        def _make(container):
            from fastapi import FastAPI
            from fastapi.testclient import TestClient

            return stack.enter_context(TestClient(container.get(FastAPI)))

        yield _make