Rev gateway tests

This commit is contained in:
Cyber MacGeddon 2025-07-14 12:51:06 +01:00
parent 808cdfa64d
commit 57bea86e18
2 changed files with 4 additions and 14 deletions

View file

@ -51,9 +51,9 @@ class TestConfigRequestor:
# Setup translator response
mock_request_translator.to_pulsar.return_value = "translated_request"
# Temporarily patch ServiceRequestor async methods to prevent coroutine warnings
with patch.object(ServiceRequestor, 'start', new_callable=AsyncMock), \
patch.object(ServiceRequestor, 'process', new_callable=AsyncMock):
# 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(
pulsar_client=MagicMock(),
consumer="test-consumer",

View file

@ -10,11 +10,7 @@ import pulsar
from trustgraph.gateway.service import Api, run, default_pulsar_host, default_prometheus_url, default_timeout, default_port, default_api_token
# Store the original method before patching
_original_app_factory = Api.app_factory
# Replace with AsyncMock to prevent coroutine warnings during introspection
Api.app_factory = AsyncMock()
# Tests for Gateway Service API
class TestApi:
@ -128,9 +124,6 @@ class TestApi:
api = Api()
# Temporarily restore the real app_factory for this test
api.app_factory = _original_app_factory.__get__(api, Api)
# Mock the dependencies
api.config_receiver = Mock()
api.config_receiver.start = AsyncMock()
@ -158,9 +151,6 @@ class TestApi:
api = Api()
# Temporarily restore the real app_factory for this test
api.app_factory = _original_app_factory.__get__(api, Api)
# Mock custom endpoints
mock_endpoint1 = Mock()
mock_endpoint1.add_routes = Mock()