mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 16:36:21 +02:00
feeb92b33f
Derive consumer behaviour from queue class, remove consumer_type parameter The queue class prefix (flow, request, response, notify) now fully determines consumer behaviour in both RabbitMQ and Pulsar backends. Added 'notify' class for ephemeral broadcast (config push notifications). Response and notify classes always create per-subscriber auto-delete queues, eliminating orphaned queues that accumulated on service restarts. Change init-trustgraph to set up the 'notify' namespace in Pulsar instead of old hangover 'state'. Fixes 'stuck backlog' on RabbitMQ config notification queue.
107 lines
3.6 KiB
Python
107 lines
3.6 KiB
Python
"""
|
|
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_notify_is_not_durable(self, backend):
|
|
name, durable = backend.map_queue_name('notify:tg:config')
|
|
assert durable is False
|
|
assert name == 'tg.notify.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 == '/'
|