mirror of
https://github.com/trustgraph-ai/trustgraph.git
synced 2026-04-25 08:26:21 +02:00
cf0daedefa
* Changed schema for Value -> Term, majorly breaking change * Following the schema change, Value -> Term into all processing * Updated Cassandra for g, p, s, o index patterns (7 indexes) * Reviewed and updated all tests * Neo4j, Memgraph and FalkorDB remain broken, will look at once settled down
89 lines
No EOL
2.7 KiB
Python
89 lines
No EOL
2.7 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 = {"t": "i", "i": "http://example.com/resource"}
|
|
|
|
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 = {"t": "l", "v": "literal string"}
|
|
|
|
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": {"t": "i", "i": "subject1"},
|
|
"p": {"t": "i", "i": "predicate1"},
|
|
"o": {"t": "l", "v": "object1"}
|
|
},
|
|
{
|
|
"s": {"t": "l", "v": "subject2"},
|
|
"p": {"t": "i", "i": "predicate2"},
|
|
"o": {"t": "i", "i": "object2"}
|
|
}
|
|
]
|
|
|
|
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 == {"t": "i", "i": "http://example.com/test"}
|
|
|
|
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 == {"t": "l", "v": "test literal"} |