mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 16:36:21 +02:00
89 lines
No EOL
2.9 KiB
Python
89 lines
No EOL
2.9 KiB
Python
"""
|
|
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} |