mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-07-22 19:51:02 +02:00
89 lines
No EOL
2.8 KiB
Python
89 lines
No EOL
2.8 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 Term, Triple, IRI, LITERAL
|
|
|
|
|
|
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, Term)
|
|
assert result.iri == "http://example.com/resource"
|
|
assert result.type == IRI
|
|
|
|
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, Term)
|
|
assert result.value == "literal string"
|
|
assert result.type == LITERAL
|
|
|
|
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.iri == "subject1"
|
|
assert result[0].s.type == IRI
|
|
assert result[0].p.iri == "predicate1"
|
|
assert result[0].p.type == IRI
|
|
assert result[0].o.value == "object1"
|
|
assert result[0].o.type == LITERAL
|
|
|
|
# Check second triple
|
|
assert result[1].s.value == "subject2"
|
|
assert result[1].s.type == LITERAL
|
|
|
|
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"""
|
|
term = Term(type=IRI, iri="http://example.com/test")
|
|
|
|
result = serialize_value(term)
|
|
|
|
assert result == {"v": "http://example.com/test", "e": True}
|
|
|
|
def test_serialize_value_with_literal(self):
|
|
"""Test serialize_value function with literal value"""
|
|
term = Term(type=LITERAL, value="test literal")
|
|
|
|
result = serialize_value(term)
|
|
|
|
assert result == {"v": "test literal", "e": False} |