from __future__ import annotations

from unittest.mock import MagicMock

import pytest

from yuxi.channel.application.pipeline.builder import build_inbound_pipeline
from yuxi.channel.application.service.auth_service import AuthService
from yuxi.channel.domain.service.pipeline import Pipeline


class TestBuildInboundPipeline:
    @pytest.fixture
    def mock_auth_service(self):
        return MagicMock(spec=AuthService)

    @pytest.fixture
    def mock_cache(self):
        return MagicMock()

    @pytest.fixture
    def mock_rate_limiter(self):
        return MagicMock()

    @pytest.fixture
    def mock_queue(self):
        return MagicMock()

    def test_build_basic_pipeline(self, mock_auth_service, mock_cache, mock_rate_limiter, mock_queue):
        pipeline = build_inbound_pipeline(
            auth_service=mock_auth_service,
            cache_port=mock_cache,
            rate_limit_port=mock_rate_limiter,
            queue_port=mock_queue,
        )
        assert isinstance(pipeline, Pipeline)
        assert len(pipeline.middlewares) == 8

    def test_build_pipeline_with_config(self, mock_auth_service, mock_cache, mock_rate_limiter, mock_queue):
        config = {
            "keyword_blocklist": ["bad"],
            "allow_from": {"web": ["user1"]},
            "access_policies": {"web": {"dm_policy": "open"}},
        }
        pipeline = build_inbound_pipeline(
            auth_service=mock_auth_service,
            cache_port=mock_cache,
            rate_limit_port=mock_rate_limiter,
            queue_port=mock_queue,
            channel_config=config,
        )
        assert isinstance(pipeline, Pipeline)
        assert len(pipeline.middlewares) == 8

    def test_build_pipeline_with_signature_verifier(self, mock_auth_service, mock_cache, mock_rate_limiter, mock_queue):
        verifier = MagicMock()
        pipeline = build_inbound_pipeline(
            auth_service=mock_auth_service,
            cache_port=mock_cache,
            rate_limit_port=mock_rate_limiter,
            queue_port=mock_queue,
            signature_verifier=verifier,
        )
        assert isinstance(pipeline, Pipeline)

    def test_build_pipeline_with_metrics(self, mock_auth_service, mock_cache, mock_rate_limiter, mock_queue):
        metrics = MagicMock()
        pipeline = build_inbound_pipeline(
            auth_service=mock_auth_service,
            cache_port=mock_cache,
            rate_limit_port=mock_rate_limiter,
            queue_port=mock_queue,
            metrics=metrics,
        )
        assert isinstance(pipeline, Pipeline)

    def test_build_pipeline_with_bot_id(self, mock_auth_service, mock_cache, mock_rate_limiter, mock_queue):
        pipeline = build_inbound_pipeline(
            auth_service=mock_auth_service,
            cache_port=mock_cache,
            rate_limit_port=mock_rate_limiter,
            queue_port=mock_queue,
            bot_id="bot123",
        )
        assert isinstance(pipeline, Pipeline)

    def test_build_pipeline_with_keyword_matcher(self, mock_auth_service, mock_cache, mock_rate_limiter, mock_queue):
        matcher = MagicMock()
        pipeline = build_inbound_pipeline(
            auth_service=mock_auth_service,
            cache_port=mock_cache,
            rate_limit_port=mock_rate_limiter,
            queue_port=mock_queue,
            keyword_matcher=matcher,
        )
        assert isinstance(pipeline, Pipeline)

    def test_pipeline_middleware_order(self, mock_auth_service, mock_cache, mock_rate_limiter, mock_queue):
        pipeline = build_inbound_pipeline(
            auth_service=mock_auth_service,
            cache_port=mock_cache,
            rate_limit_port=mock_rate_limiter,
            queue_port=mock_queue,
        )
        names = [mw.name for mw in pipeline.middlewares]
        expected = [
            "auth",
            "validation",
            "keyword_filter",
            "dedup",
            "rate_limit",
            "access_policy",
            "mention_gate",
            "enqueue_mq",
        ]
        assert names == expected