"""
Unit tests for RabbitMQ backend — queue name mapping and factory dispatch.
Does not require a running RabbitMQ instance.
"""

import pytest
import argparse

pika = pytest.importorskip("pika", reason="pika not installed")

from trustgraph.base.rabbitmq_backend import RabbitMQBackend
from trustgraph.base.pubsub import get_pubsub, add_pubsub_args


class TestRabbitMQMapQueueName:

    @pytest.fixture
    def backend(self):
        b = object.__new__(RabbitMQBackend)
        return b

    def test_flow_is_durable(self, backend):
        name, durable = backend.map_queue_name('flow:tg:text-completion-request')
        assert durable is True
        assert name == 'tg.flow.text-completion-request'

    def test_state_is_durable(self, backend):
        name, durable = backend.map_queue_name('state:tg:config')
        assert durable is True
        assert name == 'tg.state.config'

    def test_request_is_not_durable(self, backend):
        name, durable = backend.map_queue_name('request:tg:config')
        assert durable is False
        assert name == 'tg.request.config'

    def test_response_is_not_durable(self, backend):
        name, durable = backend.map_queue_name('response:tg:librarian')
        assert durable is False
        assert name == 'tg.response.librarian'

    def test_custom_topicspace(self, backend):
        name, durable = backend.map_queue_name('flow:prod:my-queue')
        assert name == 'prod.flow.my-queue'
        assert durable is True

    def test_no_colon_defaults_to_flow(self, backend):
        name, durable = backend.map_queue_name('simple-queue')
        assert name == 'tg.simple-queue'
        assert durable is False

    def test_invalid_class_raises(self, backend):
        with pytest.raises(ValueError, match="Invalid queue class"):
            backend.map_queue_name('unknown:tg:topic')

    def test_flow_with_flow_suffix(self, backend):
        """Queue names with flow suffix (e.g. :default) are preserved."""
        name, durable = backend.map_queue_name('request:tg:prompt:default')
        assert name == 'tg.request.prompt:default'


class TestGetPubsubRabbitMQ:

    def test_factory_creates_rabbitmq_backend(self):
        backend = get_pubsub(pubsub_backend='rabbitmq')
        assert isinstance(backend, RabbitMQBackend)

    def test_factory_passes_config(self):
        backend = get_pubsub(
            pubsub_backend='rabbitmq',
            rabbitmq_host='myhost',
            rabbitmq_port=5673,
            rabbitmq_username='user',
            rabbitmq_password='pass',
            rabbitmq_vhost='/test',
        )
        assert isinstance(backend, RabbitMQBackend)
        # Verify connection params were set
        params = backend._connection_params
        assert params.host == 'myhost'
        assert params.port == 5673
        assert params.virtual_host == '/test'


class TestAddPubsubArgsRabbitMQ:

    def test_rabbitmq_args_present(self):
        parser = argparse.ArgumentParser()
        add_pubsub_args(parser)
        args = parser.parse_args([
            '--pubsub-backend', 'rabbitmq',
            '--rabbitmq-host', 'myhost',
            '--rabbitmq-port', '5673',
        ])
        assert args.pubsub_backend == 'rabbitmq'
        assert args.rabbitmq_host == 'myhost'
        assert args.rabbitmq_port == 5673

    def test_rabbitmq_defaults_container(self):
        parser = argparse.ArgumentParser()
        add_pubsub_args(parser)
        args = parser.parse_args([])
        assert args.rabbitmq_host == 'rabbitmq'
        assert args.rabbitmq_port == 5672
        assert args.rabbitmq_username == 'guest'
        assert args.rabbitmq_password == 'guest'
        assert args.rabbitmq_vhost == '/'