trustgraph/tests/unit/test_cli/conftest.py

"""
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"
    }

def iri(v):
    """Wire-format IRI term dict, as triples_query_stream yields them."""
    return {"t": "i", "i": v}


def lit(v, d=None, lang=None):
    """Wire-format literal term dict (optional datatype / language)."""
    t = {"t": "l", "v": v}
    if d:
        t["d"] = d
    if lang:
        t["l"] = lang
    return t