trustgraph/tests/unit/test_gateway/test_dispatch_serialize.py

"""
Tests for Gateway Dispatch Serialization
"""

import pytest
from unittest.mock import MagicMock

from trustgraph.gateway.dispatch.serialize import to_value, to_subgraph, serialize_value
from trustgraph.schema import Value, Triple


class TestDispatchSerialize:
    """Test cases for dispatch serialization functions"""

    def test_to_value_with_uri(self):
        """Test to_value function with URI"""
        input_data = {"v": "http://example.com/resource", "e": True}
        
        result = to_value(input_data)
        
        assert isinstance(result, Value)
        assert result.value == "http://example.com/resource"
        assert result.is_uri is True

    def test_to_value_with_literal(self):
        """Test to_value function with literal value"""
        input_data = {"v": "literal string", "e": False}
        
        result = to_value(input_data)
        
        assert isinstance(result, Value)
        assert result.value == "literal string"
        assert result.is_uri is False

    def test_to_subgraph_with_multiple_triples(self):
        """Test to_subgraph function with multiple triples"""
        input_data = [
            {
                "s": {"v": "subject1", "e": True},
                "p": {"v": "predicate1", "e": True},
                "o": {"v": "object1", "e": False}
            },
            {
                "s": {"v": "subject2", "e": False},
                "p": {"v": "predicate2", "e": True},
                "o": {"v": "object2", "e": True}
            }
        ]
        
        result = to_subgraph(input_data)
        
        assert len(result) == 2
        assert all(isinstance(triple, Triple) for triple in result)
        
        # Check first triple
        assert result[0].s.value == "subject1"
        assert result[0].s.is_uri is True
        assert result[0].p.value == "predicate1"
        assert result[0].p.is_uri is True
        assert result[0].o.value == "object1"
        assert result[0].o.is_uri is False
        
        # Check second triple
        assert result[1].s.value == "subject2"
        assert result[1].s.is_uri is False

    def test_to_subgraph_with_empty_list(self):
        """Test to_subgraph function with empty input"""
        input_data = []
        
        result = to_subgraph(input_data)
        
        assert result == []

    def test_serialize_value_with_uri(self):
        """Test serialize_value function with URI value"""
        value = Value(value="http://example.com/test", is_uri=True)
        
        result = serialize_value(value)
        
        assert result == {"v": "http://example.com/test", "e": True}

    def test_serialize_value_with_literal(self):
        """Test serialize_value function with literal value"""
        value = Value(value="test literal", is_uri=False)
        
        result = serialize_value(value)
        
        assert result == {"v": "test literal", "e": False}