apunkt/trustgraph
1
0
Fork 0
mirror of https://github.com/trustgraph-ai/trustgraph.git synced 2026-07-23 12:11:02 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

Tests

Browse source
This commit is contained in:
Cyber MacGeddon 2026-01-27 00:38:11 +00:00
parent 1d6c58f8cd
commit baa5a9347f
1 changed files with 7 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/integration/test_import_export_graceful_shutdown.py
View file

@ -54,7 +54,7 @@ class MockWebSocket:
"user": "test-user", "user": "test-user",
"collection": "test-collection" "collection": "test-collection"
}, },
"triples": [{"s": {"v": "subject", "e": False}, "p": {"v": "predicate", "e": False}, "o": {"v": "object", "e": False}}] "triples": [{"s": {"t": "l", "v": "subject"}, "p": {"t": "l", "v": "predicate"}, "o": {"t": "l", "v": "object"}}]
} }
@ -118,7 +118,7 @@ async def test_import_graceful_shutdown_integration(mock_backend):
"user": "test-user", "user": "test-user",
"collection": "test-collection" "collection": "test-collection"
}, },
"triples": [{"s": {"v": f"subject-{i}", "e": False}, "p": {"v": "predicate", "e": False}, "o": {"v": f"object-{i}", "e": False}}] "triples": [{"s": {"t": "l", "v": f"subject-{i}"}, "p": {"t": "l", "v": "predicate"}, "o": {"t": "l", "v": f"object-{i}"}}]
} }
messages.append(msg_data) messages.append(msg_data)
@ -163,7 +163,7 @@ async def test_export_no_message_loss_integration(mock_backend):
"user": "test-user", "user": "test-user",
"collection": "test-collection" "collection": "test-collection"
}, },
"triples": [{"s": {"v": f"export-subject-{i}", "e": False}, "p": {"v": "predicate", "e": False}, "o": {"v": f"export-object-{i}", "e": False}}] "triples": [{"s": {"t": "l", "v": f"export-subject-{i}"}, "p": {"t": "l", "v": "predicate"}, "o": {"t": "l", "v": f"export-object-{i}"}}]
} }
# Create Triples object instead of raw dict # Create Triples object instead of raw dict
from trustgraph.schema import Triples, Metadata from trustgraph.schema import Triples, Metadata
@ -302,7 +302,7 @@ async def test_concurrent_import_export_shutdown():
"user": "test-user", "user": "test-user",
"collection": "test-collection" "collection": "test-collection"
}, },
"triples": [{"s": {"v": f"concurrent-subject-{i}", "e": False}, "p": {"v": "predicate", "e": False}, "o": {"v": "object", "e": False}}] "triples": [{"s": {"t": "l", "v": f"concurrent-subject-{i}"}, "p": {"t": "l", "v": "predicate"}, "o": {"t": "l", "v": "object"}}]
} }
await import_handler.receive(msg) await import_handler.receive(msg)
@ -359,7 +359,7 @@ async def test_websocket_close_during_message_processing():
"user": "test-user", "user": "test-user",
"collection": "test-collection" "collection": "test-collection"
}, },
"triples": [{"s": {"v": f"slow-subject-{i}", "e": False}, "p": {"v": "predicate", "e": False}, "o": {"v": "object", "e": False}}] "triples": [{"s": {"t": "l", "v": f"slow-subject-{i}"}, "p": {"t": "l", "v": "predicate"}, "o": {"t": "l", "v": "object"}}]
} }
task = asyncio.create_task(import_handler.receive(msg)) task = asyncio.create_task(import_handler.receive(msg))
message_tasks.append(task) message_tasks.append(task)
@ -423,7 +423,7 @@ async def test_backpressure_during_shutdown():
# Simulate receiving and processing a message # Simulate receiving and processing a message
msg_data = { msg_data = {
"metadata": {"id": f"msg-{i}"}, "metadata": {"id": f"msg-{i}"},
"triples": [{"s": {"v": "subject", "e": False}, "p": {"v": "predicate", "e": False}, "o": {"v": "object", "e": False}}] "triples": [{"s": {"t": "l", "v": "subject"}, "p": {"t": "l", "v": "predicate"}, "o": {"t": "l", "v": "object"}}]
} }
await ws.send_json(msg_data) await ws.send_json(msg_data)
# Check if we should stop # Check if we should stop