mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
4fb0b4d8e8
Remove Pulsar-specific concepts from application code so that the pub/sub backend is swappable via configuration. Rename translators: - to_pulsar/from_pulsar → decode/encode across all translator classes, dispatch handlers, and tests (55+ files) - from_response_with_completion → encode_with_completion - Remove pulsar.schema.Record from translator base class Queue naming (CLASS:TOPICSPACE:TOPIC): - Replace topic() helper with queue() using new format: flow:tg:name, request:tg:name, response:tg:name, state:tg:name - Queue class implies persistence/TTL (no QoS in names) - Update Pulsar backend map_topic() to parse new format - Librarian queues use flow class (persistent, for chunking) - Config push uses state class (persistent, last-value) - Remove 15 dead topic imports from schema files - Update init_trustgraph.py namespace: config → state Confine Pulsar to pulsar_backend.py: - Delete legacy PulsarClient class from pubsub.py - Move add_args to add_pubsub_args() with standalone flag for CLI tools (defaults to localhost) - PulsarBackendConsumer.receive() catches _pulsar.Timeout, raises standard TimeoutError - Remove Pulsar imports from: async_processor, flow_processor, log_level, all 11 client files, 4 storage writers, gateway service, gateway config receiver - Remove log_level/LoggerLevel from client API - Rewrite tg-monitor-prompts to use backend abstraction - Update tg-dump-queues to use add_pubsub_args Also: pubsub-abstraction.md tech spec covering problem statement, design goals, as-is requirements, candidate broker assessment, approach, and implementation order.
93 lines
No EOL
3.7 KiB
Python
93 lines
No EOL
3.7 KiB
Python
"""
|
|
Tests for Gateway Config Dispatch
|
|
"""
|
|
|
|
import pytest
|
|
from unittest.mock import MagicMock, patch, AsyncMock, Mock
|
|
|
|
from trustgraph.gateway.dispatch.config import ConfigRequestor
|
|
|
|
# Import parent class for local patching
|
|
from trustgraph.gateway.dispatch.requestor import ServiceRequestor
|
|
|
|
|
|
class TestConfigRequestor:
|
|
"""Test cases for ConfigRequestor class"""
|
|
|
|
@patch('trustgraph.gateway.dispatch.config.TranslatorRegistry')
|
|
def test_config_requestor_initialization(self, mock_translator_registry):
|
|
"""Test ConfigRequestor initialization"""
|
|
# Mock translators
|
|
mock_request_translator = Mock()
|
|
mock_response_translator = Mock()
|
|
mock_translator_registry.get_request_translator.return_value = mock_request_translator
|
|
mock_translator_registry.get_response_translator.return_value = mock_response_translator
|
|
|
|
# Mock dependencies
|
|
mock_backend = Mock()
|
|
|
|
requestor = ConfigRequestor(
|
|
backend=mock_backend,
|
|
consumer="test-consumer",
|
|
subscriber="test-subscriber",
|
|
timeout=60
|
|
)
|
|
|
|
# Verify translator setup
|
|
mock_translator_registry.get_request_translator.assert_called_once_with("config")
|
|
mock_translator_registry.get_response_translator.assert_called_once_with("config")
|
|
|
|
assert requestor.request_translator == mock_request_translator
|
|
assert requestor.response_translator == mock_response_translator
|
|
|
|
@patch('trustgraph.gateway.dispatch.config.TranslatorRegistry')
|
|
def test_config_requestor_to_request(self, mock_translator_registry):
|
|
"""Test ConfigRequestor to_request method"""
|
|
# Mock translators
|
|
mock_request_translator = Mock()
|
|
mock_translator_registry.get_request_translator.return_value = mock_request_translator
|
|
mock_translator_registry.get_response_translator.return_value = Mock()
|
|
|
|
# Setup translator response
|
|
mock_request_translator.decode.return_value = "translated_request"
|
|
|
|
# Patch ServiceRequestor async methods with regular mocks (not AsyncMock)
|
|
with patch.object(ServiceRequestor, 'start', return_value=None), \
|
|
patch.object(ServiceRequestor, 'process', return_value=None):
|
|
requestor = ConfigRequestor(
|
|
backend=Mock(),
|
|
consumer="test-consumer",
|
|
subscriber="test-subscriber"
|
|
)
|
|
|
|
# Call to_request
|
|
result = requestor.to_request({"test": "body"})
|
|
|
|
# Verify translator was called correctly
|
|
mock_request_translator.decode.assert_called_once_with({"test": "body"})
|
|
assert result == "translated_request"
|
|
|
|
@patch('trustgraph.gateway.dispatch.config.TranslatorRegistry')
|
|
def test_config_requestor_from_response(self, mock_translator_registry):
|
|
"""Test ConfigRequestor from_response method"""
|
|
# Mock translators
|
|
mock_response_translator = Mock()
|
|
mock_translator_registry.get_request_translator.return_value = Mock()
|
|
mock_translator_registry.get_response_translator.return_value = mock_response_translator
|
|
|
|
# Setup translator response
|
|
mock_response_translator.encode_with_completion.return_value = "translated_response"
|
|
|
|
requestor = ConfigRequestor(
|
|
backend=Mock(),
|
|
consumer="test-consumer",
|
|
subscriber="test-subscriber"
|
|
)
|
|
|
|
# Call from_response
|
|
mock_message = Mock()
|
|
result = requestor.from_response(mock_message)
|
|
|
|
# Verify translator was called correctly
|
|
mock_response_translator.encode_with_completion.assert_called_once_with(mock_message)
|
|
assert result == "translated_response" |