From adc43119881e6b6c94004bf9b63f13102b8c880b Mon Sep 17 00:00:00 2001 From: Cyber MacGeddon Date: Sat, 12 Jul 2025 22:04:35 +0100 Subject: [PATCH] Rev gateway tests --- tests/unit/test_gateway/test_dispatch_manager.py | 3 +++ tests/unit/test_gateway/test_service.py | 4 ++-- tests/unit/test_rev_gateway/test_rev_gateway_service.py | 4 ++-- 3 files changed, 7 insertions(+), 4 deletions(-) diff --git a/tests/unit/test_gateway/test_dispatch_manager.py b/tests/unit/test_gateway/test_dispatch_manager.py index 51c52bab..69e95d7a 100644 --- a/tests/unit/test_gateway/test_dispatch_manager.py +++ b/tests/unit/test_gateway/test_dispatch_manager.py @@ -406,6 +406,9 @@ class TestDispatcherManager: mock_config_receiver = Mock() manager = DispatcherManager(mock_pulsar_client, mock_config_receiver) + # Add flow to the flows dictionary + manager.flows["test_flow"] = {"services": {"agent": {}}} + # Pre-populate with existing dispatcher mock_dispatcher = Mock() mock_dispatcher.process = AsyncMock(return_value="cached_result") diff --git a/tests/unit/test_gateway/test_service.py b/tests/unit/test_gateway/test_service.py index 746d5284..6b1b61ff 100644 --- a/tests/unit/test_gateway/test_service.py +++ b/tests/unit/test_gateway/test_service.py @@ -132,7 +132,7 @@ class TestApi: app = await api.app_factory() assert isinstance(app, web.Application) - assert app.client_max_size == 256 * 1024 * 1024 + assert app._client_max_size == 256 * 1024 * 1024 # Verify that config receiver was started api.config_receiver.start.assert_called_once() @@ -207,7 +207,7 @@ class TestApi: assert api.dispatcher_manager.pulsar_client == api.pulsar_client assert api.dispatcher_manager.config_receiver == api.config_receiver assert api.endpoint_manager.dispatcher_manager == api.dispatcher_manager - assert api.endpoint_manager.auth == api.auth + # EndpointManager doesn't store auth directly, it passes it to individual endpoints class TestRunFunction: diff --git a/tests/unit/test_rev_gateway/test_rev_gateway_service.py b/tests/unit/test_rev_gateway/test_rev_gateway_service.py index 547edeff..d991ba45 100644 --- a/tests/unit/test_rev_gateway/test_rev_gateway_service.py +++ b/tests/unit/test_rev_gateway/test_rev_gateway_service.py @@ -119,7 +119,7 @@ class TestReverseGateway: @patch('trustgraph.rev_gateway.service.ConfigReceiver') @patch('trustgraph.rev_gateway.service.MessageDispatcher') @patch('pulsar.Client') - @patch('aiohttp.ClientSession') + @patch('trustgraph.rev_gateway.service.ClientSession') @pytest.mark.asyncio async def test_reverse_gateway_connect_success(self, mock_session_class, mock_pulsar_client, mock_dispatcher, mock_config_receiver): """Test ReverseGateway successful connection""" @@ -143,7 +143,7 @@ class TestReverseGateway: @patch('trustgraph.rev_gateway.service.ConfigReceiver') @patch('trustgraph.rev_gateway.service.MessageDispatcher') @patch('pulsar.Client') - @patch('aiohttp.ClientSession') + @patch('trustgraph.rev_gateway.service.ClientSession') @pytest.mark.asyncio async def test_reverse_gateway_connect_failure(self, mock_session_class, mock_pulsar_client, mock_dispatcher, mock_config_receiver): """Test ReverseGateway connection failure"""