mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
48 lines
No EOL
1.1 KiB
Python
48 lines
No EOL
1.1 KiB
Python
"""
|
|
Shared fixtures for CLI unit tests.
|
|
"""
|
|
|
|
import pytest
|
|
from unittest.mock import AsyncMock, MagicMock
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_websocket_connection():
|
|
"""Mock WebSocket connection for CLI tools."""
|
|
mock_ws = MagicMock()
|
|
|
|
# Create simple async functions that don't leave coroutines hanging
|
|
async def mock_send(data):
|
|
return None
|
|
|
|
async def mock_recv():
|
|
return ""
|
|
|
|
async def mock_close():
|
|
return None
|
|
|
|
mock_ws.send = mock_send
|
|
mock_ws.recv = mock_recv
|
|
mock_ws.close = mock_close
|
|
return mock_ws
|
|
|
|
|
|
@pytest.fixture
|
|
def mock_pulsar_client():
|
|
"""Mock Pulsar client for CLI tools that use messaging."""
|
|
mock_client = MagicMock()
|
|
mock_client.create_consumer = MagicMock()
|
|
mock_client.create_producer = MagicMock()
|
|
mock_client.close = MagicMock()
|
|
return mock_client
|
|
|
|
|
|
@pytest.fixture
|
|
def sample_metadata():
|
|
"""Sample metadata structure used across CLI tools."""
|
|
return {
|
|
"id": "test-doc-123",
|
|
"metadata": [],
|
|
"user": "test-user",
|
|
"collection": "test-collection"
|
|
} |