mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-22 03:31:02 +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
|
# Setup translator response
|
||||||
mock_request_translator.to_pulsar.return_value = "translated_request"
|
mock_request_translator.to_pulsar.return_value = "translated_request"
|
||||||
|
|
||||||
# Temporarily patch ServiceRequestor async methods to prevent coroutine warnings
|
# Patch ServiceRequestor async methods with regular mocks (not AsyncMock)
|
||||||
with patch.object(ServiceRequestor, 'start', new_callable=AsyncMock), \
|
with patch.object(ServiceRequestor, 'start', return_value=None), \
|
||||||
patch.object(ServiceRequestor, 'process', new_callable=AsyncMock):
|
patch.object(ServiceRequestor, 'process', return_value=None):
|
||||||
requestor = ConfigRequestor(
|
requestor = ConfigRequestor(
|
||||||
pulsar_client=MagicMock(),
|
pulsar_client=MagicMock(),
|
||||||
consumer="test-consumer",
|
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
|
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
|
# Tests for Gateway Service API
|
||||||
_original_app_factory = Api.app_factory
|
|
||||||
|
|
||||||
# Replace with AsyncMock to prevent coroutine warnings during introspection
|
|
||||||
Api.app_factory = AsyncMock()
|
|
||||||
|
|
||||||
|
|
||||||
class TestApi:
|
class TestApi:
|
||||||
|
|
@ -128,9 +124,6 @@ class TestApi:
|
||||||
|
|
||||||
api = Api()
|
api = Api()
|
||||||
|
|
||||||
# Temporarily restore the real app_factory for this test
|
|
||||||
api.app_factory = _original_app_factory.__get__(api, Api)
|
|
||||||
|
|
||||||
# Mock the dependencies
|
# Mock the dependencies
|
||||||
api.config_receiver = Mock()
|
api.config_receiver = Mock()
|
||||||
api.config_receiver.start = AsyncMock()
|
api.config_receiver.start = AsyncMock()
|
||||||
|
|
@ -158,9 +151,6 @@ class TestApi:
|
||||||
|
|
||||||
api = Api()
|
api = Api()
|
||||||
|
|
||||||
# Temporarily restore the real app_factory for this test
|
|
||||||
api.app_factory = _original_app_factory.__get__(api, Api)
|
|
||||||
|
|
||||||
# Mock custom endpoints
|
# Mock custom endpoints
|
||||||
mock_endpoint1 = Mock()
|
mock_endpoint1 = Mock()
|
||||||
mock_endpoint1.add_routes = Mock()
|
mock_endpoint1.add_routes = Mock()
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue