mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-21 19:21:03 +02:00
Rev gateway tests
This commit is contained in:
parent
808cdfa64d
commit
57bea86e18
2 changed files with 4 additions and 14 deletions
|
|
@ -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",
|
||||
|
|
|
|||
|
|
@ -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()
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue