from __future__ import annotations

import asyncio
from collections.abc import Mapping
from typing import Any

import pytest

from cmvr_edge_ai.application import EdgeAIApplication
from cmvr_edge_ai.config import load_config_data
from cmvr_edge_ai.core import ComponentContext, Source
from cmvr_edge_ai.plugins import PluginKind, PluginRegistry, PluginSpec


class _MetadataSource(Source):
    def __init__(self, node_id: str, params: Mapping[str, Any]) -> None:
        del node_id, params
        self.seen: object | None = None

    async def setup(self, context: ComponentContext) -> None:
        self.seen = context.metadata.get("invocation_broker")

    async def messages(self):  # type: ignore[no-untyped-def]
        if False:
            yield None


def test_application_passes_embedding_metadata_to_pipeline_components() -> None:
    source = _MetadataSource("source", {})
    registry = PluginRegistry()
    registry.register(
        PluginSpec(
            plugin_id="test.metadata_source@1",
            kind=PluginKind.SOURCE,
            factory=lambda _node_id, _params: source,
            outputs={"output": "*"},
        )
    )
    config = load_config_data(
        {
            "api_version": "cmvr.edge.ai/v1",
            "pipelines": {
                "metadata": {
                    "nodes": {"source": {"uses": "test.metadata_source@1"}}
                }
            },
        }
    )
    broker = object()

    async def exercise() -> None:
        application = EdgeAIApplication(
            config,
            registry,
            metadata={"invocation_broker": broker},
        )
        await application.start()
        await application.wait()
        await application.stop()

    asyncio.run(exercise())
    assert source.seen is broker


def test_application_rejects_metadata_that_overrides_owned_resources() -> None:
    config = load_config_data(
        {
            "api_version": "cmvr.edge.ai/v1",
            "pipelines": {
                "empty": {
                    "nodes": {"source": {"uses": "test.source@1"}}
                }
            },
        }
    )

    with pytest.raises(ValueError, match="must not override owned resource"):
        EdgeAIApplication(
            config,
            PluginRegistry(),
            metadata={"thread_executor": object()},
        )